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 check clause for column builder (#1065)

Browse Source
This commit is contained in:
Ariel Mashraki
2020-12-20 10:33:34 +02:00
committed by GitHub
parent 3b4b94b6be
commit eca18b0894
2 changed files with 16 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
dialect/sql/builder.go
View File

@@ -37,6 +37,7 @@ type ColumnBuilder struct {
attr string // extra attributes.
modify bool // modify existing.
fk *ForeignKeyBuilder // foreign-key constraint.
check func(*Builder) // column checks.
}
// Column returns a new ColumnBuilder with the given name.
@@ -66,12 +67,18 @@ func (c *ColumnBuilder) Constraint(fk *ForeignKeyBuilder) *ColumnBuilder {
return c
}
// Check adds a CHECK clause to the ADD COLUMN statement.
func (c *ColumnBuilder) Check(check func(*Builder)) *ColumnBuilder {
c.check = check
return c
}
// Query returns query representation of a Column.
func (c *ColumnBuilder) Query() (string, []interface{}) {
c.Ident(c.name)
if c.typ != "" {
if c.postgres() && c.modify {
c.Pad().WriteString("TYPE")
c.WriteString(" TYPE")
}
c.Pad().WriteString(c.typ)
}
@@ -85,6 +92,10 @@ func (c *ColumnBuilder) Query() (string, []interface{}) {
c.Pad().WriteString(action)
}
}
if c.check != nil {
c.WriteString(" CHECK ")
c.Nested(c.check)
}
return c.String(), c.args
}

5
dialect/sql/builder_test.go
View File

@@ -78,11 +78,14 @@ func TestBuilder(t *testing.T) {
Columns(
Column("id").Type("int").Attr("auto_increment"),
Column("card_id").Type("int"),
Column("doc").Type("longtext").Check(func(b *Builder) {
b.WriteString("JSON_VALID(").Ident("doc").WriteByte(')')
}),
).
PrimaryKey("id", "name").
ForeignKeys(ForeignKey().Columns("card_id").
Reference(Reference().Table("cards").Columns("id")).OnDelete("SET NULL")),
wantQuery: "CREATE TABLE IF NOT EXISTS `users`(`id` int auto_increment, `card_id` int, PRIMARY KEY(`id`, `name`), FOREIGN KEY(`card_id`) REFERENCES `cards`(`id`) ON DELETE SET NULL)",
wantQuery: "CREATE TABLE IF NOT EXISTS `users`(`id` int auto_increment, `card_id` int, `doc` longtext CHECK (JSON_VALID(`doc`)), PRIMARY KEY(`id`, `name`), FOREIGN KEY(`card_id`) REFERENCES `cards`(`id`) ON DELETE SET NULL)",
},
{
input: Dialect(dialect.Postgres).CreateTable("users").