Remove eds locker and pass context

eds.go

@@ -1,14 +1,16 @@
 package eds
 
 import (
+	"context"
+
 	"github.com/aserto-dev/go-edge-ds/pkg/directory"
 	"github.com/rs/zerolog"
 )
 
-func New(config *directory.Config, logger *zerolog.Logger) (*directory.Directory, error) {
+func New(ctx context.Context, config *directory.Config, logger *zerolog.Logger) (*directory.Directory, error) {
 	newLogger := logger.With().Str("component", "edge-ds").Logger()
 
-	ds, err := directory.New(config, &newLogger)
+	ds, err := directory.New(ctx, config, &newLogger)
 	if err != nil {
 		return nil, err
 	}

pkg/directory/directory.go

@@ -1,6 +1,7 @@
 package directory
 
 import (
+	"context"
 	"errors"
 	"time"
 
@@ -53,7 +54,7 @@ type Directory struct {
 	writer3   dsw3.WriterServer
 }
 
-func New(config *Config, logger *zerolog.Logger) (*Directory, error) {
+func New(ctx context.Context, config *Config, logger *zerolog.Logger) (*Directory, error) {
 	newLogger := logger.With().Str("component", "directory").Logger()
 
 	cfg := bdb.Config{

pkg/server/test-edge-ds.go

@@ -50,7 +50,7 @@ func NewTestEdgeServer(ctx context.Context, logger *zerolog.Logger, cfg *directo
 
 	edgeDSLogger := logger.With().Str("component", "api.edge-directory").Logger()
 
-	edgeDirServer, err := eds.New(cfg, &edgeDSLogger)
+	edgeDirServer, err := eds.New(context.Background(), cfg, &edgeDSLogger)
 	if err != nil {
 		logger.Error().Err(err).Msg("failed to start edge directory server")
 	}