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 = 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 %016llx:%04x\n",
  73                 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 %016llx:%04x\n",
  85                         filesystem,
  86                         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                "    %lld btree nodes scanned\n"
 100                "    %lld btree nodes deleted\n"
 101                "    %lld collision retries\n"
 102                "    %lld btree nodes rebalanced\n",
 103                rebal.stat_ncount,
 104                rebal.stat_deletions,
 105                rebal.stat_collisions,
 106                rebal.stat_nrebal
 107         );
 108 }
 109
 110 static
 111 void
 112 rebalance_usage(int exit_code)
 113 {
 114         fprintf(stderr,
 115                 "hammer rebalance <filesystem> "
 116                 "[saturation-percentage 50-100]\n"
 117                 "By default 75%% is used.\n");
 118         exit(exit_code);
 119 }