types: add connector manager

Author: c9s

pkg/types/connectivity.go

@@ -1,9 +1,15 @@
 package types
 
 import (
+	"context"
+	"fmt"
 	"sync"
 )
 
+type Connector interface {
+	Connect(ctx context.Context) error
+}
+
 func allTrue(bools []bool) bool {
 	for _, b := range bools {
 		if !b {
@@ -125,3 +131,27 @@ func (c *Connectivity) Bind(stream Stream) {
 	stream.OnAuth(c.setAuthed)
 	c.stream = stream
 }
+
+type ConnectorManager struct {
+	connectors map[Connector]Connector
+}
+
+func NewConnectorManager() *ConnectorManager {
+	return &ConnectorManager{
+		connectors: make(map[Connector]Connector),
+	}
+}
+
+func (cm *ConnectorManager) Add(connector Connector) {
+	cm.connectors[connector] = connector
+}
+
+func (cm *ConnectorManager) Connect(ctx context.Context) error {
+	for _, connector := range cm.connectors {
+		if err := connector.Connect(ctx); err != nil {
+			return fmt.Errorf("connector %T connect error: %w", connector, err)
+		}
+	}
+
+	return nil
+}