From c69e638f68ff6af525face29e59c1c869b1104d1 Mon Sep 17 00:00:00 2001
From: Jack Christensen <jack@jackchristensen.com>
Date: Fri, 12 Jul 2013 18:21:40 -0400
Subject: [PATCH] Added (remove|restore)BinaryEncoders

Makes it so the oid doesn't have to be hard coded into text
benchmarks that disable binary encoding.
---
 bench_test.go | 45 +++++++++++++++++++++++++++------------------
 1 file changed, 27 insertions(+), 18 deletions(-)

diff --git a/bench_test.go b/bench_test.go
index b6aebd01..35192315 100644
--- a/bench_test.go
+++ b/bench_test.go
@@ -52,6 +52,21 @@ func createNarrowTestData(b *testing.B, conn *Connection) {
 	narrowTestDataLoaded = true
 }
 
+func removeBinaryEncoders() (encoders map[oid]func(*MessageReader, int32) interface{}) {
+	encoders = make(map[oid]func(*MessageReader, int32) interface{})
+	for k, v := range valueTranscoders {
+		encoders[k] = v.DecodeBinary
+		valueTranscoders[k].DecodeBinary = nil
+	}
+	return
+}
+
+func restoreBinaryEncoders(encoders map[oid]func(*MessageReader, int32) interface{}) {
+	for k, v := range encoders {
+		valueTranscoders[k].DecodeBinary = v
+	}
+}
+
 func BenchmarkSelectRowSimpleNarrow(b *testing.B) {
 	conn := getSharedConnection()
 	createNarrowTestData(b, conn)
@@ -262,9 +277,8 @@ func BenchmarkInt2Text(b *testing.B) {
 	conn := getSharedConnection()
 	createInt2TextVsBinaryTestData(b, conn)
 
-	binaryDecoder := valueTranscoders[oid(21)].DecodeBinary
-	valueTranscoders[oid(21)].DecodeBinary = nil
-	defer func() { valueTranscoders[oid(21)].DecodeBinary = binaryDecoder }()
+	encoders := removeBinaryEncoders()
+	defer func() { restoreBinaryEncoders(encoders) }()
 
 	mustPrepare(b, conn, "selectInt16", "select * from t")
 	defer func() { conn.Deallocate("selectInt16") }()
@@ -322,9 +336,8 @@ func BenchmarkInt4Text(b *testing.B) {
 	conn := getSharedConnection()
 	createInt4TextVsBinaryTestData(b, conn)
 
-	binaryDecoder := valueTranscoders[oid(23)].DecodeBinary
-	valueTranscoders[oid(23)].DecodeBinary = nil
-	defer func() { valueTranscoders[oid(23)].DecodeBinary = binaryDecoder }()
+	encoders := removeBinaryEncoders()
+	defer func() { restoreBinaryEncoders(encoders) }()
 
 	mustPrepare(b, conn, "selectInt32", "select * from t")
 	defer func() { conn.Deallocate("selectInt32") }()
@@ -382,9 +395,8 @@ func BenchmarkInt8Text(b *testing.B) {
 	conn := getSharedConnection()
 	createInt8TextVsBinaryTestData(b, conn)
 
-	binaryDecoder := valueTranscoders[oid(20)].DecodeBinary
-	valueTranscoders[oid(20)].DecodeBinary = nil
-	defer func() { valueTranscoders[oid(20)].DecodeBinary = binaryDecoder }()
+	encoders := removeBinaryEncoders()
+	defer func() { restoreBinaryEncoders(encoders) }()
 
 	mustPrepare(b, conn, "selectInt64", "select * from t")
 	defer func() { conn.Deallocate("selectInt64") }()
@@ -442,9 +454,8 @@ func BenchmarkFloat4Text(b *testing.B) {
 	conn := getSharedConnection()
 	createFloat4TextVsBinaryTestData(b, conn)
 
-	binaryDecoder := valueTranscoders[oid(700)].DecodeBinary
-	valueTranscoders[oid(700)].DecodeBinary = nil
-	defer func() { valueTranscoders[oid(700)].DecodeBinary = binaryDecoder }()
+	encoders := removeBinaryEncoders()
+	defer func() { restoreBinaryEncoders(encoders) }()
 
 	mustPrepare(b, conn, "selectFloat32", "select * from t")
 	defer func() { conn.Deallocate("selectFloat32") }()
@@ -502,9 +513,8 @@ func BenchmarkFloat8Text(b *testing.B) {
 	conn := getSharedConnection()
 	createFloat8TextVsBinaryTestData(b, conn)
 
-	binaryDecoder := valueTranscoders[oid(701)].DecodeBinary
-	valueTranscoders[oid(701)].DecodeBinary = nil
-	defer func() { valueTranscoders[oid(701)].DecodeBinary = binaryDecoder }()
+	encoders := removeBinaryEncoders()
+	defer func() { restoreBinaryEncoders(encoders) }()
 
 	mustPrepare(b, conn, "selectFloat32", "select * from t")
 	defer func() { conn.Deallocate("selectFloat32") }()
@@ -562,9 +572,8 @@ func BenchmarkBoolText(b *testing.B) {
 	conn := getSharedConnection()
 	createBoolTextVsBinaryTestData(b, conn)
 
-	binaryDecoder := valueTranscoders[oid(16)].DecodeBinary
-	valueTranscoders[oid(16)].DecodeBinary = nil
-	defer func() { valueTranscoders[oid(16)].DecodeBinary = binaryDecoder }()
+	encoders := removeBinaryEncoders()
+	defer func() { restoreBinaryEncoders(encoders) }()
 
 	mustPrepare(b, conn, "selectBool", "select * from t")
 	defer func() { conn.Deallocate("selectBool") }()