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),属既有缺陷,本仓未引入亦未修复。
91 lines
2.4 KiB
Go
91 lines
2.4 KiB
Go
package supply
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"sort"
|
|
"sync"
|
|
"sync/atomic"
|
|
)
|
|
|
|
// Tracker routes optional consumer events to the local producer attempt.
|
|
type Tracker struct {
|
|
mu sync.RWMutex
|
|
runs map[string]*Observer
|
|
LocalConsumer bool
|
|
ConsumerActive atomic.Bool
|
|
}
|
|
|
|
func NewTracker(localConsumer bool) *Tracker {
|
|
return &Tracker{runs: map[string]*Observer{}, LocalConsumer: localConsumer}
|
|
}
|
|
func (t *Tracker) Put(run string, o *Observer) { t.mu.Lock(); t.runs[run] = o; t.mu.Unlock() }
|
|
func (t *Tracker) Remove(run string) { t.mu.Lock(); delete(t.runs, run); t.mu.Unlock() }
|
|
func (t *Tracker) Empty(run, key string) {
|
|
t.mu.RLock()
|
|
defer t.mu.RUnlock()
|
|
if o := t.runs[run]; o != nil {
|
|
o.Empty(key)
|
|
}
|
|
}
|
|
func (t *Tracker) Processing(run, key string, delta int) {
|
|
t.mu.RLock()
|
|
defer t.mu.RUnlock()
|
|
if o := t.runs[run]; o != nil {
|
|
o.Processing(key, delta)
|
|
}
|
|
}
|
|
func (o *Observer) Restore(previous Snapshot) {
|
|
o.mu.Lock()
|
|
defer o.mu.Unlock()
|
|
o.partial = true
|
|
o.retained = previous.Summary
|
|
o.truncated = previous.Truncated
|
|
for _, c := range previous.Channels {
|
|
if s := o.channels[c.Key]; s != nil {
|
|
if c.Startup.TotalMS+c.Confirmed.TotalMS > 0 {
|
|
o.retained.Confirmed--
|
|
}
|
|
o.retained.TotalMS -= c.Startup.TotalMS + c.Confirmed.TotalMS
|
|
o.retained.UnknownMS -= c.UnknownMS
|
|
o.retained.UnconfirmedMS -= c.Unconfirmed.TotalMS
|
|
o.retained.ConsumerMS -= c.Consumer.TotalMS
|
|
s.Channel = c
|
|
if s.Stages == nil {
|
|
s.Stages = map[string]int64{}
|
|
}
|
|
s.Termination = ""
|
|
s.MissingSamples++
|
|
s.close()
|
|
}
|
|
}
|
|
}
|
|
func (o *Observer) Encode() ([]byte, error) {
|
|
snapshot := o.Snapshot()
|
|
encoded, err := json.Marshal(snapshot)
|
|
if err != nil || len(encoded) <= 256*1024 {
|
|
return encoded, err
|
|
}
|
|
snapshot.Truncated = true
|
|
snapshot.Partial = true
|
|
sort.Slice(snapshot.Channels, func(i, j int) bool {
|
|
a, b := snapshot.Channels[i], snapshot.Channels[j]
|
|
av, bv := max(a.Startup.LongestMS, a.Confirmed.LongestMS), max(b.Startup.LongestMS, b.Confirmed.LongestMS)
|
|
if av == bv {
|
|
return a.Key < b.Key
|
|
}
|
|
return av > bv
|
|
})
|
|
snapshot.Channels = snapshot.Channels[:min(100, len(snapshot.Channels))]
|
|
for {
|
|
encoded, err = json.Marshal(snapshot)
|
|
if err != nil || len(encoded) <= 256*1024 {
|
|
return encoded, err
|
|
}
|
|
if len(snapshot.Channels) == 0 {
|
|
return nil, fmt.Errorf("supply summary exceeds snapshot limit")
|
|
}
|
|
snapshot.Channels = snapshot.Channels[:len(snapshot.Channels)-1]
|
|
}
|
|
}
|