feat: add start and trigger savepoint routes

internal/crd/finalizer.go

@@ -17,6 +17,8 @@ func (crd Crd) manageFinalizer(jobEventObservable rxgo.Observable) {
 
 	finalizerName := "flink-operator.logicamp.tech/finalizer"
 	for j := range jobEventObservable.Observe() {
+		go func() {
+
 			jobEvent := j.V.(*FlinkJobCrdEvent)
 
 			if jobEvent.Job.GetDeletionTimestamp() != nil {
@@ -51,6 +53,8 @@ func (crd Crd) manageFinalizer(jobEventObservable rxgo.Observable) {
 					return
 				}
 			}
+
+		}()
 	}
 }
 

internal/managed_job/savepoint.go

@@ -80,3 +80,12 @@ func (job ManagedJob) trackSavepoint() error {
 
 	return nil
 }
+
+func (job ManagedJob) TriggerSavepoint() error {
+	err := job.createSavepoint()
+	if err != nil {
+		return err
+	}
+	err = job.trackSavepoint()
+	return err
+}

internal/rest/controller/crd.go

@@ -1,6 +1,7 @@
 package controller
 
 import (
+	"archive/zip"
 	"bufio"
 	"context"
 	"errors"
@@ -12,7 +13,6 @@ import (
 	"fmt"
 	"io"
 	"os"
-	"os/exec"
 	"path/filepath"
 
 	"github.com/danielgtaylor/huma/v2"
@@ -62,6 +62,18 @@ func StopJob(ctx context.Context, req *StopJobReq) (*StopJobResp, error) {
 }
 
 func StartJob(ctx context.Context, req *StopJobReq) (*StopJobResp, error) {
+	mgr := managed_job.GetManager()
+	job := mgr.GetJob(k8sTypes.UID(req.JobUId))
+	err := job.Run(false)
+	if err != nil {
+		return nil, err
+	}
+	return &StopJobResp{Body: StopJobRespBody{
+		Success: true,
+	}}, nil
+}
+
+func ResumeJob(ctx context.Context, req *StopJobReq) (*StopJobResp, error) {
 	mgr := managed_job.GetManager()
 	job := mgr.GetJob(k8sTypes.UID(req.JobUId))
 	err := job.Run(true)
@@ -91,6 +103,30 @@ func PauseJob(ctx context.Context, req *StopJobReq) (*StopJobResp, error) {
 	}}, nil
 }
 
+type JobTriggerSavepointReq struct {
+	JobUId string `path:"uid"`
+}
+
+type JobTriggerSavepointRespBody struct {
+	Success bool `json:"success"`
+}
+
+type JobTriggerSavepointResp struct {
+	Body JobTriggerSavepointRespBody
+}
+
+func TriggerSavepoint(ctx context.Context, req *JobTriggerSavepointReq) (*JobTriggerSavepointResp, error) {
+	mgr := managed_job.GetManager()
+	job := mgr.GetJob(k8sTypes.UID(req.JobUId))
+	err := job.TriggerSavepoint()
+	if err != nil {
+		return nil, err
+	}
+	return &JobTriggerSavepointResp{Body: JobTriggerSavepointRespBody{
+		Success: true,
+	}}, nil
+}
+
 func DownloadSavepoint(ctx context.Context, req *types.SavepointDownloadReq) (*huma.StreamResponse, error) {
 	folderPath := req.SavepointPath // Change this to your folder path
 
@@ -102,15 +138,66 @@ func DownloadSavepoint(ctx context.Context, req *types.SavepointDownloadReq) (*h
 		zap.String("folderPath", folderPath),
 	)
 
-	// Run the zip command
-	cmd := exec.Command("zip", "-r", zipFilePath, folderPath)
-
-	// Capture any output or errors
-	output, err := cmd.CombinedOutput()
+	// Create the zip file
+	zipFile, err := os.Create(zipFilePath)
 	if err != nil {
-		return nil, fmt.Errorf("failed to run zip command: %v\nOutput: %s", err, string(output))
+		fmt.Println("Error creating zip file:", err)
+		return nil, nil
 	}
-	pkg.Logger.Debug("[controller] [savepoint]", zap.Any("zip command output", string(output)))
+	defer zipFile.Close()
+
+	// Create a new zip writer
+	zipWriter := zip.NewWriter(zipFile)
+	defer zipWriter.Close()
+
+	// Walk through the source directory and add files to the zip
+	err = filepath.Walk(folderPath, func(filePath string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+
+		// Create a new file header
+		header, err := zip.FileInfoHeader(info)
+		if err != nil {
+			return err
+		}
+
+		// Set the header name to the relative path
+		header.Name, err = filepath.Rel(folderPath, filePath)
+		if err != nil {
+			return err
+		}
+
+		// If it's a directory, add a trailing slash
+		if info.IsDir() {
+			header.Name += "/"
+		} else {
+			// Set the compression method
+			header.Method = zip.Deflate
+		}
+
+		// Create a new writer for the file
+		writer, err := zipWriter.CreateHeader(header)
+		if err != nil {
+			return err
+		}
+
+		// If it's a directory, we're done
+		if info.IsDir() {
+			return nil
+		}
+
+		// Open the file to be zipped
+		file, err := os.Open(filePath)
+		if err != nil {
+			return err
+		}
+		defer file.Close()
+
+		// Copy the file content to the zip writer
+		_, err = io.Copy(writer, file)
+		return err
+	})
 
 	// Open the zip file for reading
 	zipFileReader, err := os.Open(zipFilePath)

internal/rest/router.go

@@ -35,6 +35,15 @@ func initRouter(api huma.API) {
 		Tags:        []string{"Job"},
 	}, controller.StartJob)
 
+	huma.Register(api, huma.Operation{
+		OperationID: "resume-job",
+		Method:      http.MethodPost,
+		Path:        "/jobs/{uid}/resume",
+		Summary:     "Resume Job",
+		Description: "Resume Flink Job",
+		Tags:        []string{"Job"},
+	}, controller.ResumeJob)
+
 	huma.Register(api, huma.Operation{
 		OperationID: "remove-jar",
 		Method:      http.MethodPost,
@@ -62,4 +71,12 @@ func initRouter(api huma.API) {
 		Tags:        []string{"Savepoint"},
 	}, controller.DownloadSavepoint)
 
+	huma.Register(api, huma.Operation{
+		OperationID: "trigger-savepoint",
+		Method:      http.MethodPost,
+		Path:        "/jobs/{uid}/trigger-savepoint",
+		Summary:     "Trigger Savepoint",
+		Description: "Trigger Savepoint",
+		Tags:        []string{"Savepoint"},
+	}, controller.TriggerSavepoint)
 }