From e88c357d8899e9cdff8d8cadbbc999dfe90193f8 Mon Sep 17 00:00:00 2001
From: wuaho <wuaho@legu.cc>
Date: Tue, 26 Oct 2021 16:50:28 +0800
Subject: [PATCH] =?UTF-8?q?=E9=81=BF=E5=85=8D=E9=87=8D=E5=B9=B3=E8=A1=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app.py             | 2 +-
 utils/post_data.py | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/app.py b/app.py
index 06466b5..56189e1 100644
--- a/app.py
+++ b/app.py
@@ -27,7 +27,7 @@ def handler_userid(properties):
 
 
 def run():
-    post_data = PostData()
+    post_data = PostData(client)
     schedule.every(60).seconds.do(post_data.post)
     idx = get_data_id()
 
diff --git a/utils/post_data.py b/utils/post_data.py
index 0ad4cf3..94994f8 100644
--- a/utils/post_data.py
+++ b/utils/post_data.py
@@ -14,8 +14,9 @@ requests.packages.urllib3.disable_warnings()
 
 
 class PostData:
-    def __init__(self):
+    def __init__(self, kafka_client):
         self.data_list = []
+        self.kafka_client = kafka_client
 
     def add(self, data):
         self.data_list.append(data)
@@ -53,6 +54,7 @@ class PostData:
         except Exception as e:
             print(e)
         finally:
+            self.kafka_client.commit()
             self.set_idx()
             self.clear_data()