1717
1818package org .apache .zeppelin .graph .neo4j ;
1919
20- import java .util .HashMap ;
21- import java .util .HashSet ;
22- import java .util .List ;
23- import java .util .Map ;
24- import java .util .Properties ;
25- import java .util .Set ;
26- import java .util .regex .Matcher ;
27- import java .util .regex .Pattern ;
2820import org .apache .commons .lang3 .StringUtils ;
29- import org .apache .zeppelin .interpreter .InterpreterContext ;
30- import org .apache .zeppelin .resource .Resource ;
31- import org .apache .zeppelin .resource .ResourcePool ;
3221import org .neo4j .driver .AuthToken ;
3322import org .neo4j .driver .AuthTokens ;
3423import org .neo4j .driver .Config ;
3524import org .neo4j .driver .Driver ;
3625import org .neo4j .driver .GraphDatabase ;
3726import org .neo4j .driver .Record ;
38- import org .neo4j .driver .Result ;
3927import org .neo4j .driver .Session ;
28+ import org .neo4j .driver .Result ;
4029import org .neo4j .driver .SessionConfig ;
4130import org .slf4j .Logger ;
4231import org .slf4j .LoggerFactory ;
4332
33+ import java .util .HashMap ;
34+ import java .util .HashSet ;
35+ import java .util .List ;
36+ import java .util .Map ;
37+ import java .util .Properties ;
38+ import java .util .Set ;
39+ import java .util .regex .Matcher ;
40+ import java .util .regex .Pattern ;
41+
42+ import org .apache .zeppelin .interpreter .InterpreterContext ;
43+ import org .apache .zeppelin .resource .Resource ;
44+ import org .apache .zeppelin .resource .ResourcePool ;
45+
4446/**
4547 * Neo4j connection manager for Zeppelin.
4648 */
4749public class Neo4jConnectionManager {
48-
4950 private static final Logger LOGGER = LoggerFactory .getLogger (Neo4jConnectionManager .class );
5051
5152 public static final String NEO4J_SERVER_URL = "neo4j.url" ;
@@ -123,11 +124,11 @@ public void close() {
123124
124125 private Session getSession () {
125126 return getDriver ().session (StringUtils .isNotEmpty (database ) ?
126- SessionConfig .forDatabase (database ) : SessionConfig .defaultConfig ());
127+ SessionConfig .forDatabase (database ) : SessionConfig .defaultConfig ());
127128 }
128129
129130 public List <Record > execute (String cypherQuery ,
130- InterpreterContext interpreterContext ) {
131+ InterpreterContext interpreterContext ) {
131132 Map <String , Object > params = new HashMap <>();
132133 if (interpreterContext != null ) {
133134 ResourcePool resourcePool = interpreterContext .getResourcePool ();
@@ -143,7 +144,7 @@ public List<Record> execute(String cypherQuery,
143144 LOGGER .debug ("Executing cypher query {} with params {}" , cypherQuery , params );
144145 try (Session session = getSession ()) {
145146 final Result result = params .isEmpty ()
146- ? session .run (cypherQuery ) : session .run (cypherQuery , params );
147+ ? session .run (cypherQuery ) : session .run (cypherQuery , params );
147148 return result .list ();
148149 }
149150 }
@@ -161,3 +162,4 @@ private Set<String> extractParams(String cypherQuery, Pattern pattern, String re
161162 return keys ;
162163 }
163164}
165+
0 commit comments