mirror of
https://github.com/m1k1o/neko.git
synced 2024-07-24 14:40:50 +12:00
move session database to member folder.
This commit is contained in:
35
internal/member/database/dummy/adapter.go
Normal file
35
internal/member/database/dummy/adapter.go
Normal file
@ -0,0 +1,35 @@
|
||||
package dummy
|
||||
|
||||
import (
|
||||
"demodesk/neko/internal/types"
|
||||
)
|
||||
|
||||
func New() types.MembersDatabase {
|
||||
return &MembersDatabaseCtx{}
|
||||
}
|
||||
|
||||
type MembersDatabaseCtx struct{}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Connect() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Disconnect() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Insert(id string, profile types.MemberProfile) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Update(id string, profile types.MemberProfile) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Delete(id string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Select() (map[string]types.MemberProfile, error) {
|
||||
return map[string]types.MemberProfile{}, nil
|
||||
}
|
128
internal/member/database/file/adapter.go
Normal file
128
internal/member/database/file/adapter.go
Normal file
@ -0,0 +1,128 @@
|
||||
package file
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"sync"
|
||||
|
||||
"demodesk/neko/internal/types"
|
||||
)
|
||||
|
||||
func New(file string) types.MembersDatabase {
|
||||
return &MembersDatabaseCtx{
|
||||
file: file,
|
||||
mu: sync.Mutex{},
|
||||
}
|
||||
}
|
||||
|
||||
type MembersDatabaseCtx struct {
|
||||
file string
|
||||
mu sync.Mutex
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Connect() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Disconnect() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Insert(id string, profile types.MemberProfile) error {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
profiles, err := manager.deserialize()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, ok := profiles[id]
|
||||
if ok {
|
||||
return fmt.Errorf("Member ID already exists.")
|
||||
}
|
||||
|
||||
profiles[id] = profile
|
||||
|
||||
return manager.serialize(profiles)
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Update(id string, profile types.MemberProfile) error {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
profiles, err := manager.deserialize()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, ok := profiles[id]
|
||||
if !ok {
|
||||
return fmt.Errorf("Member ID does not exist.")
|
||||
}
|
||||
|
||||
profiles[id] = profile
|
||||
|
||||
return manager.serialize(profiles)
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Delete(id string) error {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
profiles, err := manager.deserialize()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, ok := profiles[id]
|
||||
if !ok {
|
||||
return fmt.Errorf("Member ID does not exist.")
|
||||
}
|
||||
|
||||
delete(profiles, id)
|
||||
|
||||
return manager.serialize(profiles)
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Select() (map[string]types.MemberProfile, error) {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
profiles, err := manager.deserialize()
|
||||
return profiles, err
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) deserialize() (map[string]types.MemberProfile, error) {
|
||||
file, err := os.OpenFile(manager.file, os.O_RDONLY|os.O_CREATE, os.ModePerm)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
raw, err := ioutil.ReadAll(file)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(raw) == 0 {
|
||||
return map[string]types.MemberProfile{}, nil
|
||||
}
|
||||
|
||||
var profiles map[string]types.MemberProfile
|
||||
if err := json.Unmarshal([]byte(raw), &profiles); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return profiles, nil
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) serialize(data map[string]types.MemberProfile) error {
|
||||
raw, err := json.Marshal(data)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return ioutil.WriteFile(manager.file, raw, os.ModePerm)
|
||||
}
|
22
internal/member/database/manager.go
Normal file
22
internal/member/database/manager.go
Normal file
@ -0,0 +1,22 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"demodesk/neko/internal/config"
|
||||
"demodesk/neko/internal/session/database/dummy"
|
||||
"demodesk/neko/internal/session/database/file"
|
||||
"demodesk/neko/internal/session/database/object"
|
||||
"demodesk/neko/internal/types"
|
||||
)
|
||||
|
||||
func New(config *config.Session) types.MembersDatabase {
|
||||
switch config.DatabaseAdapter {
|
||||
case "file":
|
||||
return file.New(config.FilePath)
|
||||
case "object":
|
||||
return object.New()
|
||||
case "dummy":
|
||||
return dummy.New()
|
||||
}
|
||||
|
||||
return dummy.New()
|
||||
}
|
74
internal/member/database/object/adapter.go
Normal file
74
internal/member/database/object/adapter.go
Normal file
@ -0,0 +1,74 @@
|
||||
package object
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"sync"
|
||||
|
||||
"demodesk/neko/internal/types"
|
||||
)
|
||||
|
||||
func New() types.MembersDatabase {
|
||||
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) Connect() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (manager *MembersDatabaseCtx) Disconnect() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
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() (map[string]types.MemberProfile, error) {
|
||||
manager.mu.Lock()
|
||||
defer manager.mu.Unlock()
|
||||
|
||||
return manager.profiles, nil
|
||||
}
|
Reference in New Issue
Block a user