init

internal/natsobj/nats_store.go

@@ -0,0 +1,69 @@
+package natsobj
+
+import (
+	"github.com/nats-io/nats.go"
+)
+
+type Storage struct {
+	store nats.ObjectStore
+	conn  *nats.Conn
+}
+
+func Open(bucket, url string) (*Storage, error) {
+	nc, err := nats.Connect(url)
+	if err != nil {
+		return nil, err
+	}
+
+	js, err := nc.JetStream()
+	if err != nil {
+		return nil, err
+	}
+
+	cfg := &nats.ObjectStoreConfig{
+		Bucket:      bucket,
+		Description: "microkv bucket",
+		MaxBytes:    -1,
+		Storage:     nats.FileStorage,
+		Compression: true,
+	}
+
+	store, err := js.CreateObjectStore(cfg)
+	if err != nil {
+		return nil, err
+	}
+
+	st := &Storage{store: store, conn: nc}
+
+	return st, nil
+}
+
+func (n *Storage) Save(key string, data []byte) error {
+	if _, err := n.store.PutBytes(key, data); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (n *Storage) Load(key string) ([]byte, error) {
+	b, err := n.store.GetBytes(key)
+	if err != nil {
+		return nil, err
+	}
+
+	return b, nil
+}
+
+func (n *Storage) Delete(key string) error {
+	if err := n.store.Delete(key); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (n *Storage) Close() error {
+	n.conn.Close()
+
+	return nil
+}