Unbreak output buffer logging over the network
[fio.git] / diskutil.h
1 #ifndef FIO_DISKUTIL_H
2 #define FIO_DISKUTIL_H
3 #include "json.h"
4 #define FIO_DU_NAME_SZ          64
5
6 #include "lib/output_buffer.h"
7
8 extern volatile int helper_exit;
9
10 struct disk_util_stats {
11         uint64_t ios[2];
12         uint64_t merges[2];
13         uint64_t sectors[2];
14         uint64_t ticks[2];
15         uint64_t io_ticks;
16         uint64_t time_in_queue;
17         uint64_t msec;
18 };
19
20 /*
21  * Disk utils as read in /sys/block/<dev>/stat
22  */
23 struct disk_util_stat {
24         uint8_t name[FIO_DU_NAME_SZ];
25         struct disk_util_stats s;
26 };
27
28 struct disk_util_agg {
29         uint64_t ios[2];
30         uint64_t merges[2];
31         uint64_t sectors[2];
32         uint64_t ticks[2];
33         uint64_t io_ticks;
34         uint64_t time_in_queue;
35         uint32_t slavecount;
36         uint32_t pad;
37         fio_fp64_t max_util;
38 };
39
40 /*
41  * Per-device disk util management
42  */
43 struct disk_util {
44         struct flist_head list;
45         /* If this disk is a slave, hook it into the master's
46          * list using this head.
47          */
48         struct flist_head slavelist;
49
50         char *name;
51         char *sysfs_root;
52         char path[PATH_MAX];
53         int major, minor;
54
55         struct disk_util_stat dus;
56         struct disk_util_stat last_dus;
57
58         struct disk_util_agg agg;
59
60         /* For software raids, this entry maintains pointers to the
61          * entries for the slave devices. The disk_util entries for
62          * the slaves devices should primarily be maintained through
63          * the disk_list list, i.e. for memory allocation and
64          * de-allocation, etc. Whereas this list should be used only
65          * for aggregating a software RAID's disk util figures.
66          */
67         struct flist_head slaves;
68
69         struct timeval time;
70
71         struct fio_mutex *lock;
72         unsigned long users;
73 };
74
75 static inline void disk_util_mod(struct disk_util *du, int val)
76 {
77         if (du) {
78                 struct flist_head *n;
79
80                 fio_mutex_down(du->lock);
81                 du->users += val;
82
83                 flist_for_each(n, &du->slavelist) {
84                         struct disk_util *slave;
85
86                         slave = flist_entry(n, struct disk_util, slavelist);
87                         slave->users += val;
88                 }
89                 fio_mutex_up(du->lock);
90         }
91 }
92 static inline void disk_util_inc(struct disk_util *du)
93 {
94         disk_util_mod(du, 1);
95 }
96
97 static inline void disk_util_dec(struct disk_util *du)
98 {
99         disk_util_mod(du, -1);
100 }
101
102 #define DISK_UTIL_MSEC  (250)
103
104 extern struct flist_head disk_list;
105
106 /*
107  * disk util stuff
108  */
109 #ifdef FIO_HAVE_DISK_UTIL
110 extern void print_disk_util(struct disk_util_stat *, struct disk_util_agg *, int terse, struct buf_output *);
111 extern void show_disk_util(int terse, struct json_object *parent, struct buf_output *);
112 extern void json_array_add_disk_util(struct disk_util_stat *dus,
113                 struct disk_util_agg *agg, struct json_array *parent);
114 extern void init_disk_util(struct thread_data *);
115 extern int update_io_ticks(void);
116 extern void setup_disk_util(void);
117 extern void disk_util_prune_entries(void);
118 #else
119 static inline void print_disk_util(struct disk_util_stat *du,
120                                    struct disk_util_agg *agg, int terse,
121                                    struct buf_output *out)
122 {
123 }
124 #define show_disk_util(terse, parent, out)
125 #define disk_util_prune_entries()
126 #define init_disk_util(td)
127 #define setup_disk_util()
128 #define json_array_add_disk_util(dus, agg, parent)
129
130 static inline int update_io_ticks(void)
131 {
132         return helper_exit;
133 }
134 #endif
135
136 #endif