From 0b8bd7da5d1a71a79ce2589fdc37ef1bef513f5f Mon Sep 17 00:00:00 2001
From: Matthew Dillon <dillon@apollo.backplane.com>
Date: Sun, 20 Sep 2009 10:29:32 -0700
Subject: [PATCH] HAMMER - Add rebalancing to automatic cleanup sequence

* The hammer cleanup directive now also rebalances the B-Tree.
---
 sbin/hammer/cmd_cleanup.c | 48 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/sbin/hammer/cmd_cleanup.c b/sbin/hammer/cmd_cleanup.c
index db46e2d2fd..0c5a630cec 100644
--- a/sbin/hammer/cmd_cleanup.c
+++ b/sbin/hammer/cmd_cleanup.c
@@ -75,6 +75,8 @@ static int check_expired(const char *fpath, int arg2);
 
 static int create_snapshot(const char *path, const char *snapshots_path,
 			      int arg1, int arg2);
+static int cleanup_rebalance(const char *path, const char *snapshots_path,
+			      int arg1, int arg2);
 static int cleanup_prune(const char *path, const char *snapshots_path,
 			      int arg1, int arg2, int snapshots_disabled);
 static int cleanup_reblock(const char *path, const char *snapshots_path,
@@ -149,6 +151,7 @@ do_cleanup(const char *path)
 	int prune_warning = 0;
 	int fd;
 	int r;
+	int found_rebal = 0;
 
 	bzero(&pfs, sizeof(pfs));
 	bzero(&mrec_tmp, sizeof(mrec_tmp));
@@ -235,6 +238,7 @@ do_cleanup(const char *path)
 		}
 		fprintf(fp, 
 			"prune     1d 5m\n"
+			"rebalance 1d 5m\n"
 			"reblock   1d 5m\n"
 			"recopy    30d 10m\n");
 		fclose(fp);
@@ -319,6 +323,18 @@ do_cleanup(const char *path)
 			} else {
 				printf("skip\n");
 			}
+		} else if (strcmp(cmd, "rebalance") == 0) {
+			found_rebal = 1;
+			if (check_period(snapshots_path, cmd, arg1, &savet)) {
+				printf("run");
+				fflush(stdout);
+				if (VerboseOpt)
+					printf("\n");
+				r = cleanup_rebalance(path, snapshots_path,
+						arg1, arg2);
+			} else {
+				printf("skip\n");
+			}
 		} else if (strcmp(cmd, "reblock") == 0) {
 			if (check_period(snapshots_path, cmd, arg1, &savet)) {
 				printf("run");
@@ -349,6 +365,17 @@ do_cleanup(const char *path)
 			save_period(snapshots_path, cmd, savet);
 	}
 	fclose(fp);
+
+	/*
+	 * Add new rebalance feature if the config doesn't have it
+	 */
+	if (found_rebal == 0) {
+		if ((fp = fopen(config_path, "r+")) != NULL) {
+			fseek(fp, 0L, 2);
+			fprintf(fp, "rebalance 1d 5m\n");
+			fclose(fp);
+		}
+	}
 	usleep(1000);
 }
 
@@ -626,6 +653,27 @@ cleanup_prune(const char *path __unused, const char *snapshots_path,
 	return(0);
 }
 
+static int
+cleanup_rebalance(const char *path, const char *snapshots_path,
+		  int arg1 __unused, int arg2)
+{
+	if (VerboseOpt == 0) {
+		printf(".");
+		fflush(stdout);
+	}
+
+	runcmd(NULL,
+	       "hammer -c %s/.rebalance.cycle -t %d rebalance %s",
+	       snapshots_path, arg2, path);
+	if (VerboseOpt == 0) {
+		printf(".");
+		fflush(stdout);
+	}
+	if (VerboseOpt == 0)
+		printf("\n");
+	return(0);
+}
+
 static int
 cleanup_reblock(const char *path, const char *snapshots_path,
 		  int arg1 __unused, int arg2)
-- 
2.41.0