mirror of
https://github.com/m1k1o/neko.git
synced 2024-07-24 14:40:50 +12:00
add database to Sessions.
This commit is contained in:
67
internal/session/database/manager.go
Normal file
67
internal/session/database/manager.go
Normal file
@ -0,0 +1,67 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"sync"
|
||||
|
||||
"demodesk/neko/internal/types"
|
||||
)
|
||||
|
||||
func New() *MembersDatabaseCtx {
|
||||
return &MembersDatabaseCtx{
|
||||
profiles: make(map[string]types.MemberProfile),
|
||||
mu: sync.Mutex{},
|
||||
}
|
||||
}
|
||||
|
||||
type MembersDatabaseCtx struct {
|
||||
profiles map[string]types.MemberProfile
|
||||
mu sync.Mutex
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Insert(id string, profile types.MemberProfile) error {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
_, ok := manager.profiles[id]
|
||||
if ok {
|
||||
return fmt.Errorf("Member ID already exists.")
|
||||
}
|
||||
|
||||
manager.profiles[id] = profile
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Update(id string, profile types.MemberProfile) error {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
_, ok := manager.profiles[id]
|
||||
if !ok {
|
||||
return fmt.Errorf("Member ID does not exist.")
|
||||
}
|
||||
|
||||
manager.profiles[id] = profile
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Delete(id string) error {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
_, ok := manager.profiles[id]
|
||||
if !ok {
|
||||
return fmt.Errorf("Member ID does not exist.")
|
||||
}
|
||||
|
||||
delete(manager.profiles, id)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Select(id string) (types.MemberProfile, bool) {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
profile, ok := manager.profiles[id]
|
||||
return profile, ok
|
||||
}
|
Reference in New Issue
Block a user