git.sesse.net Git - bcachefs-tools-debian/blob - libbcachefs/bkey_sort.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 #ifndef _BCACHEFS_BKEY_SORT_H
   3 #define _BCACHEFS_BKEY_SORT_H
   4
   5 struct sort_iter {
   6         struct btree            *b;
   7         unsigned                used;
   8         unsigned                size;
   9
  10         struct sort_iter_set {
  11                 struct bkey_packed *k, *end;
  12         } data[MAX_BSETS + 1];
  13 };
  14
  15 static inline void sort_iter_init(struct sort_iter *iter, struct btree *b)
  16 {
  17         iter->b = b;
  18         iter->used = 0;
  19         iter->size = ARRAY_SIZE(iter->data);
  20 }
  21
  22 static inline void sort_iter_add(struct sort_iter *iter,
  23                                  struct bkey_packed *k,
  24                                  struct bkey_packed *end)
  25 {
  26         BUG_ON(iter->used >= iter->size);
  27
  28         if (k != end)
  29                 iter->data[iter->used++] = (struct sort_iter_set) { k, end };
  30 }
  31
  32 struct btree_nr_keys
  33 bch2_key_sort_fix_overlapping(struct bch_fs *, struct bset *,
  34                               struct sort_iter *);
  35 struct btree_nr_keys
  36 bch2_extent_sort_fix_overlapping(struct bch_fs *, struct bset *,
  37                                  struct sort_iter *);
  38
  39 struct btree_nr_keys
  40 bch2_sort_repack(struct bset *, struct btree *,
  41                  struct btree_node_iter *,
  42                  struct bkey_format *, bool);
  43 struct btree_nr_keys
  44 bch2_sort_repack_merge(struct bch_fs *,
  45                        struct bset *, struct btree *,
  46                        struct btree_node_iter *,
  47                        struct bkey_format *, bool);
  48
  49 unsigned bch2_sort_keys(struct bkey_packed *,
  50                         struct sort_iter *, bool);
  51 unsigned bch2_sort_extents(struct bkey_packed *,
  52                            struct sort_iter *, bool);
  53
  54 unsigned bch2_sort_extent_whiteouts(struct bkey_packed *,
  55                                     struct sort_iter *);
  56
  57 #endif /* _BCACHEFS_BKEY_SORT_H */