fs/bcachefs/rebalance.h

   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 rebalance_wakeup(struct bch_fs *c)
  41 {
  42         struct task_struct *p;
  43
  44         rcu_read_lock();
  45         p = rcu_dereference(c->rebalance.thread);
  46         if (p)
  47                 wake_up_process(p);
  48         rcu_read_unlock();
  49 }
  50
  51 void bch2_rebalance_status_to_text(struct printbuf *, struct bch_fs *);
  52
  53 void bch2_rebalance_stop(struct bch_fs *);
  54 int bch2_rebalance_start(struct bch_fs *);
  55 void bch2_fs_rebalance_init(struct bch_fs *);
  56
  57 #endif /* _BCACHEFS_REBALANCE_H */