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
  36 struct btree_nr_keys
  37 bch2_sort_repack(struct bset *, struct btree *,
  38                  struct btree_node_iter *,
  39                  struct bkey_format *, bool);
  40
  41 unsigned bch2_sort_keys(struct bkey_packed *,
  42                         struct sort_iter *, bool);
  43
  44 #endif /* _BCACHEFS_BKEY_SORT_H */