feat: change db to buntdb

internal/managed_job/state.go

@@ -6,21 +6,18 @@ import (
 	"time"
 
 	"github.com/dgraph-io/badger/v4"
+	"github.com/tidwall/buntdb"
 )
 
 // 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 {
+		func(tx *buntdb.Tx) error {
+			if val, err := tx.Get(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)
-				})
+			} else {
+				return json.Unmarshal([]byte(val), job.state)
 			}
-			return nil
 		})
 	if errors.Is(err, badger.ErrKeyNotFound) {
 		err = nil
@@ -32,12 +29,12 @@ 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)
+	job.db.Update(func(tx *buntdb.Tx) error {
+		_, _, err := tx.Set(job.def.Key, string(value), nil)
 		if err != nil {
 			return err
 		}
-		return txn.Commit()
+		return tx.Commit()
 	})
 }