fix: purge

Signed-off-by: Luka Brecic <[email protected]>
daytonaio · Dec 23, 2024 · b6bdf4c · b6bdf4c
1 parent b67a139
commit b6bdf4c
Show file tree

Hide file tree

Showing 6 changed files with 130 additions and 0 deletions.
diff --git a/pkg/cmd/purge.go b/pkg/cmd/purge.go
@@ -164,6 +164,24 @@ var purgeCmd = &cobra.Command{
 			return err
 		}
 
+		err = server.Start()
+		if err != nil {
+			return err
+		}
+
+		log.Info("Starting job runner...")
+		jobRunner, err := bootstrap.GetJobRunner(serverConfig, serverConfigDir, internal.Version, telemetryService)
+		if err != nil {
+			return err
+		}
+
+		// TODO: context?
+		err = jobRunner.StartRunner(context.Background())
+		if err != nil {
+			return err
+		}
+		log.Info("Job runner started")
+
 		errs := server.Purge(ctx, forceFlag)
 		if len(errs) > 0 {
 			errMessage := ""

diff --git a/pkg/server/purge.go b/pkg/server/purge.go
@@ -29,6 +29,44 @@ func (s *Server) Purge(ctx context.Context, force bool) []error {
 		}
 	}
 
+	fmt.Println("Deleting all workspaces...")
+
+	workspaces, err := s.WorkspaceService.ListWorkspaces(ctx, services.WorkspaceRetrievalParams{})
+	if err != nil {
+		s.trackPurgeError(ctx, force, err)
+		if !force {
+			return []error{err}
+		}
+	}
+
+	if err == nil {
+		for _, workspace := range workspaces {
+			err := s.WorkspaceService.RemoveWorkspace(ctx, workspace.Id)
+			if err != nil {
+				s.trackPurgeError(ctx, force, err)
+				if !force {
+					return []error{err}
+				} else {
+					fmt.Printf("Failed to delete %s: %v\n", workspace.Name, err)
+				}
+			} else {
+				fmt.Printf("Workspace %s deleted\n", workspace.Name)
+			}
+		}
+	} else {
+		fmt.Printf("Failed to list workspaces: %v\n", err)
+	}
+
+	err = s.WorkspaceService.AwaitEmptyList(ctx, time.Minute)
+	if err != nil {
+		s.trackPurgeError(ctx, force, err)
+		if !force {
+			return []error{err}
+		} else {
+			fmt.Printf("Failed to await empty workspace list: %v\n", err)
+		}
+	}
+
 	fmt.Println("Deleting all targets...")
 
 	targets, err := s.TargetService.ListTargets(ctx, nil, services.TargetRetrievalParams{})
@@ -57,6 +95,26 @@ func (s *Server) Purge(ctx context.Context, force bool) []error {
 		fmt.Printf("Failed to list targets: %v\n", err)
 	}
 
+	err = s.TargetService.AwaitEmptyList(ctx, time.Minute)
+	if err != nil {
+		s.trackPurgeError(ctx, force, err)
+		if !force {
+			return []error{err}
+		} else {
+			fmt.Printf("Failed to await empty target list: %v\n", err)
+		}
+	}
+
+	err = s.TargetService.AwaitEmptyList(ctx, time.Minute)
+	if err != nil {
+		s.trackPurgeError(ctx, force, err)
+		if !force {
+			return []error{err}
+		} else {
+			fmt.Printf("Failed to await empty target list: %v\n", err)
+		}
+	}
+
 	// FIXME: todo
 	// fmt.Println("Purging providers...")
 	// err = s.ProviderManager.Purge()

diff --git a/pkg/server/targets/service.go b/pkg/server/targets/service.go
@@ -5,7 +5,9 @@ package targets
 
 import (
 	"context"
+	"errors"
 	"io"
+	"time"
 
 	"github.com/daytonaio/daytona/pkg/logs"
 	"github.com/daytonaio/daytona/pkg/models"
@@ -88,3 +90,26 @@ func (s *TargetService) UpdateTargetProviderMetadata(ctx context.Context, target
 	tg.ProviderMetadata = &metadata
 	return s.targetStore.Save(ctx, tg)
 }
+
+func (s *TargetService) AwaitEmptyList(ctx context.Context, waitTime time.Duration) error {
+	timeout := time.NewTimer(waitTime)
+	defer timeout.Stop()
+
+	for {
+		select {
+		case <-timeout.C:
+			return errors.New("awaiting empty build list timed out")
+		default:
+			targets, err := s.ListTargets(ctx, nil, services.TargetRetrievalParams{})
+			if err != nil {
+				return err
+			}
+
+			if len(targets) == 0 {
+				return nil
+			}
+
+			time.Sleep(time.Second)
+		}
+	}
+}
diff --git a/pkg/server/workspaces/service.go b/pkg/server/workspaces/service.go
@@ -5,7 +5,9 @@ package workspaces
 
 import (
 	"context"
+	"errors"
 	"io"
+	"time"
 
 	"github.com/daytonaio/daytona/pkg/gitprovider"
 	"github.com/daytonaio/daytona/pkg/logs"
@@ -103,3 +105,26 @@ func (s *WorkspaceService) UpdateWorkspaceProviderMetadata(ctx context.Context,
 	w.ProviderMetadata = &metadata
 	return s.workspaceStore.Save(ctx, w)
 }
+
+func (s *WorkspaceService) AwaitEmptyList(ctx context.Context, waitTime time.Duration) error {
+	timeout := time.NewTimer(waitTime)
+	defer timeout.Stop()
+
+	for {
+		select {
+		case <-timeout.C:
+			return errors.New("awaiting empty build list timed out")
+		default:
+			workspaces, err := s.ListWorkspaces(ctx, services.WorkspaceRetrievalParams{})
+			if err != nil {
+				return err
+			}
+
+			if len(workspaces) == 0 {
+				return nil
+			}
+
+			time.Sleep(time.Second)
+		}
+	}
+}
diff --git a/pkg/services/target.go b/pkg/services/target.go
@@ -7,6 +7,7 @@ import (
 	"context"
 	"errors"
 	"io"
+	"time"
 
 	"github.com/daytonaio/daytona/pkg/models"
 	"github.com/daytonaio/daytona/pkg/stores"
@@ -24,6 +25,7 @@ type ITargetService interface {
 	RemoveTarget(ctx context.Context, targetId string) error
 	ForceRemoveTarget(ctx context.Context, targetId string) error
 	HandleSuccessfulCreation(ctx context.Context, targetId string) error
+	AwaitEmptyList(ctx context.Context, waitTime time.Duration) error
 
 	GetTargetLogReader(ctx context.Context, targetId string) (io.Reader, error)
 	GetTargetLogWriter(ctx context.Context, targetId string) (io.WriteCloser, error)

diff --git a/pkg/services/workspace.go b/pkg/services/workspace.go
@@ -7,6 +7,7 @@ import (
 	"context"
 	"errors"
 	"io"
+	"time"
 
 	"github.com/daytonaio/daytona/pkg/gitprovider"
 	"github.com/daytonaio/daytona/pkg/models"
@@ -21,6 +22,7 @@ type IWorkspaceService interface {
 	RemoveWorkspace(ctx context.Context, workspaceId string) error
 	ForceRemoveWorkspace(ctx context.Context, workspaceId string) error
 	UpdateWorkspaceProviderMetadata(ctx context.Context, workspaceId, metadata string) error
+	AwaitEmptyList(ctx context.Context, waitTime time.Duration) error
 
 	GetWorkspaceLogReader(ctx context.Context, workspaceId string) (io.Reader, error)
 	GetWorkspaceLogWriter(ctx context.Context, workspaceId string) (io.WriteCloser, error)