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 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
  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 85%%.\n");
 119         exit(exit_code);
 120 }