MiFaZhan
2bec257289
iotlcm-worker-svr：异步执行中枢（RabbitMQ 消费者 + 全量同步编排）。 - Go module：iotlcm-worker-svr → git.5iot.com/iotlcm-public/iotlcm-worker-svr（214 文件） - 凭据与内部标识脱敏：MySQL/Redis 凭据、私网与生产 IP、内网域名、 库名 xiujie_admin → iotlcm、哨兵名占位 - 159 处真实手机号 → 合法 11 位编造号（保留号段与后 4 位，维持互异， 不破坏「校验 MSISDN 格式」等测试断言） - 移除内部资产：CI 触发器、Jenkins 触发测试文档、预演门禁与镜像 digest 记录 - Dockerfile 改用公共基础镜像（golang:1.25-alpine → alpine:3.23）， 移除写死的 git 凭据 - README 重写（架构/全量同步模型/配置/已知失败），补 Apache-2.0 LICENSE，LF 行尾 + .gitattributes 已知：3 个测试包在源仓同一提交上同样失败（client/upstream、 service/allocate、service/inbound），属既有缺陷，本仓未引入亦未修复。
89 lines
2.0 KiB
Go
89 lines
2.0 KiB
Go
package allocate
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
const (
|
|
defaultAllocateDeadlockMaxRetries = 3
|
|
defaultAllocateDeadlockRetryDelay = 200 * time.Millisecond
|
|
)
|
|
|
|
type allocateParentLock struct {
|
|
mu sync.Mutex
|
|
refs int
|
|
}
|
|
|
|
var allocateParentLocks = struct {
|
|
mu sync.Mutex
|
|
values map[string]*allocateParentLock
|
|
}{values: make(map[string]*allocateParentLock)}
|
|
|
|
func acquireAllocateParentLock(batchNo string) func() {
|
|
allocateParentLocks.mu.Lock()
|
|
lock := allocateParentLocks.values[batchNo]
|
|
if lock == nil {
|
|
lock = &allocateParentLock{}
|
|
allocateParentLocks.values[batchNo] = lock
|
|
}
|
|
lock.refs++
|
|
allocateParentLocks.mu.Unlock()
|
|
|
|
lock.mu.Lock()
|
|
return func() {
|
|
lock.mu.Unlock()
|
|
allocateParentLocks.mu.Lock()
|
|
lock.refs--
|
|
if lock.refs == 0 {
|
|
delete(allocateParentLocks.values, batchNo)
|
|
}
|
|
allocateParentLocks.mu.Unlock()
|
|
}
|
|
}
|
|
|
|
func allocateDeadlockMaxRetries(ctx context.Context) int {
|
|
retries := g.Cfg().MustGet(ctx, "batch.allocate.deadlockMaxRetries", defaultAllocateDeadlockMaxRetries).Int()
|
|
if retries < 0 {
|
|
return 0
|
|
}
|
|
return retries
|
|
}
|
|
|
|
func allocateDeadlockRetryDelay(ctx context.Context, attempt int) time.Duration {
|
|
delay := g.Cfg().MustGet(ctx, "batch.allocate.deadlockRetryDelay", defaultAllocateDeadlockRetryDelay).Duration()
|
|
if delay <= 0 {
|
|
delay = defaultAllocateDeadlockRetryDelay
|
|
}
|
|
for i := 0; i < attempt; i++ {
|
|
if delay >= time.Second {
|
|
return time.Second
|
|
}
|
|
delay *= 2
|
|
}
|
|
return delay
|
|
}
|
|
|
|
func waitAllocateDeadlockRetry(ctx context.Context, attempt int) error {
|
|
timer := time.NewTimer(allocateDeadlockRetryDelay(ctx, attempt))
|
|
defer timer.Stop()
|
|
select {
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
case <-timer.C:
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func isAllocateDeadlock(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
message := strings.ToLower(err.Error())
|
|
return strings.Contains(message, "deadlock found") || strings.Contains(message, "error 1213") || strings.Contains(message, "1213 (40001)")
|
|
}
|