Compare commits
3 Commits
Author | SHA1 | Date | |
---|---|---|---|
34acd0fa8d | |||
71d3bd77d0 | |||
1451abcca4 |
1
main.go
1
main.go
@ -78,6 +78,7 @@ func main() {
|
|||||||
signal.Notify(interrupt, os.Interrupt, syscall.SIGTERM)
|
signal.Notify(interrupt, os.Interrupt, syscall.SIGTERM)
|
||||||
stop := make(chan struct{})
|
stop := make(chan struct{})
|
||||||
go app.RunScheduledTasks(&appLoaded, 100, stop)
|
go app.RunScheduledTasks(&appLoaded, 100, stop)
|
||||||
|
|
||||||
<-interrupt
|
<-interrupt
|
||||||
log.Println("Interrupt signal received. Shutting down server...")
|
log.Println("Interrupt signal received. Shutting down server...")
|
||||||
|
|
||||||
|
@ -64,6 +64,18 @@ func CreateSession(app *app.App, w http.ResponseWriter, userId int64, remember b
|
|||||||
return session, nil
|
return session, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func GetSessionByAuthToken(app *app.App, authToken string) (Session, error) {
|
||||||
|
session := Session{}
|
||||||
|
|
||||||
|
err := app.Db.QueryRow(selectSessionByAuthToken, authToken).Scan(&session.Id, &session.UserId, &session.AuthToken, &session.RememberMe, &session.CreatedAt)
|
||||||
|
if err != nil {
|
||||||
|
log.Println("Error getting session by auth token")
|
||||||
|
return Session{}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return session, nil
|
||||||
|
}
|
||||||
|
|
||||||
// Generates a random 64-byte string
|
// Generates a random 64-byte string
|
||||||
func generateAuthToken(app *app.App) string {
|
func generateAuthToken(app *app.App) string {
|
||||||
// Generate random bytes
|
// Generate random bytes
|
||||||
|
@ -23,7 +23,6 @@ const userColumns = "\"Id\", " + userColumnsNoId
|
|||||||
const userTable = "public.\"User\""
|
const userTable = "public.\"User\""
|
||||||
|
|
||||||
const (
|
const (
|
||||||
selectSessionIdByAuthToken = "SELECT \"Id\" FROM public.\"Session\" WHERE \"AuthToken\" = $1"
|
|
||||||
selectUserById = "SELECT " + userColumns + " FROM " + userTable + " WHERE \"Id\" = $1"
|
selectUserById = "SELECT " + userColumns + " FROM " + userTable + " WHERE \"Id\" = $1"
|
||||||
selectUserByUsername = "SELECT " + userColumns + " FROM " + userTable + " WHERE \"Username\" = $1"
|
selectUserByUsername = "SELECT " + userColumns + " FROM " + userTable + " WHERE \"Username\" = $1"
|
||||||
insertUser = "INSERT INTO " + userTable + " (" + userColumnsNoId + ") VALUES ($1, $2, $3, $4) RETURNING \"Id\""
|
insertUser = "INSERT INTO " + userTable + " (" + userColumnsNoId + ") VALUES ($1, $2, $3, $4) RETURNING \"Id\""
|
||||||
@ -37,16 +36,13 @@ func GetCurrentUser(app *app.App, r *http.Request) (User, error) {
|
|||||||
return User{}, err
|
return User{}, err
|
||||||
}
|
}
|
||||||
|
|
||||||
var userId int64
|
session, err := GetSessionByAuthToken(app, cookie.Value)
|
||||||
|
|
||||||
// Query row by AuthToken
|
|
||||||
err = app.Db.QueryRow(selectSessionIdByAuthToken, cookie.Value).Scan(&userId)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Println("Error querying session row with session: " + cookie.Value)
|
log.Println("Error getting session by auth token")
|
||||||
return User{}, err
|
return User{}, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return GetUserById(app, userId)
|
return GetUserById(app, session.UserId)
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetUserById finds a User table row in the database by id and returns a struct representing this row
|
// GetUserById finds a User table row in the database by id and returns a struct representing this row
|
||||||
|
Reference in New Issue
Block a user