Function do_repeat

tools/cdb2sql/cdb2sql.cpp:1759–1816

struct winsize win_size;

static int do_repeat(char *sql)
{
    int64_t start_time_ms_tot = 0, run_time_ms_tot = 0;
    int start_time_ms, run_time_ms;

    /* Trim whitespace and then ignore comments and empty lines. */
    while (isspace(*sql))
        sql++;

    if (sql[0] == '#' || sql[0] == '\0' || (sql[0] == '-' && sql[1] == '-'))
        return 0;

    if (sql[0] == '@') {
        return process_bind(sql);
    }

    printmode |= DISP_NONE;

    ioctl(STDOUT_FILENO, TIOCGWINSZ, &win_size);
    int progress_bar_width = win_size.ws_col - 20;
    int spaces = 0;
    int stars = 0;
    float progress_pct = 0;
    int last_progress_pct = 0;

    /* Execute the query specified number of times */
    for (int i = 1; i <= repeat; i++) {
        start_time_ms = 0;
        run_time_ms = 0;
        process_line(sql, 0, NULL, &start_time_ms, &run_time_ms);
        start_time_ms_tot += start_time_ms;
        run_time_ms_tot += run_time_ms;

        progress_pct = ((float)i / repeat) * 100;
        if ((int)progress_pct > last_progress_pct) {
            last_progress_pct = (int)progress_pct;
            stars = (progress_pct * progress_bar_width) / 100;
            spaces = progress_bar_width - stars;
            printf("executing: [%s%s] %.d%%\r", std::string(stars, '*').c_str(),
                   std::string(spaces, ' ').c_str(), (int)progress_pct);
            fflush(stdout);
        }
    }
    cdb2_clearbindings(cdb2h);

    printmode &= (~DISP_NONE);

    printf("\n");

    /* Print the summary */
    if (time_mode) {
        printf("summary:\n");
        printf("  total prep time  %" PRId64 " ms\n", start_time_ms_tot);
        printf("  total run time   %" PRId64 " ms\n", run_time_ms_tot);
    }

    return 0;
}