diff --git a/dtable_events/automations/actions.py b/dtable_events/automations/actions.py index 5f2e70e6..83879286 100644 --- a/dtable_events/automations/actions.py +++ b/dtable_events/automations/actions.py @@ -1693,13 +1693,12 @@ def get_columns_dict(self, table_id): return column_dict def query_table_rows(self, table_name, column_names): - sql_columns = ', '.join(column_names) - limit = 0 + sql_columns = ', '.join(['`' + col_name + '`' for col_name in column_names]) + start = 0 step = 10000 result_rows = [] while True: - # sql = f"select `_id`, {sql_columns} from `{table_name}` limit {limit},{step}" - sql = f"select * from `{table_name}`" + sql = f"select `_id`, {sql_columns} from `{table_name}` limit {start},{step}" try: results = self.auto_rule.dtable_db_api.query(sql) except Exception as e: @@ -1707,7 +1706,7 @@ def query_table_rows(self, table_name, column_names): logger.error('query dtable: %s, table name: %s, error: %s', self.auto_rule.dtable_uuid, table_name, e) return [] result_rows += results - limit += step + start += step if len(results) < step: break return result_rows