Add login

This commit is contained in:
2020-01-05 23:20:38 +01:00
parent dc396a300f
commit c0926ffea0
35 changed files with 1647 additions and 39 deletions

View File

@@ -0,0 +1,37 @@
package fake
import (
"encoding/json"
"hamster-tycoon/storage/dao"
"hamster-tycoon/utils"
"time"
"github.com/allegro/bigcache"
)
type DatabaseFake struct {
Cache *bigcache.BigCache
}
func NewDatabaseFake() dao.Database {
cache, err := bigcache.NewBigCache(bigcache.DefaultConfig(time.Minute))
if err != nil {
utils.GetLogger().WithError(err).Fatal("Error while instantiate cache")
}
return &DatabaseFake{
Cache: cache,
}
}
func (db *DatabaseFake) save(key string, data []interface{}) {
b, err := json.Marshal(data)
if err != nil {
utils.GetLogger().WithError(err).Errorf("Error while marshal fake %s", key)
db.Cache.Set(key, []byte("[]"))
return
}
err = db.Cache.Set(key, b)
if err != nil {
utils.GetLogger().WithError(err).Errorf("Error while saving fake %s", key)
}
}

View File

@@ -0,0 +1,106 @@
package fake
import (
"encoding/json"
"errors"
"hamster-tycoon/storage/dao"
"hamster-tycoon/storage/model"
"hamster-tycoon/utils"
"time"
"github.com/satori/go.uuid"
)
const (
cacheKeyUsers = "users"
)
func (db *DatabaseFake) saveUsers(users []*model.User) {
data := make([]interface{}, 0)
for _, v := range users {
data = append(data, v)
}
db.save(cacheKeyUsers, data)
}
func (db *DatabaseFake) loadUsers() []*model.User {
users := make([]*model.User, 0)
b, err := db.Cache.Get(cacheKeyUsers)
if err != nil {
return users
}
err = json.Unmarshal(b, &users)
if err != nil {
utils.GetLogger().WithError(err).Error("Error while unmarshal fake users")
}
return users
}
func (db *DatabaseFake) GetAllUsers() ([]*model.User, error) {
return db.loadUsers(), nil
}
func (db *DatabaseFake) GetUsersByID(userID string) (*model.User, error) {
users := db.loadUsers()
for _, u := range users {
if u.ID == userID {
return u, nil
}
}
return nil, dao.NewDAOError(dao.ErrTypeNotFound, errors.New("user not found"))
}
func (db *DatabaseFake) CreateUser(user *model.User) error {
user.ID = uuid.NewV4().String()
user.CreatedAt = time.Now()
users := db.loadUsers()
users = append(users, user)
db.saveUsers(users)
return nil
}
func (db *DatabaseFake) DeleteUser(userID string) error {
users := db.loadUsers()
newUsers := make([]*model.User, 0)
for _, u := range users {
if u.ID != userID {
newUsers = append(newUsers, u)
}
}
db.saveUsers(newUsers)
return nil
}
func (db *DatabaseFake) UpdateUser(user *model.User) error {
users := db.loadUsers()
var foundUser *model.User
for _, u := range users {
if u.ID == user.ID {
foundUser = u
break
}
}
if foundUser == nil {
return dao.NewDAOError(dao.ErrTypeNotFound, errors.New("user not found"))
}
foundUser.UserEditable = user.UserEditable
now := time.Now()
foundUser.UpdatedAt = &now
db.saveUsers(users)
*user = *foundUser
return nil
}
func (db *DatabaseFake) GetUsersByGoogleID(userID string) (*model.User, error) {
users := db.loadUsers()
for _, u := range users {
if u.GoogleId == userID {
return u, nil
}
}
return nil, dao.NewDAOError(dao.ErrTypeNotFound, errors.New("user not found"))
}