From b987942b0e836c6d3d5bb7e25490e779dd21c79e Mon Sep 17 00:00:00 2001
From: dingbo <dingbo8128@163.com>
Date: Thu, 14 Jul 2022 17:43:51 +0800
Subject: [PATCH] docs: sql_writer.py

---
 docs/examples/python/mockdatasoruce.py | 6 +++---
 docs/examples/python/sql_writer.py     | 8 +++++++-
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/docs/examples/python/mockdatasoruce.py b/docs/examples/python/mockdatasoruce.py
index 12c18e57f1..f6c54a8f96 100644
--- a/docs/examples/python/mockdatasoruce.py
+++ b/docs/examples/python/mockdatasoruce.py
@@ -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()
diff --git a/docs/examples/python/sql_writer.py b/docs/examples/python/sql_writer.py
index 886f257001..3daaa7d201 100644
--- a/docs/examples/python/sql_writer.py
+++ b/docs/examples/python/sql_writer.py
@@ -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)
-- 
GitLab