Files
iotlcm/internal/logic/upstreamimpl/tibiot_rate_limiter.go
T
MiFaZhan 96b1edff45 chore: 初始开源发布
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
2026-10-22 16:08:06 +08:00

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()
}
}
}