mirror of https://github.com/veypi/OneAuth.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
713 B
Go
30 lines
713 B
Go
package models
|
|
|
|
import "github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
|
|
var _ TokenModel = (*customTokenModel)(nil)
|
|
|
|
type (
|
|
// TokenModel is an interface to be customized, add more methods here,
|
|
// and implement the added methods in customTokenModel.
|
|
TokenModel interface {
|
|
tokenModel
|
|
withSession(session sqlx.Session) TokenModel
|
|
}
|
|
|
|
customTokenModel struct {
|
|
*defaultTokenModel
|
|
}
|
|
)
|
|
|
|
// NewTokenModel returns a model for the database table.
|
|
func NewTokenModel(conn sqlx.SqlConn) TokenModel {
|
|
return &customTokenModel{
|
|
defaultTokenModel: newTokenModel(conn),
|
|
}
|
|
}
|
|
|
|
func (m *customTokenModel) withSession(session sqlx.Session) TokenModel {
|
|
return NewTokenModel(sqlx.NewSqlConnFromSession(session))
|
|
}
|