xref: /dragonfly/sbin/hammer/cmd_rebalance.c (revision 85088528)
1 /*
2  * Copyright (c) 2009 The DragonFly Project.  All rights reserved.
3  *
4  * This code is derived from software contributed to The DragonFly Project
5  * by Matthew Dillon <dillon@backplane.com>
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  *
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in
15  *    the documentation and/or other materials provided with the
16  *    distribution.
17  * 3. Neither the name of The DragonFly Project nor the names of its
18  *    contributors may be used to endorse or promote products derived
19  *    from this software without specific, prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
25  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
26  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
27  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
28  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
29  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
30  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
31  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32  * SUCH DAMAGE.
33  */
34 
35 #include "hammer.h"
36 
37 static void rebalance_usage(int exit_code);
38 
39 /*
40  * rebalance <filesystem> [saturation_percentage] (default 75%)
41  */
42 void
43 hammer_cmd_rebalance(char **av, int ac)
44 {
45 	struct hammer_ioc_rebalance rebal;
46 	const char *filesystem;
47 	int fd;
48 	int perc;
49 
50 	if (TimeoutOpt > 0)
51 		alarm(TimeoutOpt);
52 
53 	bzero(&rebal, sizeof(rebal));
54 
55 	rebal.key_beg.localization = HAMMER_MIN_LOCALIZATION;
56 	rebal.key_beg.obj_id = HAMMER_MIN_OBJID;
57 	hammer_get_cycle(&rebal.key_beg, NULL);
58 
59 	rebal.key_end.localization = HAMMER_MAX_LOCALIZATION;
60 	rebal.key_end.obj_id = HAMMER_MAX_OBJID;
61 
62 	if (ac == 0)
63 		rebalance_usage(1);
64 	filesystem = av[0];
65 	if (ac == 1) {
66 		perc = 75;
67 	} else {
68 		perc = strtol(av[1], NULL, 0);
69 		if (perc < 50 || perc > 100)
70 			rebalance_usage(1);
71 	}
72 	printf("rebalance start %016jx:%04x\n",
73 		(uintmax_t)rebal.key_beg.obj_id,
74 		rebal.key_beg.localization);
75 
76 	fd = open(filesystem, O_RDONLY);
77 	if (fd < 0)
78 		err(1, "Unable to open %s", filesystem);
79 	RunningIoctl = 1;
80 	if (ioctl(fd, HAMMERIOC_REBALANCE, &rebal) < 0) {
81 		printf("Rebalance %s failed: %s\n",
82 		       filesystem, strerror(errno));
83 	} else if (rebal.head.flags & HAMMER_IOC_HEAD_INTR) {
84 		printf("Rebalance %s interrupted by timer at %016jx:%04x\n",
85 			filesystem,
86 			(uintmax_t)rebal.key_cur.obj_id,
87 			rebal.key_cur.localization);
88 		if (CyclePath) {
89 			hammer_set_cycle(&rebal.key_cur, 0);
90 		}
91 	} else {
92 		if (CyclePath)
93 			hammer_reset_cycle();
94 		printf("Rebalance %s succeeded\n", filesystem);
95 	}
96 	RunningIoctl = 0;
97 	close(fd);
98 	printf("Rebalance:\n"
99 	       "    %jd btree nodes scanned\n"
100 	       "    %jd btree nodes deleted\n"
101 	       "    %jd collision retries\n"
102 	       "    %jd btree nodes rebalanced\n",
103 	       (intmax_t)rebal.stat_ncount,
104 	       (intmax_t)rebal.stat_deletions,
105 	       (intmax_t)rebal.stat_collisions,
106 	       (intmax_t)rebal.stat_nrebal
107 	);
108 }
109 
110 static
111 void
112 rebalance_usage(int exit_code)
113 {
114 	fprintf(stderr,
115 		"hammer rebalance <filesystem> [saturation_percentage]\n"
116 		"saturation_percentage is 50%%-100%%, default is 75%%.\n");
117 	exit(exit_code);
118 }
119