bcachefs-tools/libbcache/alloc.h
2017-03-10 12:40:01 -09:00

86 lines
2.2 KiB
C

#ifndef _BCACHE_ALLOC_H
#define _BCACHE_ALLOC_H
#include "alloc_types.h"
struct bkey;
struct bucket;
struct bch_dev;
struct bch_fs;
struct dev_group;
static inline size_t prios_per_bucket(const struct bch_dev *ca)
{
return (bucket_bytes(ca) - sizeof(struct prio_set)) /
sizeof(struct bucket_disk);
}
static inline size_t prio_buckets(const struct bch_dev *ca)
{
return DIV_ROUND_UP((size_t) (ca)->mi.nbuckets, prios_per_bucket(ca));
}
void bch_dev_group_remove(struct dev_group *, struct bch_dev *);
void bch_dev_group_add(struct dev_group *, struct bch_dev *);
int bch_prio_read(struct bch_dev *);
size_t bch_bucket_alloc(struct bch_dev *, enum alloc_reserve);
void bch_open_bucket_put(struct bch_fs *, struct open_bucket *);
struct open_bucket *bch_alloc_sectors_start(struct bch_fs *,
struct write_point *,
unsigned, unsigned,
enum alloc_reserve,
struct closure *);
void bch_alloc_sectors_append_ptrs(struct bch_fs *, struct bkey_i_extent *,
unsigned, struct open_bucket *, unsigned);
void bch_alloc_sectors_done(struct bch_fs *, struct write_point *,
struct open_bucket *);
struct open_bucket *bch_alloc_sectors(struct bch_fs *, struct write_point *,
struct bkey_i_extent *, unsigned, unsigned,
enum alloc_reserve, struct closure *);
static inline void bch_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 bch_recalc_capacity(struct bch_fs *);
void bch_dev_allocator_stop(struct bch_dev *);
int bch_dev_allocator_start(struct bch_dev *);
void bch_fs_allocator_init(struct bch_fs *);
#endif /* _BCACHE_ALLOC_H */