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