refs huandu#18. add CreateTableBuilder

Author: huandu

README.md

@@ -38,6 +38,7 @@ fmt.Println(args)
 Following builders are implemented right now. API document and examples are provided in the `godoc` document.
 
 * [Struct](https://godoc.org/github.com/huandu/go-sqlbuilder#Struct): Builder factory for a struct.
+* [CreateTableBuilder](https://godoc.org/github.com/huandu/go-sqlbuilder#CreateTableBuilder): Builder for CREATE TABLE.
 * [SelectBuilder](https://godoc.org/github.com/huandu/go-sqlbuilder#SelectBuilder): Builder for SELECT.
 * [InsertBuilder](https://godoc.org/github.com/huandu/go-sqlbuilder#InsertBuilder): Builder for INSERT.
 * [UpdateBuilder](https://godoc.org/github.com/huandu/go-sqlbuilder#UpdateBuilder): Builder for UPDATE.

createtable.go

@@ -0,0 +1,121 @@
+// Copyright 2018 Huan Du. All rights reserved.
+// Licensed under the MIT license that can be found in the LICENSE file.
+
+package sqlbuilder
+
+import (
+	"bytes"
+	"strings"
+)
+
+// NewCreateTableBuilder creates a new CREATE TABLE builder.
+func NewCreateTableBuilder() *CreateTableBuilder {
+	return DefaultFlavor.NewCreateTableBuilder()
+}
+
+func newCreateTableBuilder() *CreateTableBuilder {
+	args := &Args{}
+	return &CreateTableBuilder{
+		verb: "CREATE TABLE",
+		args: args,
+	}
+}
+
+// CreateTableBuilder is a builder to build CREATE TABLE.
+type CreateTableBuilder struct {
+	verb        string
+	ifNotExists bool
+	table       string
+	defs        [][]string
+	options     [][]string
+
+	args *Args
+}
+
+// CreateTable sets the table name in CREATE TABLE.
+func (ctb *CreateTableBuilder) CreateTable(table string) *CreateTableBuilder {
+	ctb.table = Escape(table)
+	return ctb
+}
+
+// CreateTempTable sets the table name and changes the verb of ctb to CREATE TEMPORARY TABLE.
+func (ctb *CreateTableBuilder) CreateTempTable(table string) *CreateTableBuilder {
+	ctb.verb = "CREATE TEMPORARY TABLE"
+	ctb.table = Escape(table)
+	return ctb
+}
+
+// IfNotExists adds IF NOT EXISTS before table name in CREATE TABLE.
+func (ctb *CreateTableBuilder) IfNotExists() *CreateTableBuilder {
+	ctb.ifNotExists = true
+	return ctb
+}
+
+// Define adds definition of a column or index in CREATE TABLE.
+func (ctb *CreateTableBuilder) Define(def ...string) *CreateTableBuilder {
+	ctb.defs = append(ctb.defs, def)
+	return ctb
+}
+
+// Option adds a table option in CREATE TABLE.
+func (ctb *CreateTableBuilder) Option(opt ...string) *CreateTableBuilder {
+	ctb.options = append(ctb.options, opt)
+	return ctb
+}
+
+// String returns the compiled INSERT string.
+func (ctb *CreateTableBuilder) String() string {
+	s, _ := ctb.Build()
+	return s
+}
+
+// Build returns compiled CREATE TABLE string and args.
+// They can be used in `DB#Query` of package `database/sql` directly.
+func (ctb *CreateTableBuilder) Build() (sql string, args []interface{}) {
+	return ctb.BuildWithFlavor(ctb.args.Flavor)
+}
+
+// BuildWithFlavor returns compiled CREATE TABLE string and args with flavor and initial args.
+// They can be used in `DB#Query` of package `database/sql` directly.
+func (ctb *CreateTableBuilder) BuildWithFlavor(flavor Flavor, initialArg ...interface{}) (sql string, args []interface{}) {
+	buf := &bytes.Buffer{}
+	buf.WriteString(ctb.verb)
+
+	if ctb.ifNotExists {
+		buf.WriteString(" IF NOT EXISTS")
+	}
+
+	buf.WriteRune(' ')
+	buf.WriteString(ctb.table)
+	buf.WriteString(" (")
+
+	defs := make([]string, 0, len(ctb.defs))
+
+	for _, def := range ctb.defs {
+		defs = append(defs, strings.Join(def, " "))
+	}
+
+	buf.WriteString(strings.Join(defs, ", "))
+	buf.WriteRune(')')
+
+	if len(ctb.options) > 0 {
+		buf.WriteRune(' ')
+
+		opts := make([]string, 0, len(ctb.options))
+
+		for _, opt := range ctb.options {
+			opts = append(opts, strings.Join(opt, " "))
+		}
+
+		buf.WriteString(strings.Join(opts, ", "))
+	}
+
+	return ctb.args.CompileWithFlavor(buf.String(), flavor, initialArg...)
+}
+
+// SetFlavor sets the flavor of compiled sql.
+func (ctb *CreateTableBuilder) SetFlavor(flavor Flavor) (old Flavor) {
+	old = ctb.args.Flavor
+	ctb.args.Flavor = flavor
+	return
+}

createtable_test.go

@@ -0,0 +1,40 @@
+// Copyright 2018 Huan Du. All rights reserved.
+// Licensed under the MIT license that can be found in the LICENSE file.
+
+package sqlbuilder
+
+import (
+	"fmt"
+)
+
+func ExampleCreateTableBuilder() {
+	ctb := NewCreateTableBuilder()
+	ctb.CreateTable("demo.user").IfNotExists()
+	ctb.Define("id", "BIGINT(20)", "NOT NULL", "AUTO_INCREMENT", "PRIMARY KEY", `COMMENT "user id"`)
+	ctb.Define("name", "VARCHAR(255)", "NOT NULL", `COMMENT "user name"`)
+	ctb.Define("created_at", "DATETIME", "NOT NULL", `COMMENT "user create time"`)
+	ctb.Define("modified_at", "DATETIME", "NOT NULL", `COMMENT "user modify time"`)
+	ctb.Define("KEY", "idx_name_modified_at", "name, modified_at")
+	ctb.Option("DEFAULT CHARACTER SET", "utf8mb4")
+
+	fmt.Println(ctb)
+
+	// Output:
+	// CREATE TABLE IF NOT EXISTS demo.user (id BIGINT(20) NOT NULL AUTO_INCREMENT PRIMARY KEY COMMENT "user id", name VARCHAR(255) NOT NULL COMMENT "user name", created_at DATETIME NOT NULL COMMENT "user create time", modified_at DATETIME NOT NULL COMMENT "user modify time", KEY idx_name_modified_at name, modified_at) DEFAULT CHARACTER SET utf8mb4
+}
+
+func ExampleCreateTableBuilder_tempTable() {
+	ctb := NewCreateTableBuilder()
+	ctb.CreateTempTable("demo.user").IfNotExists()
+	ctb.Define("id", "BIGINT(20)", "NOT NULL", "AUTO_INCREMENT", "PRIMARY KEY", `COMMENT "user id"`)
+	ctb.Define("name", "VARCHAR(255)", "NOT NULL", `COMMENT "user name"`)
+	ctb.Define("created_at", "DATETIME", "NOT NULL", `COMMENT "user create time"`)
+	ctb.Define("modified_at", "DATETIME", "NOT NULL", `COMMENT "user modify time"`)
+	ctb.Define("KEY", "idx_name_modified_at", "name, modified_at")
+	ctb.Option("DEFAULT CHARACTER SET", "utf8mb4")
+
+	fmt.Println(ctb)
+
+	// Output:
+	// CREATE TEMPORARY TABLE IF NOT EXISTS demo.user (id BIGINT(20) NOT NULL AUTO_INCREMENT PRIMARY KEY COMMENT "user id", name VARCHAR(255) NOT NULL COMMENT "user name", created_at DATETIME NOT NULL COMMENT "user create time", modified_at DATETIME NOT NULL COMMENT "user modify time", KEY idx_name_modified_at name, modified_at) DEFAULT CHARACTER SET utf8mb4
+}

flavor.go

@@ -33,6 +33,13 @@ func (f Flavor) String() string {
 	return "<invalid>"
 }
 
+// NewCreateTableBuilder creates a new CREATE TABLE builder with flavor.
+func (f Flavor) NewCreateTableBuilder() *CreateTableBuilder {
+	b := newCreateTableBuilder()
+	b.SetFlavor(f)
+	return b
+}
+
 // NewDeleteBuilder creates a new DELETE builder with flavor.
 func (f Flavor) NewDeleteBuilder() *DeleteBuilder {
 	b := newDeleteBuilder()