bagas
f8a6f0bafe
- Implement thread-safe session registry with sync.RWMutex - Add Registry interface for session management operations - Support Get, Register, Update, and Remove session operations - Enable dynamic slug updates for existing sessions
67 lines
1.2 KiB
Go
67 lines
1.2 KiB
Go
package session
|
|
|
|
import "sync"
|
|
|
|
type Registry interface {
|
|
Get(slug string) (session *SSHSession, exist bool)
|
|
Update(oldSlug, newSlug string) (success bool)
|
|
Register(slug string, session *SSHSession) (success bool)
|
|
Remove(slug string)
|
|
}
|
|
type registry struct {
|
|
mu sync.RWMutex
|
|
clients map[string]*SSHSession
|
|
}
|
|
|
|
func NewRegistry() Registry {
|
|
return ®istry{
|
|
clients: make(map[string]*SSHSession),
|
|
}
|
|
}
|
|
|
|
func (r *registry) Get(slug string) (session *SSHSession, exist bool) {
|
|
r.mu.RLock()
|
|
defer r.mu.RUnlock()
|
|
|
|
session, exist = r.clients[slug]
|
|
return
|
|
}
|
|
|
|
func (r *registry) Update(oldSlug, newSlug string) (success bool) {
|
|
r.mu.Lock()
|
|
defer r.mu.Unlock()
|
|
|
|
if _, exists := r.clients[newSlug]; exists && newSlug != oldSlug {
|
|
return false
|
|
}
|
|
|
|
client, ok := r.clients[oldSlug]
|
|
if !ok {
|
|
return false
|
|
}
|
|
|
|
delete(r.clients, oldSlug)
|
|
client.slugManager.Set(newSlug)
|
|
r.clients[newSlug] = client
|
|
return true
|
|
}
|
|
|
|
func (r *registry) Register(slug string, session *SSHSession) (success bool) {
|
|
r.mu.Lock()
|
|
defer r.mu.Unlock()
|
|
|
|
if _, exists := r.clients[slug]; exists {
|
|
return false
|
|
}
|
|
|
|
r.clients[slug] = session
|
|
return true
|
|
}
|
|
|
|
func (r *registry) Remove(slug string) {
|
|
r.mu.Lock()
|
|
defer r.mu.Unlock()
|
|
|
|
delete(r.clients, slug)
|
|
}
|