new: Added Flavor method to Builder interface

Author: Vasily Rodionov

builder.go

@@ -13,6 +13,7 @@ import (
 type Builder interface {
 	Build() (sql string, args []interface{})
 	BuildWithFlavor(flavor Flavor, initialArg ...interface{}) (sql string, args []interface{})
+	Flavor() Flavor
 }
 
 type compiledBuilder struct {
@@ -30,6 +31,12 @@ func (cb *compiledBuilder) BuildWithFlavor(flavor Flavor, initialArg ...interfac
 	return cb.args.CompileWithFlavor(cb.format, flavor, initialArg...)
 }
 
+// Flavor returns flavor of builder
+// Always returns DefaultFlavor
+func (cb *compiledBuilder) Flavor() Flavor {
+	return cb.args.Flavor
+}
+
 type flavoredBuilder struct {
 	builder Builder
 	flavor  Flavor
@@ -43,6 +50,11 @@ func (fb *flavoredBuilder) BuildWithFlavor(flavor Flavor, initialArg ...interfac
 	return fb.builder.BuildWithFlavor(flavor, initialArg...)
 }
 
+// Flavor returns flavor of builder
+func (fb *flavoredBuilder) Flavor() Flavor {
+	return fb.flavor
+}
+
 // WithFlavor creates a new Builder based on builder with a default flavor.
 func WithFlavor(builder Builder, flavor Flavor) Builder {
 	return &flavoredBuilder{

builder_test.go

@@ -132,3 +132,22 @@ func TestBuildWithCQL(t *testing.T) {
 	a.Equal(sql, "BEGIN BATCH USING TIMESTAMP ? INSERT INTO t1 (col1, col2) VALUES (?, ?); INSERT INTO t2 (col3, col4) VALUES (?, ?); APPLY BATCH;")
 	a.Equal(args, []interface{}{1481124356754405, 1, 2, 3, 4})
 }
+
+func TestBuilderGetFlavor(t *testing.T) {
+	a := assert.New(t)
+
+	defaultBuilder := Build("SELECT * FROM foo WHERE id = $0", 1234)
+	a.Equal(DefaultFlavor, defaultBuilder.Flavor())
+
+	buildfBuilder := Buildf("SELECT * FROM foo WHERE id = %v", 1234)
+	a.Equal(DefaultFlavor, buildfBuilder.Flavor())
+
+	namedBuilder := Buildf("SELECT * FROM ${table} WHERE id = 1234", map[string]interface{}{
+		"table": "foo",
+	})
+	a.Equal(DefaultFlavor, namedBuilder.Flavor())
+
+	flavoredBuilder := WithFlavor(Build("SELECT * FROM foo WHERE id = $0", 1234), PostgreSQL)
+	a.Equal(PostgreSQL, flavoredBuilder.Flavor())
+
+}

createtable.go

@@ -158,7 +158,8 @@ func (ctb *CreateTableBuilder) SetFlavor(flavor Flavor) (old Flavor) {
 	return
 }
 
-func (ctb *CreateTableBuilder) GetFlavor() Flavor {
+// Flavor returns flavor of builder
+func (ctb *CreateTableBuilder) Flavor() Flavor {
 	return ctb.args.Flavor
 }
 

createtable_test.go

@@ -93,13 +93,12 @@ func ExampleCreateTableBuilder_NumDefine() {
 func TestCreateTableGetFlavor(t *testing.T) {
 	a := assert.New(t)
 	ctb := newCreateTableBuilder()
-	postgresFlavor := PostgreSQL
-	clickhouseFlavor := ClickHouse
-	ctb.SetFlavor(postgresFlavor)
-	flavor := ctb.GetFlavor()
-	a.Equal(postgresFlavor, flavor)
-
-	ctbClick := clickhouseFlavor.NewCreateTableBuilder()
-	flavor = ctbClick.GetFlavor()
-	a.Equal(clickhouseFlavor, flavor)
+
+	ctb.SetFlavor(PostgreSQL)
+	flavor := ctb.Flavor()
+	a.Equal(PostgreSQL, flavor)
+
+	ctbClick := ClickHouse.NewCreateTableBuilder()
+	flavor = ctbClick.Flavor()
+	a.Equal(ClickHouse, flavor)
 }

cte.go

@@ -112,7 +112,8 @@ func (cteb *CTEBuilder) SetFlavor(flavor Flavor) (old Flavor) {
 	return
 }
 
-func (cteb *CTEBuilder) GetFlavor() Flavor {
+// Flavor returns flavor of builder
+func (cteb *CTEBuilder) Flavor() Flavor {
 	return cteb.args.Flavor
 }
 

cte_test.go

@@ -134,27 +134,25 @@ func TestRecursiveCTEBuilder(t *testing.T) {
 func TestCTEGetFlavor(t *testing.T) {
 	a := assert.New(t)
 	cteb := newCTEBuilder()
-	postgresFlavor := PostgreSQL
-	clickhouseFlavor := ClickHouse
-	cteb.SetFlavor(postgresFlavor)
-	flavor := cteb.GetFlavor()
-	a.Equal(postgresFlavor, flavor)
-
-	ctebClick := clickhouseFlavor.NewCTEBuilder()
-	flavor = ctebClick.GetFlavor()
-	a.Equal(clickhouseFlavor, flavor)
+
+	cteb.SetFlavor(PostgreSQL)
+	flavor := cteb.Flavor()
+	a.Equal(PostgreSQL, flavor)
+
+	ctebClick := ClickHouse.NewCTEBuilder()
+	flavor = ctebClick.Flavor()
+	a.Equal(ClickHouse, flavor)
 }
 
 func TestCTEQueryBuilderGetFlavor(t *testing.T) {
 	a := assert.New(t)
 	ctetb := newCTEQueryBuilder()
-	postgresFlavor := PostgreSQL
-	clickhouseFlavor := ClickHouse
-	ctetb.SetFlavor(postgresFlavor)
-	flavor := ctetb.GetFlavor()
-	a.Equal(postgresFlavor, flavor)
-
-	ctetbClick := clickhouseFlavor.NewCTEQueryBuilder()
-	flavor = ctetbClick.GetFlavor()
-	a.Equal(clickhouseFlavor, flavor)
+
+	ctetb.SetFlavor(PostgreSQL)
+	flavor := ctetb.Flavor()
+	a.Equal(PostgreSQL, flavor)
+
+	ctetbClick := ClickHouse.NewCTEQueryBuilder()
+	flavor = ctetbClick.Flavor()
+	a.Equal(ClickHouse, flavor)
 }

ctequery.go

@@ -124,7 +124,8 @@ func (ctetb *CTEQueryBuilder) SetFlavor(flavor Flavor) (old Flavor) {
 	return
 }
 
-func (ctetb *CTEQueryBuilder) GetFlavor() Flavor {
+// Flavor returns flavor of builder
+func (ctetb *CTEQueryBuilder) Flavor() Flavor {
 	return ctetb.args.Flavor
 }
 

delete.go

@@ -200,7 +200,8 @@ func (db *DeleteBuilder) SetFlavor(flavor Flavor) (old Flavor) {
 	return
 }
 
-func (db *DeleteBuilder) GetFlavor() Flavor {
+// Flavor returns flavor of builder
+func (db *DeleteBuilder) Flavor() Flavor {
 	return db.args.Flavor
 }
 

delete_test.go

@@ -87,13 +87,12 @@ func ExampleDeleteBuilder_With() {
 func TestDeleteBuilderGetFlavor(t *testing.T) {
 	a := assert.New(t)
 	db := newDeleteBuilder()
-	postgresFlavor := PostgreSQL
-	clickhouseFlavor := ClickHouse
-	db.SetFlavor(postgresFlavor)
-	flavor := db.GetFlavor()
-	a.Equal(postgresFlavor, flavor)
-
-	dbClick := clickhouseFlavor.NewDeleteBuilder()
-	flavor = dbClick.GetFlavor()
-	a.Equal(clickhouseFlavor, flavor)
+
+	db.SetFlavor(PostgreSQL)
+	flavor := db.Flavor()
+	a.Equal(PostgreSQL, flavor)
+
+	dbClick := ClickHouse.NewDeleteBuilder()
+	flavor = dbClick.Flavor()
+	a.Equal(ClickHouse, flavor)
 }

insert.go

@@ -213,7 +213,8 @@ func (ib *InsertBuilder) SetFlavor(flavor Flavor) (old Flavor) {
 	return
 }
 
-func (ib *InsertBuilder) GetFlavor() Flavor {
+// Flavor returns flavor of builder
+func (ib *InsertBuilder) Flavor() Flavor {
 	return ib.args.Flavor
 }