This commit is contained in:
baifuyu 2023-12-21 21:02:30 +08:00
parent 093e78415e
commit 60d9099ac5
3 changed files with 5 additions and 7 deletions

View File

@ -60,9 +60,9 @@ public class SubGraphMappingProcessor extends BaseMappingProcessor<SubGraphMappi
BuilderRecord mappedRecord = mapping(record, config.getMappingConfigs()); BuilderRecord mappedRecord = mapping(record, config.getMappingConfigs());
BaseAdvancedRecord advancedRecord = toSPGRecord(mappedRecord, spgType); BaseAdvancedRecord advancedRecord = toSPGRecord(mappedRecord, spgType);
if (advancedRecord != null) { if (advancedRecord != null) {
subGraphFusing.subGraphFusing(advancedRecord); List<BaseAdvancedRecord> fusedRecords = subGraphFusing.subGraphFusing(advancedRecord);
recordPredicating.propertyPredicating(advancedRecord); recordPredicating.propertyPredicating(advancedRecord);
advancedRecords.add(advancedRecord); advancedRecords.addAll(fusedRecords);
} }
} }
return (List) subjectFusing.subjectFusing(advancedRecords); return (List) subjectFusing.subjectFusing(advancedRecords);

View File

@ -4,12 +4,11 @@ import com.antgroup.openspg.builder.core.runtime.BuilderContext;
import com.antgroup.openspg.builder.model.exception.BuilderException; import com.antgroup.openspg.builder.model.exception.BuilderException;
import com.antgroup.openspg.builder.model.exception.FusingException; import com.antgroup.openspg.builder.model.exception.FusingException;
import com.antgroup.openspg.builder.model.record.BaseAdvancedRecord; import com.antgroup.openspg.builder.model.record.BaseAdvancedRecord;
import com.antgroup.openspg.builder.model.record.BaseSPGRecord;
import java.util.List; import java.util.List;
public interface SubGraphFusing { public interface SubGraphFusing {
void init(BuilderContext context) throws BuilderException; void init(BuilderContext context) throws BuilderException;
List<BaseSPGRecord> subGraphFusing(BaseAdvancedRecord advancedRecord) throws FusingException; List<BaseAdvancedRecord> subGraphFusing(BaseAdvancedRecord advancedRecord) throws FusingException;
} }

View File

@ -6,7 +6,6 @@ import com.antgroup.openspg.builder.model.exception.FusingException;
import com.antgroup.openspg.builder.model.pipeline.config.BaseMappingNodeConfig; import com.antgroup.openspg.builder.model.pipeline.config.BaseMappingNodeConfig;
import com.antgroup.openspg.builder.model.pipeline.config.fusing.BaseFusingConfig; import com.antgroup.openspg.builder.model.pipeline.config.fusing.BaseFusingConfig;
import com.antgroup.openspg.builder.model.record.BaseAdvancedRecord; import com.antgroup.openspg.builder.model.record.BaseAdvancedRecord;
import com.antgroup.openspg.builder.model.record.BaseSPGRecord;
import com.antgroup.openspg.builder.model.record.property.BasePropertyRecord; import com.antgroup.openspg.builder.model.record.property.BasePropertyRecord;
import com.antgroup.openspg.cloudext.interfaces.graphstore.adapter.util.VertexRecordConvertor; import com.antgroup.openspg.cloudext.interfaces.graphstore.adapter.util.VertexRecordConvertor;
import com.antgroup.openspg.common.util.CollectionsUtils; import com.antgroup.openspg.common.util.CollectionsUtils;
@ -45,9 +44,9 @@ public class SubGraphFusingImpl implements SubGraphFusing {
} }
@Override @Override
public List<BaseSPGRecord> subGraphFusing(BaseAdvancedRecord advancedRecord) public List<BaseAdvancedRecord> subGraphFusing(BaseAdvancedRecord advancedRecord)
throws FusingException { throws FusingException {
List<BaseSPGRecord> results = new ArrayList<>(); List<BaseAdvancedRecord> results = new ArrayList<>();
for (BasePropertyRecord propertyRecord : advancedRecord.getProperties()) { for (BasePropertyRecord propertyRecord : advancedRecord.getProperties()) {
if (propertyRecord.isSemanticProperty()) { if (propertyRecord.isSemanticProperty()) {
EntityFusing entityFusing = semanticEntityFusing.get(propertyRecord.getName()); EntityFusing entityFusing = semanticEntityFusing.get(propertyRecord.getName());