bcachefs: bch_acct_rebalance_work

Kent Overstreet · Kent Overstreet · commit 20ac515a9cc7 · 2024-07-14T19:00:14.000-04:00
Signed-off-by: Kent Overstreet &lt;kent.overstreet@linux.dev&gt;
diff --git a/fs/bcachefs/buckets.c b/fs/bcachefs/buckets.c
@@ -808,6 +808,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,

Original file line number	Diff line number	Diff line change
`@@ -808,6 +808,15 @@ static int __trigger_extent(struct btree_trans *trans,`
`808`	`808`	`return ret;`
`809`	`809`	`}`
`810`	`810`
	`811`	`+ if (bch2_bkey_rebalance_opts(k)) {`
	`812`	`+ struct disk_accounting_pos acc = {`
	`813`	`+ .type = BCH_DISK_ACCOUNTING_rebalance_work,`
	`814`	`+ };`
	`815`	`+ ret = bch2_disk_accounting_mod(trans, &acc, &replicas_sectors, 1, gc);`
	`816`	`+ if (ret)`
	`817`	`+ return ret;`
	`818`	`+ }`
	`819`	`+`
`811`	`820`	`return 0;`
`812`	`821`	`}`
`813`	`822`