From c0daa5b8ecdf5f12dd31a73609a304d39dba2687 Mon Sep 17 00:00:00 2001
From: Ben Hood <0x6e6562@gmail.com>
Date: Thu, 27 Mar 2014 18:50:49 +0000
Subject: [PATCH] Add support for decimals with the SliceMap() interface

---
 cassandra_test.go | 21 +++++++++++++++++++--
 helpers.go        |  3 +++
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/cassandra_test.go b/cassandra_test.go
index a0ed38f50..e9319a4c3 100644
--- a/cassandra_test.go
+++ b/cassandra_test.go
@@ -346,6 +346,7 @@ func TestSliceMap(t *testing.T) {
 			testfloat	   float,
 			testdouble	   double,
 			testint        int,
+			testdecimal    decimal,
 			testset        set<int>,
 			testmap        map<varchar, varchar>
 		)`).Exec(); err != nil {
@@ -361,11 +362,12 @@ func TestSliceMap(t *testing.T) {
 	m["testfloat"] = float32(4.564)
 	m["testdouble"] = float64(4.815162342)
 	m["testint"] = 2343
+	m["testdecimal"] = inf.NewDec(100, 0)
 	m["testset"] = []int{1, 2, 3, 4, 5, 6, 7, 8, 9}
 	m["testmap"] = map[string]string{"field1": "val1", "field2": "val2", "field3": "val3"}
 	sliceMap := []map[string]interface{}{m}
-	if err := session.Query(`INSERT INTO slice_map_table (testuuid, testtimestamp, testvarchar, testbigint, testblob, testbool, testfloat, testdouble, testint, testset, testmap) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
-		m["testuuid"], m["testtimestamp"], m["testvarchar"], m["testbigint"], m["testblob"], m["testbool"], m["testfloat"], m["testdouble"], m["testint"], m["testset"], m["testmap"]).Exec(); err != nil {
+	if err := session.Query(`INSERT INTO slice_map_table (testuuid, testtimestamp, testvarchar, testbigint, testblob, testbool, testfloat, testdouble, testint, testdecimal, testset, testmap) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+		m["testuuid"], m["testtimestamp"], m["testvarchar"], m["testbigint"], m["testblob"], m["testbool"], m["testfloat"], m["testdouble"], m["testint"], m["testdecimal"], m["testset"], m["testmap"]).Exec(); err != nil {
 		t.Fatal("insert:", err)
 	}
 	if returned, retErr := session.Query(`SELECT * FROM slice_map_table`).Iter().SliceMap(); retErr != nil {
@@ -398,6 +400,13 @@ func TestSliceMap(t *testing.T) {
 		if sliceMap[0]["testint"] != returned[0]["testint"] {
 			t.Fatal("returned testint did not match")
 		}
+
+		expectedDecimal := sliceMap[0]["testdecimal"].(*inf.Dec)
+		returnedDecimal := returned[0]["testdecimal"].(*inf.Dec)
+
+		if expectedDecimal.Cmp(returnedDecimal) != 0 {
+			t.Fatal("returned testdecimal did not match")
+		}
 		if !reflect.DeepEqual(sliceMap[0]["testset"], returned[0]["testset"]) {
 			t.Fatal("returned testset did not match")
 		}
@@ -438,6 +447,14 @@ func TestSliceMap(t *testing.T) {
 	if sliceMap[0]["testint"] != testMap["testint"] {
 		t.Fatal("returned testint did not match")
 	}
+
+	expectedDecimal := sliceMap[0]["testdecimal"].(*inf.Dec)
+	returnedDecimal := testMap["testdecimal"].(*inf.Dec)
+
+	if expectedDecimal.Cmp(returnedDecimal) != 0 {
+		t.Fatal("returned testdecimal did not match")
+	}
+
 	if !reflect.DeepEqual(sliceMap[0]["testset"], testMap["testset"]) {
 		t.Fatal("returned testset did not match")
 	}
diff --git a/helpers.go b/helpers.go
index 2c696cf45..868662098 100644
--- a/helpers.go
+++ b/helpers.go
@@ -6,6 +6,7 @@ package gocql
 
 import (
 	"reflect"
+	"speter.net/go/exp/math/dec/inf"
 	"time"
 )
 
@@ -38,6 +39,8 @@ func goType(t *TypeInfo) reflect.Type {
 		return reflect.TypeOf(*new(float64))
 	case TypeInt:
 		return reflect.TypeOf(*new(int))
+	case TypeDecimal:
+		return reflect.TypeOf(*new(*inf.Dec))
 	case TypeUUID, TypeTimeUUID:
 		return reflect.TypeOf(*new(UUID))
 	case TypeList, TypeSet: