eta.c - platform/external/fio - Gitiles

 /*
  * Status and ETA code
  */
 #include <unistd.h>
 #include <fcntl.h>
 #include <string.h>

 #include "fio.h"

 static char run_str[MAX_JOBS + 1];

 /*
  * Sets the status of the 'td' in the printed status map.
  */
 static void check_str_update(struct thread_data *td)
 {
 	char c = run_str[td->thread_number - 1];

 	switch (td->runstate) {
 		case TD_REAPED:
 			c = '_';
 			break;
 		case TD_EXITED:
 			c = 'E';
 			break;
 		case TD_RUNNING:
 			if (td_rw(td)) {
 				if (td_random(td))
 					c = 'm';
 				else
 					c = 'M';
 			} else if (td_read(td)) {
 				if (td_random(td))
 					c = 'r';
 				else
 					c = 'R';
 			} else {
 				if (td_random(td))
 					c = 'w';
 				else
 					c = 'W';
 			}
 			break;
 		case TD_VERIFYING:
 			c = 'V';
 			break;
 		case TD_FSYNCING:
 			c = 'F';
 			break;
 		case TD_CREATED:
 			c = 'C';
 			break;
 		case TD_INITIALIZED:
 			c = 'I';
 			break;
 		case TD_NOT_CREATED:
 			c = 'P';
 			break;
 		default:
 			log_err("state %d\n", td->runstate);
 	}

 	run_str[td->thread_number - 1] = c;
 }

 /*
  * Convert seconds to a printable string.
  */
 static void eta_to_str(char *str, int eta_sec)
 {
 	unsigned int d, h, m, s;
 	int disp_hour = 0;

 	d = h = m = s = 0;

 	s = eta_sec % 60;
 	eta_sec /= 60;
 	m = eta_sec % 60;
 	eta_sec /= 60;
 	h = eta_sec % 24;
 	eta_sec /= 24;
 	d = eta_sec;

 	if (d) {
 		disp_hour = 1;
 		str += sprintf(str, "%02ud:", d);
 	}

 	if (h || disp_hour)
 		str += sprintf(str, "%02uh:", h);

 	str += sprintf(str, "%02um:", m);
 	str += sprintf(str, "%02us", s);
 }

 /*
  * Best effort calculation of the estimated pending runtime of a job.
  */
 static int thread_eta(struct thread_data *td, unsigned long elapsed)
 {
 	unsigned long long bytes_total, bytes_done;
 	unsigned long eta_sec = 0;

 	bytes_total = td->total_io_size;

 	/*
 	 * if writing, bytes_total will be twice the size. If mixing,
 	 * assume a 50/50 split and thus bytes_total will be 50% larger.
 	 */
 	if (td->o.do_verify && td->o.verify && td_write(td)) {
 		if (td_rw(td))
 			bytes_total = bytes_total * 3 / 2;
 		else
 			bytes_total <<= 1;
 	}

 	if (td->o.zone_size && td->o.zone_skip)
 		bytes_total /= (td->o.zone_skip / td->o.zone_size);

 	if (td->runstate == TD_RUNNING || td->runstate == TD_VERIFYING) {
 		double perc, perc_t;

 		bytes_done = td->io_bytes[DDIR_READ] + td->io_bytes[DDIR_WRITE];
 		perc = (double) bytes_done / (double) bytes_total;
 		if (perc > 1.0)
 			perc = 1.0;

 		if (td->o.time_based) {
 			perc_t = (double) elapsed / (double) td->o.timeout;
 			if (perc_t < perc)
 				perc = perc_t;
 		}

 		eta_sec = (unsigned long) (elapsed * (1.0 / perc)) - elapsed;

 		if (td->o.timeout && eta_sec > (td->o.timeout - elapsed))
 			eta_sec = td->o.timeout - elapsed;
 	} else if (td->runstate == TD_NOT_CREATED || td->runstate == TD_CREATED
 			|| td->runstate == TD_INITIALIZED) {
 		int t_eta = 0, r_eta = 0;

 		/*
 		 * We can only guess - assume it'll run the full timeout
 		 * if given, otherwise assume it'll run at the specified rate.
 		 */
 		if (td->o.timeout)
 			t_eta = td->o.timeout + td->o.start_delay - elapsed;
 		if (td->o.rate) {
 			r_eta = (bytes_total / 1024) / td->o.rate;
 			r_eta += td->o.start_delay - elapsed;
 		}

 		if (r_eta && t_eta)
 			eta_sec = min(r_eta, t_eta);
 		else if (r_eta)
 			eta_sec = r_eta;
 		else if (t_eta)
 			eta_sec = t_eta;
 		else
 			eta_sec = 0;
 	} else {
 		/*
 		 * thread is already done or waiting for fsync
 		 */
 		eta_sec = 0;
 	}

 	return eta_sec;
 }

 static void calc_rate(unsigned long mtime, unsigned long long *io_bytes,
 		      unsigned long long *prev_io_bytes, unsigned int *rate)
 {
 	rate[0] = (io_bytes[0] - prev_io_bytes[0]) / mtime;
 	rate[1] = (io_bytes[1] - prev_io_bytes[1]) / mtime;
 	prev_io_bytes[0] = io_bytes[0];
 	prev_io_bytes[1] = io_bytes[1];
 }

 /*
  * Print status of the jobs we know about. This includes rate estimates,
  * ETA, thread state, etc.
  */
 void print_thread_status(void)
 {
 	unsigned long elapsed = mtime_since_genesis() / 1000;
 	int i, nr_running, nr_pending, t_rate, m_rate, *eta_secs, eta_sec;
 	int t_iops, m_iops, files_open;
 	struct thread_data *td;
 	char eta_str[128];
 	double perc = 0.0;
 	unsigned long long io_bytes[2];
 	unsigned long rate_time, disp_time, bw_avg_time;
 	struct timeval now;
 	int linelen;

 	static unsigned long long rate_io_bytes[2];
 	static unsigned long long disp_io_bytes[2];
 	static struct timeval rate_prev_time, disp_prev_time;
 	static unsigned int rate[2];
 	static int linelen_last;

 	if (temp_stall_ts || terse_output || eta_print == FIO_ETA_NEVER)
 		return;

 	if (!isatty(STDOUT_FILENO) && (eta_print != FIO_ETA_ALWAYS))
 		return;

 	if (!rate_io_bytes[0] && !rate_io_bytes[1])
 		fill_start_time(&rate_prev_time);
 	if (!disp_io_bytes[0] && !disp_io_bytes[1])
 		fill_start_time(&disp_prev_time);

 	eta_secs = malloc(thread_number * sizeof(int));
 	memset(eta_secs, 0, thread_number * sizeof(int));

 	io_bytes[0] = io_bytes[1] = 0;
 	nr_pending = nr_running = t_rate = m_rate = t_iops = m_iops = 0;
 	bw_avg_time = ULONG_MAX;
 	files_open = 0;
 	for_each_td(td, i) {
 		if (td->o.bw_avg_time < bw_avg_time)
 			bw_avg_time = td->o.bw_avg_time;
 		if (td->runstate == TD_RUNNING || td->runstate == TD_VERIFYING
 		    || td->runstate == TD_FSYNCING) {
 			nr_running++;
 			t_rate += td->o.rate;
 			m_rate += td->o.ratemin;
 			t_iops += td->o.rate_iops;
 			m_iops += td->o.rate_iops_min;
 			files_open += td->nr_open_files;
 		} else if (td->runstate < TD_RUNNING)
 			nr_pending++;

 		if (elapsed >= 3)
 			eta_secs[i] = thread_eta(td, elapsed);
 		else
 			eta_secs[i] = INT_MAX;

 		check_str_update(td);
 		io_bytes[0] += td->io_bytes[0];
 		io_bytes[1] += td->io_bytes[1];
 	}

 	if (exitall_on_terminate)
 		eta_sec = INT_MAX;
 	else
 		eta_sec = 0;

 	for_each_td(td, i) {
 		if (exitall_on_terminate) {
 			if (eta_secs[i] < eta_sec)
 				eta_sec = eta_secs[i];
 		} else {
 			if (eta_secs[i] > eta_sec)
 				eta_sec = eta_secs[i];
 		}
 	}

 	free(eta_secs);

 	if (eta_sec != INT_MAX && elapsed) {
 		perc = (double) elapsed / (double) (elapsed + eta_sec);
 		eta_to_str(eta_str, eta_sec);
 	}

 	fio_gettime(&now, NULL);
 	rate_time = mtime_since(&rate_prev_time, &now);

 	if (write_bw_log && rate_time> bw_avg_time) {
 		calc_rate(rate_time, io_bytes, rate_io_bytes, rate);
 		memcpy(&rate_prev_time, &now, sizeof(now));
 		add_agg_sample(rate[DDIR_READ], DDIR_READ);
 		add_agg_sample(rate[DDIR_WRITE], DDIR_WRITE);
 	}

 	disp_time = mtime_since(&disp_prev_time, &now);
 	if (disp_time < 1000)
 		return;

 	calc_rate(disp_time, io_bytes, disp_io_bytes, rate);
 	memcpy(&disp_prev_time, &now, sizeof(now));

 	if (!nr_running && !nr_pending)
 		return;

 	printf("Jobs: %d (f=%d)", nr_running, files_open);
 	if (m_rate || t_rate)
 		printf(", CR=%d/%d KiB/s", t_rate, m_rate);
 	else if (m_iops || t_iops)
 		printf(", CR=%d/%d IOPS", t_iops, m_iops);
 	if (eta_sec != INT_MAX && nr_running) {
 		perc *= 100.0;
 		linelen = printf(": [%s] [%3.1f%% done] [%6u/%6u kb/s] [eta %s]",
 				 run_str, perc, rate[0], rate[1], eta_str);
 		if (linelen >= 0 && linelen < linelen_last)
 			printf("%*s", linelen_last-linelen, "");
 		linelen_last = linelen;
 	}
 	printf("\r");
 	fflush(stdout);
 }

 void print_status_init(int thread_number)
 {
 	run_str[thread_number] = 'P';
 }
	/*
	* Status and ETA code
	*/
	#include <unistd.h>
	#include <fcntl.h>
	#include <string.h>

	#include "fio.h"

	static char run_str[MAX_JOBS + 1];

	/*
	* Sets the status of the 'td' in the printed status map.
	*/
	static void check_str_update(struct thread_data *td)
	{
	char c = run_str[td->thread_number - 1];

	switch (td->runstate) {
	case TD_REAPED:
	c = '_';
	break;
	case TD_EXITED:
	c = 'E';
	break;
	case TD_RUNNING:
	if (td_rw(td)) {
	if (td_random(td))
	c = 'm';
	else
	c = 'M';
	} else if (td_read(td)) {
	if (td_random(td))
	c = 'r';
	else
	c = 'R';
	} else {
	if (td_random(td))
	c = 'w';
	else
	c = 'W';
	}
	break;
	case TD_VERIFYING:
	c = 'V';
	break;
	case TD_FSYNCING:
	c = 'F';
	break;
	case TD_CREATED:
	c = 'C';
	break;
	case TD_INITIALIZED:
	c = 'I';
	break;
	case TD_NOT_CREATED:
	c = 'P';
	break;
	default:
	log_err("state %d\n", td->runstate);
	}

	run_str[td->thread_number - 1] = c;
	}

	/*
	* Convert seconds to a printable string.
	*/
	static void eta_to_str(char *str, int eta_sec)
	{
	unsigned int d, h, m, s;
	int disp_hour = 0;

	d = h = m = s = 0;

	s = eta_sec % 60;
	eta_sec /= 60;
	m = eta_sec % 60;
	eta_sec /= 60;
	h = eta_sec % 24;
	eta_sec /= 24;
	d = eta_sec;

	if (d) {
	disp_hour = 1;
	str += sprintf(str, "%02ud:", d);
	}

	if (h \|\| disp_hour)
	str += sprintf(str, "%02uh:", h);

	str += sprintf(str, "%02um:", m);
	str += sprintf(str, "%02us", s);
	}

	/*
	* Best effort calculation of the estimated pending runtime of a job.
	*/
	static int thread_eta(struct thread_data *td, unsigned long elapsed)
	{
	unsigned long long bytes_total, bytes_done;
	unsigned long eta_sec = 0;

	bytes_total = td->total_io_size;

	/*
	* if writing, bytes_total will be twice the size. If mixing,
	* assume a 50/50 split and thus bytes_total will be 50% larger.
	*/
	if (td->o.do_verify && td->o.verify && td_write(td)) {
	if (td_rw(td))
	bytes_total = bytes_total * 3 / 2;
	else
	bytes_total <<= 1;
	}

	if (td->o.zone_size && td->o.zone_skip)
	bytes_total /= (td->o.zone_skip / td->o.zone_size);

	if (td->runstate == TD_RUNNING \|\| td->runstate == TD_VERIFYING) {
	double perc, perc_t;

	bytes_done = td->io_bytes[DDIR_READ] + td->io_bytes[DDIR_WRITE];
	perc = (double) bytes_done / (double) bytes_total;
	if (perc > 1.0)
	perc = 1.0;

	if (td->o.time_based) {
	perc_t = (double) elapsed / (double) td->o.timeout;
	if (perc_t < perc)
	perc = perc_t;
	}

	eta_sec = (unsigned long) (elapsed * (1.0 / perc)) - elapsed;

	if (td->o.timeout && eta_sec > (td->o.timeout - elapsed))
	eta_sec = td->o.timeout - elapsed;
	} else if (td->runstate == TD_NOT_CREATED \|\| td->runstate == TD_CREATED
	\|\| td->runstate == TD_INITIALIZED) {
	int t_eta = 0, r_eta = 0;

	/*
	* We can only guess - assume it'll run the full timeout
	* if given, otherwise assume it'll run at the specified rate.
	*/
	if (td->o.timeout)
	t_eta = td->o.timeout + td->o.start_delay - elapsed;
	if (td->o.rate) {
	r_eta = (bytes_total / 1024) / td->o.rate;
	r_eta += td->o.start_delay - elapsed;
	}

	if (r_eta && t_eta)
	eta_sec = min(r_eta, t_eta);
	else if (r_eta)
	eta_sec = r_eta;
	else if (t_eta)
	eta_sec = t_eta;
	else
	eta_sec = 0;
	} else {
	/*
	* thread is already done or waiting for fsync
	*/
	eta_sec = 0;
	}

	return eta_sec;
	}

	static void calc_rate(unsigned long mtime, unsigned long long *io_bytes,
	unsigned long long prev_io_bytes, unsigned int rate)
	{
	rate[0] = (io_bytes[0] - prev_io_bytes[0]) / mtime;
	rate[1] = (io_bytes[1] - prev_io_bytes[1]) / mtime;
	prev_io_bytes[0] = io_bytes[0];
	prev_io_bytes[1] = io_bytes[1];
	}

	/*
	* Print status of the jobs we know about. This includes rate estimates,
	* ETA, thread state, etc.
	*/
	void print_thread_status(void)
	{
	unsigned long elapsed = mtime_since_genesis() / 1000;
	int i, nr_running, nr_pending, t_rate, m_rate, *eta_secs, eta_sec;
	int t_iops, m_iops, files_open;
	struct thread_data *td;
	char eta_str[128];
	double perc = 0.0;
	unsigned long long io_bytes[2];
	unsigned long rate_time, disp_time, bw_avg_time;
	struct timeval now;
	int linelen;

	static unsigned long long rate_io_bytes[2];
	static unsigned long long disp_io_bytes[2];
	static struct timeval rate_prev_time, disp_prev_time;
	static unsigned int rate[2];
	static int linelen_last;

	if (temp_stall_ts \|\| terse_output \|\| eta_print == FIO_ETA_NEVER)
	return;

	if (!isatty(STDOUT_FILENO) && (eta_print != FIO_ETA_ALWAYS))
	return;

	if (!rate_io_bytes[0] && !rate_io_bytes[1])
	fill_start_time(&rate_prev_time);
	if (!disp_io_bytes[0] && !disp_io_bytes[1])
	fill_start_time(&disp_prev_time);

	eta_secs = malloc(thread_number * sizeof(int));
	memset(eta_secs, 0, thread_number * sizeof(int));

	io_bytes[0] = io_bytes[1] = 0;
	nr_pending = nr_running = t_rate = m_rate = t_iops = m_iops = 0;
	bw_avg_time = ULONG_MAX;
	files_open = 0;
	for_each_td(td, i) {
	if (td->o.bw_avg_time < bw_avg_time)
	bw_avg_time = td->o.bw_avg_time;
	if (td->runstate == TD_RUNNING \|\| td->runstate == TD_VERIFYING
	\|\| td->runstate == TD_FSYNCING) {
	nr_running++;
	t_rate += td->o.rate;
	m_rate += td->o.ratemin;
	t_iops += td->o.rate_iops;
	m_iops += td->o.rate_iops_min;
	files_open += td->nr_open_files;
	} else if (td->runstate < TD_RUNNING)
	nr_pending++;

	if (elapsed >= 3)
	eta_secs[i] = thread_eta(td, elapsed);
	else
	eta_secs[i] = INT_MAX;

	check_str_update(td);
	io_bytes[0] += td->io_bytes[0];
	io_bytes[1] += td->io_bytes[1];
	}

	if (exitall_on_terminate)
	eta_sec = INT_MAX;
	else
	eta_sec = 0;

	for_each_td(td, i) {
	if (exitall_on_terminate) {
	if (eta_secs[i] < eta_sec)
	eta_sec = eta_secs[i];
	} else {
	if (eta_secs[i] > eta_sec)
	eta_sec = eta_secs[i];
	}
	}

	free(eta_secs);

	if (eta_sec != INT_MAX && elapsed) {
	perc = (double) elapsed / (double) (elapsed + eta_sec);
	eta_to_str(eta_str, eta_sec);
	}

	fio_gettime(&now, NULL);
	rate_time = mtime_since(&rate_prev_time, &now);

	if (write_bw_log && rate_time> bw_avg_time) {
	calc_rate(rate_time, io_bytes, rate_io_bytes, rate);
	memcpy(&rate_prev_time, &now, sizeof(now));
	add_agg_sample(rate[DDIR_READ], DDIR_READ);
	add_agg_sample(rate[DDIR_WRITE], DDIR_WRITE);
	}

	disp_time = mtime_since(&disp_prev_time, &now);
	if (disp_time < 1000)
	return;

	calc_rate(disp_time, io_bytes, disp_io_bytes, rate);
	memcpy(&disp_prev_time, &now, sizeof(now));

	if (!nr_running && !nr_pending)
	return;

	printf("Jobs: %d (f=%d)", nr_running, files_open);
	if (m_rate \|\| t_rate)
	printf(", CR=%d/%d KiB/s", t_rate, m_rate);
	else if (m_iops \|\| t_iops)
	printf(", CR=%d/%d IOPS", t_iops, m_iops);
	if (eta_sec != INT_MAX && nr_running) {
	perc *= 100.0;
	linelen = printf(": [%s] [%3.1f%% done] [%6u/%6u kb/s] [eta %s]",
	run_str, perc, rate[0], rate[1], eta_str);
	if (linelen >= 0 && linelen < linelen_last)
	printf("%*s", linelen_last-linelen, "");
	linelen_last = linelen;
	}
	printf("\r");
	fflush(stdout);
	}

	void print_status_init(int thread_number)
	{
	run_str[thread_number] = 'P';
	}