diff --git a/hub/internal/service/connection_manager/store/manager.go b/hub/internal/service/connection_manager/store/manager.go new file mode 100644 index 0000000..b690f36 --- /dev/null +++ b/hub/internal/service/connection_manager/store/manager.go @@ -0,0 +1,42 @@ +package store + +import ( + "sync" + + "github.com/lorsanstand/HomeOps-Hub/hub/internal/service/connection_manager" +) + +type AgentConnStore struct { + mutex sync.RWMutex + store map[string]*connection_manager.AgentConnection +} + +func NewAgentConnStore() *AgentConnStore { + return &AgentConnStore{store: make(map[string]*connection_manager.AgentConnection)} +} + +func (a *AgentConnStore) Get(agentID string) *connection_manager.AgentConnection { + a.mutex.RLock() + defer a.mutex.RUnlock() + return a.store[agentID] +} + +func (a *AgentConnStore) Add(agentConn *connection_manager.AgentConnection) { + a.mutex.Lock() + defer a.mutex.Unlock() + a.store[agentConn.AgentID] = agentConn +} + +func (a *AgentConnStore) Delete(agentID string) { + a.mutex.Lock() + defer a.mutex.Unlock() + delete(a.store, agentID) +} + +func (a *AgentConnStore) Pop(agentID string) *connection_manager.AgentConnection { + a.mutex.Lock() + defer a.mutex.Unlock() + agent := a.store[agentID] + delete(a.store, agentID) + return agent +}