new kv, ctx in nfs handler

src/export/nfs/kvhandler.go

@@ -2,6 +2,7 @@ package nfs
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"path"
 	"strings"
@@ -10,6 +11,7 @@ import (
 
 	"git.kmsign.ru/royalcat/tstor/pkg/go-nfs"
 	"git.kmsign.ru/royalcat/tstor/src/config"
+	"git.kmsign.ru/royalcat/tstor/src/log"
 	"go.opentelemetry.io/otel"
 	"go.opentelemetry.io/otel/metric"
 
@@ -50,6 +52,7 @@ var kvhandlerMeter = otel.Meter("git.kmsign.ru/royalcat/tstor/src/export/nfs.kvh
 func NewKvHandler(h nfs.Handler, fs nfs.Filesystem, config config.NFS) (nfs.Handler, error) {
 	opts := kvbadger.DefaultOptions(path.Join(config.CachePath, "handlers"))
 	opts.DefaultTTL = time.Hour
+	opts.BadgerOptions.Logger = log.BadgerLogger("nfs", "kvhandler")
 
 	activeHandles, err := kvbadger.NewBagerKVBinaryKey[uuid.UUID, handle](opts)
 	if err != nil {
@@ -98,8 +101,7 @@ type CachingHandler struct {
 // ToHandle takes a file and represents it with an opaque handle to reference it.
 // In stateless nfs (when it's serving a unix fs) this can be the device + inode
 // but we can generalize with a stateful local cache of handed out IDs.
-func (c *CachingHandler) ToHandle(_ nfs.Filesystem, path []string) []byte {
-	ctx := context.Background()
+func (c *CachingHandler) ToHandle(ctx context.Context, _ nfs.Filesystem, path []string) []byte {
 	var id uuid.UUID
 
 	cacheKey := handle(path).String()
@@ -123,31 +125,29 @@ func (c *CachingHandler) ToHandle(_ nfs.Filesystem, path []string) []byte {
 }
 
 // FromHandle converts from an opaque handle to the file it represents
-func (c *CachingHandler) FromHandle(fh []byte) (nfs.Filesystem, []string, error) {
+func (c *CachingHandler) FromHandle(ctx context.Context, fh []byte) (nfs.Filesystem, []string, error) {
 	c.mu.Lock()
 	defer c.mu.Unlock()
 
-	ctx := context.Background()
-
 	id, err := uuid.FromBytes(fh)
 	if err != nil {
 		return nil, nil, err
 	}
 
-	paths, found, err := c.activeHandles.Get(ctx, id)
+	paths, err := c.activeHandles.Get(ctx, id)
 	if err != nil {
+		if errors.Is(err, kv.ErrKeyNotFound) {
+			return nil, nil, &nfs.NFSStatusError{NFSStatus: nfs.NFSStatusStale}
+		}
+
 		return nil, nil, fmt.Errorf("kv error: %w", err)
 	}
 
-	if found {
-		return c.fs, paths, nil
-	}
+	return c.fs, paths, nil
 
-	return nil, nil, &nfs.NFSStatusError{NFSStatus: nfs.NFSStatusStale}
 }
 
-func (c *CachingHandler) InvalidateHandle(fs nfs.Filesystem, handle []byte) error {
-	ctx := context.Background()
+func (c *CachingHandler) InvalidateHandle(ctx context.Context, fs nfs.Filesystem, handle []byte) error {
 	//Remove from cache
 	id, err := uuid.FromBytes(handle)
 	if err != nil {