import com.alibaba.fastjson.JSON;
import com.google.common.base.Throwables;
import com.google.common.collect.Lists;
import org.neo4j.ogm.model.QueryStatistics;
import org.neo4j.ogm.model.Result;
import org.neo4j.ogm.session.Session;
import org.neo4j.ogm.session.SessionFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@Component
public class CyberQueryExecuteUtil {
private static final Logger LOGGER = LoggerFactory.getLogger(CyberQueryExecuteUtil.class);
private static SessionFactory SESSION_FACTORY;
@Autowired
public void init(SessionFactory sessionFactory) {
SESSION_FACTORY = sessionFactory;
}
/**
* 执行统计方法
*
* @param cql cypher查询语句
* @return 统计结果
*/
public static Long executeCountQuery(String cql) {
try {
Session session = SESSION_FACTORY.openSession();
Long count = session.queryForObject(Long.class, cql, new HashMap<>());
session.clear();
LOGGER.debug("Neo4j cypher query: [{}], result: [{}]", cql, count);
return count;
} catch (Exception e) {
LOGGER.error(Throwables.getStackTraceAsString(e));
return 0L;
}
}
/**
* 执行实体查询方法
*
* @param cql cypher查询语句
* @param className 实体类型
* @return 查询结果
*/
public static <T> Iterable<T> executeEntityQuery(String cql, Class<T> className) {
try {
Session session = SESSION_FACTORY.openSession();
Iterable<T> response = session.query(className, cql, new HashMap<>());
session.clear();
LOGGER.debug("Neo4j cypher query: [{}]", cql);
return response;
} catch (Exception e) {
LOGGER.error(Throwables.getStackTraceAsString(e));
return new ArrayList<>();
}
}
/**
* 执行实体查询方法
*
* @param cql cypher查询语句
* @return 查询原始结果
*/
public static List<Map<String, Object>> executeOriginResultQuery(String cql) {
try {
Session session = SESSION_FACTORY.openSession();
Result response = session.query(cql, new HashMap<>());
session.clear();
LOGGER.debug("Neo4j cypher query: [{}], status: [{}]", cql, JSON.toJSONString(response.queryStatistics()));
return Lists.newArrayList(response.queryResults());
} catch (Exception e) {
LOGGER.error(Throwables.getStackTraceAsString(e));
return new ArrayList<>();
}
}
/**
* 执行neo4j更新语句
*
* @param cql cypher更新语句
*/
public static void executeUpdateQuery(String cql) {
try {
Session session = SESSION_FACTORY.openSession();
Result result = session.query(cql, new HashMap<>());
QueryStatistics queryStatistics = result.queryStatistics();
session.clear();
LOGGER.debug("Neo4j execute cql: [{}] contain update:[{}], status: [{}]", cql, queryStatistics.containsUpdates(), JSON.toJSON(queryStatistics));
} catch (Exception e) {
LOGGER.error(Throwables.getStackTraceAsString(e));
}
}
/**
* 执行neo4j批量更新语句
*
* @param cql cypher批量更新语句
*/
public static void executeBatchUpdateQuery(String cql) {
try {
Session session = SESSION_FACTORY.openSession();
Result result = session.query(cql, new HashMap<>());
QueryStatistics queryStatistics = result.queryStatistics();
session.clear();
LOGGER.debug("Neo4j batch save cql: [{}] contain update:[{}], status: [{}]", cql, queryStatistics.containsUpdates(), JSON.toJSON(queryStatistics));
} catch (Exception e) {
LOGGER.error(Throwables.getStackTraceAsString(e));
}
}
/**
* 执行neo4j批量更新语句
*
* @param cql cypher批量更新语句
*/
public static void executeBatchUpdateQuery(String cql, Map<String, Object> param) {
try {
Session session = SESSION_FACTORY.openSession();
Result result = session.query(cql, param);
QueryStatistics queryStatistics = result.queryStatistics();
session.clear();
LOGGER.debug("Neo4j batch save cql: [{}] contain update:[{}], status: [{}]", cql, queryStatistics.containsUpdates(), JSON.toJSON(queryStatistics));
} catch (Exception e) {
LOGGER.error(Throwables.getStackTraceAsString(e));
}
}
}