yujunjun/openspg
1
0
Fork 0
mirror of https://github.com/OpenSPG/openspg.git synced 2025-06-27 03:20:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix one hot expand with belongTo

Browse Source
This commit is contained in:
wenchengyao 2024-08-22 13:30:59 +08:00
parent 0fde581e87
commit 67d8e514ef
2 changed files with 27 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
reasoner/lube-logical/src/main/scala/com/antgroup/openspg/reasoner/lube/logical/planning/SubQueryMerger.scala
View File

@ -88,8 +88,16 @@ class SubQueryMerger(val dag: Dag[LogicalOperator])(implicit context: LogicalPla
.filter(_.alias.equals(name))
.filter(conn => {
val types =
if (conn.direction == Direction.OUT) pattern.getNode(conn.target).typeNames
else pattern.getNode(conn.source).typeNames
if (conn.direction == Direction.OUT) {
pattern.getNode(conn.target).typeNames
}
else if (conn.direction == Direction.IN) {
pattern.getNode(conn.source).typeNames
} else {
pattern
.getNode(conn.source)
.typeNames ++ pattern.getNode(conn.target).typeNames
}
conn.relTypes.contains(spo.getP) && (types.contains(
LabelTypeUtils.getMetaType(spo.getO)) || types.contains(spo.getO))
})

20
reasoner/lube-logical/src/main/scala/com/antgroup/openspg/reasoner/lube/logical/planning/SubQueryPlanner.scala
View File

@ -22,7 +22,13 @@ import com.antgroup.openspg.reasoner.lube.Logging
import com.antgroup.openspg.reasoner.lube.block._
import com.antgroup.openspg.reasoner.lube.catalog.{Catalog, SemanticRule, TemplateSemanticRule}
import com.antgroup.openspg.reasoner.lube.common.pattern.Pattern
import com.antgroup.openspg.reasoner.lube.logical.{EdgeVar, NodeVar, RepeatPathVar, SolvedModel, Var}
import com.antgroup.openspg.reasoner.lube.logical.{
EdgeVar,
NodeVar,
RepeatPathVar,
SolvedModel,
Var
}
import com.antgroup.openspg.reasoner.lube.logical.operators._
import com.antgroup.openspg.reasoner.lube.logical.planning.SubQueryPlanner.nodeName
import com.antgroup.openspg.reasoner.lube.logical.validate.Dag
@ -148,8 +154,16 @@ class SubQueryPlanner(val dag: Dag[Block])(implicit context: LogicalPlannerConte
.filter(_.alias.equals(name))
.filter(conn => {
val types =
if (conn.direction == Direction.OUT) pattern.getNode(conn.target).typeNames
else pattern.getNode(conn.source).typeNames
if (conn.direction == Direction.OUT) {
pattern.getNode(conn.target).typeNames
}
else if (conn.direction == Direction.IN) {
pattern.getNode(conn.source).typeNames
} else {
pattern
.getNode(conn.source)
.typeNames ++ pattern.getNode(conn.target).typeNames
}
conn.relTypes.contains(spo.getP) && (types.contains(
LabelTypeUtils.getMetaType(spo.getO)) || types.contains(spo.getO))
})