跳至主要內容
  • Hostloc 空間訪問刷分
  • 售賣場
  • 廣告位
  • 賣站?

4563博客

全新的繁體中文 WordPress 網站
  • 首頁
  • uber-go/ratelimit 菜鸡求助
未分類
14 12 月 2020

uber-go/ratelimit 菜鸡求助

uber-go/ratelimit 菜鸡求助

資深大佬 : bigHave 6

uber 开源的限流算法,代码不多,尝试理解但是没怎么理解。有没有大佬指点一二.主要不理解的点在于这个 for 循环

// Take blocks to ensure that the time spent between multiple // Take calls is on average time.Second/rate. func (t *limiter) Take() time.Time {  newState := state{}  taken := false  for !taken {   now := t.clock.Now()    previousStatePointer := atomic.LoadPointer(&t.state)   oldState := (*state)(previousStatePointer)    newState = state{}   newState.last = now    // If this is our first request, then we allow it.   if oldState.last.IsZero() {    taken = atomic.CompareAndSwapPointer(&t.state, previousStatePointer, unsafe.Pointer(&newState))    continue   }    // sleepFor calculates how much time we should sleep based on   // the perRequest budget and how long the last request took.   // Since the request may take longer than the budget, this number   // can get negative, and is summed across requests.   newState.sleepFor += t.perRequest - now.Sub(oldState.last)   // We shouldn't allow sleepFor to get too negative, since it would mean that   // a service that slowed down a lot for a short period of time would get   // a much higher RPS following that.   if newState.sleepFor < t.maxSlack {    newState.sleepFor = t.maxSlack   }   if newState.sleepFor > 0 {    newState.last = newState.last.Add(newState.sleepFor)   }   taken = atomic.CompareAndSwapPointer(&t.state, previousStatePointer, unsafe.Pointer(&newState))  }  t.clock.Sleep(newState.sleepFor)  return newState.last } 

大佬有話說 (2)

  • 資深大佬 : CRVV

    var counter int64

    然后如果有多个线程执行下面这个循环,每次 break 都会给 counter 加一
    for {
    counter0 := atomic.LoadInt64(&counter)
    if atomic.CompareAndSwap(&counter, counter0, counter+1) {
    break
    }
    }

    这种就是所谓的 lock-free,把它写成用 lock 的形式,再把那几个 if 去掉,这个函数其实就是

    func (t *limiter) Take() time.Time {

    mutex.Lock()

    oldState := t.state
    newState = state{}
    now := t.clock.Now()
    newState.last = now
    newState.sleepFor += t.perRequest – now.Sub(oldState.last)

    t.state = &newState
    mutex.Unlock()

    t.clock.Sleep(newState.sleepFor)
    return newState.last
    }

    然后应该很好懂了

  • 主 資深大佬 : bigHave

    @CRVV 感谢。第一次在 V2EX 发帖就得到了高效帮助,太棒了

文章導覽

上一篇文章
下一篇文章

AD

其他操作

  • 登入
  • 訂閱網站內容的資訊提供
  • 訂閱留言的資訊提供
  • WordPress.org 台灣繁體中文

51la

4563博客

全新的繁體中文 WordPress 網站
返回頂端
本站採用 WordPress 建置 | 佈景主題採用 GretaThemes 所設計的 Memory
4563博客
  • Hostloc 空間訪問刷分
  • 售賣場
  • 廣告位
  • 賣站?
在這裡新增小工具