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

entc/gen: move set-neighbors logic from codegen to sqlgraph

Browse Source 
Summary: Pull Request resolved: https://github.com/facebookincubator/ent/pull/183

Reviewed By: alexsn

Differential Revision: D18644179

fbshipit-source-id: 2f2b0c03811a09c4c11240e5161187afbb2ac553
This commit is contained in:
Ariel Mashraki
2019-11-24 02:43:21 -08:00
committed by Facebook Github Bot
parent 1e49561634
commit c355ed49be
34 changed files with 659 additions and 659 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
examples/o2mrecur/ent/node_query.go
View File

@@ -56,30 +56,34 @@ func (nq *NodeQuery) Order(o ...Order) *NodeQuery {
// QueryParent chains the current query on the parent edge.
func (nq *NodeQuery) QueryParent() *NodeQuery {
query := &NodeQuery{config: nq.config}
builder := sql.Dialect(nq.driver.Dialect())
t1 := builder.Table(node.Table)
t2 := nq.sqlQuery()
t2.Select(t2.C(node.ParentColumn))
query.sql = builder.Select(t1.Columns(node.Columns...)...).
From(t1).
Join(t2).
On(t1.C(node.FieldID), t2.C(node.ParentColumn))
step := &sql.Step{}
step.From.V = nq.sqlQuery()
step.From.Table = node.Table
step.From.Column = node.FieldID
step.To.Table = node.Table
step.To.Column = node.FieldID
step.Edge.Rel = sql.M2O
step.Edge.Inverse = true
step.Edge.Table = node.ParentTable
step.Edge.Columns = append(step.Edge.Columns, node.ParentColumn)
query.sql = sql.SetNeighbors(nq.driver.Dialect(), step)
return query
}
// QueryChildren chains the current query on the children edge.
func (nq *NodeQuery) QueryChildren() *NodeQuery {
query := &NodeQuery{config: nq.config}
builder := sql.Dialect(nq.driver.Dialect())
t1 := builder.Table(node.Table)
t2 := nq.sqlQuery()
t2.Select(t2.C(node.FieldID))
query.sql = builder.Select().
From(t1).
Join(t2).
On(t1.C(node.ChildrenColumn), t2.C(node.FieldID))
step := &sql.Step{}
step.From.V = nq.sqlQuery()
step.From.Table = node.Table
step.From.Column = node.FieldID
step.To.Table = node.Table
step.To.Column = node.FieldID
step.Edge.Rel = sql.O2M
step.Edge.Inverse = false
step.Edge.Table = node.ChildrenTable
step.Edge.Columns = append(step.Edge.Columns, node.ChildrenColumn)
query.sql = sql.SetNeighbors(nq.driver.Dialect(), step)
return query
}