Rename StatsBuffer to TimedStore in utils

This commit is contained in:
Victor Marmol 2015-04-23 11:12:09 -07:00
parent a32015579e
commit 08186b6297
3 changed files with 19 additions and 18 deletions

View File

@ -22,13 +22,14 @@ import (
"github.com/golang/glog"
info "github.com/google/cadvisor/info/v1"
"github.com/google/cadvisor/storage"
"github.com/google/cadvisor/utils"
)
// TODO(vmarmol): See about refactoring this class, we have an unecessary redirection of containerStorage and InMemoryStorage.
// containerStorage is used to store per-container information
type containerStorage struct {
ref info.ContainerReference
recentStats *StatsBuffer
recentStats *utils.TimedStore
maxAge time.Duration
lock sync.RWMutex
}
@ -51,7 +52,7 @@ func (self *containerStorage) RecentStats(start, end time.Time, maxStats int) ([
func newContainerStore(ref info.ContainerReference, maxAge time.Duration) *containerStorage {
return &containerStorage{
ref: ref,
recentStats: NewStatsBuffer(maxAge),
recentStats: utils.NewTimedStore(maxAge),
maxAge: maxAge,
}
}

View File

@ -12,7 +12,7 @@
// See the License for the specific language governing permissions and
// limitations under the License.
package memory
package utils
import (
"sort"
@ -22,21 +22,21 @@ import (
)
// A time-based buffer for ContainerStats. Holds information for a specific time period.
type StatsBuffer struct {
type TimedStore struct {
buffer []*info.ContainerStats
age time.Duration
}
// Returns a new thread-compatible StatsBuffer.
func NewStatsBuffer(age time.Duration) *StatsBuffer {
return &StatsBuffer{
// Returns a new thread-compatible TimedStore.
func NewTimedStore(age time.Duration) *TimedStore {
return &TimedStore{
buffer: make([]*info.ContainerStats, 0),
age: age,
}
}
// Adds an element to the start of the buffer (removing one from the end if necessary).
func (self *StatsBuffer) Add(item *info.ContainerStats) {
func (self *TimedStore) Add(item *info.ContainerStats) {
// Remove any elements before the eviction time.
evictTime := item.Timestamp.Add(-self.age)
index := sort.Search(len(self.buffer), func(index int) bool {
@ -53,7 +53,7 @@ func (self *StatsBuffer) Add(item *info.ContainerStats) {
// Returns up to maxResult elements in the specified time period (inclusive).
// Results are from first to last. maxResults of -1 means no limit. When first
// and last are specified, maxResults is ignored.
func (self *StatsBuffer) InTimeRange(start, end time.Time, maxResults int) []*info.ContainerStats {
func (self *TimedStore) InTimeRange(start, end time.Time, maxResults int) []*info.ContainerStats {
// No stats, return empty.
if len(self.buffer) == 0 {
return []*info.ContainerStats{}
@ -117,10 +117,10 @@ func (self *StatsBuffer) InTimeRange(start, end time.Time, maxResults int) []*in
}
// Gets the element at the specified index. Note that elements are output in LIFO order.
func (self *StatsBuffer) Get(index int) *info.ContainerStats {
func (self *TimedStore) Get(index int) *info.ContainerStats {
return self.buffer[len(self.buffer)-index-1]
}
func (self *StatsBuffer) Size() int {
func (self *TimedStore) Size() int {
return len(self.buffer)
}

View File

@ -12,7 +12,7 @@
// See the License for the specific language governing permissions and
// limitations under the License.
package memory
package utils
import (
"strconv"
@ -38,13 +38,13 @@ func createStats(id int32) *info.ContainerStats {
}
}
func expectSize(t *testing.T, sb *StatsBuffer, expectedSize int) {
func expectSize(t *testing.T, sb *TimedStore, expectedSize int) {
if sb.Size() != expectedSize {
t.Errorf("Expected size %v, got %v", expectedSize, sb.Size())
}
}
func expectAllElements(t *testing.T, sb *StatsBuffer, expected []int32) {
func expectAllElements(t *testing.T, sb *TimedStore, expected []int32) {
size := sb.Size()
els := make([]*info.ContainerStats, size)
for i := 0; i < size; i++ {
@ -81,7 +81,7 @@ func expectElement(t *testing.T, stat *info.ContainerStats, expected int32) {
}
func TestAdd(t *testing.T) {
sb := NewStatsBuffer(5 * time.Second)
sb := NewTimedStore(5 * time.Second)
// Add 1.
sb.Add(createStats(0))
@ -110,7 +110,7 @@ func TestAdd(t *testing.T) {
}
func TestGet(t *testing.T) {
sb := NewStatsBuffer(5 * time.Second)
sb := NewTimedStore(5 * time.Second)
sb.Add(createStats(1))
sb.Add(createStats(2))
sb.Add(createStats(3))
@ -122,7 +122,7 @@ func TestGet(t *testing.T) {
}
func TestInTimeRange(t *testing.T) {
sb := NewStatsBuffer(5 * time.Second)
sb := NewTimedStore(5 * time.Second)
assert := assert.New(t)
var empty time.Time
@ -199,7 +199,7 @@ func TestInTimeRange(t *testing.T) {
}
func TestInTimeRangeWithLimit(t *testing.T) {
sb := NewStatsBuffer(5 * time.Second)
sb := NewTimedStore(5 * time.Second)
sb.Add(createStats(1))
sb.Add(createStats(2))
sb.Add(createStats(3))