xref: /linux/drivers/md/dm-io-tracker.h (revision dc4fa29f)
1 /*
2  * Copyright (C) 2021 Red Hat, Inc. All rights reserved.
3  *
4  * This file is released under the GPL.
5  */
6 
7 #ifndef DM_IO_TRACKER_H
8 #define DM_IO_TRACKER_H
9 
10 #include <linux/jiffies.h>
11 
12 struct dm_io_tracker {
13 	spinlock_t lock;
14 
15 	/*
16 	 * Sectors of in-flight IO.
17 	 */
18 	sector_t in_flight;
19 
20 	/*
21 	 * The time, in jiffies, when this device became idle
22 	 * (if it is indeed idle).
23 	 */
24 	unsigned long idle_time;
25 	unsigned long last_update_time;
26 };
27 
28 static inline void dm_iot_init(struct dm_io_tracker *iot)
29 {
30 	spin_lock_init(&iot->lock);
31 	iot->in_flight = 0ul;
32 	iot->idle_time = 0ul;
33 	iot->last_update_time = jiffies;
34 }
35 
36 static inline bool dm_iot_idle_for(struct dm_io_tracker *iot, unsigned long j)
37 {
38 	bool r = false;
39 
40 	spin_lock_irq(&iot->lock);
41 	if (!iot->in_flight)
42 		r = time_after(jiffies, iot->idle_time + j);
43 	spin_unlock_irq(&iot->lock);
44 
45 	return r;
46 }
47 
48 static inline void dm_iot_io_begin(struct dm_io_tracker *iot, sector_t len)
49 {
50 	spin_lock_irq(&iot->lock);
51 	iot->in_flight += len;
52 	spin_unlock_irq(&iot->lock);
53 }
54 
55 static inline void dm_iot_io_end(struct dm_io_tracker *iot, sector_t len)
56 {
57 	unsigned long flags;
58 
59 	if (!len)
60 		return;
61 
62 	spin_lock_irqsave(&iot->lock, flags);
63 	iot->in_flight -= len;
64 	if (!iot->in_flight)
65 		iot->idle_time = jiffies;
66 	spin_unlock_irqrestore(&iot->lock, flags);
67 }
68 
69 #endif
70