From f897164bcfe463926f7479c18c8c17abd0eb1cbe Mon Sep 17 00:00:00 2001
From: "Matt T. Proud" <matt.proud@gmail.com>
Date: Tue, 7 May 2013 10:18:19 +0200
Subject: [PATCH] Expose TieredStorage.DiskStorage.

---
 storage/metric/tiered.go      | 18 ++++++++++--------
 storage/metric/tiered_test.go |  4 ++--
 2 files changed, 12 insertions(+), 10 deletions(-)

diff --git a/storage/metric/tiered.go b/storage/metric/tiered.go
index 9994260c8..f82307978 100644
--- a/storage/metric/tiered.go
+++ b/storage/metric/tiered.go
@@ -30,10 +30,12 @@ import (
 // TieredStorage both persists samples and generates materialized views for
 // queries.
 type TieredStorage struct {
+	// BUG(matt): This introduces a Law of Demeter violation.  Ugh.
+	DiskStorage *LevelDBMetricPersistence
+
 	appendToDiskQueue   chan model.Samples
 	appendToMemoryQueue chan model.Samples
 	diskFrontier        *diskFrontier
-	diskStorage         *LevelDBMetricPersistence
 	draining            chan chan bool
 	flushMemoryInterval time.Duration
 	memoryArena         memorySeriesStorage
@@ -60,7 +62,7 @@ func NewTieredStorage(appendToMemoryQueueDepth, appendToDiskQueueDepth, viewQueu
 	storage = &TieredStorage{
 		appendToDiskQueue:   make(chan model.Samples, appendToDiskQueueDepth),
 		appendToMemoryQueue: make(chan model.Samples, appendToMemoryQueueDepth),
-		diskStorage:         diskStorage,
+		DiskStorage:         diskStorage,
 		draining:            make(chan chan bool),
 		flushMemoryInterval: flushMemoryInterval,
 		memoryArena:         NewMemorySeriesStorage(),
@@ -210,7 +212,7 @@ func (t TieredStorage) Flush() {
 func (t TieredStorage) Close() {
 	log.Println("Closing tiered storage...")
 	t.Drain()
-	t.diskStorage.Close()
+	t.DiskStorage.Close()
 	t.memoryArena.Close()
 
 	close(t.appendToDiskQueue)
@@ -323,7 +325,7 @@ func (t *TieredStorage) flushMemory() {
 	defer t.mutex.Unlock()
 
 	flusher := &memoryToDiskFlusher{
-		disk:        t.diskStorage,
+		disk:        t.DiskStorage,
 		olderThan:   time.Now().Add(-1 * t.memoryTTL),
 		toDiskQueue: t.appendToDiskQueue,
 	}
@@ -351,7 +353,7 @@ func (t TieredStorage) renderView(viewJob viewJob) {
 		scans = viewJob.builder.ScanJobs()
 		view  = newView()
 		// Get a single iterator that will be used for all data extraction below.
-		iterator = t.diskStorage.MetricSamples.NewIterator(true)
+		iterator = t.DiskStorage.MetricSamples.NewIterator(true)
 	)
 	defer iterator.Close()
 
@@ -541,7 +543,7 @@ func (t TieredStorage) loadChunkAroundTime(iterator leveldb.Iterator, frontier *
 
 // Get all label values that are associated with the provided label name.
 func (t TieredStorage) GetAllValuesForLabel(labelName model.LabelName) (values model.LabelValues, err error) {
-	diskValues, err := t.diskStorage.GetAllValuesForLabel(labelName)
+	diskValues, err := t.DiskStorage.GetAllValuesForLabel(labelName)
 	if err != nil {
 		return
 	}
@@ -568,7 +570,7 @@ func (t TieredStorage) GetFingerprintsForLabelSet(labelSet model.LabelSet) (fing
 	if err != nil {
 		return
 	}
-	diskFingerprints, err := t.diskStorage.GetFingerprintsForLabelSet(labelSet)
+	diskFingerprints, err := t.DiskStorage.GetFingerprintsForLabelSet(labelSet)
 	if err != nil {
 		return
 	}
@@ -590,7 +592,7 @@ func (t TieredStorage) GetMetricForFingerprint(f model.Fingerprint) (m *model.Me
 		return
 	}
 	if m == nil {
-		m, err = t.diskStorage.GetMetricForFingerprint(f)
+		m, err = t.DiskStorage.GetMetricForFingerprint(f)
 	}
 	return
 }
diff --git a/storage/metric/tiered_test.go b/storage/metric/tiered_test.go
index 8ce35eeff..721b9400a 100644
--- a/storage/metric/tiered_test.go
+++ b/storage/metric/tiered_test.go
@@ -485,7 +485,7 @@ func TestGetAllValuesForLabel(t *testing.T) {
 				}
 			}
 			if metric.appendToDisk {
-				if err := tiered.diskStorage.AppendSample(sample); err != nil {
+				if err := tiered.DiskStorage.AppendSample(sample); err != nil {
 					t.Fatalf("%d.%d. failed to add fixture data: %s", i, j, err)
 				}
 			}
@@ -520,7 +520,7 @@ func TestGetFingerprintsForLabelSet(t *testing.T) {
 	if err := tiered.memoryArena.AppendSample(memorySample); err != nil {
 		t.Fatalf("Failed to add fixture data: %s", err)
 	}
-	if err := tiered.diskStorage.AppendSample(diskSample); err != nil {
+	if err := tiered.DiskStorage.AppendSample(diskSample); err != nil {
 		t.Fatalf("Failed to add fixture data: %s", err)
 	}
 	tiered.Flush()