MiFaZhan
88dfe5c640
iotlcm-openapi：对外 OpenAPI 网关（V3 接口 + 旧版兼容接口）。
- Go module：iotlcm-openapi → git.5iot.com/iotlcm-public/iotlcm-openapi（178 文件）
- 凭据与地址脱敏：JWT signingKey、MySQL DSN、Redis 口令、
RabbitMQ DSN 全部改为 ${*} 占位；私网 IP → 文档地址段
- 7 处真实手机号 → 合法 11 位编造号
- 启动 banner 统一 IOTLCM 方块字 + 社区惯例署名
- 移除内部资产：CI 触发器
- Dockerfile 改用公共基础镜像，移除 GOPRIVATE 内网设置
- README 重写（原为 GoFrame 模板占位文案），补 Apache-2.0 LICENSE，
LF 行尾 + .gitattributes
84 lines
2.2 KiB
Go
84 lines
2.2 KiB
Go
package batchimport
|
|
|
|
import (
|
|
"context"
|
|
"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:"
|
|
)
|
|
|
|
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 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
|
|
}
|