bcachefs: bch_acct_rebalance_work

Signed-off-by: Kent Overstreet <[email protected]>
koverstreet · May 29, 2024 · fd1b0a9 · fd1b0a9
1 parent 1cc24d3
commit fd1b0a9
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/fs/bcachefs/buckets.c b/fs/bcachefs/buckets.c
@@ -770,6 +770,15 @@ static int __trigger_extent(struct btree_trans *trans,
 			return ret;
 	}
 
+	if (bch2_bkey_rebalance_opts(k)) {
+		struct disk_accounting_pos acc = {
+			.type		= BCH_DISK_ACCOUNTING_rebalance_work,
+		};
+		ret = bch2_disk_accounting_mod(trans, &acc, &replicas_sectors, 1, gc);
+		if (ret)
+			return ret;
+	}
+
 	return 0;
 }
 

diff --git a/fs/bcachefs/disk_accounting_format.h b/fs/bcachefs/disk_accounting_format.h
@@ -102,7 +102,8 @@ static inline bool data_type_is_hidden(enum bch_data_type type)
 	x(dev_data_type,	3)		\
 	x(compression,		4)		\
 	x(snapshot,		5)		\
-	x(btree,		6)
+	x(btree,		6)		\
+	x(rebalance_work,	7)
 
 enum disk_accounting_type {
 #define x(f, nr)	BCH_DISK_ACCOUNTING_##f	= nr,