54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package code
|
||
|
||
import (
|
||
"fmt"
|
||
"samba/util/model"
|
||
"samba/util/util"
|
||
"strings"
|
||
)
|
||
|
||
// 定义字符集，排除L、O、I、0、1
|
||
var charset = []rune("ABCDEFGHJKMNPQRSTUVWXYZ23456789")
|
||
|
||
// RedeemCodeGenerator 兑换码生成器
|
||
type RedeemCodeGenerator struct {
|
||
}
|
||
|
||
// UnsafeNext 生成下一个兑换码
|
||
func (g RedeemCodeGenerator) UnsafeNext() string {
|
||
var b strings.Builder
|
||
for i := 0; i < 8; i++ {
|
||
b.WriteRune(util.RandChoice(charset))
|
||
}
|
||
return b.String()
|
||
}
|
||
|
||
// Next 生成下一个兑换码,附带去重判断
|
||
func (g RedeemCodeGenerator) Next() string {
|
||
op := model.NewRedeemCodeOp()
|
||
code := g.UnsafeNext()
|
||
for op.IsExist(code) {
|
||
code = g.UnsafeNext()
|
||
}
|
||
return code
|
||
}
|
||
|
||
// IsValid 验证兑换码是否合法
|
||
func (g RedeemCodeGenerator) IsValid(code string) error {
|
||
if len(code) != 8 {
|
||
return fmt.Errorf("%w:length must be 8", ErrCodeFormat)
|
||
}
|
||
charsetMp := make(map[rune]struct{}, len(charset))
|
||
for _, r := range charset {
|
||
charsetMp[r] = struct{}{}
|
||
}
|
||
for _, r := range code {
|
||
if _, ok := charsetMp[r]; !ok {
|
||
return fmt.Errorf("%w: invalid char %s", ErrCodeFormat, string(r))
|
||
}
|
||
}
|
||
|
||
return nil
|
||
|
||
}
|