feat: implement badgerdb database with basic CRUD operations

Signed-off-by: tbxark <[email protected]>

Author: TBXark

cache/badgerdb/database.go

@@ -0,0 +1,110 @@
+package badgerdb
+
+import (
+	"context"
+	"github.com/dgraph-io/badger/v4"
+	"time"
+)
+
+type Config struct {
+	Path string `json:"path"`
+}
+
+type Database struct {
+	db *badger.DB
+}
+
+func NewDatabase(config *Config) (*Database, error) {
+	db, err := badger.Open(badger.DefaultOptions("/tmp/badger"))
+	if err != nil {
+		return nil, err
+	}
+	return &Database{
+		db: db,
+	}, nil
+}
+
+func (d *Database) Set(ctx context.Context, key string, val []byte, expiration time.Duration) error {
+	return d.db.Update(func(txn *badger.Txn) error {
+		return txn.Set([]byte(key), val)
+	})
+}
+
+func (d *Database) Get(ctx context.Context, key string) (*[]byte, error) {
+	var val []byte
+	err := d.db.View(func(txn *badger.Txn) error {
+		item, err := txn.Get([]byte(key))
+		if err != nil {
+			return err
+		}
+		val, err = item.ValueCopy(nil)
+		if err != nil {
+			return err
+		}
+		return nil
+	})
+	if err != nil {
+		return nil, err
+	}
+	return &val, nil
+}
+
+func (d *Database) Del(ctx context.Context, key string) error {
+	return d.db.Update(func(txn *badger.Txn) error {
+		return txn.Delete([]byte(key))
+	})
+}
+
+func (d *Database) MultiSet(ctx context.Context, valMap map[string][]byte, expiration time.Duration) error {
+	return d.db.Update(func(txn *badger.Txn) error {
+		for k, v := range valMap {
+			err := txn.SetEntry(badger.NewEntry([]byte(k), v).WithTTL(expiration))
+			if err != nil {
+				return err
+			}
+		}
+		return nil
+	})
+}
+
+func (d *Database) MultiGet(ctx context.Context, keys []string) (map[string][]byte, error) {
+	res := make(map[string][]byte)
+	err := d.db.View(func(txn *badger.Txn) error {
+		for _, key := range keys {
+			item, err := txn.Get([]byte(key))
+			if err != nil {
+				return err
+			}
+			val, err := item.ValueCopy(nil)
+			if err != nil {
+				return err
+			}
+			res[key] = val
+		}
+		return nil
+	})
+	if err != nil {
+		return nil, err
+	}
+	return res, nil
+}
+
+func (d *Database) MultiDel(ctx context.Context, keys []string) error {
+	return d.db.Update(func(txn *badger.Txn) error {
+		for _, key := range keys {
+			err := d.Del(ctx, key)
+			if err != nil {
+				return err
+			}
+		}
+		return nil
+	})
+}
+
+func (d *Database) DelAll(ctx context.Context) error {
+	return d.db.DropAll()
+}
+
+func (d *Database) Close() error {
+	return d.db.Close()
+}

go.mod

@@ -5,6 +5,7 @@ go 1.23.2
 require (
 	entgo.io/ent v0.14.2
 	github.com/TBXark/jsoncompressor v0.0.0-20250220090015-07c767467512
+	github.com/dgraph-io/badger/v4 v4.5.1
 	github.com/gin-contrib/cors v1.7.3
 	github.com/gin-contrib/pprof v1.5.2
 	github.com/gin-contrib/zap v1.1.4
@@ -25,6 +26,7 @@ require (
 	golang.org/x/crypto v0.33.0
 	golang.org/x/exp v0.0.0-20250218142911-aa4b98e5adaa
 	golang.org/x/sync v0.11.0
+	golang.org/x/text v0.22.0
 	golang.org/x/time v0.10.0
 	gopkg.in/natefinch/lumberjack.v2 v2.2.1
 	modernc.org/sqlite v1.35.0
@@ -40,6 +42,7 @@ require (
 	github.com/cespare/xxhash/v2 v2.3.0 // indirect
 	github.com/cloudwego/base64x v0.1.5 // indirect
 	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
+	github.com/dgraph-io/ristretto/v2 v2.1.0 // indirect
 	github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
 	github.com/dustin/go-humanize v1.0.1 // indirect
 	github.com/gabriel-vasile/mimetype v1.4.8 // indirect
@@ -54,10 +57,11 @@ require (
 	github.com/go-playground/locales v0.14.1 // indirect
 	github.com/go-playground/universal-translator v0.18.1 // indirect
 	github.com/go-playground/validator/v10 v10.25.0 // indirect
-	github.com/go-redis/redis/v8 v8.11.5 // indirect
 	github.com/goccy/go-json v0.10.5 // indirect
 	github.com/gofrs/flock v0.12.1 // indirect
 	github.com/golang-jwt/jwt/v4 v4.5.1 // indirect
+	github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect
+	github.com/google/flatbuffers v24.12.23+incompatible // indirect
 	github.com/google/uuid v1.6.0 // indirect
 	github.com/josharian/intern v1.0.0 // indirect
 	github.com/json-iterator/go v1.1.12 // indirect
@@ -72,16 +76,17 @@ require (
 	github.com/modern-go/reflect2 v1.0.2 // indirect
 	github.com/ncruces/go-strftime v0.1.9 // indirect
 	github.com/pelletier/go-toml/v2 v2.2.3 // indirect
+	github.com/pkg/errors v0.9.1 // indirect
 	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
 	github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
 	github.com/rs/xid v1.6.0 // indirect
 	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
 	github.com/ugorji/go/codec v1.2.12 // indirect
+	go.opencensus.io v0.24.0 // indirect
 	go.uber.org/multierr v1.11.0 // indirect
 	golang.org/x/arch v0.14.0 // indirect
 	golang.org/x/net v0.35.0 // indirect
 	golang.org/x/sys v0.30.0 // indirect
-	golang.org/x/text v0.22.0 // indirect
 	golang.org/x/tools v0.30.0 // indirect
 	google.golang.org/protobuf v1.36.5 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect