[feat][hbase] 任务结束时将数据写出

AndrewYang9112 · Dec 16, 2020 · 3d91393 · 3d91393
1 parent fea8a43
commit 3d91393
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/hbase/hbase-sink/src/main/java/com/dtstack/flink/sql/sink/hbase/HbaseOutputFormat.java b/hbase/hbase-sink/src/main/java/com/dtstack/flink/sql/sink/hbase/HbaseOutputFormat.java
@@ -300,9 +300,8 @@ private Map<String, Object> rowConvertMap(Row record) {
 
     @Override
     public void close() throws IOException {
-        if (conn != null) {
-            conn.close();
-            conn = null;
+        if (!records.isEmpty()) {
+            dealBatchOperation(records);
         }
 
         if (scheduledFuture != null) {
@@ -311,6 +310,12 @@ public void close() throws IOException {
                 executor.shutdownNow();
             }
         }
+
+        if (conn != null) {
+            conn.close();
+            conn = null;
+        }
+
     }
 
     private HbaseOutputFormat() {