bcachefs-tools/libbcachefs/alloc.h

85 lines
2.3 KiB
C

#ifndef _BCACHE_ALLOC_H
#define _BCACHE_ALLOC_H
#include "bcachefs.h"
#include "alloc_types.h"
struct bkey;
struct bucket;
struct bch_dev;
struct bch_fs;
struct dev_group;
void bch2_dev_group_remove(struct dev_group *, struct bch_dev *);
void bch2_dev_group_add(struct dev_group *, struct bch_dev *);
int bch2_alloc_read(struct bch_fs *, struct list_head *);
int bch2_alloc_write(struct bch_fs *, struct bch_dev *, u64 *);
int bch2_alloc_replay_key(struct bch_fs *, struct bpos);
long bch2_bucket_alloc(struct bch_fs *, struct bch_dev *, enum alloc_reserve);
void bch2_open_bucket_put(struct bch_fs *, struct open_bucket *);
struct open_bucket *bch2_alloc_sectors_start(struct bch_fs *,
struct write_point *,
unsigned, unsigned,
enum alloc_reserve,
struct closure *);
void bch2_alloc_sectors_append_ptrs(struct bch_fs *, struct bkey_i_extent *,
unsigned, struct open_bucket *, unsigned);
void bch2_alloc_sectors_done(struct bch_fs *, struct write_point *,
struct open_bucket *);
struct open_bucket *bch2_alloc_sectors(struct bch_fs *, struct write_point *,
struct bkey_i_extent *, unsigned, unsigned,
enum alloc_reserve, struct closure *);
static inline void bch2_wake_allocator(struct bch_dev *ca)
{
struct task_struct *p;
rcu_read_lock();
if ((p = ACCESS_ONCE(ca->alloc_thread)))
wake_up_process(p);
rcu_read_unlock();
}
static inline struct bch_dev *dev_group_next(struct dev_group *devs,
unsigned *iter)
{
struct bch_dev *ret = NULL;
while (*iter < devs->nr &&
!(ret = rcu_dereference_check(devs->d[*iter].dev,
lockdep_is_held(&devs->lock))))
(*iter)++;
return ret;
}
#define group_for_each_dev(ca, devs, iter) \
for ((iter) = 0; \
((ca) = dev_group_next((devs), &(iter))); \
(iter)++)
#define open_bucket_for_each_ptr(_ob, _ptr) \
for ((_ptr) = (_ob)->ptrs; \
(_ptr) < (_ob)->ptrs + (_ob)->nr_ptrs; \
(_ptr)++)
void bch2_recalc_capacity(struct bch_fs *);
void bch2_dev_allocator_remove(struct bch_fs *, struct bch_dev *);
void bch2_dev_allocator_add(struct bch_fs *, struct bch_dev *);
void bch2_dev_allocator_stop(struct bch_dev *);
int bch2_dev_allocator_start(struct bch_dev *);
void bch2_fs_allocator_init(struct bch_fs *);
extern const struct bkey_ops bch2_bkey_alloc_ops;
#endif /* _BCACHE_ALLOC_H */