Skip to content

Commit

Permalink
Rename StatsBuffer to TimedStore in utils
Browse files Browse the repository at this point in the history
  • Loading branch information
vmarmol committed Apr 23, 2015
1 parent a320155 commit 08186b6
Show file tree
Hide file tree
Showing 3 changed files with 19 additions and 18 deletions.
5 changes: 3 additions & 2 deletions storage/memory/memory.go
Original file line number Diff line number Diff line change
Expand Up @@ -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
}
Expand All @@ -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,
}
}
Expand Down
18 changes: 9 additions & 9 deletions storage/memory/stats_buffer.go → utils/timed_store.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
// See the License for the specific language governing permissions and
// limitations under the License.

package memory
package utils

import (
"sort"
Expand All @@ -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 {
Expand All @@ -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{}
Expand Down Expand Up @@ -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)
}
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
// See the License for the specific language governing permissions and
// limitations under the License.

package memory
package utils

import (
"strconv"
Expand All @@ -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++ {
Expand Down Expand Up @@ -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))
Expand Down Expand Up @@ -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))
Expand All @@ -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
Expand Down Expand Up @@ -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))
Expand Down

0 comments on commit 08186b6

Please sign in to comment.