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

entc: cleanup defaults ID (#2709)

Browse Source
This commit is contained in:
Giau. Tran Minh
2022-07-02 03:30:18 +07:00
committed by GitHub
parent 904b649120
commit 4f7b1739cd
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
entc/gen/graph.go
View File

@@ -175,25 +175,23 @@ func (g *Graph) defaults() {
if g.IDType != nil {
return
}
g.IDType = defaultIDType
if len(g.Nodes) == 0 {
g.IDType = defaultIDType
return
}
nodes := make([]*Type, 0, len(g.Nodes))
idTypes := make([]*field.TypeInfo, 0, len(g.Nodes))
for _, n := range g.Nodes {
if n.HasOneFieldID() {
nodes = append(nodes, n)
idTypes = append(idTypes, n.ID.Type)
}
}
// Check that all nodes have the same type for the ID field.
for i := 0; i < len(nodes)-1; i++ {
cid, nid := nodes[i].ID.Type, nodes[i+1].ID.Type
if cid.Type != nid.Type {
g.IDType = defaultIDType
for i := 0; i < len(idTypes)-1; i++ {
if idTypes[i].Type != idTypes[i+1].Type {
return
}
}
g.IDType = nodes[0].ID.Type
g.IDType = idTypes[0]
}
// Gen generates the artifacts for the graph.