feat(savepoint): create and track for getting savepoint location

This commit is contained in:
2024-11-30 02:11:30 +03:30
parent 3b0aff5688
commit 8f4cb093b1
7 changed files with 66 additions and 22 deletions

View File

@@ -1,7 +1,10 @@
package managed_job
import (
"strings"
"gitea.com/logicamp/lc"
api "github.com/logi-camp/go-flink-client"
"go.uber.org/zap"
)
@@ -16,5 +19,29 @@ func (job ManagedJob) createSavepoint() error {
return err
}
lc.Logger.Debug("[managed-job] [savepoint]", zap.Any("savepoint-resp", resp))
job.setSavepointTriggerId(resp.RequestID)
return nil
}
func (job ManagedJob) trackSavepoint() error {
if job.state.JobId == nil {
lc.Logger.Debug("[managed-job] [savepoint] no job id")
return ErrNoJobId
}
if job.state.SavepointTriggerId == nil {
lc.Logger.Debug("[managed-job] [savepoint] no job id")
return ErrNoSavepointTriggerId
}
resp, err := job.client.TrackSavepoint(*job.state.JobId, *job.state.SavepointTriggerId)
lc.Logger.Debug("[managed-job] [savepoint] track savepoint", zap.Any("resp", resp), zap.Error(err))
if err != nil {
if strings.IndexAny(err.Error(), "http status not 2xx: 404") == 0 {
job.removeSavepointTriggerId()
}
}
if resp.Status.Id == api.SavepointStatusInCompleted {
job.setSavepointLocation(resp.Operation.Location)
}
return nil
}