2025-05-31 12:03:22 +08:00
|
|
|
|
package impl
|
2025-05-31 16:14:34 +08:00
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
|
"time"
|
|
|
|
|
|
|
|
|
|
|
|
"122.51.31.227/go-course/go18/devcloud/mcenter/apps/token"
|
|
|
|
|
|
"122.51.31.227/go-course/go18/devcloud/mcenter/apps/user"
|
|
|
|
|
|
"github.com/infraboard/mcube/v2/ioc"
|
|
|
|
|
|
"github.com/infraboard/mcube/v2/ioc/config/datasource"
|
|
|
|
|
|
"github.com/infraboard/mcube/v2/ioc/config/log"
|
|
|
|
|
|
"github.com/rs/zerolog"
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
func init() {
|
|
|
|
|
|
ioc.Controller().Registry(&TokenServiceImpl{
|
|
|
|
|
|
AutoRefresh: true,
|
|
|
|
|
|
RereshTTLSecond: 1 * 60 * 60,
|
|
|
|
|
|
})
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
var _ token.Service = (*TokenServiceImpl)(nil)
|
|
|
|
|
|
|
|
|
|
|
|
type TokenServiceImpl struct {
|
|
|
|
|
|
ioc.ObjectImpl
|
|
|
|
|
|
user user.Service
|
|
|
|
|
|
log *zerolog.Logger
|
|
|
|
|
|
// policy policy.PermissionService
|
|
|
|
|
|
|
|
|
|
|
|
// 自动刷新, 直接刷新Token的过期时间，而不是生成一个新Token
|
|
|
|
|
|
AutoRefresh bool `json:"auto_refresh" toml:"auto_refresh" yaml:"auto_refresh" env:"AUTO_REFRESH"`
|
|
|
|
|
|
// 刷新TTL
|
|
|
|
|
|
RereshTTLSecond uint64 `json:"refresh_ttl" toml:"refresh_ttl" yaml:"refresh_ttl" env:"REFRESH_TTL"`
|
|
|
|
|
|
// Api最多多少个, 这种Token往往过期时间比较长, 为了安全不要申请太多
|
|
|
|
|
|
MaxActiveApiToken uint8 `json:"max_active_api_token" toml:"max_active_api_token" yaml:"max_active_api_token" env:"MAX_ACTIVE_API_TOKEN"`
|
|
|
|
|
|
|
|
|
|
|
|
refreshDuration time.Duration
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func (i *TokenServiceImpl) Init() error {
|
|
|
|
|
|
i.log = log.Sub(i.Name())
|
|
|
|
|
|
i.user = user.GetService()
|
|
|
|
|
|
// i.policy = policy.GetService()
|
|
|
|
|
|
i.refreshDuration = time.Duration(i.RereshTTLSecond) * time.Second
|
|
|
|
|
|
|
|
|
|
|
|
if datasource.Get().AutoMigrate {
|
|
|
|
|
|
err := datasource.DB().AutoMigrate(&token.Token{})
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
return err
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
return nil
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func (i *TokenServiceImpl) Name() string {
|
|
|
|
|
|
return token.APP_NAME
|
|
|
|
|
|
}
|
