2020-03-04 11:07:12 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"crypto/rand"
|
|
|
|
"time"
|
|
|
|
)
|
|
|
|
|
|
|
|
type Session struct {
|
2022-07-09 17:42:00 +00:00
|
|
|
Id []byte `json:"id"`
|
2020-03-04 11:07:12 +00:00
|
|
|
IdUser *int64 `json:"login"`
|
2022-07-09 17:42:00 +00:00
|
|
|
Time time.Time `json:"time"`
|
2020-03-04 11:07:12 +00:00
|
|
|
}
|
|
|
|
|
2022-07-09 17:42:00 +00:00
|
|
|
func getSession(id []byte) (s *Session, err error) {
|
|
|
|
s = new(Session)
|
2020-03-04 11:07:12 +00:00
|
|
|
err = DBQueryRow("SELECT id_session, id_user, time FROM user_sessions WHERE id_session=?", id).Scan(&s.Id, &s.IdUser, &s.Time)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2022-07-09 17:42:00 +00:00
|
|
|
func NewSession() (*Session, error) {
|
2020-03-04 11:07:12 +00:00
|
|
|
session_id := make([]byte, 255)
|
|
|
|
if _, err := rand.Read(session_id); err != nil {
|
2022-07-09 17:42:00 +00:00
|
|
|
return nil, err
|
2020-03-04 11:07:12 +00:00
|
|
|
} else if _, err := DBExec("INSERT INTO user_sessions (id_session, time) VALUES (?, ?)", session_id, time.Now()); err != nil {
|
2022-07-09 17:42:00 +00:00
|
|
|
return nil, err
|
2020-03-04 11:07:12 +00:00
|
|
|
} else {
|
2022-07-09 17:42:00 +00:00
|
|
|
return &Session{session_id, nil, time.Now()}, nil
|
2020-03-04 11:07:12 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-07-09 17:42:00 +00:00
|
|
|
func (user User) NewSession() (*Session, error) {
|
2020-03-04 11:07:12 +00:00
|
|
|
session_id := make([]byte, 255)
|
|
|
|
if _, err := rand.Read(session_id); err != nil {
|
2022-07-09 17:42:00 +00:00
|
|
|
return nil, err
|
2020-03-04 11:07:12 +00:00
|
|
|
} else if _, err := DBExec("INSERT INTO user_sessions (id_session, id_user, time) VALUES (?, ?, ?)", session_id, user.Id, time.Now()); err != nil {
|
2022-07-09 17:42:00 +00:00
|
|
|
return nil, err
|
2020-03-04 11:07:12 +00:00
|
|
|
} else {
|
2022-07-09 17:42:00 +00:00
|
|
|
return &Session{session_id, &user.Id, time.Now()}, nil
|
2020-03-04 11:07:12 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-07-09 17:42:00 +00:00
|
|
|
func (s Session) SetUser(user *User) (*Session, error) {
|
2020-03-04 11:07:12 +00:00
|
|
|
s.IdUser = &user.Id
|
|
|
|
_, err := s.Update()
|
2022-07-09 17:42:00 +00:00
|
|
|
return &s, err
|
2020-03-04 11:07:12 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (s Session) Update() (int64, error) {
|
|
|
|
if res, err := DBExec("UPDATE user_sessions SET id_user = ?, time = ? WHERE id_session = ?", s.IdUser, s.Time, s.Id); err != nil {
|
|
|
|
return 0, err
|
|
|
|
} else if nb, err := res.RowsAffected(); err != nil {
|
|
|
|
return 0, err
|
|
|
|
} else {
|
|
|
|
return nb, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s Session) Delete() (int64, error) {
|
|
|
|
if res, err := DBExec("DELETE FROM user_sessions WHERE id_session = ?", s.Id); err != nil {
|
|
|
|
return 0, err
|
|
|
|
} else if nb, err := res.RowsAffected(); err != nil {
|
|
|
|
return 0, err
|
|
|
|
} else {
|
|
|
|
return nb, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func ClearSession() (int64, error) {
|
|
|
|
if res, err := DBExec("DELETE FROM user_sessions"); err != nil {
|
|
|
|
return 0, err
|
|
|
|
} else if nb, err := res.RowsAffected(); err != nil {
|
|
|
|
return 0, err
|
|
|
|
} else {
|
|
|
|
return nb, err
|
|
|
|
}
|
|
|
|
}
|