first commit

2020-06-28 18:20:38 +08:00
parent 20af47f202
commit 6f86960a70
24 changed files with 1456 additions and 0 deletions
--- a/IP-learning-graph/src/main/java/cn/ac/iie/utils/ClickhouseConnect.java
+++ b/IP-learning-graph/src/main/java/cn/ac/iie/utils/ClickhouseConnect.java
@@ -0,0 +1,109 @@
+package cn.ac.iie.utils;
+
+import com.alibaba.druid.pool.DruidDataSource;
+import com.alibaba.druid.pool.DruidPooledConnection;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.sql.Connection;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.sql.Statement;
+import java.util.Properties;
+
+public class ClickhouseConnect {
+    private static final Logger LOG = LoggerFactory.getLogger(ClickhouseConnect.class);
+    private static DruidDataSource dataSource = null;
+    private static ClickhouseConnect dbConnect = null;
+    private static Properties props = new Properties();
+
+    static {
+        getDbConnect();
+    }
+
+    private static void getDbConnect() {
+        try {
+            if (dataSource == null) {
+                dataSource = new DruidDataSource();
+                props.load(ClickhouseConnect.class.getClassLoader().getResourceAsStream("clickhouse.properties"));
+                //设置连接参数
+                dataSource.setUrl("jdbc:clickhouse://" + props.getProperty("db.id"));
+                dataSource.setDriverClassName(props.getProperty("drivers"));
+                dataSource.setUsername(props.getProperty("mdb.user"));
+                dataSource.setPassword(props.getProperty("mdb.password"));
+                //配置初始化大小、最小、最大
+                dataSource.setInitialSize(Integer.parseInt(props.getProperty("initialsize")));
+                dataSource.setMinIdle(Integer.parseInt(props.getProperty("minidle")));
+                dataSource.setMaxActive(Integer.parseInt(props.getProperty("maxactive")));
+                //配置获取连接等待超时的时间
+                dataSource.setMaxWait(30000);
+                //配置间隔多久才进行一次检测，检测需要关闭的空闲连接，单位是毫秒
+                dataSource.setTimeBetweenEvictionRunsMillis(2000);
+                //防止过期
+                dataSource.setValidationQuery("SELECT 1");
+                dataSource.setTestWhileIdle(true);
+                dataSource.setTestOnBorrow(true);
+                dataSource.setKeepAlive(true);
+            }
+        } catch (Exception e) {
+            LOG.error(e.getMessage());
+
+        }
+    }
+
+    /**
+     * 数据库连接池单例
+     *
+     * @return dbConnect
+     */
+    public static synchronized ClickhouseConnect getInstance() {
+        if (null == dbConnect) {
+            dbConnect = new ClickhouseConnect();
+        }
+        return dbConnect;
+    }
+
+    /**
+     * 返回druid数据库连接
+     *
+     * @return 连接
+     * @throws SQLException sql异常
+     */
+    public DruidPooledConnection getConnection() throws SQLException {
+        return dataSource.getConnection();
+    }
+
+    /**
+     * 清空PreparedStatement、Connection对象，未定义的置空。
+     *
+     * @param pstmt      PreparedStatement对象
+     * @param connection Connection对象
+     */
+    public void clear(Statement pstmt, Connection connection) {
+        try {
+            if (pstmt != null) {
+                pstmt.close();
+            }
+            if (connection != null) {
+                connection.close();
+            }
+        } catch (SQLException e) {
+            LOG.error(e.getMessage());
+        }
+
+    }
+
+    public ResultSet executorQuery(String query,Connection connection,Statement pstm){
+//        Connection connection = null;
+//        Statement pstm = null;
+        try {
+            connection = getConnection();
+            pstm = connection.createStatement();
+            return pstm.executeQuery(query);
+        }catch (Exception e){
+            LOG.error(e.getMessage());
+            return null;
+        }
+    }
+
+}