snapshot of https://github.com/centrifugal/centrifugo.git (master, 2026-09-16; blobs > 131072B omitted)

internal/controllers/controllers.go

@@ -0,0 +1,48 @@
+package controllers
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/centrifugal/centrifugo/v6/internal/configtypes"
+
+	"github.com/centrifugal/centrifuge"
+	"github.com/rs/zerolog/log"
+)
+
+// New creates and initializes a controller from the given config.
+// Returns nil, nil when config.Enabled is false.
+func New(node *centrifuge.Node, config configtypes.Controller) (centrifuge.Controller, error) {
+	if !config.Enabled {
+		return nil, nil
+	}
+
+	if config.Type == "postgres" {
+		controller, err := NewPostgresController(node, PostgresControllerConfig{
+			DSN:                      config.Postgres.DSN,
+			TLS:                      config.Postgres.TLS,
+			PoolSize:                 config.Postgres.PoolSize,
+			NumShards:                config.Postgres.NumShards,
+			TablePrefix:              config.Postgres.TablePrefix,
+			PollInterval:             config.Postgres.PollInterval.ToDuration(),
+			UseNotify:                config.Postgres.UseNotify,
+			NotifyDSN:                config.Postgres.NotifyDSN,
+			PartitionRetentionDays:   config.Postgres.PartitionRetentionDays,
+			PartitionLookaheadDays:   config.Postgres.PartitionLookaheadDays,
+			PartitionCleanupInterval: config.Postgres.PartitionCleanupInterval.ToDuration(),
+			BatchSize:                config.Postgres.BatchSize,
+		})
+		if err != nil {
+			return nil, fmt.Errorf("error initializing Postgres controller: %w", err)
+		}
+		if !config.Postgres.SkipSchemaInit {
+			if err := controller.EnsureSchema(context.Background()); err != nil {
+				return nil, fmt.Errorf("error initializing Postgres controller schema: %w", err)
+			}
+		}
+		log.Info().Str("controller_type", config.Type).Msg("controller is enabled")
+		return controller, nil
+	}
+
+	return nil, fmt.Errorf("unknown controller type: %s", config.Type)
+}