chore: refactor storage layer

cmd/resticui/resticui.go

@@ -13,7 +13,7 @@ import (
 
 	"github.com/garethgeorge/resticui/internal/api"
 	"github.com/garethgeorge/resticui/internal/config"
-	"github.com/garethgeorge/resticui/internal/oplog"
+	"github.com/garethgeorge/resticui/internal/database/oplog"
 	"github.com/garethgeorge/resticui/internal/orchestrator"
 	static "github.com/garethgeorge/resticui/webui"
 	"github.com/mattn/go-colorable"

internal/api/server.go

@@ -10,7 +10,7 @@ import (
 	"github.com/garethgeorge/resticui/gen/go/types"
 	v1 "github.com/garethgeorge/resticui/gen/go/v1"
 	"github.com/garethgeorge/resticui/internal/config"
-	"github.com/garethgeorge/resticui/internal/oplog"
+	"github.com/garethgeorge/resticui/internal/database/oplog"
 	"github.com/garethgeorge/resticui/internal/orchestrator"
 	"github.com/garethgeorge/resticui/pkg/restic"
 	"go.uber.org/zap"

internal/database/indexutil/indexutil.go

@@ -0,0 +1,55 @@
+package indexutil
+
+import (
+	"bytes"
+
+	"github.com/garethgeorge/resticui/internal/database/serializationutil"
+	bolt "go.etcd.io/bbolt"
+)
+
+func IndexByteValue(b *bolt.Bucket, value []byte, recordId int64) error {
+	key := serializationutil.BytesToKey(value)
+	key = append(key, serializationutil.Itob(recordId)...)
+	return b.Put(key, []byte{})
+}
+
+func IndexSearchByteValue(b *bolt.Bucket, value []byte) *IndexSearchIterator {
+	return newSearchIterator(b, serializationutil.BytesToKey(value))
+}
+
+func IndexSearchIntValue(b *bolt.Bucket, value int64) *IndexSearchIterator {
+	return newSearchIterator(b, serializationutil.Itob(value))
+}
+
+type IndexSearchIterator struct {
+	c *bolt.Cursor
+	k []byte
+	prefix []byte
+}
+
+func newSearchIterator(b *bolt.Bucket, prefix []byte) *IndexSearchIterator {
+	c := b.Cursor()
+	k, _ := c.Seek(prefix)
+	return &IndexSearchIterator{
+		c: c,
+		k: k,
+		prefix: prefix,
+	}
+}
+
+func (i *IndexSearchIterator) Next() (int64, bool) {
+	if i.k == nil || !bytes.HasPrefix(i.k, i.prefix) {
+		return 0, false
+	}
+	id := serializationutil.Btoi(i.k[len(i.prefix):])
+	i.k, _ = i.c.Next()
+	return id, true
+}
+
+func (i *IndexSearchIterator) ToSlice() []int64 {
+	var ids []int64
+	for id, ok := i.Next(); ok; id, ok = i.Next() {
+		ids = append(ids, id)
+	}
+	return ids
+}