[fio.git] / eta.c

/*
 * 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';
}
Commit	Line	Data
263e529f JA	1	/*
	2	* Status and ETA code
	3	*/
	4	#include <unistd.h>
	5	#include <fcntl.h>
	6	#include <string.h>
	7
	8	#include "fio.h"
263e529f JA	9
	10	static char run_str[MAX_JOBS + 1];
	11
	12	/*
	13	* Sets the status of the 'td' in the printed status map.
	14	*/
	15	static void check_str_update(struct thread_data *td)
	16	{
	17	char c = run_str[td->thread_number - 1];
	18
	19	switch (td->runstate) {
	20	case TD_REAPED:
	21	c = '_';
	22	break;
	23	case TD_EXITED:
	24	c = 'E';
	25	break;
	26	case TD_RUNNING:
	27	if (td_rw(td)) {
413dd459	28	if (td_random(td))
263e529f	29	c = 'm';
263e529f	30	else
413dd459 JA	31	c = 'M';
	32	} else if (td_read(td)) {
	33	if (td_random(td))
263e529f	34	c = 'r';
263e529f	35	else
413dd459 JA	36	c = 'R';
	37	} else {
	38	if (td_random(td))
263e529f	39	c = 'w';
413dd459 JA	40	else
413dd459 JA	41	c = 'W';
263e529f JA	42	}
	43	break;
	44	case TD_VERIFYING:
	45	c = 'V';
	46	break;
	47	case TD_FSYNCING:
	48	c = 'F';
	49	break;
	50	case TD_CREATED:
	51	c = 'C';
	52	break;
	53	case TD_INITIALIZED:
	54	c = 'I';
	55	break;
	56	case TD_NOT_CREATED:
	57	c = 'P';
	58	break;
	59	default:
	60	log_err("state %d\n", td->runstate);
	61	}
	62
	63	run_str[td->thread_number - 1] = c;
	64	}
	65
	66	/*
	67	* Convert seconds to a printable string.
	68	*/
	69	static void eta_to_str(char *str, int eta_sec)
	70	{
	71	unsigned int d, h, m, s;
165faf16	72	int disp_hour = 0;
263e529f JA	73
	74	d = h = m = s = 0;
	75
	76	s = eta_sec % 60;
	77	eta_sec /= 60;
	78	m = eta_sec % 60;
	79	eta_sec /= 60;
	80	h = eta_sec % 24;
	81	eta_sec /= 24;
	82	d = eta_sec;
	83
165faf16 JA	84	if (d) {
165faf16 JA	85	disp_hour = 1;
1e97cce9	86	str += sprintf(str, "%02ud:", d);
263e529f	87	}
165faf16 JA	88
165faf16 JA	89	if (h \|\| disp_hour)
1e97cce9	90	str += sprintf(str, "%02uh:", h);
263e529f	91
1e97cce9 JA	92	str += sprintf(str, "%02um:", m);
1e97cce9 JA	93	str += sprintf(str, "%02us", s);
263e529f JA	94	}
	95
	96	/*
	97	* Best effort calculation of the estimated pending runtime of a job.
	98	*/
	99	static int thread_eta(struct thread_data *td, unsigned long elapsed)
	100	{
	101	unsigned long long bytes_total, bytes_done;
1e97cce9	102	unsigned long eta_sec = 0;
263e529f JA	103
	104	bytes_total = td->total_io_size;
	105
74939e38 JA	106	/*
	107	* if writing, bytes_total will be twice the size. If mixing,
	108	* assume a 50/50 split and thus bytes_total will be 50% larger.
	109	*/
0dd8377f	110	if (td->o.do_verify && td->o.verify && td_write(td)) {
74939e38 JA	111	if (td_rw(td))
	112	bytes_total = bytes_total * 3 / 2;
	113	else
	114	bytes_total <<= 1;
	115	}
	116
2dc1bbeb JA	117	if (td->o.zone_size && td->o.zone_skip)
2dc1bbeb JA	118	bytes_total /= (td->o.zone_skip / td->o.zone_size);
263e529f JA	119
263e529f JA	120	if (td->runstate == TD_RUNNING \|\| td->runstate == TD_VERIFYING) {
cf4464ca	121	double perc, perc_t;
263e529f JA	122
	123	bytes_done = td->io_bytes[DDIR_READ] + td->io_bytes[DDIR_WRITE];
	124	perc = (double) bytes_done / (double) bytes_total;
	125	if (perc > 1.0)
	126	perc = 1.0;
	127
cf4464ca JA	128	if (td->o.time_based) {
	129	perc_t = (double) elapsed / (double) td->o.timeout;
	130	if (perc_t < perc)
	131	perc = perc_t;
	132	}
	133
1e97cce9	134	eta_sec = (unsigned long) (elapsed * (1.0 / perc)) - elapsed;
263e529f	135
2dc1bbeb JA	136	if (td->o.timeout && eta_sec > (td->o.timeout - elapsed))
2dc1bbeb JA	137	eta_sec = td->o.timeout - elapsed;
263e529f JA	138	} else if (td->runstate == TD_NOT_CREATED \|\| td->runstate == TD_CREATED
	139	\|\| td->runstate == TD_INITIALIZED) {
	140	int t_eta = 0, r_eta = 0;
	141
	142	/*
	143	* We can only guess - assume it'll run the full timeout
	144	* if given, otherwise assume it'll run at the specified rate.
	145	*/
2dc1bbeb JA	146	if (td->o.timeout)
	147	t_eta = td->o.timeout + td->o.start_delay - elapsed;
	148	if (td->o.rate) {
	149	r_eta = (bytes_total / 1024) / td->o.rate;
	150	r_eta += td->o.start_delay - elapsed;
263e529f JA	151	}
	152
	153	if (r_eta && t_eta)
	154	eta_sec = min(r_eta, t_eta);
	155	else if (r_eta)
	156	eta_sec = r_eta;
	157	else if (t_eta)
	158	eta_sec = t_eta;
	159	else
	160	eta_sec = 0;
	161	} else {
	162	/*
	163	* thread is already done or waiting for fsync
	164	*/
	165	eta_sec = 0;
	166	}
	167
	168	return eta_sec;
	169	}
	170
46fda8d0 JA	171	static void calc_rate(unsigned long mtime, unsigned long long *io_bytes,
	172	unsigned long long prev_io_bytes, unsigned int rate)
	173	{
	174	rate[0] = (io_bytes[0] - prev_io_bytes[0]) / mtime;
	175	rate[1] = (io_bytes[1] - prev_io_bytes[1]) / mtime;
	176	prev_io_bytes[0] = io_bytes[0];
	177	prev_io_bytes[1] = io_bytes[1];
	178	}
	179
263e529f JA	180	/*
	181	* Print status of the jobs we know about. This includes rate estimates,
	182	* ETA, thread state, etc.
	183	*/
	184	void print_thread_status(void)
	185	{
	186	unsigned long elapsed = mtime_since_genesis() / 1000;
	187	int i, nr_running, nr_pending, t_rate, m_rate, *eta_secs, eta_sec;
e6e41602	188	int t_iops, m_iops, files_open;
34572e28	189	struct thread_data *td;
2eaa41ce	190	char eta_str[128];
263e529f	191	double perc = 0.0;
6043c579	192	unsigned long long io_bytes[2];
46fda8d0 JA	193	unsigned long rate_time, disp_time, bw_avg_time;
46fda8d0 JA	194	struct timeval now;
43819609	195	int linelen;
46fda8d0 JA	196
	197	static unsigned long long rate_io_bytes[2];
	198	static unsigned long long disp_io_bytes[2];
	199	static struct timeval rate_prev_time, disp_prev_time;
	200	static unsigned int rate[2];
43819609	201	static int linelen_last;
6043c579	202
e592a06b AC	203	if (temp_stall_ts \|\| terse_output \|\| eta_print == FIO_ETA_NEVER)
	204	return;
	205
	206	if (!isatty(STDOUT_FILENO) && (eta_print != FIO_ETA_ALWAYS))
263e529f JA	207	return;
263e529f JA	208
46fda8d0 JA	209	if (!rate_io_bytes[0] && !rate_io_bytes[1])
	210	fill_start_time(&rate_prev_time);
	211	if (!disp_io_bytes[0] && !disp_io_bytes[1])
	212	fill_start_time(&disp_prev_time);
6043c579	213
263e529f JA	214	eta_secs = malloc(thread_number * sizeof(int));
	215	memset(eta_secs, 0, thread_number * sizeof(int));
	216
6043c579	217	io_bytes[0] = io_bytes[1] = 0;
ac71feb1	218	nr_pending = nr_running = t_rate = m_rate = t_iops = m_iops = 0;
bb3884d8	219	bw_avg_time = ULONG_MAX;
e6e41602	220	files_open = 0;
34572e28	221	for_each_td(td, i) {
2dc1bbeb JA	222	if (td->o.bw_avg_time < bw_avg_time)
2dc1bbeb JA	223	bw_avg_time = td->o.bw_avg_time;
46fda8d0 JA	224	if (td->runstate == TD_RUNNING \|\| td->runstate == TD_VERIFYING
46fda8d0 JA	225	\|\| td->runstate == TD_FSYNCING) {
263e529f	226	nr_running++;
2dc1bbeb JA	227	t_rate += td->o.rate;
2dc1bbeb JA	228	m_rate += td->o.ratemin;
ac71feb1 JA	229	t_iops += td->o.rate_iops;
ac71feb1 JA	230	m_iops += td->o.rate_iops_min;
e6e41602	231	files_open += td->nr_open_files;
263e529f JA	232	} else if (td->runstate < TD_RUNNING)
	233	nr_pending++;
	234
	235	if (elapsed >= 3)
	236	eta_secs[i] = thread_eta(td, elapsed);
	237	else
	238	eta_secs[i] = INT_MAX;
	239
	240	check_str_update(td);
6043c579 JA	241	io_bytes[0] += td->io_bytes[0];
6043c579 JA	242	io_bytes[1] += td->io_bytes[1];
263e529f JA	243	}
	244
	245	if (exitall_on_terminate)
	246	eta_sec = INT_MAX;
	247	else
	248	eta_sec = 0;
	249
34572e28	250	for_each_td(td, i) {
263e529f JA	251	if (exitall_on_terminate) {
	252	if (eta_secs[i] < eta_sec)
	253	eta_sec = eta_secs[i];
	254	} else {
	255	if (eta_secs[i] > eta_sec)
	256	eta_sec = eta_secs[i];
	257	}
	258	}
	259
eecf272f JA	260	free(eta_secs);
eecf272f JA	261
263e529f JA	262	if (eta_sec != INT_MAX && elapsed) {
	263	perc = (double) elapsed / (double) (elapsed + eta_sec);
	264	eta_to_str(eta_str, eta_sec);
	265	}
	266
46fda8d0 JA	267	fio_gettime(&now, NULL);
	268	rate_time = mtime_since(&rate_prev_time, &now);
	269
	270	if (write_bw_log && rate_time> bw_avg_time) {
	271	calc_rate(rate_time, io_bytes, rate_io_bytes, rate);
	272	memcpy(&rate_prev_time, &now, sizeof(now));
	273	add_agg_sample(rate[DDIR_READ], DDIR_READ);
	274	add_agg_sample(rate[DDIR_WRITE], DDIR_WRITE);
6043c579 JA	275	}
6043c579 JA	276
46fda8d0 JA	277	disp_time = mtime_since(&disp_prev_time, &now);
	278	if (disp_time < 1000)
	279	return;
	280
	281	calc_rate(disp_time, io_bytes, disp_io_bytes, rate);
	282	memcpy(&disp_prev_time, &now, sizeof(now));
	283
263e529f JA	284	if (!nr_running && !nr_pending)
	285	return;
	286
e6e41602	287	printf("Jobs: %d (f=%d)", nr_running, files_open);
263e529f	288	if (m_rate \|\| t_rate)
2cb8d960	289	printf(", CR=%d/%d KiB/s", t_rate, m_rate);
ac71feb1 JA	290	else if (m_iops \|\| t_iops)
ac71feb1 JA	291	printf(", CR=%d/%d IOPS", t_iops, m_iops);
263e529f JA	292	if (eta_sec != INT_MAX && nr_running) {
263e529f JA	293	perc *= 100.0;
43819609 AC	294	linelen = printf(": [%s] [%3.1f%% done] [%6u/%6u kb/s] [eta %s]",
	295	run_str, perc, rate[0], rate[1], eta_str);
	296	if (linelen >= 0 && linelen < linelen_last)
	297	printf("%*s", linelen_last-linelen, "");
	298	linelen_last = linelen;
263e529f JA	299	}
	300	printf("\r");
	301	fflush(stdout);
263e529f JA	302	}
	303
	304	void print_status_init(int thread_number)
	305	{
	306	run_str[thread_number] = 'P';
	307	}