mirror of
https://github.com/m1k1o/neko.git
synced 2024-07-24 14:40:50 +12:00
implemented member manager.
This commit is contained in:
@ -1,74 +0,0 @@
|
||||
package object
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"sync"
|
||||
|
||||
"demodesk/neko/internal/types"
|
||||
)
|
||||
|
||||
func New() types.MemberManager {
|
||||
return &MemberManagerCtx{
|
||||
profiles: make(map[string]types.MemberProfile),
|
||||
mu: sync.Mutex{},
|
||||
}
|
||||
}
|
||||
|
||||
type MemberManagerCtx struct {
|
||||
profiles map[string]types.MemberProfile
|
||||
mu sync.Mutex
|
||||
}
|
||||
|
||||
func (manager *MemberManagerCtx) Connect() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MemberManagerCtx) Disconnect() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MemberManagerCtx) 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 *MemberManagerCtx) 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 *MemberManagerCtx) 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 *MemberManagerCtx) Select() (map[string]types.MemberProfile, error) {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
return manager.profiles, nil
|
||||
}
|
149
internal/member/object/manager.go
Normal file
149
internal/member/object/manager.go
Normal file
@ -0,0 +1,149 @@
|
||||
package object
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"sync"
|
||||
|
||||
"demodesk/neko/internal/types"
|
||||
)
|
||||
|
||||
func New() types.MemberManager {
|
||||
return &MemberManagerCtx{
|
||||
entries: make(map[string]MemberEntry),
|
||||
mu: sync.Mutex{},
|
||||
}
|
||||
}
|
||||
|
||||
type MemberManagerCtx struct {
|
||||
entries map[string]MemberEntry
|
||||
mu sync.Mutex
|
||||
}
|
||||
|
||||
type MemberEntry struct {
|
||||
Password string `json:"password"`
|
||||
Profile types.MemberProfile `json:"profile"`
|
||||
}
|
||||
|
||||
func (manager *MemberManagerCtx) Connect() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MemberManagerCtx) Disconnect() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MemberManagerCtx) Authenticate(username string, password string) (string, types.MemberProfile, error) {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
// id will be also username
|
||||
id := username
|
||||
|
||||
entry, ok := manager.entries[id]
|
||||
if !ok {
|
||||
return "", types.MemberProfile{}, fmt.Errorf("Member ID does not exist.")
|
||||
}
|
||||
|
||||
// TODO: Use hash function.
|
||||
if entry.Password != password {
|
||||
return "", types.MemberProfile{}, fmt.Errorf("Invalid password.")
|
||||
}
|
||||
|
||||
return id, entry.Profile, nil
|
||||
}
|
||||
|
||||
func (manager *MemberManagerCtx) Insert(username string, password string, profile types.MemberProfile) (string, error) {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
// id will be also username
|
||||
id := username
|
||||
|
||||
entry, ok := manager.entries[id]
|
||||
if ok {
|
||||
return "", fmt.Errorf("Member ID already exists.")
|
||||
}
|
||||
|
||||
// TODO: Use hash function.
|
||||
entry.Password = password
|
||||
entry.Profile = profile
|
||||
manager.entries[id] = entry
|
||||
|
||||
return id, nil
|
||||
}
|
||||
|
||||
func (manager *MemberManagerCtx) Update(id string, profile types.MemberProfile) error {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
entry, ok := manager.entries[id]
|
||||
if !ok {
|
||||
return fmt.Errorf("Member ID does not exist.")
|
||||
}
|
||||
|
||||
entry.Profile = profile
|
||||
manager.entries[id] = entry
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MemberManagerCtx) UpdatePassword(id string, password string) error {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
entry, ok := manager.entries[id]
|
||||
if !ok {
|
||||
return fmt.Errorf("Member ID does not exist.")
|
||||
}
|
||||
|
||||
// TODO: Use hash function.
|
||||
entry.Password = password
|
||||
manager.entries[id] = entry
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MemberManagerCtx) Select(id string) (types.MemberProfile, error) {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
entry, ok := manager.entries[id]
|
||||
if ok {
|
||||
return types.MemberProfile{}, fmt.Errorf("Member ID already exists.")
|
||||
}
|
||||
|
||||
return entry.Profile, nil
|
||||
}
|
||||
|
||||
func (manager *MemberManagerCtx) SelectAll(limit int, offset int) (map[string]types.MemberProfile, error) {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
profiles := map[string]types.MemberProfile{}
|
||||
|
||||
i := 0
|
||||
for id, entry := range manager.entries {
|
||||
if i < offset || i > offset + limit {
|
||||
continue
|
||||
}
|
||||
|
||||
profiles[id] = entry.Profile
|
||||
i = i + 1
|
||||
}
|
||||
|
||||
return profiles, nil
|
||||
}
|
||||
|
||||
func (manager *MemberManagerCtx) Delete(id string) error {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
_, ok := manager.entries[id]
|
||||
if !ok {
|
||||
return fmt.Errorf("Member ID does not exist.")
|
||||
}
|
||||
|
||||
delete(manager.entries, id)
|
||||
|
||||
return nil
|
||||
}
|
Reference in New Issue
Block a user