mirrors/owncast
1
0
Fork 0
mirror of https://github.com/owncast/owncast.git synced 2024-11-21 12:18:02 +03:00
Code Issues Projects Releases Packages Wiki Activity

Merge fixes

Browse source
This commit is contained in:
Gabe Kangas 2022-04-30 21:51:55 -07:00
parent 3bfa260e5e
commit a507eba350
No known key found for this signature in database
GPG key ID: 9A56337728BC81EA
1 changed files with 13 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
core/user/user.go
View file

@ -17,7 +17,10 @@ import (
log "github.com/sirupsen/logrus"
)
var _datastore *data.Datastore
var (
_datastore *data.Datastore
_cachedGetUserByTokenStatement *sql.Stmt
)
const (
moderatorScopeKey = "MODERATOR"
@ -139,14 +142,14 @@ func create(user *User) error {
defer _datastore.DbLock.Unlock()
if createUserStatement == nil {
stmt, err := _datastore.DB.Prepare("INSERT INTO users(id, access_token, display_name, display_color, previous_names, created_at) values(?, ?, ?, ?, ?, ?)")
stmt, err := _datastore.DB.Prepare("INSERT INTO users(id, display_name, display_color, previous_names, created_at) values(?, ?, ?, ?, ?, ?)")
if err != nil {
return errors.Wrap(err, "error preparing create user statement")
}
createUserStatement = stmt
}
_, err := createUserStatement.Exec(user.ID, user.AccessToken, user.DisplayName, user.DisplayColor, user.DisplayName, user.CreatedAt)
_, err := createUserStatement.Exec(user.ID, user.DisplayName, user.DisplayColor, user.DisplayName, user.CreatedAt)
if err != nil {
return errors.Wrap(err, "error creating user")
}
@ -186,8 +189,6 @@ func SetEnabled(userID string, enabled bool) error {
return tx.Commit()
}
var cachedGUserByTokenStatement *sql.Stmt
// GetUserByToken will return a user by an access token.
func GetUserByToken(token string) *User {
u, err := _datastore.GetQueries().GetUserByAccessToken(context.Background(), token)
@ -195,15 +196,19 @@ func GetUserByToken(token string) *User {
return nil
}
if cachedGUserByTokenStatement == nil {
if _cachedGetUserByTokenStatement == nil {
stmt, err := _datastore.DB.Prepare("SELECT id, display_name, display_color, created_at, disabled_at, previous_names, namechanged_at, scopes FROM users WHERE access_token = ?")
if err != nil {
log.Errorln(err)
return nil
}
cachedGUserByTokenStatement = stmt
_cachedGetUserByTokenStatement = stmt
}
var scopes []string
if u.Scopes.Valid {
scopes = strings.Split(u.Scopes.String, ",")
}
row := cachedGUserByTokenStatement.QueryRow(token)
var disabledAt *time.Time
if u.DisabledAt.Valid {