Golang 业务限流器

xiaowei520 · · 1185 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

github地址 https://github.com/xiaowei520...

package rate

import (
    "math/rand"
    "sync"
    "sync/atomic"
    "time"
)

//使用cas 锁来实现、减少开销
//业务限流器
const (
    Normal = iota
    AbNormal
)

//标示, 可以是appId  uid 唯一性的代表值
type Limit float64

type Limiter struct {
    limit     Limit
    EventName string
    //cas
    done uint32

    //时间
    retryTime int
    //重试次数
    retryLen int

    //最大缓冲时间
    maxTime int64
    //最小缓冲时间
    minTime int64

    mu sync.Mutex
    // last is the last time the limiter's tokens field was updated
    last time.Time
    // lastEvent is the latest time of a rate-limited event (past or future)
    lastEvent time.Time
}

func NewLimiter(r Limit, n string) *Limiter {

    return &Limiter{
        limit:     r,
        EventName: n,
    }
}

func DefaultLimiter(r Limit, n string) *Limiter {
    return &Limiter{
        limit:     r,
        EventName: n,
        maxTime:   1000,
        minTime:   100,
    }
}

func (lim *Limiter) Allow() bool {
    //no limit rate
    if atomic.LoadUint32(&lim.done) == Normal {
        return true
    }
    //rand sleep, avoid conflict
    randTime := rand.Int63n(lim.maxTime-lim.minTime) + lim.minTime
    time.Sleep(time.Duration(randTime) * time.Millisecond)
    return false
}

//where use stop
func (lim *Limiter) Stop() bool {
    lim.mu.Lock()
    defer lim.mu.Unlock()
    if lim.done == Normal {
        defer atomic.StoreUint32(&lim.done, AbNormal)
    }

    return true
}

func (lim *Limiter) Recover() {

    if atomic.LoadUint32(&lim.done) == AbNormal {
        return
    }
    lim.mu.Lock()
    defer lim.mu.Unlock()
    if lim.done == AbNormal {
        defer atomic.StoreUint32(&lim.done, Normal)
    }
}


有疑问加站长微信联系(非本文作者)

本文来自:Segmentfault

感谢作者:xiaowei520

查看原文:Golang 业务限流器

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:701969077

1185 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传