From 5fb01d41aa3906abdc37438bc95b032df7f9c41a Mon Sep 17 00:00:00 2001
From: Fabian Reinartz <fab.reinartz@gmail.com>
Date: Mon, 16 Jan 2017 21:29:53 +0100
Subject: [PATCH] Use new Prometheus text format parser

---
 cmd/tsdb/Makefile |  4 +--
 cmd/tsdb/main.go  | 81 ++++++++++++++++-------------------------------
 head.go           |  1 -
 3 files changed, 29 insertions(+), 57 deletions(-)

diff --git a/cmd/tsdb/Makefile b/cmd/tsdb/Makefile
index 71fc993fd..3c0f760fe 100644
--- a/cmd/tsdb/Makefile
+++ b/cmd/tsdb/Makefile
@@ -1,9 +1,9 @@
 build:
-	@go build .
+	@go1.8rc1 build .
 	
 bench: build 
 	@echo ">> running benchmark"
-	@./tsdb bench write --metrics=$(NUM_METRICS) testdata.100k
+	@./tsdb bench write --metrics=$(NUM_METRICS) testdata.1m
 	@go tool pprof -svg ./tsdb benchout/cpu.prof > benchout/cpuprof.svg
 	@go tool pprof -svg ./tsdb benchout/mem.prof > benchout/memprof.svg
 	@go tool pprof -svg ./tsdb benchout/block.prof > benchout/blockprof.svg
diff --git a/cmd/tsdb/main.go b/cmd/tsdb/main.go
index fb65ae503..62b1139fc 100644
--- a/cmd/tsdb/main.go
+++ b/cmd/tsdb/main.go
@@ -9,15 +9,14 @@ import (
 	"path/filepath"
 	"runtime"
 	"runtime/pprof"
-	"sort"
 	"sync"
 	"time"
+	"unsafe"
 
 	"github.com/fabxc/tsdb"
 	"github.com/fabxc/tsdb/labels"
-	dto "github.com/prometheus/client_model/go"
-	"github.com/prometheus/common/expfmt"
-	"github.com/prometheus/common/model"
+	promlabels "github.com/prometheus/prometheus/pkg/labels"
+	"github.com/prometheus/prometheus/pkg/textparse"
 	"github.com/spf13/cobra"
 )
 
@@ -97,7 +96,7 @@ func (b *writeBenchmark) run(cmd *cobra.Command, args []string) {
 	}
 	b.storage = st
 
-	var metrics []model.Metric
+	var metrics []labels.Labels
 
 	measureTime("readData", func() {
 		f, err := os.Open(args[0])
@@ -133,23 +132,11 @@ func (b *writeBenchmark) run(cmd *cobra.Command, args []string) {
 	})
 }
 
-func (b *writeBenchmark) ingestScrapes(metrics []model.Metric, scrapeCount int) error {
+func (b *writeBenchmark) ingestScrapes(lbls []labels.Labels, scrapeCount int) error {
 	var wg sync.WaitGroup
 	var mu sync.Mutex
 	var total uint64
 
-	lbls := make([]labels.Labels, 0, len(metrics))
-
-	for _, m := range metrics {
-		lset := make(labels.Labels, 0, len(m))
-		for k, v := range m {
-			lset = append(lset, labels.Label{Name: string(k), Value: string(v)})
-		}
-		sort.Sort(lset)
-
-		lbls = append(lbls, lset)
-	}
-
 	for i := 0; i < scrapeCount; i += 50 {
 		lbls := lbls
 		for len(lbls) > 0 {
@@ -305,44 +292,30 @@ func measureTime(stage string, f func()) {
 	fmt.Printf(">> completed stage=%s duration=%s\n", stage, time.Since(start))
 }
 
-func readPrometheusLabels(r io.Reader, n int) ([]model.Metric, error) {
-	dec := expfmt.NewDecoder(r, expfmt.FmtProtoText)
-
-	var mets []model.Metric
-	fps := map[model.Fingerprint]struct{}{}
-	var mf dto.MetricFamily
-	var dups int
-
-	for i := 0; i < n; {
-		if err := dec.Decode(&mf); err != nil {
-			if err == io.EOF {
-				break
-			}
-			return nil, err
-		}
-
-		for _, m := range mf.GetMetric() {
-			met := make(model.Metric, len(m.GetLabel())+1)
-			met["__name__"] = model.LabelValue(mf.GetName())
-
-			for _, l := range m.GetLabel() {
-				met[model.LabelName(l.GetName())] = model.LabelValue(l.GetValue())
-			}
-			if _, ok := fps[met.Fingerprint()]; ok {
-				dups++
-			} else {
-				mets = append(mets, met)
-				fps[met.Fingerprint()] = struct{}{}
-			}
-			i++
-		}
+func readPrometheusLabels(r io.Reader, n int) ([]labels.Labels, error) {
+	b, err := ioutil.ReadAll(r)
+	if err != nil {
+		return nil, err
 	}
-	if dups > 0 {
-		fmt.Println("dropped duplicate metrics:", dups)
-	}
-	fmt.Println("read metrics", len(mets[:n]))
 
-	return mets[:n], nil
+	p := textparse.New(b)
+	i := 0
+	var mets []labels.Labels
+	hashes := map[uint64]struct{}{}
+
+	for p.Next() && i < n {
+		m := make(labels.Labels, 0, 10)
+		p.Metric((*promlabels.Labels)(unsafe.Pointer(&m)))
+
+		h := m.Hash()
+		if _, ok := hashes[h]; ok {
+			continue
+		}
+		mets = append(mets, m)
+		hashes[h] = struct{}{}
+		i++
+	}
+	return mets, p.Err()
 }
 
 func exitWithError(err error) {
diff --git a/head.go b/head.go
index 184d0344d..7c355cdd8 100644
--- a/head.go
+++ b/head.go
@@ -427,7 +427,6 @@ func (h *headBlock) create(hash uint64, lset labels.Labels) *memSeries {
 		lset: lset,
 		ref:  uint32(len(h.series)),
 	}
-	fmt.Println("create", lset)
 
 	// Allocate empty space until we can insert at the given index.
 	h.series = append(h.series, s)