Merge remote-tracking branch 'upstream/master'

# Conflicts:
#	README.md
#	args.go
#	flavor.go
#	insert_test.go
#	interpolate.go
#	interpolate_test.go
#	select_test.go
#	struct.go
#	struct_test.go

Author: SpencerC

README.md

@@ -9,7 +9,7 @@
 - [Usage](#usage)
   - [Basic usage](#basic-usage)
   - [Pre-defined SQL builders](#pre-defined-sql-builders)
-  - [Build SQL for MySQL, PostgreSQL, SQLServer or SQLite](#build-sql-for-mysql-postgresql-sqlserver-or-sqlite)
+  - [Build SQL for MySQL, PostgreSQL, SQLServer, SQLite, CQL, or ClickHouse](#build-sql-for-mysql-postgresql-sqlserve-sqlite-or-clickhouse)
   - [Using `Struct` as a light weight ORM](#using-struct-as-a-light-weight-orm)
   - [Nested SQL](#nested-sql)
   - [Use `sql.Named` in a builder](#use-sqlnamed-in-a-builder)
@@ -110,7 +110,7 @@ Following are some utility methods to deal with special cases.
 
 To learn how to use builders, check out [examples on GoDoc](https://pkg.go.dev/github.com/huandu/go-sqlbuilder#pkg-examples).
 
-### Build SQL for MySQL, PostgreSQL, SQLServer, SQLite, or CQL
+### Build SQL for MySQL, PostgreSQL, SQLServer, SQLite, CQL, or ClickHouse
 
 Parameter markers are different in MySQL, PostgreSQL, SQLServer and SQLite. This package provides some methods to set the type of markers (we call it "flavor") in all builders.
 

args.go

@@ -233,7 +233,7 @@ func (args *Args) compileArg(buf *bytes.Buffer, flavor Flavor, values []interfac
 		}
 	default:
 		switch flavor {
-		case MySQL, SQLite, CQL:
+		case MySQL, SQLite, CQL, ClickHouse:
 			buf.WriteRune('?')
 		case PostgreSQL:
 			fmt.Fprintf(buf, "$%d", len(values)+1)

flavor.go

@@ -17,6 +17,7 @@ const (
 	SQLite
 	SQLServer
 	CQL
+	ClickHouse
 )
 
 var (
@@ -52,6 +53,8 @@ func (f Flavor) String() string {
 		return "SQLServer"
 	case CQL:
 		return "CQL"
+	case ClickHouse:
+		return "ClickHouse"
 	}
 
 	return "<invalid>"
@@ -74,6 +77,8 @@ func (f Flavor) Interpolate(sql string, args []interface{}) (string, error) {
 		return sqlserverInterpolate(sql, args...)
 	case CQL:
 		return cqlInterpolate(sql, args...)
+	case ClickHouse:
+		return clickhouseInterpolate(sql, args...)
 	}
 
 	return "", ErrInterpolateNotImplemented
@@ -124,11 +129,11 @@ func (f Flavor) NewUnionBuilder() *UnionBuilder {
 // Quote adds quote for name to make sure the name can be used safely
 // as table name or field name.
 //
-//     * For MySQL, use back quote (`) to quote name;
-//     * For PostgreSQL, SQL Server and SQLite, use double quote (") to quote name.
+//   - For MySQL, use back quote (`) to quote name;
+//   - For PostgreSQL, SQL Server and SQLite, use double quote (") to quote name.
 func (f Flavor) Quote(name string) string {
 	switch f {
-	case MySQL:
+	case MySQL, ClickHouse:
 		return fmt.Sprintf("`%s`", name)
 	case PostgreSQL, SQLServer, SQLite:
 		return fmt.Sprintf(`"%s"`, name)
@@ -153,6 +158,9 @@ func (f Flavor) PrepareInsertIgnore(table string, ib *InsertBuilder) {
 	case SQLite:
 		// see https://www.sqlite.org/lang_insert.html
 		ib.verb = "INSERT OR IGNORE"
+	case ClickHouse:
+		// see https://clickhouse.tech/docs/en/sql-reference/statements/insert-into/
+		ib.verb = "INSERT"
 	default:
 		// panic if the db flavor is not supported
 		panic(fmt.Errorf("unsupported db flavor: %s", ib.args.Flavor.String()))

insert_test.go

@@ -113,6 +113,22 @@ func ExampleInsertBuilder_insertIgnore_sqlite() {
 	// [1 Huan Du 1 2 Charmy Liu 1 1234567890]
 }
 
+func ExampleInsertBuilder_insertIgnore_clickhouse() {
+	ib := ClickHouse.NewInsertBuilder()
+	ib.InsertIgnoreInto("demo.user")
+	ib.Cols("id", "name", "status", "created_at")
+	ib.Values(1, "Huan Du", 1, Raw("UNIX_TIMESTAMP(NOW())"))
+	ib.Values(2, "Charmy Liu", 1, 1234567890)
+
+	sql, args := ib.Build()
+	fmt.Println(sql)
+	fmt.Println(args)
+
+	// Output:
+	// INSERT INTO demo.user (id, name, status, created_at) VALUES (?, ?, ?, UNIX_TIMESTAMP(NOW())), (?, ?, ?, ?)
+	// [1 Huan Du 1 2 Charmy Liu 1 1234567890]
+}
+
 func ExampleInsertBuilder_replaceInto() {
 	ib := NewInsertBuilder()
 	ib.ReplaceInto("demo.user")

interpolate.go

@@ -4,7 +4,9 @@
 package sqlbuilder
 
 import (
+	"database/sql/driver"
 	"fmt"
+	"reflect"
 	"strconv"
 	"time"
 	"unicode"
@@ -389,83 +391,22 @@ func cqlInterpolate(query string, args ...interface{}) (string, error) {
 	return mysqlLikeInterpolate(CQL, query, args...)
 }
 
+func clickhouseInterpolate(query string, args ...interface{}) (string, error) {
+	return mysqlLikeInterpolate(ClickHouse, query, args...)
+}
+
 func encodeValue(buf []byte, arg interface{}, flavor Flavor) ([]byte, error) {
 	switch v := arg.(type) {
 	case nil:
 		buf = append(buf, "NULL"...)
 
-	case bool:
-		if v {
-			buf = append(buf, "TRUE"...)
+	case driver.Valuer:
+		if val, err := v.Value(); err != nil {
+			return nil, err
 		} else {
-			buf = append(buf, "FALSE"...)
+			return encodeValue(buf, val, flavor)
 		}
 
-	case int:
-		buf = strconv.AppendInt(buf, int64(v), 10)
-
-	case int8:
-		buf = strconv.AppendInt(buf, int64(v), 10)
-
-	case int16:
-		buf = strconv.AppendInt(buf, int64(v), 10)
-
-	case int32:
-		buf = strconv.AppendInt(buf, int64(v), 10)
-
-	case int64:
-		buf = strconv.AppendInt(buf, v, 10)
-
-	case uint:
-		buf = strconv.AppendUint(buf, uint64(v), 10)
-
-	case uint8:
-		buf = strconv.AppendUint(buf, uint64(v), 10)
-
-	case uint16:
-		buf = strconv.AppendUint(buf, uint64(v), 10)
-
-	case uint32:
-		buf = strconv.AppendUint(buf, uint64(v), 10)
-
-	case uint64:
-		buf = strconv.AppendUint(buf, v, 10)
-
-	case float32:
-		buf = strconv.AppendFloat(buf, float64(v), 'g', -1, 32)
-
-	case float64:
-		buf = strconv.AppendFloat(buf, v, 'g', -1, 64)
-
-	case []byte:
-		if v == nil {
-			buf = append(buf, "NULL"...)
-			break
-		}
-
-		switch flavor {
-		case MySQL:
-			buf = append(buf, "_binary"...)
-			buf = quoteStringValue(buf, *(*string)(unsafe.Pointer(&v)), flavor)
-
-		case PostgreSQL:
-			buf = append(buf, "E'\\\\x"...)
-			buf = appendHex(buf, v)
-			buf = append(buf, "'::bytea"...)
-
-		case SQLite:
-			buf = append(buf, "X'"...)
-			buf = appendHex(buf, v)
-			buf = append(buf, '\'')
-
-		case SQLServer, CQL:
-			buf = append(buf, "0x"...)
-			buf = appendHex(buf, v)
-		}
-
-	case string:
-		buf = quoteStringValue(buf, v, flavor)
-
 	case time.Time:
 		if v.IsZero() {
 			buf = append(buf, "'0000-00-00'"...)
@@ -492,6 +433,9 @@ func encodeValue(buf []byte, arg interface{}, flavor Flavor) ([]byte, error) {
 
 		case CQL:
 			buf = append(buf, v.Format("2006-01-02 15:04:05.999999Z0700")...)
+
+		case ClickHouse:
+			buf = append(buf, v.Format("2006-01-02 15:04:05.999999")...)
 		}
 
 		buf = append(buf, '\'')
@@ -500,7 +444,108 @@ func encodeValue(buf []byte, arg interface{}, flavor Flavor) ([]byte, error) {
 		buf = quoteStringValue(buf, v.String(), flavor)
 
 	default:
-		return nil, ErrInterpolateUnsupportedArgs
+		primative := reflect.ValueOf(arg)
+
+		switch k := primative.Kind(); k {
+		case reflect.Bool:
+			if primative.Bool() {
+				buf = append(buf, "TRUE"...)
+			} else {
+				buf = append(buf, "FALSE"...)
+			}
+
+		case reflect.Int:
+			buf = strconv.AppendInt(buf, primative.Int(), 10)
+
+		case reflect.Int8:
+			buf = strconv.AppendInt(buf, primative.Int(), 10)
+
+		case reflect.Int16:
+			buf = strconv.AppendInt(buf, primative.Int(), 10)
+
+		case reflect.Int32:
+			buf = strconv.AppendInt(buf, primative.Int(), 10)
+
+		case reflect.Int64:
+			buf = strconv.AppendInt(buf, primative.Int(), 10)
+
+		case reflect.Uint:
+			buf = strconv.AppendUint(buf, primative.Uint(), 10)
+
+		case reflect.Uint8:
+			buf = strconv.AppendUint(buf, primative.Uint(), 10)
+
+		case reflect.Uint16:
+			buf = strconv.AppendUint(buf, primative.Uint(), 10)
+
+		case reflect.Uint32:
+			buf = strconv.AppendUint(buf, primative.Uint(), 10)
+
+		case reflect.Uint64:
+			buf = strconv.AppendUint(buf, primative.Uint(), 10)
+
+		case reflect.Float32:
+			buf = strconv.AppendFloat(buf, primative.Float(), 'g', -1, 32)
+
+		case reflect.Float64:
+			buf = strconv.AppendFloat(buf, primative.Float(), 'g', -1, 64)
+
+		case reflect.String:
+			buf = quoteStringValue(buf, primative.String(), flavor)
+
+		case reflect.Slice, reflect.Array:
+			if k == reflect.Slice && primative.IsNil() {
+				buf = append(buf, "NULL"...)
+				break
+			}
+
+			if elem := primative.Type().Elem(); elem.Kind() != reflect.Uint8 {
+				return nil, ErrInterpolateUnsupportedArgs
+			}
+
+			var data []byte
+
+			// Bytes() will panic if primative is an array and cannot be addressed.
+			// Copy all bytes to data as a fallback.
+			if k == reflect.Array && !primative.CanAddr() {
+				l := primative.Len()
+				data = make([]byte, l)
+
+				for i := 0; i < l; i++ {
+					data[i] = byte(primative.Index(i).Uint())
+				}
+			} else {
+				data = primative.Bytes()
+			}
+
+			switch flavor {
+			case MySQL:
+				buf = append(buf, "_binary"...)
+				buf = quoteStringValue(buf, *(*string)(unsafe.Pointer(&data)), flavor)
+
+			case PostgreSQL:
+				buf = append(buf, "E'\\\\x"...)
+				buf = appendHex(buf, data)
+				buf = append(buf, "'::bytea"...)
+
+			case SQLite:
+				buf = append(buf, "X'"...)
+				buf = appendHex(buf, data)
+				buf = append(buf, '\'')
+
+			case SQLServer, CQL:
+				buf = append(buf, "0x"...)
+				buf = appendHex(buf, data)
+
+			case ClickHouse:
+				buf = append(buf, "unhex('"...)
+				buf = appendHex(buf, data)
+				buf = append(buf, "')"...)
+			}
+
+		default:
+			return nil, ErrInterpolateUnsupportedArgs
+		}
 	}
 
 	return buf, nil