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

dialect/entsql: add support for table checks in schema/migration

Browse Source
This commit is contained in:
Ariel Mashraki
2021-06-17 23:11:12 +03:00
committed by Ariel Mashraki
parent b8f4614bfd
commit 42a2c67cc4
11 changed files with 111 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
dialect/entsql/annotation.go
View File

@@ -87,13 +87,23 @@ type Annotation struct {
//
OnDelete ReferenceOption `json:"on_delete,omitempty"`
// Check allows injecting custom "DDL" for setting the "CHECK" clause in "CREATE TABLE".
// Check allows injecting custom "DDL" for setting an unnamed "CHECK" clause in "CREATE TABLE".
//
// entsql.Annotation{
// Check: "age < 10",
// }
//
Check string `json:"check,omitempty"`
// Checks allows injecting custom "DDL" for setting named "CHECK" clauses in "CREATE TABLE".
//
// entsql.Annotation{
// Checks: map[string]string{
// "valid_discount": "price > discount_price",
// },
// }
//
Checks map[string]string `json:"checks,omitempty"`
}
// Name describes the annotation name.
@@ -135,8 +145,16 @@ func (a Annotation) Merge(other schema.Annotation) schema.Annotation {
if s := ant.OnDelete; s != "" {
a.OnDelete = s
}
if s := ant.Check; s != "" {
a.Check = s
if c := ant.Check; c != "" {
a.Check = c
}
if checks := ant.Checks; len(checks) > 0 {
if a.Checks == nil {
a.Checks = make(map[string]string)
}
for name, check := range checks {
a.Checks[name] = check
}
}
return a
}