108 lines
3.4 KiB
Go
108 lines
3.4 KiB
Go
package dto
|
||
|
||
import (
|
||
"errors"
|
||
"regexp"
|
||
"time"
|
||
|
||
validation "github.com/go-ozzo/ozzo-validation/v4"
|
||
)
|
||
|
||
// TeamKey 结构体定义
|
||
type TeamKey struct {
|
||
ID *int64 `json:"id,omitempty"`
|
||
UserID *int64 `json:"userID,omitempty"`
|
||
Name *string `json:"name,omitempty"` // 必须
|
||
Key *string `json:"key,omitempty"`
|
||
Status *int64 `json:"status,omitempty"` // 默认1 允许，0禁止
|
||
Quota *int64 `json:"quota,omitempty"` // UnlimitedQuota不为1 的时候必须
|
||
UnlimitedQuota *bool `json:"unlimitedQuota,omitempty"` // 默认1 不限制，0限制
|
||
UsedQuota *int64 `json:"usedQuota,omitempty"`
|
||
CreatedAt *int64 `json:"createdAt,omitempty"`
|
||
ExpiredAt *int64 `json:"expiredAt,omitempty"` // 可选
|
||
}
|
||
|
||
// DefaultTeamKey 创建一个具有默认值的 TeamKey
|
||
func DefaultTeamKey() TeamKey {
|
||
status := int64(1) // 默认允许
|
||
unlimitedQuota := true // 默认不限制
|
||
createdAt := time.Now().Unix()
|
||
|
||
return TeamKey{
|
||
Status: &status,
|
||
UnlimitedQuota: &unlimitedQuota,
|
||
CreatedAt: &createdAt,
|
||
}
|
||
}
|
||
|
||
// Validate 验证 TeamKey 结构体
|
||
func (t TeamKey) Validate() error {
|
||
// 自定义验证规则
|
||
var quotaRule validation.Rule = validation.Skip
|
||
if t.UnlimitedQuota != nil && !*t.UnlimitedQuota {
|
||
quotaRule = validation.Required.Error("当 UnlimitedQuota 为 false 时，Quota 是必填项")
|
||
}
|
||
|
||
// 过期时间校验
|
||
var expiredAtRule validation.Rule = validation.Skip
|
||
if t.ExpiredAt != nil {
|
||
expiredAtRule = validation.Min(time.Now().Unix()).Error("过期时间不能早于当前时间")
|
||
}
|
||
|
||
return validation.ValidateStruct(&t,
|
||
// ID 通常由系统生成，不需要验证
|
||
|
||
// UserID 可选，但如果提供必须大于 0
|
||
validation.Field(&t.UserID,
|
||
validation.When(t.UserID != nil, validation.Min(int64(1)).Error("用户 ID 必须大于 0"))),
|
||
|
||
// Name 是必填字段
|
||
validation.Field(&t.Name,
|
||
validation.Required.Error("名称不能为空"),
|
||
validation.When(t.Name != nil, validation.Length(1, 100).Error("名称长度应在 1-100 之间"))),
|
||
|
||
// Key 可选，但如果提供需要符合特定格式
|
||
validation.Field(&t.Key,
|
||
validation.When(t.Key != nil,
|
||
validation.Length(1, 255).Error("Key 长度应在 1-255 之间")),
|
||
validation.Match(regexp.MustCompile(`^[^\s]+$`)).Error("Key 不能包含空格"),
|
||
),
|
||
|
||
// Status 只能是 0 或 1
|
||
validation.Field(&t.Status,
|
||
validation.When(t.Status != nil, validation.In(int64(0), int64(1)).Error("状态只能是 0(禁止) 或 1(允许)"))),
|
||
|
||
// Quota 要求依赖于 UnlimitedQuota
|
||
validation.Field(&t.Quota, quotaRule,
|
||
validation.When(t.Quota != nil, validation.Min(int64(1)).Error("配额必须大于 0"))),
|
||
|
||
// UnlimitedQuota 是否限制配额
|
||
validation.Field(&t.UnlimitedQuota),
|
||
|
||
// UsedQuota 系统维护，不需要验证
|
||
validation.Field(&t.UsedQuota,
|
||
validation.When(t.UsedQuota != nil, validation.Min(int64(0)).Error("已使用配额不能为负数"))),
|
||
|
||
// CreatedAt 系统维护，不需要验证
|
||
validation.Field(&t.CreatedAt),
|
||
|
||
// ExpiredAt 可选，但如果提供必须大于当前时间
|
||
validation.Field(&t.ExpiredAt, expiredAtRule),
|
||
)
|
||
}
|
||
|
||
// ValidateCreate 创建时的特殊验证
|
||
func (t TeamKey) ValidateCreate() error {
|
||
// 首先进行基本验证
|
||
if err := t.Validate(); err != nil {
|
||
return err
|
||
}
|
||
|
||
// 创建时的额外验证
|
||
if t.Name == nil {
|
||
return errors.New("创建时必须提供名称")
|
||
}
|
||
|
||
return nil
|
||
}
|