More playground architecture changes

Author: 0x19

playground/entrypoint.go

@@ -10,6 +10,7 @@ import (
 	"github.com/unpackdev/fdb/pkg/rbac"
 	"github.com/unpackdev/fdb/pkg/shutdown"
 	"github.com/unpackdev/fdb/pkg/types"
+	"github.com/unpackdev/fdb/playground/registry"
 	"github.com/unpackdev/fdb/playground/strategies"
 	"github.com/unpackdev/fdb/playground/suite"
 	"github.com/urfave/cli/v2"
@@ -73,7 +74,7 @@ func Run(cliCtx *cli.Context, cfg Config) error {
 	shutdownMgr.Start()
 
 	// Create the strategy registry
-	registry := NewRegistry(testLogger)
+	registry := registry.New(testLogger)
 
 	// Register all available strategies
 	registry.RegisterAll()
@@ -171,6 +172,6 @@ func Run(cliCtx *cli.Context, cfg Config) error {
 		testLogger.Error("Shutdown completed with errors", "error", err.Error())
 	}
 
-	testLogger.Info("Playground shutdown complete. Cause: %s", "cause", shutdownMgr.ShutdownCause())
+	testLogger.Info("Playground shutdown complete.", "cause", shutdownMgr.ShutdownCause())
 	return nil
 }

playground/registry.go playground/registry/registry.goplayground/registry.go renamed to playground/registry/registry.go

@@ -1,4 +1,4 @@
-package playground
+package registry
 
 import (
 	"context"
@@ -23,8 +23,8 @@ type Registry struct {
 	logger     logger.Logger
 }
 
-// NewRegistry creates a new strategy registry
-func NewRegistry(logger logger.Logger) *Registry {
+// New creates a new strategy registry
+func New(logger logger.Logger) *Registry {
 	return &Registry{
 		logger:     logger,
 		strategies: make(map[string]strategyEntry),
@@ -107,10 +107,10 @@ func (r *Registry) CreateStrategy(name string, logger logger.Logger, nodes suite
 	return entry.Strategy.CreateFn()(logger, nodes, mergedArgs)
 }
 
-// RegisterAll registers all available strategies from the strategies package
+// RegisterAll registers all available strategies from the registry package
 func (r *Registry) RegisterAll() {
 	// Register all strategies from the centralized map
-	for name, factory := range strategies.AvailableStrategies {
+	for name, factory := range AvailableStrategies {
 		// Create a prototype instance using the factory
 		strategy := factory(r.logger)
 
@@ -144,7 +144,7 @@ func (r *Registry) RunStrategy(ctx context.Context, name string, logger logger.L
 	case <-ctx.Done():
 		// External cancellation (shutdown manager)
 		logger.Info("Strategy stopping due to context cancellation")
-		
+
 	case <-completionCh:
 		// Strategy has completed naturally
 		logger.Info("Strategy completed its work successfully")

playground/registry/strategies.go

@@ -0,0 +1,17 @@
+package registry
+
+import (
+	"github.com/unpackdev/fdb/pkg/logger"
+	"github.com/unpackdev/fdb/playground/strategies"
+)
+
+// RegistryFn is a function that creates a strategy prototype
+type RegistryFn func(logger logger.Logger) strategies.Strategy
+
+// AvailableStrategies is a map of all available strategies and their factory functions
+// Add new strategies to this map to make them available in the playground
+var AvailableStrategies = map[string]RegistryFn{
+	"write": func(logger logger.Logger) strategies.Strategy {
+		return strategies.NewWriteStrategy(logger, nil) // Nodes will be provided later
+	},
+}

playground/strategies/strategies.go

@@ -96,14 +96,3 @@ func ParseCliArgs(cliCtx *cli.Context, info Info) map[string]any {
 
 	return args
 }
-
-// StrategyFactory is a function that creates a strategy prototype
-type StrategyFactory func(logger logger.Logger) Strategy
-
-// AvailableStrategies is a map of all available strategies and their factory functions
-// Add new strategies to this map to make them available in the playground
-var AvailableStrategies = map[string]StrategyFactory{
-	"write": func(logger logger.Logger) Strategy {
-		return NewWriteStrategy(logger, nil) // Nodes will be provided later
-	},
-}