Skip to content

Commit

Permalink
merge master
Browse files Browse the repository at this point in the history
Signed-off-by: Cabinfever_B <cabinfeveroier@gmail.com>
  • Loading branch information
CabinfeverB committed Jan 11, 2023
1 parent 69a3ac0 commit 62b275c
Showing 1 changed file with 0 additions and 35 deletions.
35 changes: 0 additions & 35 deletions pkg/mcs/resource_manager/server/resource_group.go
Original file line number Diff line number Diff line change
Expand Up @@ -172,60 +172,25 @@ func FromProtoResourceGroup(group *rmpb.ResourceGroup) *ResourceGroup {
return rg
}

<<<<<<< HEAD
// UpdateRRU updates the RRU of the resource group.
func (rg *ResourceGroup) UpdateRRU(now time.Time) {
rg.Lock()
defer rg.Unlock()
if rg.RUSettings != nil {
rg.RUSettings.RRU.update(now)
}
}

// UpdateWRU updates the WRU of the resource group.
func (rg *ResourceGroup) UpdateWRU(now time.Time) {
rg.Lock()
defer rg.Unlock()
if rg.RUSettings != nil {
rg.RUSettings.WRU.update(now)
}
}

// RequestRRU requests the RRU of the resource group.
func (rg *ResourceGroup) RequestRRU(neededTokens float64, targetPeriodMs uint64) *rmpb.GrantedRUTokenBucket {
=======
// RequestRRU requests the RRU of the resource group.
func (rg *ResourceGroup) RequestRRU(now time.Time, neededTokens float64, targetPeriodMs uint64) *rmpb.GrantedRUTokenBucket {
>>>>>>> master
rg.Lock()
defer rg.Unlock()
if rg.RUSettings == nil {
return nil
}
<<<<<<< HEAD
tb, trickleTimeMs := rg.RUSettings.RRU.request(neededTokens, targetPeriodMs)
=======
tb, trickleTimeMs := rg.RUSettings.RRU.request(now, neededTokens, targetPeriodMs)
>>>>>>> master
return &rmpb.GrantedRUTokenBucket{Type: rmpb.RequestUnitType_RRU, GrantedTokens: tb, TrickleTimeMs: trickleTimeMs}
}

// RequestWRU requests the WRU of the resource group.
<<<<<<< HEAD
func (rg *ResourceGroup) RequestWRU(neededTokens float64, targetPeriodMs uint64) *rmpb.GrantedRUTokenBucket {
=======
func (rg *ResourceGroup) RequestWRU(now time.Time, neededTokens float64, targetPeriodMs uint64) *rmpb.GrantedRUTokenBucket {
>>>>>>> master
rg.Lock()
defer rg.Unlock()
if rg.RUSettings == nil {
return nil
}
<<<<<<< HEAD
tb, trickleTimeMs := rg.RUSettings.WRU.request(neededTokens, targetPeriodMs)
=======
tb, trickleTimeMs := rg.RUSettings.WRU.request(now, neededTokens, targetPeriodMs)
>>>>>>> master
return &rmpb.GrantedRUTokenBucket{Type: rmpb.RequestUnitType_WRU, GrantedTokens: tb, TrickleTimeMs: trickleTimeMs}
}

Expand Down

0 comments on commit 62b275c

Please sign in to comment.