@ -4,16 +4,17 @@ import cn.hutool.core.util.StrUtil;
import cn.hutool.poi.excel.ExcelReader ;
import cn.hutool.poi.excel.ExcelUtil ;
import com.supervision.common.domain.R ;
import com.supervision.common.utils.StringUtils ;
import com.supervision.neo4j.domain.CaseNode ;
import com.supervision.neo4j.domain.Rel ;
import com.supervision.neo4j.dto.WebRelDTO ;
import com.supervision.neo4j.service.Neo4jService ;
import com.supervision.neo4j.utils.Neo4jUtils ;
import com.supervision.police.vo.GraphReqVO ;
import lombok.Data ;
import lombok.extern.slf4j.Slf4j ;
import org. neo4j.driver.* ;
import org. apache.commons.lang3.StringUtils ;
import org.neo4j.driver.Record ;
import org.neo4j.driver.* ;
import org.springframework.beans.factory.annotation.Autowired ;
import org.springframework.stereotype.Service ;
@ -50,7 +51,7 @@ public class Neo4jServiceImpl implements Neo4jService {
CaseNode res = null ;
try {
Session session = driver . session ( ) ;
StringBu ffer cql = new StringBuff er( ) ;
StringBu ilder cql = new StringBuild er( ) ;
Map < String , Object > params = new HashMap < > ( ) ;
cql . append ( "CREATE (n:" ) . append ( caseNode . getNodeType ( ) ) . append ( "{name:$name" ) ;
params . put ( "name" , caseNode . getName ( ) ) ;
@ -74,7 +75,7 @@ public class Neo4jServiceImpl implements Neo4jService {
Result run = session . run ( cql . toString ( ) , params ) ;
res = Neo4jUtils . getOneNode ( run ) ;
} catch ( Exception e ) {
e. printStackTrace ( ) ;
log. error ( e . getMessage ( ) , e ) ;
}
return res ;
}
@ -88,13 +89,12 @@ public class Neo4jServiceImpl implements Neo4jService {
public void delNode ( Long id ) {
try {
Session session = driver . session ( ) ;
StringBu ffer cql = new StringBuff er( ) ;
StringBu ilder cql = new StringBuild er( ) ;
cql . append ( "MATCH (n) where id(n) = " ) . append ( id ) . append ( " DELETE n" ) ;
log . info ( cql . toString ( ) ) ;
Result run = session . run ( cql . toString ( ) ) ;
while ( run . hasNext ( ) ) {
Record next = run . next ( ) ;
// log.info(next.toString());
run . next ( ) ;
}
} catch ( Exception e ) {
log . error ( e . getMessage ( ) , e ) ;
@ -105,13 +105,12 @@ public class Neo4jServiceImpl implements Neo4jService {
public void deleteNoRelationNode ( Long id ) {
try {
Session session = driver . session ( ) ;
StringBu ffer cql = new StringBuff er( ) ;
cql . append ( "MATCH (n) WHERE n.id = " ) . append ( id ) . append ( " AND NOT (n)--() DELETE n" ) ;
StringBu ilder cql = new StringBuild er( ) ;
cql . append ( "MATCH (n) WHERE n.id = " ) . append ( id ) . append ( " AND NOT (n)--() DELETE n" ) ;
log . info ( cql . toString ( ) ) ;
Result run = session . run ( cql . toString ( ) ) ;
while ( run . hasNext ( ) ) {
Record next = run . next ( ) ;
// log.info(next.toString());
run . next ( ) ;
}
} catch ( Exception e ) {
log . error ( e . getMessage ( ) , e ) ;
@ -131,11 +130,10 @@ public class Neo4jServiceImpl implements Neo4jService {
log . info ( cql . toString ( ) ) ;
Result run = session . run ( cql . toString ( ) ) ;
while ( run . hasNext ( ) ) {
Record next = run . next ( ) ;
// log.info(next.toString());
run . next ( ) ;
}
} catch ( Exception e ) {
e. printStackTrace ( ) ;
log. error ( e . getMessage ( ) , e ) ;
}
}
@ -144,12 +142,10 @@ public class Neo4jServiceImpl implements Neo4jService {
CaseNode node = null ;
try {
Session session = driver . session ( ) ;
StringBuffer cql = new StringBuffer ( ) ;
cql . append ( "MATCH (n) where id(n) = " ) . append ( id ) . append ( Neo4jUtils . NODE_RETURN ) ;
Result run = session . run ( cql . toString ( ) ) ;
Result run = session . run ( "MATCH (n) where id(n) = " + id + Neo4jUtils . NODE_RETURN ) ;
node = Neo4jUtils . getOneNode ( run ) ;
} catch ( Exception e ) {
e. printStackTrace ( ) ;
log. error ( e . getMessage ( ) , e ) ;
}
return node ;
}
@ -159,7 +155,7 @@ public class Neo4jServiceImpl implements Neo4jService {
List < CaseNode > list = new ArrayList < > ( ) ;
try {
Session session = driver . session ( ) ;
StringBu ffer cql = new StringBuff er( ) ;
StringBu ilder cql = new StringBuild er( ) ;
cql . append ( "MATCH (n" ) ;
if ( StringUtils . isNotEmpty ( nodeType ) ) {
cql . append ( ":" ) ;
@ -187,7 +183,7 @@ public class Neo4jServiceImpl implements Neo4jService {
Result run = session . run ( cql . toString ( ) ) ;
list = Neo4jUtils . getNodeList ( run ) ;
} catch ( Exception e ) {
e. printStackTrace ( ) ;
log. error ( e . getMessage ( ) , e ) ;
}
return list ;
}
@ -197,7 +193,7 @@ public class Neo4jServiceImpl implements Neo4jService {
CaseNode node = null ;
try {
Session session = driver . session ( ) ;
StringBu ffer cql = new StringBuff er( ) ;
StringBu ilder cql = new StringBuild er( ) ;
Map < String , Object > params = new HashMap < > ( ) ;
cql . append ( "MATCH (n" ) ;
if ( StringUtils . isNotEmpty ( nodeType ) ) {
@ -225,7 +221,7 @@ public class Neo4jServiceImpl implements Neo4jService {
Result run = session . run ( cql . toString ( ) , params ) ;
node = Neo4jUtils . getOneNode ( run ) ;
} catch ( Exception e ) {
e. printStackTrace ( ) ;
log. error ( e . getMessage ( ) , e ) ;
}
return node ;
}
@ -234,17 +230,16 @@ public class Neo4jServiceImpl implements Neo4jService {
public Rel findRelation ( Rel rel ) {
try {
Session session = driver . session ( ) ;
StringBuffer cql = new StringBuffer ( ) ;
Map < String , Object > params = new HashMap < > ( ) ;
cql. append ( "MATCH (a)-[rel:" ) . append ( rel . getName ( ) ) . append ( "]->(b) where id(a) = $sourceId and id(b) = $targetId" )
. append ( Neo4jUtils . REL_RETURN ) ;
String cql = "MATCH (a)-[rel:" + rel . getName ( ) + "]->(b) where id(a) = $sourceId and id(b) = $targetId" +
Neo4jUtils . REL_RETURN ;
params . put ( "sourceId" , rel . getSourceId ( ) ) ;
params . put ( "targetId" , rel . getTargetId ( ) ) ;
Result run = session . run ( cql .toString ( ) , params ) ;
Result run = session . run ( cql params ) ;
rel = Neo4jUtils . getOneRel ( run ) ;
} catch ( Exception e ) {
e. printStackTrace ( ) ;
log. error ( e . getMessage ( ) , e ) ;
}
if ( rel ! = null & & rel . getId ( ) ! = null ) {
return rel ;
@ -255,64 +250,83 @@ public class Neo4jServiceImpl implements Neo4jService {
@Override
public Rel saveRelation ( Rel rel ) {
Rel res = null ;
try {
Session session = driver . session ( ) ;
StringBuffer cql = new StringBuffer ( ) ;
Map < String , Object > params = new HashMap < > ( ) ;
cql. append ( "MATCH (a), (b) where id(a) = $sourceId and id(b) = $targetId CREATE(a)-[rel:" ) . append ( rel . getName ( ) )
. append ( "]->(b) " ) . append ( Neo4jUtils . REL_RETURN ) ;
String cql = "MATCH (a), (b) where id(a) = $sourceId and id(b) = $targetId CREATE(a)-[rel:" + rel . getName ( ) +
"]->(b) " + Neo4jUtils . REL_RETURN ;
params . put ( "sourceId" , rel . getSourceId ( ) ) ;
params . put ( "targetId" , rel . getTargetId ( ) ) ;
Result run = session . run ( cql .toString ( ) , params ) ;
Result run = session . run ( cql params ) ;
rel = Neo4jUtils . getOneRel ( run ) ;
} catch ( Exception e ) {
e. printStackTrace ( ) ;
log. error ( e . getMessage ( ) , e ) ;
}
return rel ;
}
@Override
public R < ? > get Node( String picType , String caseId ) {
public R < ? > get CaseGraph( GraphReqVO graphReqVO ) {
Map < String , Object > map = new HashMap < > ( ) ;
List < WebRelDTO > list = new ArrayList < > ( ) ;
List < Map < String , String > > nodes = new ArrayList < > ( ) ;
try {
Session session = driver . session ( ) ;
StringBuilder relQuery = new StringBuilder ( "MATCH (n)-[rel]->(m) " +
"WHERE n.picType = $picType " +
"AND n.caseId = $caseId " ) ;
Map < String , Object > params = new HashMap < > ( ) ;
params . put ( "picType" , picType ) ;
params . put ( "caseId" , caseId ) ;
Result run = session . run ( "MATCH (n)-[rel]->(r) where n.picType = r.picType = $picType and n.caseId = r.caseId = $caseId" +
" RETURN id(rel) as id, n.name as source, id(n) as sourceId, type(rel) as name, r.name as target, id(r) as targetId" , params ) ;
params . put ( "picType" , graphReqVO . getPicType ( ) ) ;
params . put ( "caseId" , graphReqVO . getCaseId ( ) ) ;
if ( StringUtils . isNotEmpty ( graphReqVO . getQueryStr ( ) ) ) {
params . put ( "queryStr" , graphReqVO . getQueryStr ( ) ) ;
relQuery . append ( "AND (n.name CONTAINS $queryStr OR type(rel) CONTAINS $queryStr OR m.name CONTAINS $queryStr) " ) ;
}
if ( graphReqVO . getNodeLabels ( ) ! = null & & ! graphReqVO . getNodeLabels ( ) . isEmpty ( ) ) {
params . put ( "nodeLabels" , graphReqVO . getNodeLabels ( ) ) ;
relQuery . append ( "AND (ANY(label IN $nodeLabels WHERE label IN labels(n)) " +
"OR ANY(label IN $nodeLabels WHERE label IN labels(m))) " ) ;
}
if ( graphReqVO . getRelTypes ( ) ! = null & & ! graphReqVO . getRelTypes ( ) . isEmpty ( ) ) {
params . put ( "relTypes" , graphReqVO . getRelTypes ( ) ) ;
relQuery . append ( "AND ANY(type IN $relTypes WHERE type = type(rel)) " ) ;
}
relQuery . append ( "RETURN id(rel) as id, n.name as source, id(n) as sourceId, n.name as sourceName, type(rel) as relName, m.name as target, id(m) as targetId, m.name as targetName" ) ;
log . info ( "relQuery:{}" , relQuery ) ;
Result run = session . run ( relQuery . toString ( ) , params ) ;
while ( run . hasNext ( ) ) {
Record record = run . next ( ) ;
//long id = record.get("id").asLong();
//String source = record.get("source").asString();
// 组织边
long sourceId = record . get ( "sourceId" ) . asLong ( ) ;
String name = record . get ( "name" ) . asString ( ) ;
//String target = record.get("target").asString();
String relName = record . get ( "relName" ) . asString ( ) ;
long targetId = record . get ( "targetId" ) . asLong ( ) ;
list . add ( new WebRelDTO ( sourceId , targetId , name ) ) ;
}
Result node = session . run ( "MATCH (n) where n.picType = $picType and n.caseId = $caseId RETURN id(n) as id, n.name as name" , params ) ;
while ( node . hasNext ( ) ) {
Record record = node . next ( ) ;
String name = record . get ( "name" ) . asString ( ) ;
long idLong = record . get ( "id" ) . asLong ( ) ;
Map < String , String > nodeMap = new HashMap < > ( ) ;
nodeMap . put ( "name" , name ) ;
nodeMap . put ( "entityName" , name ) ;
nodeMap . put ( "id" , String . valueOf ( idLong ) ) ;
nodes . add ( nodeMap ) ;
list . add ( new WebRelDTO ( sourceId , targetId , relName ) ) ;
// 组织节点
Map < String , String > sourceNodeMap = new HashMap < > ( ) ;
sourceNodeMap . put ( "name" , record . get ( "sourceName" ) . asString ( ) ) ;
sourceNodeMap . put ( "id" , String . valueOf ( sourceId ) ) ;
nodes . add ( sourceNodeMap ) ;
Map < String , String > targetNodeMap = new HashMap < > ( ) ;
targetNodeMap . put ( "name" , record . get ( "targetName" ) . asString ( ) ) ;
targetNodeMap . put ( "id" , String . valueOf ( targetId ) ) ;
nodes . add ( targetNodeMap ) ;
}
} catch ( Exception e ) {
e. printStackTrace ( ) ;
log. error ( "查询失败" , e ) ;
}
// 根据节点ID去重
List < Map < String , String > > distinctNodes = new ArrayList < > ( nodes . stream ( )
. collect ( Collectors . toMap (
node - > node . get ( "id" ) , // 以 ID 为唯一键
node - > node , // 保留整个 Map 作为值
( existing , replacement ) - > existing ) ) // 如果有重复 ID，
. values ( ) ) ;
// 节点和关系合并
Map < String , NodeMapRecord > nodeRecordMap = electNodeRecord ( nodes ) ;
list = mergerWebRel ( list , nodeRecordMap ) ;
Map < String , NodeMapRecord > nodeRecordMap = electNodeRecord ( disti nctN odes) ;
list = mergerWebRel ( list , nodeRecordMap ) ;
nodes = mergeNode ( nodes , nodeRecordMap ) ;
map . put ( "list" , list ) ;
@ -320,6 +334,40 @@ public class Neo4jServiceImpl implements Neo4jService {
return R . ok ( map ) ;
}
@Override
public R < ? > getNodeAndRelationListByCaseId ( String picType , String caseId ) {
Map < String , Set < String > > map = new HashMap < > ( ) ;
Set < String > nodeLabels = new HashSet < > ( ) ;
Set < String > relTypes = new HashSet < > ( ) ;
try ( Session session = driver . session ( ) ) {
// 查询所有匹配caseId的节点及其关联关系
String query = "MATCH (n)-[r]->(m) WHERE n.caseId = $caseId AND m.caseId = $caseId RETURN labels(n) as sourceLabels, type(r) as relName, labels(m) as targetLabels" ;
session . executeRead ( tx - > {
Result result = tx . run ( query , Values . parameters ( "caseId" , caseId ) ) ;
while ( result . hasNext ( ) ) {
Record record = result . next ( ) ;
List < String > sourceLabels = record . get ( "sourceLabels" ) . asList ( Value : : asString ) ;
List < String > targetLabels = record . get ( "targetLabels" ) . asList ( Value : : asString ) ;
if ( ! sourceLabels . isEmpty ( ) ) {
nodeLabels . add ( sourceLabels . get ( 0 ) ) ;
}
if ( ! targetLabels . isEmpty ( ) ) {
nodeLabels . add ( targetLabels . get ( 0 ) ) ;
}
relTypes . add ( record . get ( "relName" ) . asString ( ) ) ;
}
return null ;
} ) ;
} catch ( Exception e ) {
log . error ( "查询失败" , e ) ;
}
log . info ( "查询到的节点类型{}个：{}" , nodeLabels . size ( ) , nodeLabels ) ;
log . info ( "查询到的关系类型{}个：{}" , relTypes . size ( ) , relTypes ) ;
map . put ( "nodeLabels" , nodeLabels ) ;
map . put ( "relTypes" , relTypes ) ;
return R . ok ( map ) ;
}
record NodeMapRecord ( String name , String id , Set < String > idSet ) {
}
@ -327,91 +375,93 @@ public class Neo4jServiceImpl implements Neo4jService {
/ * *
* 推 选 出 代 表 节 点 信 息
*
* @param nodes key : name , entityName , id 节 点 信 息
* @return
* @return key : name , value : NodeMapRecord
* /
private Map < String , NodeMapRecord > electNodeRecord ( List < Map < String , String > > nodes ) {
private Map < String , NodeMapRecord > electNodeRecord ( List < Map < String , String > > nodes ) {
Map < String , NodeMapRecord > nodeRecordMap = new HashMap < > ( ) ;
for ( Map < String , String > node : nodes ) {
String name = node . get ( "name" ) ;
String id = node . get ( "id" ) ;
NodeMapRecord nodeMapRecord = nodeRecordMap . get ( name ) ;
if ( nodeMapRecord = = null ) {
if ( nodeMapRecord = = null ) {
Set < String > idSet = new HashSet < > ( ) ;
idSet . add ( id ) ;
nodeRecordMap . put ( name , new NodeMapRecord ( name , id , idSet ) ) ;
} else {
nodeRecordMap . put ( name , new NodeMapRecord ( name , id , idSet ) ) ;
} else {
nodeMapRecord . idSet . add ( id ) ;
}
}
return nodeRecordMap ;
}
/ * *
* 合 并 节 点 信 息
* 合 并 节 点 信 息
* 合 并 依 据 ：
* name 为 唯 一 标 识
* @param nodes key : name , entityName , id
* name 为 唯 一 标 识
*
* @param nodes key : name , entityName , id
* @param nodeRecordMap 代 表 节 点 信 息
* @return
* @return 合 并 后 的 节 点 信 息
* /
private List < Map < String , String > > mergeNode ( List < Map < String , String > > nodes , Map < String , NodeMapRecord > nodeRecordMap ) {
private List < Map < String , String > > mergeNode ( List < Map < String , String > > nodes , Map < String , NodeMapRecord > nodeRecordMap ) {
return nodes . stream ( ) . map ( map - > {
Map < String , String > nodeMap = new HashMap < > ( ) ;
nodeMap . put ( "name" , map . get ( "name" ) ) ;
nodeMap . put ( "entityName" , map . get ( "entityName" ) ) ;
NodeMapRecord nodeMapRecord = nodeRecordMap . get ( map . get ( "name" ) ) ;
if ( null = = nodeMapRecord ) {
log . warn ( "mergeNode:节点信息异常， ： , map . get ( "name" ) ) ;
return nodeMap ;
}
if ( ! nodeMapRecord . idSet . contains ( map . get ( "id" ) ) ) {
log . warn ( "mergeNode:节点信息异常， , map . get ( "id" ) , map . get ( "name" ) ) ;
return nodeMap ;
}
nodeMap . put ( "id" , nodeMapRecord . id ) ;
return nodeMap ;
} ) . filter ( map - > StrUtil . isNotEmpty ( map . get ( "id" ) ) )
. filter ( distinctPredicate ( m - > m . get ( "id" ) ) ) . collect ( Collectors . toList ( ) ) ;
Map < String , String > nodeMap = new HashMap < > ( ) ;
nodeMap . put ( "name" , map . get ( "name" ) ) ;
NodeMapRecord nodeMapRecord = nodeRecordMap . get ( map . get ( "name" ) ) ;
if ( null = = nodeMapRecord ) {
log . warn ( "mergeNode:节点信息异常， ： , map . get ( "name" ) ) ;
return nodeMap ;
}
if ( ! nodeMapRecord . idSet . contains ( map . get ( "id" ) ) ) {
log . warn ( "mergeNode:节点信息异常， , map . get ( "id" ) , map . get ( "name" ) ) ;
return nodeMap ;
}
nodeMap . put ( "id" , nodeMapRecord . id ) ;
return nodeMap ;
} ) . filter ( map - > StrUtil . isNotEmpty ( map . get ( "id" ) ) )
. filter ( distinctPredicate ( m - > m . get ( "id" ) ) ) . collect ( Collectors . toList ( ) ) ;
}
/ * *
* 合 并 关 系 信 息
*
* @param webRelDTOList 关 系 信 息
* @param nodeRecordMap 代 表 节 点 信 息
* @return
* @param nodeRecordMap 代 表 节 点 信 息
* @return 合 并 后 的 关 系 信 息
* /
private List < WebRelDTO > mergerWebRel ( List < WebRelDTO > webRelDTOList , Map < String , NodeMapRecord > nodeRecordMap ) {
Map < String , NodeMapRecord > idNodeRecordMap = nodeRecordMap . entrySet ( ) . stream ( )
. collect ( Collectors . toMap ( entry - > entry . getValue ( ) . id , Map . Entry : : getValue ) ) ;
return webRelDTOList . stream ( ) . map ( webRelDTO - > {
String target = webRelDTO . getTarget ( ) ;
String source = webRelDTO . getSource ( ) ;
String name = webRelDTO . getName ( ) ;
return webRelDTOList . stream ( ) . map ( webRelDTO - > {
String target = webRelDTO . getTarget ( ) ;
String source = webRelDTO . getSource ( ) ;
String name = webRelDTO . getName ( ) ;
String sourceNew = idNodeRecordMap . entrySet ( ) . stream ( )
. filter ( entry - > entry . getValue ( ) . idSet . contains ( source ) )
. findAny ( ) . map ( Map . Entry : : getKey ) . orElse ( "" ) ;
String sourceNew = idNodeRecordMap . entrySet ( ) . stream ( )
. filter ( entry - > entry . getValue ( ) . idSet . contains ( source ) )
. findAny ( ) . map ( Map . Entry : : getKey ) . orElse ( "" ) ;
String targetNew = idNodeRecordMap . entrySet ( ) . stream ( )
. filter ( entry - > entry . getValue ( ) . idSet . contains ( target ) )
. findAny ( ) . map ( Map . Entry : : getKey ) . orElse ( "" ) ;
String targetNew = idNodeRecordMap . entrySet ( ) . stream ( )
. filter ( entry - > entry . getValue ( ) . idSet . contains ( target ) )
. findAny ( ) . map ( Map . Entry : : getKey ) . orElse ( "" ) ;
if ( StrUtil . isEmpty ( sourceNew ) | | StrUtil . isEmpty ( targetNew ) ) {
log . warn ( "mergerWebRel:关系信息异常， , source , target , name ) ;
}
if ( StrUtil . isEmpty ( sourceNew ) | | StrUtil . isEmpty ( targetNew ) ) {
log . warn ( "mergerWebRel:关系信息异常， , source , target , name ) ;
}
return new WebRelDTO ( sourceNew , targetNew , name ) ;
} ) . filter ( webRelDTO - > StrUtil . isNotEmpty ( webRelDTO . getSource ( ) ) & & StrUtil . isNotEmpty ( webRelDTO . getTarget ( ) ) )
return new WebRelDTO ( sourceNew , targetNew , name ) ;
} ) . filter ( webRelDTO - > StrUtil . isNotEmpty ( webRelDTO . getSource ( ) ) & & StrUtil . isNotEmpty ( webRelDTO . getTarget ( ) ) )
. filter ( distinctPredicate ( rel - > rel . getSource ( ) + rel . getTarget ( ) ) ) . toList ( ) ;
}
private < K > Predicate < K > distinctPredicate ( Function < K , Object > function ) {
private < K > Predicate < K > distinctPredicate ( Function < K , Object > function ) {
ConcurrentHashMap < Object , Boolean > map = new ConcurrentHashMap < > ( ) ;
return ( t ) - > null = = map . putIfAbsent ( function . apply ( t ) , true ) ;
return ( t ) - > null = = map . putIfAbsent ( function . apply ( t ) , true ) ;
}
xueqingkun
5 months ago
