add replace_include and replace_exclude

RickyHuo · Dec 8, 2017 · bb30049 · bb30049
1 parent 348d54a
commit bb30049
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main/java/com/sina/bip/hangout/outputs/Clickhouse.java b/src/main/java/com/sina/bip/hangout/outputs/Clickhouse.java
@@ -76,7 +76,7 @@ protected void prepare() {
 
         if (this.config.containsKey("replace_include_fields") && this.config.containsKey("replace_exclude_fields")) {
             System.out.println("Replace_include_fields and replace_exclude_fields exist at the same time," +
-                    " please use one of them");
+                    " please use one of them.");
             System.exit(1);
         }
 
@@ -136,7 +136,7 @@ protected StringBuilder makeUpSql(List<Map> events) {
                         String fieldValue = e.get(field).toString();
                         if ((this.replace_include_fields != null && this.replace_include_fields.contains(field)) ||
                                 (this.replace_exclude_fields != null && !this.replace_exclude_fields.contains(field))) {
-                            dealWithQuote(fieldValue);
+                            value += "'" + dealWithQuote(fieldValue) + "'";
                         } else {
                             value += "'" + fieldValue + "'";
                         }
@@ -171,6 +171,7 @@ protected void bulkInsert(Map event) throws Exception{
             Connection conn = balanced.getConnection();
             try {
                 conn.createStatement().execute(sqls.toString());
+                conn.close();
             } catch (SQLException e){
                 System.out.println(sqls.toString());
                 System.out.println(e.toString());