fixed bug in Remove() where the currentSum and numberOfTasksInWindow weren't getting updated.

This commit is contained in:
Pradyumna Kaushik 2016-12-20 13:53:28 -05:00 committed by Renan DelValle
parent e0a16da97a
commit 1b821735ed

View file

@ -74,6 +74,8 @@ func (rac *runningAverageCalculator) removeFromWindow(id string) (interface{}, e
for element := rac.window.Front(); element != nil; element = element.Next() {
if elementToRemove := element.Value.(Interface); elementToRemove.ID() == id {
rac.window.Remove(element)
rac.currentSum -= elementToRemove.Val()
rac.numberOfElementsInWindow--
return elementToRemove, nil
}
}