f-25: code: add db api

Signed-off-by: GnomeZworc <nicolas.boufidjeline@g3e.fr>

pkg/db/kv/admin_server.go

@@ -0,0 +1,52 @@
+package kv
+
+import (
+	"fmt"
+	"log/slog"
+	"net/http"
+	"sort"
+
+	"github.com/dgraph-io/badger/v4"
+)
+
+type AdminServer struct {
+	db     *badger.DB
+	logger *slog.Logger
+}
+
+func NewAdminServer(db *badger.DB, logger *slog.Logger) *AdminServer {
+	return &AdminServer{db: db, logger: logger}
+}
+
+func (s *AdminServer) Start(address string) {
+	mux := http.NewServeMux()
+	mux.HandleFunc("/db", s.dbHandler)
+	s.logger.Info("admin server listening", "address", address)
+	if err := http.ListenAndServe(address, mux); err != nil {
+		s.logger.Error("admin server stopped", "error", err)
+	}
+}
+
+func (s *AdminServer) dbHandler(w http.ResponseWriter, r *http.Request) {
+	if r.Method != http.MethodGet {
+		http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
+		return
+	}
+
+	entries, err := ListByPrefix(s.db, r.URL.Query().Get("prefix"))
+	if err != nil {
+		http.Error(w, "db error: "+err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	keys := make([]string, 0, len(entries))
+	for k := range entries {
+		keys = append(keys, k)
+	}
+	sort.Strings(keys)
+
+	w.Header().Set("Content-Type", "text/plain; charset=utf-8")
+	for _, k := range keys {
+		fmt.Fprintf(w, "%s=%s\n", k, entries[k])
+	}
+}

pkg/db/kv/init.go

@@ -17,7 +17,8 @@ func InitDB(conf Config, readonly bool) *badger.DB {
 	opts.NumLevelZeroTablesStall = 2
 	db, err := badger.Open(opts)
 	if err != nil {
-		log.Fatalf("kv.InitDB (readonly=%v, path=%s): %v", readonly, conf.Path, err)
+		log.Printf("kv.InitDB (readonly=%v, path=%s): %v", readonly, conf.Path, err)
+		panic(err)
 	}
 	return db
 }