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()