HouYunFei
2181b3b885
- 在管理后台设置页面添加模型算力点配置表格 - 实现后端AI接口调用时按模型扣除用户算力点 - 添加算力点余额检查和不足时的错误处理 - 在credit_logs中记录AI调用消费流水 - 更新前端远程调用后刷新用户信息显示 - 文档中补充模型算力点配置说明
102 lines
2.8 KiB
Go
102 lines
2.8 KiB
Go
package model
|
|
|
|
type UserRole string
|
|
|
|
const (
|
|
UserRoleGuest UserRole = "guest"
|
|
UserRoleUser UserRole = "user"
|
|
UserRoleAdmin UserRole = "admin"
|
|
)
|
|
|
|
type UserStatus string
|
|
|
|
const (
|
|
UserStatusActive UserStatus = "active"
|
|
UserStatusBan UserStatus = "ban"
|
|
)
|
|
|
|
// User 系统用户。
|
|
type User struct {
|
|
ID string `json:"id" gorm:"primaryKey"`
|
|
Username string `json:"username" gorm:"uniqueIndex"`
|
|
Password string `json:"password,omitempty"`
|
|
Email string `json:"email"`
|
|
DisplayName string `json:"displayName"`
|
|
AvatarURL string `json:"avatarUrl"`
|
|
Role UserRole `json:"role"`
|
|
Credits int `json:"credits"`
|
|
AffCode string `json:"affCode" gorm:"uniqueIndex"`
|
|
AffCount int `json:"affCount"`
|
|
InviterID string `json:"inviterId"`
|
|
GithubID string `json:"githubId"`
|
|
LinuxDoID string `json:"linuxDoId" gorm:"index"`
|
|
WechatID string `json:"wechatId"`
|
|
Status UserStatus `json:"status"`
|
|
LastLoginAt string `json:"lastLoginAt"`
|
|
Extra string `json:"extra" gorm:"type:text"`
|
|
CreatedAt string `json:"createdAt"`
|
|
UpdatedAt string `json:"updatedAt"`
|
|
}
|
|
|
|
// UserList 用户分页结果。
|
|
type UserList struct {
|
|
Items []User `json:"items"`
|
|
Total int `json:"total"`
|
|
}
|
|
|
|
// AuthUser 用户公开信息。
|
|
type AuthUser struct {
|
|
ID string `json:"id"`
|
|
Username string `json:"username"`
|
|
DisplayName string `json:"displayName"`
|
|
AvatarURL string `json:"avatarUrl"`
|
|
Role UserRole `json:"role"`
|
|
Credits int `json:"credits"`
|
|
CreatedAt string `json:"createdAt"`
|
|
UpdatedAt string `json:"updatedAt"`
|
|
}
|
|
|
|
// AuthSession 登录会话信息。
|
|
type AuthSession struct {
|
|
Token string `json:"token"`
|
|
User AuthUser `json:"user"`
|
|
}
|
|
|
|
func PublicUser(user User) AuthUser {
|
|
return AuthUser{
|
|
ID: user.ID,
|
|
Username: user.Username,
|
|
DisplayName: user.DisplayName,
|
|
AvatarURL: user.AvatarURL,
|
|
Role: user.Role,
|
|
Credits: user.Credits,
|
|
CreatedAt: user.CreatedAt,
|
|
UpdatedAt: user.UpdatedAt,
|
|
}
|
|
}
|
|
|
|
type CreditLogType string
|
|
|
|
const (
|
|
CreditLogTypeAdminAdjust CreditLogType = "admin_adjust"
|
|
CreditLogTypeAIConsume CreditLogType = "ai_consume"
|
|
)
|
|
|
|
// CreditLog 用户算力点变更流水。
|
|
type CreditLog struct {
|
|
ID string `json:"id" gorm:"primaryKey"`
|
|
UserID string `json:"userId" gorm:"index"`
|
|
Type CreditLogType `json:"type"`
|
|
Amount int `json:"amount"`
|
|
Balance int `json:"balance"`
|
|
RelatedID string `json:"relatedId"`
|
|
Remark string `json:"remark"`
|
|
Extra string `json:"extra" gorm:"type:text"`
|
|
CreatedAt string `json:"createdAt"`
|
|
}
|
|
|
|
type CreditLogList struct {
|
|
Items []CreditLog `json:"items"`
|
|
Total int `json:"total"`
|
|
}
|