wip: list

internal/manager/error.go

@@ -0,0 +1,7 @@
+package manager
+
+import "errors"
+
+var (
+	ErrNotFound = errors.New("not found")
+)

internal/manager/manager.go

@@ -5,14 +5,71 @@ import (
 	"github.com/loveuer/nf-disk/internal/model"
 	"github.com/loveuer/nf-disk/internal/s3"
 	"github.com/loveuer/nf/nft/log"
+	"sync"
+)
+
+type client struct {
+	conn   *model.Connection
+	client *s3.Client
+}
+
+type manager struct {
+	sync.Mutex
+	clients map[uint64]*client
+}
+
+var (
+	Manager *manager
 )
 
 func Init(ctx context.Context) error {
-	return nil
-}
-
-func Register(m *model.Connection, c *s3.Client) error {
-	log.Debug("manager: register connection-client: id = %d, name = %s", m.Id, m.Name)
+	Manager = &manager{
+		clients: make(map[uint64]*client),
+	}
 
 	return nil
 }
+
+func (m *manager) Register(c *model.Connection, s *s3.Client) error {
+	log.Debug("manager: register connection-client: id = %d, name = %s", c.Id, c.Name)
+
+	Manager.Lock()
+	defer Manager.Unlock()
+	Manager.clients[c.Id] = &client{conn: c, client: s}
+
+	return nil
+}
+
+func (m *manager) UnRegister(id uint64) error {
+	Manager.Lock()
+	defer Manager.Unlock()
+	c, ok := m.clients[id]
+	if !ok {
+		return ErrNotFound
+	}
+
+	log.Debug("manager: register connection-client: id = %d, name = %s", c.conn, c.conn.Name)
+
+	delete(m.clients, id)
+
+	return nil
+}
+
+func (m *manager) Map(fn func(*model.Connection, *s3.Client) error) error {
+	for _, item := range m.clients {
+		if err := fn(item.conn, item.client); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (m *manager) Use(id uint64) (*model.Connection, *s3.Client, error) {
+	c, ok := m.clients[id]
+	if !ok {
+		return nil, nil, ErrNotFound
+	}
+
+	return c.conn, c.client, nil
+}