96b1edff45
iotlcm-upstream-chan:运营商上游通道适配服务(统一 /sim/* 协议)。 - 内联原私有依赖:yunze-public → internal/pub/(10 包 18 文件), yunze-pb/pb → internal/pb/(9 文件);Consul 注册改用官方 gogf/gf/contrib/registry/consul/v2(API 兼容) - Go module:yz-svr → git.5iot.com/iotlcm-public/iotlcm-upstream-chan - 修复 mingqian 包测试:实现重构后 `adapter.operations` 字段已移除, 测试断言按当前实现语义重写(不改产品行为) - 凭据与内部标识脱敏:MySQL/Redis/Consul 凭据、私网与生产 IP、 内网域名、真实手机号、注释里的真实 token/transid - 移除内部资产:CI 触发器、生产通道种子 SQL、真实凭据测试资源文档 - Dockerfile 改用公共基础镜像,移除写死的 registry 凭据 - 补 Apache-2.0 LICENSE,重写 README,LF 行尾 + .gitattributes
69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
package upstreamimpl
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
const tibiotSimDataMarginInterval = 650 * time.Millisecond
|
|
|
|
// tibiotSimDataMarginLimiter spaces requests globally because Tibiot applies
|
|
// the simDataMargin quota to the gateway's public IP, not to a channel.
|
|
var tibiotSimDataMarginLimiter = newTibiotRateLimiter(tibiotSimDataMarginInterval)
|
|
|
|
type tibiotRateLimiter struct {
|
|
mu sync.Mutex
|
|
interval time.Duration
|
|
next time.Time
|
|
}
|
|
|
|
func newTibiotRateLimiter(interval time.Duration) *tibiotRateLimiter {
|
|
if interval <= 0 {
|
|
interval = tibiotSimDataMarginInterval
|
|
}
|
|
return &tibiotRateLimiter{interval: interval}
|
|
}
|
|
|
|
// wait blocks until a slot is available and only consumes the slot when the
|
|
// caller's context is still active. Waiting callers do not reserve future
|
|
// slots, so canceled requests cannot delay later retries.
|
|
func (l *tibiotRateLimiter) wait(ctx context.Context) error {
|
|
if l == nil {
|
|
return nil
|
|
}
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
|
|
for {
|
|
if err := ctx.Err(); err != nil {
|
|
return err
|
|
}
|
|
|
|
l.mu.Lock()
|
|
now := time.Now()
|
|
if !l.next.After(now) {
|
|
// Check again while holding the mutex so a cancellation racing with
|
|
// the slot boundary cannot consume a slot unnecessarily.
|
|
if err := ctx.Err(); err != nil {
|
|
l.mu.Unlock()
|
|
return err
|
|
}
|
|
l.next = now.Add(l.interval)
|
|
l.mu.Unlock()
|
|
return nil
|
|
}
|
|
wakeAt := l.next
|
|
l.mu.Unlock()
|
|
|
|
timer := time.NewTimer(time.Until(wakeAt))
|
|
select {
|
|
case <-timer.C:
|
|
case <-ctx.Done():
|
|
timer.Stop()
|
|
return ctx.Err()
|
|
}
|
|
}
|
|
}
|