提交 b987942b 编写于 作者: D dingbo

docs: sql_writer.py

上级 ad461ba7
......@@ -60,9 +60,9 @@ if __name__ == '__main__':
def consume():
global count
for data in MockDataSource("1", 1000):
for _ in data:
count += 1
for data in MockDataSource("1", 100):
for _, rows in data:
count += len(rows)
Thread(target=consume).start()
......
......@@ -38,7 +38,7 @@ class SQLWriter:
group_id = ps[-1]
tag_value = f"('{location}',{group_id})"
self._tb_tags[table_name] = tag_value
self.flush()
self.flush2()
def flush(self):
"""
......@@ -63,6 +63,12 @@ class SQLWriter:
self.execute_sql(sql)
self._tb_values.clear()
def flush2(self):
for tb_name, values in self._tb_values.items():
q = "INSERT INTO " + tb_name + " VALUES " + values
self.execute_sql(q)
self._tb_values.clear()
def execute_sql(self, sql):
try:
self._conn.execute(sql)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册