On 01.05.2011 17:47, Hugo Mills wrote: > For the impatient, this patch introduces the pot-watching --monitor > option, which checks the balance progress at regular intervals, and > updates a single status line with the current progress and an > estimated completion time. > > Signed-off-by: Hugo Mills <h...@carfax.org.uk> > --- > btrfs.c | 4 +- > btrfs_cmds.c | 109 ++++++++++++++++++++++++++++++++++++++++++++++++++----- > ioctl.h | 2 +- > man/btrfs.8.in | 7 ++-- > 4 files changed, 106 insertions(+), 16 deletions(-) > > diff --git a/btrfs.c b/btrfs.c > index 3a24019..0b6186c 100644 > --- a/btrfs.c > +++ b/btrfs.c > @@ -99,8 +99,8 @@ static struct Command commands[] = { > "balance start", "<path>\n" > "Synonym for \"btrfs filesystem balance\"." > }, > - { do_balance_progress, 1, > - "balance progress", "<path>\n" > + { do_balance_progress, -1, > + "balance progress", "[-m|--monitor] <path>\n" > "Show progress of the balance operation running on <path>." > }, > { do_scan, > diff --git a/btrfs_cmds.c b/btrfs_cmds.c > index 00dda0a..ad2300d 100644 > --- a/btrfs_cmds.c > +++ b/btrfs_cmds.c > @@ -797,22 +797,108 @@ int get_balance_progress(char *path, struct > btrfs_ioctl_balance_progress *bal) > return err; > } > > +const struct option progress_options[] = { > + { "monitor", 0, NULL, 'm' }, > + { NULL, 0, NULL, 0 } > +}; > + > int do_balance_progress(int argc, char **argv) > { > char *path; > int ret = 0; > int err = 0; > struct btrfs_ioctl_balance_progress bal; > + __u64 last_completed = -1; > + __u64 initial_completed = -1; > + struct timeval now; > + struct timeval started; > + int monitor = 0; > + > + optind = 1; > + while(1) { > + int c = getopt_long(argc, argv, "m", progress_options, NULL); > + if (c < 0) > + break; > + switch(c) { > + case 'm': > + monitor = 1; > + break; > + default: > + fprintf(stderr, "Invalid arguments for balance > progress\n"); > + free(argv); > + return 1; > + } > + } > + > + if(optind >= argc) { > + fprintf(stderr, "No filesystem path given for progress\n"); > + return 1; > + } > + > + path = argv[optind]; > + do { > + int prs = 0; > > - path = argv[1]; > + ret = get_balance_progress(path, &bal); > + if (ret) > + break; > > - ret = get_balance_progress(path, &bal); > - if (!ret) > - printf("\r%u/%u block groups moved, " > - "%0.2f%% complete.\n", > - bal.completed, > - bal.expected, > - (float)bal.completed/bal.expected*100.0); > + if (last_completed != bal.completed) { > + printf("\r%u/%u block groups moved, " > + "%0.2f%% complete.", > + bal.completed, > + bal.expected, > + (float)bal.completed/bal.expected*100.0); > + } > + > + if (initial_completed != -1 > + && initial_completed != bal.completed) { > + ret = gettimeofday(&now, NULL); > + if (ret) { > + fprintf(stderr, "Can't read current time\n"); > + return 22; > + } > + /* Seconds per block */ > + float rate = (float)(now.tv_sec - started.tv_sec) > + / (bal.completed - initial_completed); > + int secs_remaining = rate > + * (bal.expected - bal.completed); > + printf(" Time remaining"); > + if (secs_remaining >= 60*60*24) { > + printf(" %dd", secs_remaining / (60*60*24)); > + secs_remaining %= 60*60*24; > + prs = 1; > + } > + if (prs || secs_remaining >= 60*60) { > + printf(" %dh", secs_remaining / (60*60)); > + secs_remaining %= 60*60; > + prs = 1; > + } > + if (prs || secs_remaining > 60) { > + printf(" %dm", secs_remaining / 60); > + secs_remaining %= 60; > + } > + printf(" %ds\x1b[K", secs_remaining); > + } > + > + if (last_completed != -1 && last_completed != bal.completed) { > + initial_completed = bal.completed; > + ret = gettimeofday(&started, NULL); > + if (ret) { > + fprintf(stderr, "Can't read current time\n"); > + return 22; > + } > + } > + > + last_completed = bal.completed; > + > + if (monitor) { > + fflush(stdout); > + sleep(1); > + } else { > + printf("\n"); > + } > + } while(monitor); > > switch(ret) { > case 0: > @@ -821,10 +907,13 @@ int do_balance_progress(int argc, char **argv) > fprintf(stderr, "ERROR: can't access '%s'\n", path); > return 13; > case EINVAL: > - fprintf(stderr, > + if (!monitor) { > + fprintf(stderr, > "No balance operation running on '%s'.\n", > path); > - return 20; > + return 20; > + } > + break; > default: > fprintf(stderr, "ERROR: ioctl returned error %d.", err);
The output of fprintf should be terminated by a newline. Thanks, Andreas > return 21; > diff --git a/ioctl.h b/ioctl.h > index f07d3a2..3eeaa33 100644 > --- a/ioctl.h > +++ b/ioctl.h > @@ -174,6 +174,6 @@ struct btrfs_ioctl_balance_progress { > #define BTRFS_IOC_DEFAULT_SUBVOL _IOW(BTRFS_IOCTL_MAGIC, 19, u64) > #define BTRFS_IOC_SPACE_INFO _IOWR(BTRFS_IOCTL_MAGIC, 20, \ > struct btrfs_ioctl_space_args) > -#define BTRFS_IOC_BALANCE_PROGRESS _IOR(BTRFS_IOCTL_MAGIC, 25, \ > +#define BTRFS_IOC_BALANCE_PROGRESS _IOR(BTRFS_IOCTL_MAGIC, 27, \ > struct btrfs_ioctl_balance_progress) > #endif > diff --git a/man/btrfs.8.in b/man/btrfs.8.in > index 7b66fb7..5fcad89 100644 > --- a/man/btrfs.8.in > +++ b/man/btrfs.8.in > @@ -27,7 +27,7 @@ btrfs \- control a btrfs filesystem > .PP > \fBbtrfs\fP \fBbalance start\fP\fI <path> \fP > .PP > -\fBbtrfs\fP \fBbalance progress\fP\fI <path>\fP > +\fBbtrfs\fP \fBbalance progress\fP [\fB-m\fP|\fB--monitor\fP] \fI<path>\fP > .PP > \fBbtrfs\fP \fBdevice show\fP\fI <dev>|<label> [<dev>|<label>...]\fP > .PP > @@ -162,9 +162,10 @@ Add device(s) to the filesystem identified by > \fI<path>\fR. > Remove device(s) from a filesystem identified by \fI<path>\fR. > .PP > > -\fBbalance progress\fP \fI<path>\fP > +\fBbalance progress\fP [\fB-m\fP|\fB--monitor\fP] \fI<path>\fP > Report progress on the currently-running balance operation on the > -filesystem mounted at \fI<path>\fP. > +filesystem mounted at \fI<path>\fP. Use --monitor to report progress > +continually, including an estimate of completion time. > > .SH EXIT STATUS > \fBbtrfs\fR returns a zero exist status if it succeeds. Non zero is returned > in -- To unsubscribe from this list: send the line "unsubscribe linux-btrfs" in the body of a message to majord...@vger.kernel.org More majordomo info at http://vger.kernel.org/majordomo-info.html