1/* SPDX-License-Identifier: GPL-2.0 */
2#ifndef _BCACHEFS_CLOCK_H
3#define _BCACHEFS_CLOCK_H
4
5void bch2_io_timer_add(struct io_clock *, struct io_timer *);
6void bch2_io_timer_del(struct io_clock *, struct io_timer *);
7void bch2_kthread_io_clock_wait(struct io_clock *, unsigned long,
8				unsigned long);
9
10void __bch2_increment_clock(struct io_clock *, unsigned);
11
12static inline void bch2_increment_clock(struct bch_fs *c, unsigned sectors,
13					int rw)
14{
15	struct io_clock *clock = &c->io_clock[rw];
16
17	if (unlikely(this_cpu_add_return(*clock->pcpu_buf, sectors) >=
18		   IO_CLOCK_PCPU_SECTORS))
19		__bch2_increment_clock(clock, this_cpu_xchg(*clock->pcpu_buf, 0));
20}
21
22void bch2_io_clock_schedule_timeout(struct io_clock *, unsigned long);
23
24#define bch2_kthread_wait_event_ioclock_timeout(condition, clock, timeout)\
25({									\
26	long __ret = timeout;						\
27	might_sleep();							\
28	if (!___wait_cond_timeout(condition))				\
29		__ret = __wait_event_timeout(wq, condition, timeout);	\
30	__ret;								\
31})
32
33void bch2_io_timers_to_text(struct printbuf *, struct io_clock *);
34
35void bch2_io_clock_exit(struct io_clock *);
36int bch2_io_clock_init(struct io_clock *);
37
38#endif /* _BCACHEFS_CLOCK_H */
39