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