17:47:58       rv = self.invoke(ctx)
17:47:58     File "/root/.pex/installed_wheels/fffc555c02d3e106d99152a9e77adc93e581517e021fa9e5925e50c76cfc41b4/click-8.0.4-py3-none-any.whl/click/core.py", line 1659, in invoke
17:47:58       return _process_result(sub_ctx.command.invoke(sub_ctx))
17:47:58     File "/root/.pex/installed_wheels/fffc555c02d3e106d99152a9e77adc93e581517e021fa9e5925e50c76cfc41b4/click-8.0.4-py3-none-any.whl/click/core.py", line 1395, in invoke
17:47:58       return ctx.invoke(self.callback, **ctx.params)
17:47:58     File "/root/.pex/installed_wheels/fffc555c02d3e106d99152a9e77adc93e581517e021fa9e5925e50c76cfc41b4/click-8.0.4-py3-none-any.whl/click/core.py", line 754, in invoke
17:47:58       return __callback(*args, **kwargs)
17:47:58     File "/root/.pex/installed_wheels/247320f755562a5b42c5a63a4b4a99a4a5a077492554dc993344265c00bdfa2a/workload-0.1.0-py3-none-any.whl/workload/op_rqg_case.py", line 176, in run_rqg_case
17:47:58       task.result()
17:47:58     File "/usr/lib/python3.8/concurrent/futures/_base.py", line 437, in result
17:47:58       return self.__get_result()
17:47:58     File "/usr/lib/python3.8/concurrent/futures/_base.py", line 389, in __get_result
17:47:58       raise self._exception
17:47:58     File "/usr/lib/python3.8/concurrent/futures/thread.py", line 57, in run
17:47:58       result = self.fn(*self.args, **self.kwargs)
17:47:58     File "/root/.pex/installed_wheels/247320f755562a5b42c5a63a4b4a99a4a5a077492554dc993344265c00bdfa2a/workload-0.1.0-py3-none-any.whl/workload/op_rqg_case.py", line 170, in handle
17:47:58       RqgSuiteRunner(suite, reader, logger, case_workers).run()
17:47:58     File "/root/.pex/installed_wheels/247320f755562a5b42c5a63a4b4a99a4a5a077492554dc993344265c00bdfa2a/workload-0.1.0-py3-none-any.whl/workload/op_rqg_case.py", line 130, in run
17:47:58       self.prepare_data()
17:47:58     File "/root/.pex/installed_wheels/247320f755562a5b42c5a63a4b4a99a4a5a077492554dc993344265c00bdfa2a/workload-0.1.0-py3-none-any.whl/workload/op_rqg_case.py", line 77, in prepare_data
17:47:58       self.db_client.run_sql(self.suite.data_sql_content)
17:47:58     File "/root/.pex/installed_wheels/247320f755562a5b42c5a63a4b4a99a4a5a077492554dc993344265c00bdfa2a/workload-0.1.0-py3-none-any.whl/workload/op_db.py", line 112, in run_sql
17:47:58       conn.commit()
17:47:58     File "/root/.pex/installed_wheels/f770db718ad01c6034e6ddd7c28a13f2f602c5818fc18f4a04d833fe86436058/DBUtils-3.0.3-py3-none-any.whl/dbutils/steady_db.py", line 439, in commit
17:47:58       raise error  # re-raise the original error
17:47:58     File "/root/.pex/installed_wheels/f770db718ad01c6034e6ddd7c28a13f2f602c5818fc18f4a04d833fe86436058/DBUtils-3.0.3-py3-none-any.whl/dbutils/steady_db.py", line 430, in commit
17:47:58       self._con.commit()
17:47:58     File "/root/.pex/installed_wheels/ef6fb62f863fb17f59d92343c780fecb0350a0b93d549482f1646e1a00a4b100/PyMySQL-1.1.0-py3-none-any.whl/pymysql/connections.py", line 478, in commit
17:47:58       self._read_ok_packet()
17:47:58     File "/root/.pex/installed_wheels/ef6fb62f863fb17f59d92343c780fecb0350a0b93d549482f1646e1a00a4b100/PyMySQL-1.1.0-py3-none-any.whl/pymysql/connections.py", line 448, in _read_ok_packet
17:47:58       pkt = self._read_packet()
17:47:58     File "/root/.pex/installed_wheels/ef6fb62f863fb17f59d92343c780fecb0350a0b93d549482f1646e1a00a4b100/PyMySQL-1.1.0-py3-none-any.whl/pymysql/connections.py", line 752, in _read_packet
17:47:58       raise err.InternalError(
17:47:58   pymysql.err.InternalError: Packet sequence number wrong - got 2 expected 1

@dataclass
class DbClient:  # pylint: disable=too-many-instance-attributes
    """
    db client
    """

    host: str
    port: str
    stream_port: str = ""
    user: str = "root"
    password: str = ""
    version: str = ""
    database: Optional[str] = None
    max_connection: int = 1
    _pool: PooledDB = field(init=False)

    def __post_init__(self) -> None:
        self._pool = PooledDB(
            pymysql,
            maxconnections=self.max_connection * 2,
            blocking=True,
            host=self.host,
            port=int(self.port),
            user=self.user,
            passwd=self.password,
            db=self.database,
            charset="utf8mb4",
            setsession=[
                "SET SESSION net_read_timeout=600",
                "SET SESSION net_write_timeout=600",
            ],
            maxcached=-1,
            mincached=-1,
        )

    def get_db(self):
            return self._pool.connection()

    def run_sql(self, sql: str, *params: Any, cursor_cls: cursors.Cursor = None):
        conn = self.get_db()
        with conn_inuse(conn):
            with conn.cursor(cursor=cursor_cls) as cursor:
                if len(params) > 0:
                    cursor.execute(sql, params)
                else:
                    cursor.execute(sql)
                result = cursor.fetchall()
            conn.commit()
        del conn
        return result

def handle(suite):
        SuiteRunner(suite, reader, logger, case_workers).run()

    with ThreadPoolExecutor(max_workers=suite_workers) as executor:
        tasks = {executor.submit(handle, suite): suite.name for suite in suites}
        for task in as_completed(tasks):
            try:
                task.result()
            except:
                # If one suite failed due to unhandled exception, then exit.
                print(f"Suite({tasks[task]})) failed")
                raise

  def run(self):
        self.logger.on_suite_started(self.suite.name)
        try:
            self.prepare_data()
            print(
                f"Run sql for suite {self.suite.name}, workers num: {self.worker_num}"
            )

def prepare_data(self):
        print(f"Prepare database for suite {self.suite.name}")
        self.no_db_client.run_sql(f"drop database if exists {self._db_name}")
        self.no_db_client.run_sql(f"create database {self._db_name}")

        print(f"Prepare data for suite {self.suite.name}")
        self.db_client.run_sql(self.suite.data_sql_content)