mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
9cc5089af8
* Moving mlog to corelibs * Regenerating app layers * Fix golangci-lint problem * Fixing golangci-lint errors * Renaming from corelibs to shared * Renaming from corelibs to shared * Fixing import * Fixing merge problems * Fixing build
83 lines
2.4 KiB
Go
83 lines
2.4 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package sqlstore
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/mattermost/mattermost-server/v5/model"
|
|
"github.com/mattermost/mattermost-server/v5/shared/mlog"
|
|
"github.com/mattermost/mattermost-server/v5/store"
|
|
)
|
|
|
|
type SqlTokenStore struct {
|
|
*SqlStore
|
|
}
|
|
|
|
func newSqlTokenStore(sqlStore *SqlStore) store.TokenStore {
|
|
s := &SqlTokenStore{sqlStore}
|
|
|
|
for _, db := range sqlStore.GetAllConns() {
|
|
table := db.AddTableWithName(model.Token{}, "Tokens").SetKeys(false, "Token")
|
|
table.ColMap("Token").SetMaxSize(64)
|
|
table.ColMap("Type").SetMaxSize(64)
|
|
table.ColMap("Extra").SetMaxSize(2048)
|
|
}
|
|
|
|
return s
|
|
}
|
|
|
|
func (s SqlTokenStore) createIndexesIfNotExists() {
|
|
}
|
|
|
|
func (s SqlTokenStore) Save(token *model.Token) error {
|
|
if err := token.IsValid(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := s.GetMaster().Insert(token); err != nil {
|
|
return errors.Wrap(err, "failed to save Token")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s SqlTokenStore) Delete(token string) error {
|
|
if _, err := s.GetMaster().Exec("DELETE FROM Tokens WHERE Token = :Token", map[string]interface{}{"Token": token}); err != nil {
|
|
return errors.Wrapf(err, "failed to delete Token with value %s", token)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s SqlTokenStore) GetByToken(tokenString string) (*model.Token, error) {
|
|
token := &model.Token{}
|
|
|
|
if err := s.GetReplica().SelectOne(token, "SELECT * FROM Tokens WHERE Token = :Token", map[string]interface{}{"Token": tokenString}); err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, store.NewErrNotFound("Token", fmt.Sprintf("Token=%s", tokenString))
|
|
}
|
|
|
|
return nil, errors.Wrapf(err, "failed to get Token with value %s", tokenString)
|
|
}
|
|
|
|
return token, nil
|
|
}
|
|
|
|
func (s SqlTokenStore) Cleanup() {
|
|
mlog.Debug("Cleaning up token store.")
|
|
deltime := model.GetMillis() - model.MAX_TOKEN_EXIPRY_TIME
|
|
if _, err := s.GetMaster().Exec("DELETE FROM Tokens WHERE CreateAt < :DelTime", map[string]interface{}{"DelTime": deltime}); err != nil {
|
|
mlog.Error("Unable to cleanup token store.")
|
|
}
|
|
}
|
|
|
|
func (s SqlTokenStore) RemoveAllTokensByType(tokenType string) error {
|
|
if _, err := s.GetMaster().Exec("DELETE FROM Tokens WHERE Type = :TokenType", map[string]interface{}{"TokenType": tokenType}); err != nil {
|
|
return errors.Wrapf(err, "failed to remove all Tokens with Type=%s", tokenType)
|
|
}
|
|
return nil
|
|
}
|