Merge pull request #126 from zhangyongding/master

support oracle

Author: huandu

args.go

@@ -247,6 +247,8 @@ func (args *Args) compileArg(buf *stringBuilder, flavor Flavor, values []interfa
 			fmt.Fprintf(buf, "$%d", len(values)+1)
 		case SQLServer:
 			fmt.Fprintf(buf, "@p%d", len(values)+1)
+		case Oracle:
+			fmt.Fprintf(buf, ":%d", len(values)+1)
 		default:
 			panic(fmt.Errorf("Args.CompileWithFlavor: invalid flavor %v (%v)", flavor, int(flavor)))
 		}

fieldmapper.go

@@ -1,6 +1,8 @@
 package sqlbuilder
 
 import (
+	"reflect"
+
 	"github.com/huandu/xstrings"
 )
 
@@ -13,8 +15,19 @@ var (
 	//
 	// Field mapper is called only once on a Struct when the Struct is used to create builder for the first time.
 	DefaultFieldMapper FieldMapperFunc
+
+	// DefaultGetAlias is the default alias and dbtag get func
+	DefaultGetAlias GetAliasFunc
 )
 
+func init() {
+	DefaultGetAlias = func(field *reflect.StructField) (alias string, dbtag string) {
+		dbtag = field.Tag.Get(DBTag)
+		alias = dbtag
+		return
+	}
+}
+
 // FieldMapperFunc is a func to map struct field names to column names,
 // which will be used in query as columns.
 type FieldMapperFunc func(name string) string
@@ -28,3 +41,6 @@ type FieldMapperFunc func(name string) string
 func SnakeCaseMapper(field string) string {
 	return xstrings.ToSnakeCase(field)
 }
+
+// GetAliasFunc is a func to get alias and dbtag
+type GetAliasFunc func(field *reflect.StructField) (alias string, dbtag string)

flavor.go

@@ -19,6 +19,7 @@ const (
 	CQL
 	ClickHouse
 	Presto
+	Oracle
 )
 
 var (
@@ -58,6 +59,8 @@ func (f Flavor) String() string {
 		return "ClickHouse"
 	case Presto:
 		return "Presto"
+	case Oracle:
+		return "Oracle"
 	}
 
 	return "<invalid>"
@@ -84,6 +87,8 @@ func (f Flavor) Interpolate(sql string, args []interface{}) (string, error) {
 		return clickhouseInterpolate(sql, args...)
 	case Presto:
 		return prestoInterpolate(sql, args...)
+	case Oracle:
+		return oracleInterpolate(sql, args...)
 	}
 
 	return "", ErrInterpolateNotImplemented
@@ -140,7 +145,7 @@ func (f Flavor) Quote(name string) string {
 	switch f {
 	case MySQL, ClickHouse:
 		return fmt.Sprintf("`%s`", name)
-	case PostgreSQL, SQLServer, SQLite, Presto:
+	case PostgreSQL, SQLServer, SQLite, Presto, Oracle:
 		return fmt.Sprintf(`"%s"`, name)
 	case CQL:
 		return fmt.Sprintf("'%s'", name)
@@ -152,7 +157,7 @@ func (f Flavor) Quote(name string) string {
 // PrepareInsertIgnore prepares the insert builder to build insert ignore SQL statement based on the sql flavor
 func (f Flavor) PrepareInsertIgnore(table string, ib *InsertBuilder) {
 	switch ib.args.Flavor {
-	case MySQL:
+	case MySQL, Oracle:
 		ib.verb = "INSERT IGNORE"
 
 	case PostgreSQL:

flavor_test.go

@@ -138,3 +138,21 @@ func ExampleFlavor_Interpolate_cql() {
 	// SELECT name FROM user WHERE id = 1234 AND name = 'Charmy Liu'
 	// <nil>
 }
+
+func ExampleFlavor_Interpolate_oracle() {
+	sb := Oracle.NewSelectBuilder()
+	sb.Select("name").From("user").Where(
+		sb.E("id", 1234),
+		sb.E("name", "Charmy Liu"),
+		sb.E("enabled", true),
+	)
+	sql, args := sb.Build()
+	query, err := Oracle.Interpolate(sql, args)
+
+	fmt.Println(query)
+	fmt.Println(err)
+
+	// Output:
+	// SELECT name FROM user WHERE id = 1234 AND name = 'Charmy Liu' AND enabled = 1
+	// <nil>
+}

insert.go

@@ -120,6 +120,37 @@ func (ib *InsertBuilder) BuildWithFlavor(flavor Flavor, initialArg ...interface{
 	buf := newStringBuilder()
 	ib.injection.WriteTo(buf, insertMarkerInit)
 
+	if len(ib.values) > 1 && ib.args.Flavor == Oracle {
+		buf.WriteLeadingString(ib.verb)
+		buf.WriteString(" ALL")
+
+		for _, v := range ib.values {
+			if len(ib.table) > 0 {
+				buf.WriteString(" INTO ")
+				buf.WriteString(ib.table)
+			}
+			ib.injection.WriteTo(buf, insertMarkerAfterInsertInto)
+			if len(ib.cols) > 0 {
+				buf.WriteLeadingString("(")
+				buf.WriteString(strings.Join(ib.cols, ", "))
+				buf.WriteString(")")
+
+				ib.injection.WriteTo(buf, insertMarkerAfterCols)
+			}
+
+			buf.WriteLeadingString("VALUES ")
+			values := make([]string, 0, len(ib.values))
+			values = append(values, fmt.Sprintf("(%v)", strings.Join(v, ", ")))
+			buf.WriteString(strings.Join(values, ", "))
+		}
+
+		buf.WriteString(" SELECT 1 from DUAL")
+
+		ib.injection.WriteTo(buf, insertMarkerAfterValues)
+
+		return ib.args.CompileWithFlavor(buf.String(), flavor, initialArg...)
+	}
+
 	if len(ib.table) > 0 {
 		buf.WriteLeadingString(ib.verb)
 		buf.WriteString(" INTO ")

insert_test.go

@@ -65,6 +65,22 @@ func ExampleInsertBuilder() {
 	// [1 Huan Du 1 2 Charmy Liu 1 1234567890]
 }
 
+func ExampleInsertBuilder_Oracle() {
+	ib := Oracle.NewInsertBuilder()
+	ib.InsertInto("demo.user")
+	ib.Cols("id", "name", "status")
+	ib.Values(1, "Huan Du", 1)
+	ib.Values(2, "Charmy Liu", 1)
+
+	sql, args := ib.Build()
+	fmt.Println(sql)
+	fmt.Println(args)
+
+	// Output:
+	// INSERT ALL INTO demo.user (id, name, status) VALUES (:1, :2, :3) INTO demo.user (id, name, status) VALUES (:4, :5, :6) SELECT 1 from DUAL
+	// [1 Huan Du 1 2 Charmy Liu 1]
+}
+
 func ExampleInsertBuilder_insertIgnore() {
 	ib := NewInsertBuilder()
 	ib.InsertIgnoreInto("demo.user")