Function check

packages/db/src/clickhouse/migration.ts:466–512

()

            let checking = false; // Add flag to prevent multiple concurrent checks

            async function check() {
              if (checking) return; // Skip if already checking
              checking = true;

              try {
                const res = await chMigrationClient
                  .query({
                    query: `SELECT
                              query_id,
                              elapsed,
                              read_rows,
                              written_rows,
                              memory_usage
                            FROM system.processes 
                            WHERE query_id = '${activeQueryId}'`,
                    format: 'JSONEachRow',
                  })
                  .then((res) => res.json());

                const formatMemory = (bytes: number) => {
                  const units = ['B', 'KB', 'MB', 'GB'];
                  let size = bytes;
                  let unitIndex = 0;
                  while (size >= 1024 && unitIndex < units.length - 1) {
                    size /= 1024;
                    unitIndex++;
                  }
                  return `${Math.round(size * 100) / 100}${units[unitIndex]}`;
                };

                const formatNumber = (num: number) => {
                  return num.toString().replace(/\B(?=(\d{3})+(?!\d))/g, ',');
                };

                if (Array.isArray(res) && res.length > 0) {
                  const { elapsed, read_rows, written_rows, memory_usage } =
                    res[0] as any;
                  console.log(
                    `Progress: ${elapsed.toFixed(2)}s | Memory: ${formatMemory(memory_usage)} | Read: ${formatNumber(read_rows)} rows | Written: ${formatNumber(written_rows)} rows`,
                  );
                }
              } finally {
                checking = false;
              }

              timer = setTimeout(check, 5000); // Schedule next check after current one completes
            }

            // Start the first check after 5 seconds
            timer = setTimeout(check, 5000);