feat: clean all if no space left

internal/builder/compiler.go

@@ -3,8 +3,11 @@ package builder
 import (
 	"bytes"
 	"context"
+	"errors"
 	"fmt"
 	"os"
+	"syscall"
+	"time"
 
 	"github.com/x1unix/go-playground/internal/builder/storage"
 	"github.com/x1unix/go-playground/pkg/util/osutil"
@@ -94,6 +97,10 @@ func (s BuildService) Build(ctx context.Context, files map[string][]byte) (*Resu
 
 	workspace, err := s.storage.CreateWorkspace(aid, files)
 	if err != nil {
+		if errors.Is(err, syscall.ENOSPC) {
+			// Immediately schedule cleanup job!
+			s.handleNoSpaceLeft()
+		}
 		return nil, err
 	}
 
@@ -110,6 +117,19 @@ func (s BuildService) buildSource(ctx context.Context, workspace *storage.Worksp
 	return s.runGoTool(ctx, workspace.WorkDir, "build", "-o", workspace.BinaryPath, ".")
 }
 
+func (s BuildService) handleNoSpaceLeft() {
+	s.log.Warn("no space left on device, immediate clean triggered!")
+	ctx, cancelFn := context.WithTimeout(context.Background(), time.Minute)
+	defer cancelFn()
+
+	if err := s.storage.Clean(ctx); err != nil {
+		s.log.Error("failed to clear storage", zap.Error(err))
+	}
+	if err := s.Clean(ctx); err != nil {
+		s.log.Error("failed to clear Go cache", zap.Error(err))
+	}
+}
+
 func (s BuildService) runGoTool(ctx context.Context, workDir string, args ...string) error {
 	cmd := newGoToolCommand(ctx, args...)
 	cmd.Dir = workDir

internal/builder/compiler_test.go

@@ -6,6 +6,7 @@ import (
 	"io/ioutil"
 	"os"
 	"strings"
+	"syscall"
 	"testing"
 
 	"github.com/stretchr/testify/require"
@@ -27,6 +28,7 @@ type testStorage struct {
 	hasItem         func(id storage.ArtifactID) (bool, error)
 	getItem         func(id storage.ArtifactID) (storage.ReadCloseSizer, error)
 	createWorkspace func(id storage.ArtifactID, entries map[string][]byte) (*storage.Workspace, error)
+	clean           func(ctx context.Context) error
 }
 
 func (ts testStorage) HasItem(id storage.ArtifactID) (bool, error) {
@@ -41,6 +43,14 @@ func (ts testStorage) CreateWorkspace(id storage.ArtifactID, entries map[string]
 	return ts.createWorkspace(id, entries)
 }
 
+func (ts testStorage) Clean(ctx context.Context) error {
+	if ts.clean != nil {
+		return ts.clean(ctx)
+	}
+
+	return errors.New("not implemented")
+}
+
 func TestBuildService_GetArtifact(t *testing.T) {
 	cases := map[string]struct {
 		artifactID storage.ArtifactID
@@ -95,6 +105,7 @@ func TestBuildService_Build(t *testing.T) {
 	cases := map[string]struct {
 		skip         bool
 		files        map[string][]byte
+		env          osutil.EnvironmentVariables
 		wantErr      string
 		wantResult   func(files map[string][]byte) *Result
 		beforeRun    func(t *testing.T)
@@ -227,6 +238,30 @@ func TestBuildService_Build(t *testing.T) {
 				return nil, nil
 			},
 		},
+		"handle no space left": {
+			wantErr: "no space left",
+			env: osutil.EnvironmentVariables{
+				"PATH":   ".",
+				"GOPATH": "/tmp",
+			},
+			files: map[string][]byte{
+				"main.go": []byte("package main"),
+			},
+			store: func(t *testing.T, _ map[string][]byte) (storage.StoreProvider, func() error) {
+				return testStorage{
+					hasItem: func(id storage.ArtifactID) (bool, error) {
+						return false, nil
+					},
+					createWorkspace: func(id storage.ArtifactID, entries map[string][]byte) (*storage.Workspace, error) {
+						return nil, &os.PathError{Err: syscall.ENOSPC, Op: "write"}
+					},
+					clean: func(_ context.Context) error {
+						t.Log("cleanup called")
+						return nil
+					},
+				}, nil
+			},
+		},
 	}
 
 	for n, c := range cases {
@@ -249,7 +284,9 @@ func TestBuildService_Build(t *testing.T) {
 				}()
 			}
 
-			bs := NewBuildService(zaptest.NewLogger(t), BuildEnvironmentConfig{}, store)
+			bs := NewBuildService(zaptest.NewLogger(t), BuildEnvironmentConfig{
+				IncludedEnvironmentVariables: c.env,
+			}, store)
 			got, err := bs.Build(context.TODO(), c.files)
 			if c.wantErr != "" {
 				if c.onErrorCheck != nil {

internal/builder/storage/storage.go

@@ -1,6 +1,7 @@
 package storage
 
 import (
+	"context"
 	"errors"
 	"io"
 )
@@ -38,4 +39,7 @@ type StoreProvider interface {
 
 	// CreateWorkspace creates workspace entry in storage
 	CreateWorkspace(id ArtifactID, files map[string][]byte) (*Workspace, error)
+
+	// Clean truncates storage contents
+	Clean(ctx context.Context) error
 }