golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-22 09:31:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

dialect/sql: add support for table options (#925)

Browse Source
This commit is contained in:
Ariel Mashraki
2020-11-06 18:14:23 +02:00
committed by GitHub
parent d6702a49fa
commit 122ee020b7
2 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
dialect/sql/builder.go
View File

@@ -95,6 +95,7 @@ type TableBuilder struct {
exists bool // check existence.
charset string // table charset.
collation string // table collation.
options string // table options.
columns []Querier // table columns.
primary []string // primary key.
constraints []Querier // foreign keys and indices.
@@ -172,6 +173,12 @@ func (t *TableBuilder) Collate(s string) *TableBuilder {
return t
}
// Options appends additional options to to the statement (MySQL only).
func (t *TableBuilder) Options(s string) *TableBuilder {
t.options = s
return t
}
// Query returns query representation of a `CREATE TABLE` statement.
//
// CREATE TABLE [IF NOT EXISTS] name
@@ -202,6 +209,9 @@ func (t *TableBuilder) Query() (string, []interface{}) {
if t.collation != "" {
t.WriteString(" COLLATE " + t.collation)
}
if t.options != "" {
t.WriteString(" " + t.options)
}
return t.String(), t.args
}

5
dialect/sql/builder_test.go
View File

@@ -59,8 +59,9 @@ func TestBuilder(t *testing.T) {
).
PrimaryKey("id").
Charset("utf8mb4").
Collate("utf8mb4_general_ci"),
wantQuery: "CREATE TABLE `users`(`id` int auto_increment, `name` varchar(255), PRIMARY KEY(`id`)) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci",
Collate("utf8mb4_general_ci").
Options("ENGINE=InnoDB"),
wantQuery: "CREATE TABLE `users`(`id` int auto_increment, `name` varchar(255), PRIMARY KEY(`id`)) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci ENGINE=InnoDB",
},
{
input: CreateTable("users").