|
|
package schema
|
|
|
|
|
|
import (
|
|
|
"entgo.io/ent"
|
|
|
"entgo.io/ent/dialect"
|
|
|
"entgo.io/ent/dialect/entsql"
|
|
|
"entgo.io/ent/schema"
|
|
|
"entgo.io/ent/schema/edge"
|
|
|
"entgo.io/ent/schema/field"
|
|
|
"git.noahlan.cn/noahlan/ntool-biz/core/orm/nent/mixins"
|
|
|
"git.noahlan.cn/noahlan/ntool/nrandom"
|
|
|
"git.noahlan.cn/noahlan/ntool/nstd/tea"
|
|
|
)
|
|
|
|
|
|
type Token struct {
|
|
|
ent.Schema
|
|
|
}
|
|
|
|
|
|
func (Token) Fields() []ent.Field {
|
|
|
return []ent.Field{
|
|
|
field.Int64("user_id").
|
|
|
Annotations(entsql.Annotation{Incremental: tea.Bool(false)}).
|
|
|
Comment("User's ID | 用户ID"),
|
|
|
field.String("token_type").Comment("Token type | 凭证类型 [Bearer]"),
|
|
|
field.String("access_token").Comment("AccessToken | AccessToken 字符串"),
|
|
|
field.String("refresh_token").Comment("RefreshToken | RefreshToken 字符串"),
|
|
|
field.String("source").Comment("Source | Token来源，PCWeb/WechatMini/AlipayMini"),
|
|
|
field.Time("expired_at").SchemaType(map[string]string{dialect.MySQL: "datetime"}).Comment("Expire Time | 过期时间"),
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func (Token) Mixin() []ent.Mixin {
|
|
|
return []ent.Mixin{
|
|
|
mixins.NewDistributedIDMixin(func() int64 { return nrandom.SnowflakeId() }),
|
|
|
mixins.TimeMixin{},
|
|
|
mixins.StatusMixin{},
|
|
|
mixins.OptimisticLockMixin{},
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func (Token) Edges() []ent.Edge {
|
|
|
return []ent.Edge{
|
|
|
edge.From("user", User.Type).
|
|
|
Ref("token").
|
|
|
Field("user_id").
|
|
|
Unique().
|
|
|
Required(),
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func (Token) Indexes() []ent.Index {
|
|
|
return []ent.Index{}
|
|
|
}
|
|
|
|
|
|
func (Token) Annotations() []schema.Annotation {
|
|
|
return []schema.Annotation{
|
|
|
entsql.WithComments(true),
|
|
|
entsql.Annotation{Table: "sys_token"},
|
|
|
schema.Comment("Token表"),
|
|
|
}
|
|
|
}
|