mirror of
https://github.com/grafana/grafana.git
synced 2025-02-14 09:33:34 -06:00
9855e74b92
Chore: Refactor quota service (#57586) * Chore: refactore quota service * Apply suggestions from code review
80 lines
1.8 KiB
Go
80 lines
1.8 KiB
Go
package auth
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/services/quota"
|
|
)
|
|
|
|
type userAuthToken struct {
|
|
Id int64
|
|
UserId int64
|
|
AuthToken string
|
|
PrevAuthToken string
|
|
UserAgent string
|
|
ClientIp string
|
|
AuthTokenSeen bool
|
|
SeenAt int64
|
|
RotatedAt int64
|
|
CreatedAt int64
|
|
UpdatedAt int64
|
|
RevokedAt int64
|
|
UnhashedToken string `xorm:"-"`
|
|
}
|
|
|
|
func userAuthTokenFromUserToken(ut *models.UserToken) (*userAuthToken, error) {
|
|
var uat userAuthToken
|
|
err := uat.fromUserToken(ut)
|
|
return &uat, err
|
|
}
|
|
|
|
func (uat *userAuthToken) fromUserToken(ut *models.UserToken) error {
|
|
if uat == nil {
|
|
return fmt.Errorf("needs pointer to userAuthToken struct")
|
|
}
|
|
|
|
uat.Id = ut.Id
|
|
uat.UserId = ut.UserId
|
|
uat.AuthToken = ut.AuthToken
|
|
uat.PrevAuthToken = ut.PrevAuthToken
|
|
uat.UserAgent = ut.UserAgent
|
|
uat.ClientIp = ut.ClientIp
|
|
uat.AuthTokenSeen = ut.AuthTokenSeen
|
|
uat.SeenAt = ut.SeenAt
|
|
uat.RotatedAt = ut.RotatedAt
|
|
uat.CreatedAt = ut.CreatedAt
|
|
uat.UpdatedAt = ut.UpdatedAt
|
|
uat.RevokedAt = ut.RevokedAt
|
|
uat.UnhashedToken = ut.UnhashedToken
|
|
|
|
return nil
|
|
}
|
|
|
|
func (uat *userAuthToken) toUserToken(ut *models.UserToken) error {
|
|
if uat == nil {
|
|
return fmt.Errorf("needs pointer to userAuthToken struct")
|
|
}
|
|
|
|
ut.Id = uat.Id
|
|
ut.UserId = uat.UserId
|
|
ut.AuthToken = uat.AuthToken
|
|
ut.PrevAuthToken = uat.PrevAuthToken
|
|
ut.UserAgent = uat.UserAgent
|
|
ut.ClientIp = uat.ClientIp
|
|
ut.AuthTokenSeen = uat.AuthTokenSeen
|
|
ut.SeenAt = uat.SeenAt
|
|
ut.RotatedAt = uat.RotatedAt
|
|
ut.CreatedAt = uat.CreatedAt
|
|
ut.UpdatedAt = uat.UpdatedAt
|
|
ut.RevokedAt = uat.RevokedAt
|
|
ut.UnhashedToken = uat.UnhashedToken
|
|
|
|
return nil
|
|
}
|
|
|
|
const (
|
|
QuotaTargetSrv quota.TargetSrv = "auth"
|
|
QuotaTarget quota.Target = "session"
|
|
)
|