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         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 }