diff --git a/docs/examples/python/mockdatasoruce.py b/docs/examples/python/mockdatasoruce.py
index 12c18e57f13aebdf4bd0481845c2b31a93f6d63f..f6c54a8f96322f4495c91de54ce6bc2face0b5dc 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 886f257001e0ca16fd52224fe2fb6c76c54b4a8b..3daaa7d201fd46d6d24ff7f6ba724a823813bdb1 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)