From a010934299efa90867f2897f674f3b732e04f335 Mon Sep 17 00:00:00 2001
From: Nan Deng <dengnan@google.com>
Date: Tue, 17 Jun 2014 14:33:32 -0700
Subject: [PATCH] Close() error

---
 storage/memory/memory.go | 7 +++++++
 storage/storage.go       | 2 ++
 2 files changed, 9 insertions(+)

diff --git a/storage/memory/memory.go b/storage/memory/memory.go
index 1fb5887e..6155448a 100644
--- a/storage/memory/memory.go
+++ b/storage/memory/memory.go
@@ -204,6 +204,13 @@ func (self *InMemoryStorage) Percentiles(name string, cpuPercentiles, memPercent
 	return cstore.Percentiles(cpuPercentiles, memPercentiles)
 }
 
+func (self *InMemoryStorage) Close() error {
+	self.lock.Lock()
+	self.containerStorageMap = make(map[string]*containerStorage, 32)
+	self.lock.Unlock()
+	return nil
+}
+
 func New(maxNumSamples, maxNumStats int) storage.StorageDriver {
 	ret := &InMemoryStorage{
 		containerStorageMap: make(map[string]*containerStorage, 32),
diff --git a/storage/storage.go b/storage/storage.go
index 84de7b57..7d521eb0 100644
--- a/storage/storage.go
+++ b/storage/storage.go
@@ -32,4 +32,6 @@ type StorageDriver interface {
 	// the number of returned samples is implementation defined. Otherwise, the driver
 	// should return at most numSamples samples.
 	Samples(containername string, numSamples int) ([]*info.ContainerStatsSample, error)
+
+	Close() error
 }