MiFaZhan
59eb34976a
iotlcm-admin-svr：管理后台服务。 - Go module：iotlcm → git.5iot.com/iotlcm-public/iotlcm-admin-svr（729 文件） - 凭据与地址脱敏：JWT signingKey（含代码里的 MustGet 默认值）、 MySQL DSN、Redis 口令、告警邮件发件人邮箱全部改为占位； 私网 IP → 文档地址段；库名 xiujie_admin → iotlcm - 真实手机号/邮箱 → 合法 11 位编造号 / example.com - 移除内部资产：CI 触发器、过时 DDL（sql/db_iotlcm.sql）、 私有依赖 go get 行、明文 registry 登录口令 - 启动 banner 统一 IOTLCM 方块字 + 社区惯例署名 - Dockerfile 改用公共基础镜像 - 拼错的 RADEME.md → README.md，README 重写 - 需要真实数据库的用例改为可用 IOTLCM_TEST_WITH_DB=1 显式启用 - 补 Apache-2.0 LICENSE，LF 行尾 + .gitattributes 数据库 schema 不随本仓发布（生产 DDL 已过时），README 已说明。
118 lines
3.1 KiB
Go
118 lines
3.1 KiB
Go
package batchimport
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
const (
|
|
batchNoPrefixLength = 3
|
|
batchNoSequenceLength = 3
|
|
batchNoSequenceMax = 999
|
|
batchNoSequenceTTL = 48 * time.Hour
|
|
batchNoSequenceRedisPrefix = "iotlcm:batch:no:seq:"
|
|
randomNoSuffixLength = 8
|
|
randomNoAlphabet = "ABCDEFGHJKLMNPQRSTUVWXYZ23456789"
|
|
)
|
|
|
|
func NextBatchNo(prefix string) (string, error) {
|
|
return NextBatchNoContext(context.Background(), prefix)
|
|
}
|
|
|
|
func NextBatchNoContext(ctx context.Context, prefix string) (string, error) {
|
|
prefix, err := normalizeBatchNoPrefix(prefix)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
now := time.Now()
|
|
seq, err := nextDailyBatchSequence(ctx, now)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return formatBatchNo(prefix, now, seq)
|
|
}
|
|
|
|
func NextRandomNo(prefix string) (string, error) {
|
|
prefix = strings.TrimSpace(strings.ToUpper(prefix))
|
|
if prefix == "" {
|
|
return "", gerror.New("编号前缀不能为空")
|
|
}
|
|
|
|
random, err := randomNoSuffix(randomNoSuffixLength)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return fmt.Sprintf("%s%s%s", prefix, time.Now().Format("20060102150405"), random), nil
|
|
}
|
|
|
|
func normalizeBatchNoPrefix(prefix string) (string, error) {
|
|
prefix = strings.TrimSpace(strings.ToUpper(prefix))
|
|
if prefix == "" {
|
|
return "", gerror.New("批次号前缀不能为空")
|
|
}
|
|
if len(prefix) != batchNoPrefixLength {
|
|
return "", gerror.Newf("批次号前缀必须为%d位", batchNoPrefixLength)
|
|
}
|
|
return prefix, nil
|
|
}
|
|
|
|
func nextDailyBatchSequence(ctx context.Context, now time.Time) (int64, error) {
|
|
key := batchNoSequenceKey(now)
|
|
script := `
|
|
local v = redis.call("INCR", KEYS[1])
|
|
if v == 1 then
|
|
redis.call("EXPIRE", KEYS[1], ARGV[1])
|
|
end
|
|
return v`
|
|
res, err := g.Redis().Do(ctx, "EVAL", script, 1, key, int(batchNoSequenceTTL.Seconds()))
|
|
if err != nil {
|
|
return 0, gerror.Wrap(err, "生成批次号序号失败")
|
|
}
|
|
seq := res.Int64()
|
|
if seq <= 0 {
|
|
return 0, gerror.New("生成批次号序号失败")
|
|
}
|
|
if seq > batchNoSequenceMax {
|
|
return 0, gerror.Newf("当日批次号序号超过%03d", batchNoSequenceMax)
|
|
}
|
|
return seq, nil
|
|
}
|
|
|
|
func batchNoSequenceKey(now time.Time) string {
|
|
return batchNoSequenceRedisPrefix + now.Format("20060102")
|
|
}
|
|
|
|
func formatBatchNo(prefix string, now time.Time, seq int64) (string, error) {
|
|
if seq <= 0 || seq > batchNoSequenceMax {
|
|
return "", gerror.Newf("批次号序号必须在1-%03d之间", batchNoSequenceMax)
|
|
}
|
|
return fmt.Sprintf("%s%s%0*d", prefix, now.Format("20060102150405"), batchNoSequenceLength, seq), nil
|
|
}
|
|
|
|
func randomNoSuffix(length int) (string, error) {
|
|
if length <= 0 {
|
|
return "", gerror.New("随机码长度必须大于0")
|
|
}
|
|
|
|
buf := make([]byte, length)
|
|
randomBytes := make([]byte, length)
|
|
if _, err := rand.Read(randomBytes); err != nil {
|
|
return "", gerror.Wrap(err, "生成随机码失败")
|
|
}
|
|
|
|
alphabetLen := byte(len(randomNoAlphabet))
|
|
for i, b := range randomBytes {
|
|
buf[i] = randomNoAlphabet[b%alphabetLen]
|
|
}
|
|
return string(buf), nil
|
|
}
|