From f079d84728ae4fbd9cdcf23d33d3675f3b541de8 Mon Sep 17 00:00:00 2001 From: Jack Christensen Date: Tue, 6 Aug 2013 10:20:32 -0500 Subject: [PATCH] Added SanitizeSql bool support --- sanitize.go | 2 ++ sanitize_test.go | 4 ++++ 2 files changed, 6 insertions(+) diff --git a/sanitize.go b/sanitize.go index 0f60b21a..2e808ba6 100644 --- a/sanitize.go +++ b/sanitize.go @@ -60,6 +60,8 @@ func (c *Connection) SanitizeSql(sql string, args ...interface{}) (output string return strconv.FormatFloat(float64(arg), 'f', -1, 32) case float64: return strconv.FormatFloat(arg, 'f', -1, 64) + case bool: + return strconv.FormatBool(arg) case []byte: return `E'\\x` + hex.EncodeToString(arg) + `'` case []int16: diff --git a/sanitize_test.go b/sanitize_test.go index c668ece2..0c1055b5 100644 --- a/sanitize_test.go +++ b/sanitize_test.go @@ -31,6 +31,10 @@ func TestSanitizeSql(t *testing.T) { t.Errorf("Failed to pass through float: %v - %v", san, err) } + if san, err := conn.SanitizeSql("select $1", true); err != nil || san != "select true" { + t.Errorf("Failed to pass through bool: %v - %v", san, err) + } + if san, err := conn.SanitizeSql("select $1, $2, $3", "Jack's", 42, 1.23); err != nil || san != "select 'Jack''s', 42, 1.23" { t.Errorf("Failed to sanitize multiple params: %v - %v", san, err) }