1818import com .antgroup .openspg .builder .core .runtime .BuilderContext ;
1919import com .antgroup .openspg .builder .core .runtime .impl .DefaultBuilderCatalog ;
2020import com .antgroup .openspg .builder .model .pipeline .config .GraphStoreSinkNodeConfig ;
21- import com .antgroup .openspg .builder .model .pipeline .config .Neo4jSinkNodeConfig ;
2221import com .antgroup .openspg .builder .model .record .BaseRecord ;
2322import com .antgroup .openspg .builder .model .record .RecordAlterOperationEnum ;
2423import com .antgroup .openspg .builder .model .record .SubGraphRecord ;
2524import com .antgroup .openspg .builder .runner .local .physical .sink .impl .GraphStoreSinkWriter ;
26- import com .antgroup .openspg .builder .runner .local .physical .sink .impl .Neo4jSinkWriter ;
2725import com .antgroup .openspg .core .schema .model .identifier .SPGTypeIdentifier ;
2826import com .antgroup .openspg .core .schema .model .type .BaseSPGType ;
2927import com .antgroup .openspg .core .schema .model .type .ConceptList ;
4139import com .antgroup .openspg .server .biz .schema .SchemaManager ;
4240import com .antgroup .openspg .server .biz .service .GraphManager ;
4341import com .google .common .collect .Lists ;
44- import java .util .HashMap ;
45- import java .util .List ;
46- import java .util .Map ;
47- import java .util .UUID ;
42+
43+ import java .util .*;
44+
4845import lombok .extern .slf4j .Slf4j ;
4946import org .apache .commons .collections4 .CollectionUtils ;
5047import org .springframework .beans .factory .annotation .Autowired ;
@@ -179,12 +176,12 @@ public void check() {
179176 public Boolean action () {
180177 ProjectSchema projectSchema = schemaManager .getProjectSchema (request .getProjectId ());
181178 boolean enableLeadTo =
182- ( request .getEnableLeadTo () == null ) ? false : request .getEnableLeadTo ();
179+ request .getEnableLeadTo () != null && request .getEnableLeadTo ();
183180 Map <SPGTypeIdentifier , ConceptList > conceptLists =
184181 getConceptLists (enableLeadTo , projectSchema );
185- Neo4jSinkWriter writer =
186- new Neo4jSinkWriter (
187- UUID .randomUUID ().toString (), "图存储" , new Neo4jSinkNodeConfig (true ));
182+ GraphStoreSinkWriter writer =
183+ new GraphStoreSinkWriter (
184+ UUID .randomUUID ().toString (), "图存储" , new GraphStoreSinkNodeConfig (true ));
188185 BuilderContext context =
189186 new BuilderContext ()
190187 .setProjectId (request .getProjectId ())
@@ -201,12 +198,13 @@ public Boolean action() {
201198
202199 SubGraphRecord subGraph =
203200 JSON .parseObject (JSON .toJSONString (request .getSubGraph ()), SubGraphRecord .class );
204- writer .writeToNeo4j (subGraph );
201+ List <BaseRecord > records = Lists .newArrayList ();
202+ records .add (subGraph );
203+ writer .write (records );
204+
205205 if (context .isEnableLeadTo ()) {
206206 ReasonProcessor reasonProcessor = new ReasonProcessor ();
207207 reasonProcessor .init (context );
208- List <BaseRecord > records = Lists .newArrayList ();
209- records .add (subGraph );
210208 List <BaseRecord > reasonResults = reasonProcessor .process (records );
211209 if (CollectionUtils .isNotEmpty (reasonResults )) {
212210 GraphStoreSinkWriter sinkWriter =
0 commit comments