From b571812307204c595ee3dcd8f48feaa0600a960a Mon Sep 17 00:00:00 2001
From: Linh Vo <lvo@umbel.com>
Date: Fri, 27 Jan 2017 13:41:19 -0600
Subject: [PATCH] default content-type for benchmarks

---
 bench/diagonal.go | 3 +++
 bench/multidb.go  | 4 ++++
 bench/random.go   | 4 ++++
 bench/zipf.go     | 3 +++
 4 files changed, 14 insertions(+)

diff --git a/bench/diagonal.go b/bench/diagonal.go
index 4cb0e354c..393b732b3 100644
--- a/bench/diagonal.go
+++ b/bench/diagonal.go
@@ -57,6 +57,8 @@ The following arguments are available:
 	-client-type string
 		Can be 'single' (all agents hitting one host) or 'round_robin'
 
+	-content-type string
+		protobuf or pql
 `[1:]
 }
 
@@ -71,6 +73,7 @@ func (b *DiagonalSetBits) ConsumeFlags(args []string) ([]string, error) {
 	fs.IntVar(&b.Iterations, "iterations", 100, "")
 	fs.StringVar(&b.DB, "db", "benchdb", "")
 	fs.StringVar(&b.ClientType, "client-type", "single", "")
+	fs.StringVar(&b.ContentType, "content-type", "protobuf", "")
 
 	if err := fs.Parse(args); err != nil {
 		return nil, err
diff --git a/bench/multidb.go b/bench/multidb.go
index 8ea8cd135..aa275ac17 100644
--- a/bench/multidb.go
+++ b/bench/multidb.go
@@ -49,6 +49,9 @@ The following arguments are available:
 	-client-type string
 		Can be 'single' (all agents hitting one host) or 'round_robin'
 
+	-content-type string
+		protobuf or pql
+
 `[1:]
 }
 
@@ -62,6 +65,7 @@ func (b *MultiDBSetBits) ConsumeFlags(args []string) ([]string, error) {
 	fs.IntVar(&b.BaseProfileID, "base-profile-id", 0, "")
 	fs.IntVar(&b.Iterations, "iterations", 100, "")
 	fs.StringVar(&b.ClientType, "client-type", "single", "")
+	fs.StringVar(&b.ContentType, "content-type", "protobuf", "")
 
 	if err := fs.Parse(args); err != nil {
 		return nil, err
diff --git a/bench/random.go b/bench/random.go
index e3e89400c..3bb74e886 100644
--- a/bench/random.go
+++ b/bench/random.go
@@ -65,6 +65,9 @@ The following arguments are available:
 
 	-client-type string
 		Can be 'single' (all agents hitting one host) or 'round_robin'
+
+	-content-type string
+		protobuf or pql
 `[1:]
 }
 
@@ -82,6 +85,7 @@ func (b *RandomSetBits) ConsumeFlags(args []string) ([]string, error) {
 	fs.IntVar(&b.Iterations, "iterations", 100, "")
 	fs.StringVar(&b.DB, "db", "benchdb", "")
 	fs.StringVar(&b.ClientType, "client-type", "single", "")
+	fs.StringVar(&b.ContentType, "content-type", "protobuf", "")
 
 	if err := fs.Parse(args); err != nil {
 		return nil, err
diff --git a/bench/zipf.go b/bench/zipf.go
index 6343cd4c6..7bd48ed1f 100644
--- a/bench/zipf.go
+++ b/bench/zipf.go
@@ -86,6 +86,8 @@ The following arguments are available:
 
 	-client-type string
 		Can be 'single' (all agents hitting one host) or 'round_robin'
+	-content-type string
+		protobuf or pql
 `[1:]
 }
 
@@ -107,6 +109,7 @@ func (b *ZipfSetBits) ConsumeFlags(args []string) ([]string, error) {
 	fs.Float64Var(&b.ProfileExponent, "profile-exponent", 1.01, "")
 	fs.Float64Var(&b.ProfileRatio, "profile-ratio", 0.25, "")
 	fs.StringVar(&b.ClientType, "client-type", "single", "")
+	fs.StringVar(&b.ContentType, "content-type", "protobuf", "")
 
 	if err := fs.Parse(args); err != nil {
 		return nil, err