feat: initialize

internal/managed_job/state.go

@@ -0,0 +1,55 @@
+package managed_job
+
+import (
+	"encoding/json"
+	"errors"
+	"time"
+
+	"github.com/dgraph-io/badger/v4"
+)
+
+// get state of job from local db
+func (job *ManagedJob) loadState() {
+	err := job.db.View(
+		func(tx *badger.Txn) error {
+			if val, err := tx.Get([]byte(job.def.Key)); err != nil {
+				return err
+			} else if val != nil {
+				val.Value(func(val []byte) error {
+					job.state = &jobState{}
+					return json.Unmarshal(val, job.state)
+				})
+			}
+			return nil
+		})
+	if errors.Is(err, badger.ErrKeyNotFound) {
+		err = nil
+	}
+}
+
+// save state of job to local db
+func (job *ManagedJob) updateState(state jobState) {
+	job.state = &state
+
+	value, _ := json.Marshal(job.state)
+	job.db.Update(func(txn *badger.Txn) error {
+		err := txn.Set([]byte(job.def.Key), value)
+		if err != nil {
+			return err
+		}
+		return txn.Commit()
+	})
+}
+
+func (job *ManagedJob) setError(errMsg string) {
+	job.state.Error = &errMsg
+	job.state.Status = JobStatusError
+	job.updateState(*job.state)
+}
+
+func (job *ManagedJob) setSavepointId(savepointId string) {
+	job.state.LastSavepointId = &savepointId
+	n := time.Now()
+	job.state.LastSavepointDate = &n
+	job.updateState(*job.state)
+}