xref: /dragonfly/sbin/hammer/cmd_rebalance.c (revision 73610d44)
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 85%)
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 	rebal.allpfs = AllPFS;
62 
63 	if (ac == 0)
64 		rebalance_usage(1);
65 	filesystem = av[0];
66 	if (ac == 1) {
67 		perc = 85;
68 	} else {
69 		perc = strtol(av[1], NULL, 0);
70 		if (perc < 50 || perc > 100)
71 			rebalance_usage(1);
72 	}
73 	rebal.saturation = HAMMER_BTREE_INT_ELMS * perc / 100;
74 
75 	printf("rebalance start %016jx:%04x\n",
76 		(uintmax_t)rebal.key_beg.obj_id,
77 		rebal.key_beg.localization);
78 
79 	fd = open(filesystem, O_RDONLY);
80 	if (fd < 0)
81 		err(1, "Unable to open %s", filesystem);
82 	RunningIoctl = 1;
83 	if (ioctl(fd, HAMMERIOC_REBALANCE, &rebal) < 0) {
84 		printf("Rebalance %s failed: %s\n",
85 		       filesystem, strerror(errno));
86 	} else if (rebal.head.flags & HAMMER_IOC_HEAD_INTR) {
87 		printf("Rebalance %s interrupted by timer at %016jx:%04x\n",
88 			filesystem,
89 			(uintmax_t)rebal.key_cur.obj_id,
90 			rebal.key_cur.localization);
91 		if (CyclePath) {
92 			hammer_set_cycle(&rebal.key_cur, 0);
93 		}
94 	} else {
95 		if (CyclePath)
96 			hammer_reset_cycle();
97 		printf("Rebalance %s succeeded\n", filesystem);
98 	}
99 	RunningIoctl = 0;
100 	close(fd);
101 	printf("Rebalance:\n"
102 	       "    %jd B-Tree nodes scanned\n"
103 	       "    %jd B-Tree nodes deleted\n"
104 	       "    %jd collision retries\n"
105 	       "    %jd B-Tree nodes rebalanced\n",
106 	       (intmax_t)rebal.stat_ncount,
107 	       (intmax_t)rebal.stat_deletions,
108 	       (intmax_t)rebal.stat_collisions,
109 	       (intmax_t)rebal.stat_nrebal
110 	);
111 }
112 
113 static
114 void
115 rebalance_usage(int exit_code)
116 {
117 	fprintf(stderr,
118 		"hammer rebalance <filesystem> [saturation_percentage]\n"
119 		"saturation_percentage is 50%%-100%%, default is 85%%.\n");
120 	exit(exit_code);
121 }
122