lanhu/platform
2
0
Fork 0
Code Issues Actions Packages Releases 13 Activity

重构认证授权逻辑，集中到 auth 包中

Browse Source
This commit is contained in:
luorijun
2025-05-12 10:07:12 +08:00
parent cfdee98a1b
commit 2c37dcc2be
40 changed files with 905 additions and 455 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
web/queries/gen.go
View File

@@ -37,6 +37,7 @@ var (
ResourcePps *resourcePps
ResourcePsr *resourcePsr
ResourcePss *resourcePss
Session *session
Trade *trade
User *user
UserRole *userRole
@@ -67,6 +68,7 @@ func SetDefault(db *gorm.DB, opts ...gen.DOOption) {
ResourcePps = &Q.ResourcePps
ResourcePsr = &Q.ResourcePsr
ResourcePss = &Q.ResourcePss
Session = &Q.Session
Trade = &Q.Trade
User = &Q.User
UserRole = &Q.UserRole
@@ -98,6 +100,7 @@ func Use(db *gorm.DB, opts ...gen.DOOption) *Query {
ResourcePps: newResourcePps(db, opts...),
ResourcePsr: newResourcePsr(db, opts...),
ResourcePss: newResourcePss(db, opts...),
Session: newSession(db, opts...),
Trade: newTrade(db, opts...),
User: newUser(db, opts...),
UserRole: newUserRole(db, opts...),
@@ -130,6 +133,7 @@ type Query struct {
ResourcePps resourcePps
ResourcePsr resourcePsr
ResourcePss resourcePss
Session session
Trade trade
User user
UserRole userRole
@@ -163,6 +167,7 @@ func (q *Query) clone(db *gorm.DB) *Query {
ResourcePps: q.ResourcePps.clone(db),
ResourcePsr: q.ResourcePsr.clone(db),
ResourcePss: q.ResourcePss.clone(db),
Session: q.Session.clone(db),
Trade: q.Trade.clone(db),
User: q.User.clone(db),
UserRole: q.UserRole.clone(db),
@@ -203,6 +208,7 @@ func (q *Query) ReplaceDB(db *gorm.DB) *Query {
ResourcePps: q.ResourcePps.replaceDB(db),
ResourcePsr: q.ResourcePsr.replaceDB(db),
ResourcePss: q.ResourcePss.replaceDB(db),
Session: q.Session.replaceDB(db),
Trade: q.Trade.replaceDB(db),
User: q.User.replaceDB(db),
UserRole: q.UserRole.replaceDB(db),
@@ -233,6 +239,7 @@ type queryCtx struct {
ResourcePps *resourcePpsDo
ResourcePsr *resourcePsrDo
ResourcePss *resourcePssDo
Session *sessionDo
Trade *tradeDo
User *userDo
UserRole *userRoleDo
@@ -263,6 +270,7 @@ func (q *Query) WithContext(ctx context.Context) *queryCtx {
ResourcePps: q.ResourcePps.WithContext(ctx),
ResourcePsr: q.ResourcePsr.WithContext(ctx),
ResourcePss: q.ResourcePss.WithContext(ctx),
Session: q.Session.WithContext(ctx),
Trade: q.Trade.WithContext(ctx),
User: q.User.WithContext(ctx),
UserRole: q.UserRole.WithContext(ctx),

371
web/queries/session.gen.go Normal file
View File

@@ -0,0 +1,371 @@
// Code generated by gorm.io/gen. DO NOT EDIT.
// Code generated by gorm.io/gen. DO NOT EDIT.
// Code generated by gorm.io/gen. DO NOT EDIT.
package queries
import (
"context"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"gorm.io/gorm/schema"
"gorm.io/gen"
"gorm.io/gen/field"
"gorm.io/plugin/dbresolver"
"platform/web/models"
)
func newSession(db *gorm.DB, opts ...gen.DOOption) session {
_session := session{}
_session.sessionDo.UseDB(db, opts...)
_session.sessionDo.UseModel(&models.Session{})
tableName := _session.sessionDo.TableName()
_session.ALL = field.NewAsterisk(tableName)
_session.ID = field.NewInt32(tableName, "id")
_session.UserID = field.NewInt32(tableName, "user_id")
_session.ClientID = field.NewInt32(tableName, "client_id")
_session.IP = field.NewString(tableName, "ip")
_session.Ua = field.NewString(tableName, "ua")
_session.GrantType = field.NewString(tableName, "grant_type")
_session.AccessToken = field.NewString(tableName, "access_token")
_session.AccessTokenExpires = field.NewField(tableName, "access_token_expires")
_session.RefreshToken = field.NewString(tableName, "refresh_token")
_session.RefreshTokenExpires = field.NewField(tableName, "refresh_token_expires")
_session.Scopes_ = field.NewString(tableName, "scopes")
_session.CreatedAt = field.NewField(tableName, "created_at")
_session.UpdatedAt = field.NewField(tableName, "updated_at")
_session.DeletedAt = field.NewField(tableName, "deleted_at")
_session.fillFieldMap()
return _session
}
type session struct {
sessionDo
ALL field.Asterisk
ID field.Int32 // 会话ID
UserID field.Int32 // 用户ID
ClientID field.Int32 // 客户端ID
IP field.String // IP地址
Ua field.String // 用户代理
GrantType field.String // 授权类型：authorization_code-授权码模式，client_credentials-客户端凭证模式，refresh_token-刷新令牌模式，password-密码模式
AccessToken field.String // 访问令牌
AccessTokenExpires field.Field // 访问令牌过期时间
RefreshToken field.String // 刷新令牌
RefreshTokenExpires field.Field // 刷新令牌过期时间
Scopes_ field.String // 权限范围
CreatedAt field.Field // 创建时间
UpdatedAt field.Field // 更新时间
DeletedAt field.Field // 删除时间
fieldMap map[string]field.Expr
}
func (s session) Table(newTableName string) *session {
s.sessionDo.UseTable(newTableName)
return s.updateTableName(newTableName)
}
func (s session) As(alias string) *session {
s.sessionDo.DO = *(s.sessionDo.As(alias).(*gen.DO))
return s.updateTableName(alias)
}
func (s *session) updateTableName(table string) *session {
s.ALL = field.NewAsterisk(table)
s.ID = field.NewInt32(table, "id")
s.UserID = field.NewInt32(table, "user_id")
s.ClientID = field.NewInt32(table, "client_id")
s.IP = field.NewString(table, "ip")
s.Ua = field.NewString(table, "ua")
s.GrantType = field.NewString(table, "grant_type")
s.AccessToken = field.NewString(table, "access_token")
s.AccessTokenExpires = field.NewField(table, "access_token_expires")
s.RefreshToken = field.NewString(table, "refresh_token")
s.RefreshTokenExpires = field.NewField(table, "refresh_token_expires")
s.Scopes_ = field.NewString(table, "scopes")
s.CreatedAt = field.NewField(table, "created_at")
s.UpdatedAt = field.NewField(table, "updated_at")
s.DeletedAt = field.NewField(table, "deleted_at")
s.fillFieldMap()
return s
}
func (s *session) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
_f, ok := s.fieldMap[fieldName]
if !ok || _f == nil {
return nil, false
}
_oe, ok := _f.(field.OrderExpr)
return _oe, ok
}
func (s *session) fillFieldMap() {
s.fieldMap = make(map[string]field.Expr, 14)
s.fieldMap["id"] = s.ID
s.fieldMap["user_id"] = s.UserID
s.fieldMap["client_id"] = s.ClientID
s.fieldMap["ip"] = s.IP
s.fieldMap["ua"] = s.Ua
s.fieldMap["grant_type"] = s.GrantType
s.fieldMap["access_token"] = s.AccessToken
s.fieldMap["access_token_expires"] = s.AccessTokenExpires
s.fieldMap["refresh_token"] = s.RefreshToken
s.fieldMap["refresh_token_expires"] = s.RefreshTokenExpires
s.fieldMap["scopes"] = s.Scopes_
s.fieldMap["created_at"] = s.CreatedAt
s.fieldMap["updated_at"] = s.UpdatedAt
s.fieldMap["deleted_at"] = s.DeletedAt
}
func (s session) clone(db *gorm.DB) session {
s.sessionDo.ReplaceConnPool(db.Statement.ConnPool)
return s
}
func (s session) replaceDB(db *gorm.DB) session {
s.sessionDo.ReplaceDB(db)
return s
}
type sessionDo struct{ gen.DO }
func (s sessionDo) Debug() *sessionDo {
return s.withDO(s.DO.Debug())
}
func (s sessionDo) WithContext(ctx context.Context) *sessionDo {
return s.withDO(s.DO.WithContext(ctx))
}
func (s sessionDo) ReadDB() *sessionDo {
return s.Clauses(dbresolver.Read)
}
func (s sessionDo) WriteDB() *sessionDo {
return s.Clauses(dbresolver.Write)
}
func (s sessionDo) Session(config *gorm.Session) *sessionDo {
return s.withDO(s.DO.Session(config))
}
func (s sessionDo) Clauses(conds ...clause.Expression) *sessionDo {
return s.withDO(s.DO.Clauses(conds...))
}
func (s sessionDo) Returning(value interface{}, columns ...string) *sessionDo {
return s.withDO(s.DO.Returning(value, columns...))
}
func (s sessionDo) Not(conds ...gen.Condition) *sessionDo {
return s.withDO(s.DO.Not(conds...))
}
func (s sessionDo) Or(conds ...gen.Condition) *sessionDo {
return s.withDO(s.DO.Or(conds...))
}
func (s sessionDo) Select(conds ...field.Expr) *sessionDo {
return s.withDO(s.DO.Select(conds...))
}
func (s sessionDo) Where(conds ...gen.Condition) *sessionDo {
return s.withDO(s.DO.Where(conds...))
}
func (s sessionDo) Order(conds ...field.Expr) *sessionDo {
return s.withDO(s.DO.Order(conds...))
}
func (s sessionDo) Distinct(cols ...field.Expr) *sessionDo {
return s.withDO(s.DO.Distinct(cols...))
}
func (s sessionDo) Omit(cols ...field.Expr) *sessionDo {
return s.withDO(s.DO.Omit(cols...))
}
func (s sessionDo) Join(table schema.Tabler, on ...field.Expr) *sessionDo {
return s.withDO(s.DO.Join(table, on...))
}
func (s sessionDo) LeftJoin(table schema.Tabler, on ...field.Expr) *sessionDo {
return s.withDO(s.DO.LeftJoin(table, on...))
}
func (s sessionDo) RightJoin(table schema.Tabler, on ...field.Expr) *sessionDo {
return s.withDO(s.DO.RightJoin(table, on...))
}
func (s sessionDo) Group(cols ...field.Expr) *sessionDo {
return s.withDO(s.DO.Group(cols...))
}
func (s sessionDo) Having(conds ...gen.Condition) *sessionDo {
return s.withDO(s.DO.Having(conds...))
}
func (s sessionDo) Limit(limit int) *sessionDo {
return s.withDO(s.DO.Limit(limit))
}
func (s sessionDo) Offset(offset int) *sessionDo {
return s.withDO(s.DO.Offset(offset))
}
func (s sessionDo) Scopes(funcs ...func(gen.Dao) gen.Dao) *sessionDo {
return s.withDO(s.DO.Scopes(funcs...))
}
func (s sessionDo) Unscoped() *sessionDo {
return s.withDO(s.DO.Unscoped())
}
func (s sessionDo) Create(values ...*models.Session) error {
if len(values) == 0 {
return nil
}
return s.DO.Create(values)
}
func (s sessionDo) CreateInBatches(values []*models.Session, batchSize int) error {
return s.DO.CreateInBatches(values, batchSize)
}
// Save : !!! underlying implementation is different with GORM
// The method is equivalent to executing the statement: db.Clauses(clause.OnConflict{UpdateAll: true}).Create(values)
func (s sessionDo) Save(values ...*models.Session) error {
if len(values) == 0 {
return nil
}
return s.DO.Save(values)
}
func (s sessionDo) First() (*models.Session, error) {
if result, err := s.DO.First(); err != nil {
return nil, err
} else {
return result.(*models.Session), nil
}
}
func (s sessionDo) Take() (*models.Session, error) {
if result, err := s.DO.Take(); err != nil {
return nil, err
} else {
return result.(*models.Session), nil
}
}
func (s sessionDo) Last() (*models.Session, error) {
if result, err := s.DO.Last(); err != nil {
return nil, err
} else {
return result.(*models.Session), nil
}
}
func (s sessionDo) Find() ([]*models.Session, error) {
result, err := s.DO.Find()
return result.([]*models.Session), err
}
func (s sessionDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*models.Session, err error) {
buf := make([]*models.Session, 0, batchSize)
err = s.DO.FindInBatches(&buf, batchSize, func(tx gen.Dao, batch int) error {
defer func() { results = append(results, buf...) }()
return fc(tx, batch)
})
return results, err
}
func (s sessionDo) FindInBatches(result *[]*models.Session, batchSize int, fc func(tx gen.Dao, batch int) error) error {
return s.DO.FindInBatches(result, batchSize, fc)
}
func (s sessionDo) Attrs(attrs ...field.AssignExpr) *sessionDo {
return s.withDO(s.DO.Attrs(attrs...))
}
func (s sessionDo) Assign(attrs ...field.AssignExpr) *sessionDo {
return s.withDO(s.DO.Assign(attrs...))
}
func (s sessionDo) Joins(fields ...field.RelationField) *sessionDo {
for _, _f := range fields {
s = *s.withDO(s.DO.Joins(_f))
}
return &s
}
func (s sessionDo) Preload(fields ...field.RelationField) *sessionDo {
for _, _f := range fields {
s = *s.withDO(s.DO.Preload(_f))
}
return &s
}
func (s sessionDo) FirstOrInit() (*models.Session, error) {
if result, err := s.DO.FirstOrInit(); err != nil {
return nil, err
} else {
return result.(*models.Session), nil
}
}
func (s sessionDo) FirstOrCreate() (*models.Session, error) {
if result, err := s.DO.FirstOrCreate(); err != nil {
return nil, err
} else {
return result.(*models.Session), nil
}
}
func (s sessionDo) FindByPage(offset int, limit int) (result []*models.Session, count int64, err error) {
result, err = s.Offset(offset).Limit(limit).Find()
if err != nil {
return
}
if size := len(result); 0 < limit && 0 < size && size < limit {
count = int64(size + offset)
return
}
count, err = s.Offset(-1).Limit(-1).Count()
return
}
func (s sessionDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
count, err = s.Count()
if err != nil {
return
}
err = s.Offset(offset).Limit(limit).Scan(result)
return
}
func (s sessionDo) Scan(result interface{}) (err error) {
return s.DO.Scan(result)
}
func (s sessionDo) Delete(models ...*models.Session) (result gen.ResultInfo, err error) {
return s.DO.Delete(models)
}
func (s *sessionDo) withDO(do gen.Dao) *sessionDo {
s.DO = *do.(*gen.DO)
return s
}