/* SPDX-License-Identifier: GPL-2.0 */
#ifndef _BCACHEFS_LRU_H
#define _BCACHEFS_LRU_H

static inline u64 lru_pos_id(struct bpos pos)
{
	return pos.inode >> LRU_TIME_BITS;
}

static inline u64 lru_pos_time(struct bpos pos)
{
	return pos.inode & ~(~0ULL << LRU_TIME_BITS);
}

static inline struct bpos lru_pos(u16 lru_id, u64 dev_bucket, u64 time)
{
	struct bpos pos = POS(((u64) lru_id << LRU_TIME_BITS)|time, dev_bucket);

	EBUG_ON(time > LRU_TIME_MAX);
	EBUG_ON(lru_pos_id(pos) != lru_id);
	EBUG_ON(lru_pos_time(pos) != time);
	EBUG_ON(pos.offset != dev_bucket);

	return pos;
}

static inline struct bpos lru_start(u16 lru_id)
{
	return lru_pos(lru_id, 0, 0);
}

static inline struct bpos lru_end(u16 lru_id)
{
	return lru_pos(lru_id, U64_MAX, LRU_TIME_MAX);
}

static inline enum bch_lru_type lru_type(struct bkey_s_c l)
{
	u16 lru_id = l.k->p.inode >> 48;

	switch (lru_id) {
	case BCH_LRU_BUCKET_FRAGMENTATION:
		return BCH_LRU_fragmentation;
	case BCH_LRU_STRIPE_FRAGMENTATION:
		return BCH_LRU_stripes;
	default:
		return BCH_LRU_read;
	}
}

int bch2_lru_validate(struct bch_fs *, struct bkey_s_c, struct bkey_validate_context);
void bch2_lru_to_text(struct printbuf *, struct bch_fs *, struct bkey_s_c);

void bch2_lru_pos_to_text(struct printbuf *, struct bpos);

#define bch2_bkey_ops_lru ((struct bkey_ops) {	\
	.key_validate	= bch2_lru_validate,	\
	.val_to_text	= bch2_lru_to_text,	\
	.min_val_size	= 8,			\
})

int __bch2_lru_change(struct btree_trans *, u16, u64, u64, u64);

static inline int bch2_lru_change(struct btree_trans *trans,
		      u16 lru_id, u64 dev_bucket,
		      u64 old_time, u64 new_time)
{
	return old_time != new_time
		? __bch2_lru_change(trans, lru_id, dev_bucket, old_time, new_time)
		: 0;
}

int bch2_dev_remove_lrus(struct bch_fs *, struct bch_dev *);

struct wb_maybe_flush;
int bch2_lru_check_set(struct btree_trans *, u16, u64, u64, struct bkey_s_c,
		       struct wb_maybe_flush *);
int bch2_check_lrus(struct bch_fs *);

#endif /* _BCACHEFS_LRU_H */
