workqueue.h

   1 #ifndef FIO_RATE_H
   2 #define FIO_RATE_H
   3
   4 #include <inttypes.h>
   5 #include <pthread.h>
   6
   7 #include "flist.h"
   8 #include "lib/types.h"
   9
  10 struct sk_out;
  11 struct thread_data;
  12
  13 struct workqueue_work {
  14         struct flist_head list;
  15 };
  16
  17 struct submit_worker {
  18         pthread_t thread;
  19         pthread_mutex_t lock;
  20         pthread_cond_t cond;
  21         struct flist_head work_list;
  22         unsigned int flags;
  23         unsigned int index;
  24         uint64_t seq;
  25         struct workqueue *wq;
  26         void *priv;
  27         struct sk_out *sk_out;
  28 };
  29
  30 typedef int (workqueue_work_fn)(struct submit_worker *, struct workqueue_work *);
  31 typedef bool (workqueue_pre_sleep_flush_fn)(struct submit_worker *);
  32 typedef void (workqueue_pre_sleep_fn)(struct submit_worker *);
  33 typedef int (workqueue_alloc_worker_fn)(struct submit_worker *);
  34 typedef void (workqueue_free_worker_fn)(struct submit_worker *);
  35 typedef int (workqueue_init_worker_fn)(struct submit_worker *);
  36 typedef void (workqueue_exit_worker_fn)(struct submit_worker *, unsigned int *);
  37 typedef void (workqueue_update_acct_fn)(struct submit_worker *);
  38
  39 struct workqueue_ops {
  40         workqueue_work_fn *fn;
  41         workqueue_pre_sleep_flush_fn *pre_sleep_flush_fn;
  42         workqueue_pre_sleep_fn *pre_sleep_fn;
  43
  44         workqueue_update_acct_fn *update_acct_fn;
  45
  46         workqueue_alloc_worker_fn *alloc_worker_fn;
  47         workqueue_free_worker_fn *free_worker_fn;
  48
  49         workqueue_init_worker_fn *init_worker_fn;
  50         workqueue_exit_worker_fn *exit_worker_fn;
  51
  52         unsigned int nice;
  53 };
  54
  55 struct workqueue {
  56         unsigned int max_workers;
  57
  58         struct thread_data *td;
  59         struct workqueue_ops ops;
  60
  61         uint64_t work_seq;
  62         struct submit_worker *workers;
  63         unsigned int next_free_worker;
  64
  65         pthread_cond_t flush_cond;
  66         pthread_mutex_t flush_lock;
  67         pthread_mutex_t stat_lock;
  68         volatile int wake_idle;
  69 };
  70
  71 int workqueue_init(struct thread_data *td, struct workqueue *wq, struct workqueue_ops *ops, unsigned int max_workers, struct sk_out *sk_out);
  72 void workqueue_exit(struct workqueue *wq);
  73
  74 void workqueue_enqueue(struct workqueue *wq, struct workqueue_work *work);
  75 void workqueue_flush(struct workqueue *wq);
  76
  77 static inline bool workqueue_pre_sleep_check(struct submit_worker *sw)
  78 {
  79         struct workqueue *wq = sw->wq;
  80
  81         if (!wq->ops.pre_sleep_flush_fn)
  82                 return false;
  83
  84         return wq->ops.pre_sleep_flush_fn(sw);
  85 }
  86
  87 static inline void workqueue_pre_sleep(struct submit_worker *sw)
  88 {
  89         struct workqueue *wq = sw->wq;
  90
  91         if (wq->ops.pre_sleep_fn)
  92                 wq->ops.pre_sleep_fn(sw);
  93 }
  94
  95 static inline int workqueue_init_worker(struct submit_worker *sw)
  96 {
  97         struct workqueue *wq = sw->wq;
  98
  99         if (!wq->ops.init_worker_fn)
 100                 return 0;
 101
 102         return wq->ops.init_worker_fn(sw);
 103 }
 104
 105 static inline void workqueue_exit_worker(struct submit_worker *sw,
 106                                          unsigned int *sum_cnt)
 107 {
 108         struct workqueue *wq = sw->wq;
 109         unsigned int tmp = 1;
 110
 111         if (!wq->ops.exit_worker_fn)
 112                 return;
 113
 114         if (!sum_cnt)
 115                 sum_cnt = &tmp;
 116
 117         wq->ops.exit_worker_fn(sw, sum_cnt);
 118 }
 119 #endif