9 changed files with 150 additions and 178 deletions
-
27cmd/server/main.go
-
9go.mod
-
29go.sum
-
7server/models/password_token.go
-
10server/models/session.go
-
10server/models/user.go
-
114server/server.go
-
60server/token_db.go
-
62server/user_db.go
@ -0,0 +1,7 @@ |
|||||
|
package models |
||||
|
|
||||
|
type PasswordToken struct { |
||||
|
Token string |
||||
|
Expires int64 |
||||
|
UserID uint |
||||
|
} |
||||
@ -0,0 +1,10 @@ |
|||||
|
package models |
||||
|
|
||||
|
import "gorm.io/gorm" |
||||
|
|
||||
|
type Session struct { |
||||
|
gorm.Model |
||||
|
Token string |
||||
|
Expires int64 |
||||
|
UserID uint |
||||
|
} |
||||
@ -0,0 +1,10 @@ |
|||||
|
package models |
||||
|
|
||||
|
import "gorm.io/gorm" |
||||
|
|
||||
|
type User struct { |
||||
|
gorm.Model |
||||
|
Email string |
||||
|
Password string |
||||
|
Sessions []Session |
||||
|
} |
||||
@ -1,60 +0,0 @@ |
|||||
package server |
|
||||
|
|
||||
import ( |
|
||||
"errors" |
|
||||
"time" |
|
||||
) |
|
||||
|
|
||||
type Token struct { |
|
||||
UserID int64 |
|
||||
Token string |
|
||||
Expires *time.Time |
|
||||
} |
|
||||
|
|
||||
type TokenDB interface { |
|
||||
FindByUserID(id int64) (*Token, error) |
|
||||
FindByToken(token string) (*Token, error) |
|
||||
Save(token *Token) error |
|
||||
} |
|
||||
|
|
||||
type inMemoryTokenDB struct { |
|
||||
tokens []*Token |
|
||||
} |
|
||||
|
|
||||
func newInMemoryTokenDB() *inMemoryTokenDB { |
|
||||
return &inMemoryTokenDB{ |
|
||||
tokens: make([]*Token, 0), |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
func (m *inMemoryTokenDB) FindByUserID(id int64) (*Token, error) { |
|
||||
for _, t := range m.tokens { |
|
||||
if t.UserID == id { |
|
||||
return t, nil |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
return nil, errors.New("Token not found.") |
|
||||
} |
|
||||
|
|
||||
func (m *inMemoryTokenDB) FindByToken(token string) (*Token, error) { |
|
||||
for _, t := range m.tokens { |
|
||||
if t.Token == token { |
|
||||
return t, nil |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
return nil, errors.New("Token not found.") |
|
||||
} |
|
||||
|
|
||||
func (m *inMemoryTokenDB) Save(token *Token) error { |
|
||||
for i, t := range m.tokens { |
|
||||
if t.UserID == token.UserID || t.Token == token.Token { |
|
||||
m.tokens[i] = token |
|
||||
return nil |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
m.tokens = append(m.tokens, token) |
|
||||
return nil |
|
||||
} |
|
||||
@ -1,62 +0,0 @@ |
|||||
package server |
|
||||
|
|
||||
import "errors" |
|
||||
|
|
||||
type User struct { |
|
||||
ID int64 |
|
||||
Email string |
|
||||
Password string |
|
||||
} |
|
||||
|
|
||||
type UserDB interface { |
|
||||
FindByID(id int64) (*User, error) |
|
||||
FindByEmail(email string) (*User, error) |
|
||||
Save(*User) error |
|
||||
} |
|
||||
|
|
||||
type inMemoryUserDB struct { |
|
||||
UserDB |
|
||||
nextID int64 |
|
||||
users []*User |
|
||||
} |
|
||||
|
|
||||
func newInMemoryUserDB() *inMemoryUserDB { |
|
||||
return &inMemoryUserDB{ |
|
||||
nextID: 1, |
|
||||
users: make([]*User, 0), |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
func (m *inMemoryUserDB) FindByID(id int64) (*User, error) { |
|
||||
for _, u := range m.users { |
|
||||
if u.ID == id { |
|
||||
return u, nil |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
return nil, errors.New("User not found.") |
|
||||
} |
|
||||
|
|
||||
func (m *inMemoryUserDB) FindByEmail(email string) (*User, error) { |
|
||||
for _, u := range m.users { |
|
||||
if u.Email == email { |
|
||||
return u, nil |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
return nil, errors.New("User not found.") |
|
||||
} |
|
||||
|
|
||||
func (m *inMemoryUserDB) Save(user *User) error { |
|
||||
for i, u := range m.users { |
|
||||
if u.ID == user.ID || u.Email == user.Email { |
|
||||
m.users[i] = user |
|
||||
return nil |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
user.ID = m.nextID |
|
||||
m.users = append(m.users, user) |
|
||||
m.nextID++ |
|
||||
return nil |
|
||||
} |
|
||||
Write
Preview
Loading…
Cancel
Save
Reference in new issue