[linux-block.git] / fs / bcachefs / rebalance.h

/* SPDX-License-Identifier: GPL-2.0 */
#ifndef _BCACHEFS_REBALANCE_H
#define _BCACHEFS_REBALANCE_H

#include "compress.h"
#include "disk_groups.h"
#include "opts.h"
#include "rebalance_types.h"

static inline struct bch_extent_rebalance io_opts_to_rebalance_opts(struct bch_fs *c,
								    struct bch_io_opts *opts)
{
	struct bch_extent_rebalance r = {
		.type = BIT(BCH_EXTENT_ENTRY_rebalance),
#define x(_name)							\
		._name = opts->_name,					\
		._name##_from_inode = opts->_name##_from_inode,
		BCH_REBALANCE_OPTS()
#undef x
	};

	if (r.background_target &&
	    !bch2_target_accepts_data(c, BCH_DATA_user, r.background_target))
		r.background_target = 0;

	return r;
};

u64 bch2_bkey_sectors_need_rebalance(struct bch_fs *, struct bkey_s_c);
int bch2_bkey_set_needs_rebalance(struct bch_fs *, struct bch_io_opts *, struct bkey_i *);
int bch2_get_update_rebalance_opts(struct btree_trans *,
				   struct bch_io_opts *,
				   struct btree_iter *,
				   struct bkey_s_c);

int bch2_set_rebalance_needs_scan_trans(struct btree_trans *, u64);
int bch2_set_rebalance_needs_scan(struct bch_fs *, u64 inum);
int bch2_set_fs_needs_rebalance(struct bch_fs *);

static inline void bch2_rebalance_wakeup(struct bch_fs *c)
{
	c->rebalance.kick++;
	guard(rcu)();
	struct task_struct *p = rcu_dereference(c->rebalance.thread);
	if (p)
		wake_up_process(p);
}

void bch2_rebalance_status_to_text(struct printbuf *, struct bch_fs *);

void bch2_rebalance_stop(struct bch_fs *);
int bch2_rebalance_start(struct bch_fs *);

void bch2_fs_rebalance_exit(struct bch_fs *);
int bch2_fs_rebalance_init(struct bch_fs *);

int bch2_check_rebalance_work(struct bch_fs *);

#endif /* _BCACHEFS_REBALANCE_H */
Commit	Line	Data
	1	/* SPDX-License-Identifier: GPL-2.0 */
	2	#ifndef _BCACHEFS_REBALANCE_H
	3	#define _BCACHEFS_REBALANCE_H
	4
	5	#include "compress.h"
	6	#include "disk_groups.h"
	7	#include "opts.h"
	8	#include "rebalance_types.h"
	9
	10	static inline struct bch_extent_rebalance io_opts_to_rebalance_opts(struct bch_fs *c,
	11	struct bch_io_opts *opts)
	12	{
	13	struct bch_extent_rebalance r = {
	14	.type = BIT(BCH_EXTENT_ENTRY_rebalance),
	15	#define x(_name) \
	16	._name = opts->_name, \
	17	._name##_from_inode = opts->_name##_from_inode,
	18	BCH_REBALANCE_OPTS()
	19	#undef x
	20	};
	21
	22	if (r.background_target &&
	23	!bch2_target_accepts_data(c, BCH_DATA_user, r.background_target))
	24	r.background_target = 0;
	25
	26	return r;
	27	};
	28
	29	u64 bch2_bkey_sectors_need_rebalance(struct bch_fs *, struct bkey_s_c);
	30	int bch2_bkey_set_needs_rebalance(struct bch_fs , struct bch_io_opts , struct bkey_i *);
	31	int bch2_get_update_rebalance_opts(struct btree_trans *,
	32	struct bch_io_opts *,
	33	struct btree_iter *,
	34	struct bkey_s_c);
	35
	36	int bch2_set_rebalance_needs_scan_trans(struct btree_trans *, u64);
	37	int bch2_set_rebalance_needs_scan(struct bch_fs *, u64 inum);
	38	int bch2_set_fs_needs_rebalance(struct bch_fs *);
	39
	40	static inline void bch2_rebalance_wakeup(struct bch_fs *c)
	41	{
	42	c->rebalance.kick++;
	43	guard(rcu)();
	44	struct task_struct *p = rcu_dereference(c->rebalance.thread);
	45	if (p)
	46	wake_up_process(p);
	47	}
	48
	49	void bch2_rebalance_status_to_text(struct printbuf , struct bch_fs );
	50
	51	void bch2_rebalance_stop(struct bch_fs *);
	52	int bch2_rebalance_start(struct bch_fs *);
	53
	54	void bch2_fs_rebalance_exit(struct bch_fs *);
	55	int bch2_fs_rebalance_init(struct bch_fs *);
	56
	57	int bch2_check_rebalance_work(struct bch_fs *);
	58
	59	#endif /* _BCACHEFS_REBALANCE_H */