From 2688e30907eef5f761959fcf2c87a1cde633e454 Mon Sep 17 00:00:00 2001 From: Qian Wang Date: Mon, 3 Dec 2018 20:33:14 +0000 Subject: [PATCH] Inline formatting --- internal/multiplex/qos.go | 42 ++++++++++----------------------------- 1 file changed, 10 insertions(+), 32 deletions(-) diff --git a/internal/multiplex/qos.go b/internal/multiplex/qos.go index 933bb9c..4c5ebfe 100644 --- a/internal/multiplex/qos.go +++ b/internal/multiplex/qos.go @@ -31,46 +31,24 @@ func MakeValve(rxRate, txRate int64, rxCredit, txCredit *int64) *Valve { return v } -// TODO: inline formatting +func (v *Valve) SetRxRate(rate int64) { v.rxtb.Store(ratelimit.NewBucketWithRate(float64(rate), rate)) } -func (v *Valve) SetRxRate(rate int64) { - v.rxtb.Store(ratelimit.NewBucketWithRate(float64(rate), rate)) -} +func (v *Valve) SetTxRate(rate int64) { v.txtb.Store(ratelimit.NewBucketWithRate(float64(rate), rate)) } -func (v *Valve) SetTxRate(rate int64) { - v.txtb.Store(ratelimit.NewBucketWithRate(float64(rate), rate)) -} +func (v *Valve) rxWait(n int) { v.rxtb.Load().(*ratelimit.Bucket).Wait(int64(n)) } -func (v *Valve) rxWait(n int) { - v.rxtb.Load().(*ratelimit.Bucket).Wait(int64(n)) -} +func (v *Valve) txWait(n int) { v.txtb.Load().(*ratelimit.Bucket).Wait(int64(n)) } -func (v *Valve) txWait(n int) { - v.txtb.Load().(*ratelimit.Bucket).Wait(int64(n)) -} +func (v *Valve) SetRxCredit(n int64) { atomic.StoreInt64(v.rxCredit, n) } -func (v *Valve) SetRxCredit(n int64) { - atomic.StoreInt64(v.rxCredit, n) -} +func (v *Valve) SetTxCredit(n int64) { atomic.StoreInt64(v.txCredit, n) } -func (v *Valve) SetTxCredit(n int64) { - atomic.StoreInt64(v.txCredit, n) -} +func (v *Valve) GetRxCredit() int64 { return atomic.LoadInt64(v.rxCredit) } -func (v *Valve) GetRxCredit() int64 { - return atomic.LoadInt64(v.rxCredit) -} - -func (v *Valve) GetTxCredit() int64 { - return atomic.LoadInt64(v.txCredit) -} +func (v *Valve) GetTxCredit() int64 { return atomic.LoadInt64(v.txCredit) } // n can be negative -func (v *Valve) AddRxCredit(n int64) int64 { - return atomic.AddInt64(v.rxCredit, n) -} +func (v *Valve) AddRxCredit(n int64) int64 { return atomic.AddInt64(v.rxCredit, n) } // n can be negative -func (v *Valve) AddTxCredit(n int64) int64 { - return atomic.AddInt64(v.txCredit, n) -} +func (v *Valve) AddTxCredit(n int64) int64 { return atomic.AddInt64(v.txCredit, n) }