78 lines
1.4 KiB
Go
78 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"log"
|
|
"math/rand"
|
|
"sync"
|
|
)
|
|
|
|
var (
|
|
userCenter UserCenter
|
|
)
|
|
|
|
type UserCenter struct {
|
|
UsersSTOI map[string]uint16
|
|
UsersITOS map[uint16]string
|
|
Mu sync.Mutex
|
|
}
|
|
|
|
func (u *UserCenter) Init() {
|
|
u.UsersSTOI = make(map[string]uint16)
|
|
u.UsersITOS = make(map[uint16]string)
|
|
}
|
|
|
|
var (
|
|
ERROR_ALREADY_HAVE = errors.New("Already taken")
|
|
ERROR_DONT_HAVE = errors.New("Not found")
|
|
)
|
|
|
|
func (u *UserCenter) AddUser(name string) (uint16, error) {
|
|
u.Mu.Lock()
|
|
defer u.Mu.Unlock()
|
|
_, alreadyHave := u.UsersSTOI[name]
|
|
if alreadyHave {
|
|
return 0, ERROR_ALREADY_HAVE
|
|
}
|
|
var ret uint16
|
|
for {
|
|
rInt := rand.Int31()
|
|
_, taken := u.UsersITOS[uint16(rInt)]
|
|
if !taken {
|
|
ret = uint16(rInt)
|
|
break
|
|
}
|
|
}
|
|
log.Printf("Users: add %s with %d id\n", name, ret)
|
|
u.UsersITOS[ret] = name
|
|
u.UsersSTOI[name] = ret
|
|
return ret, nil
|
|
}
|
|
|
|
func (u *UserCenter) DeleteIfHaveOne(id uint16) {
|
|
name, found := u.UsersITOS[id]
|
|
if !found {
|
|
log.Printf("User with %d id is not found; Can not delete\n", id)
|
|
return
|
|
}
|
|
delete(u.UsersITOS, id)
|
|
delete(u.UsersSTOI, name)
|
|
log.Printf("User with %s name and %d id was found; User is deleted\n", name, id)
|
|
}
|
|
|
|
func (u *UserCenter) GetID(name string) (uint16, error) {
|
|
id, have := u.UsersSTOI[name]
|
|
if !have {
|
|
return 0, ERROR_DONT_HAVE
|
|
}
|
|
return id, nil
|
|
}
|
|
|
|
func (u *UserCenter) GetName(id uint16) (string, error) {
|
|
name, have := u.UsersITOS[id]
|
|
if !have {
|
|
return "", ERROR_DONT_HAVE
|
|
}
|
|
return name, nil
|
|
}
|