waitqueue.h 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. #ifndef WAITQUEUE_H__
  2. #define WAITQUEUE_H__
  3. #include <rtthread.h>
  4. #define RT_WQ_FLAG_CLEAN 0x00
  5. #define RT_WQ_FLAG_WAKEUP 0x01
  6. struct rt_wqueue_node;
  7. typedef int (*rt_wqueue_func_t)(struct rt_wqueue_node *wait, void *key);
  8. struct rt_wqueue_node
  9. {
  10. rt_thread_t polling_thread;
  11. rt_list_t list;
  12. rt_wqueue_func_t wakeup;
  13. rt_uint32_t key;
  14. };
  15. typedef struct rt_wqueue_node rt_wqueue_node_t;
  16. int __wqueue_default_wake(struct rt_wqueue_node *wait, void *key);
  17. rt_inline void rt_wqueue_init(rt_wqueue_t *queue)
  18. {
  19. RT_ASSERT(queue != RT_NULL);
  20. queue->flag = RT_WQ_FLAG_CLEAN;
  21. rt_list_init(&(queue->waiting_list));
  22. }
  23. void rt_wqueue_add(rt_wqueue_t *queue, struct rt_wqueue_node *node);
  24. void rt_wqueue_remove(struct rt_wqueue_node *node);
  25. int rt_wqueue_wait(rt_wqueue_t *queue, int condition, int timeout);
  26. void rt_wqueue_wakeup(rt_wqueue_t *queue, void *key);
  27. #define DEFINE_WAIT_FUNC(name, function) \
  28. struct rt_wqueue_node name = { \
  29. rt_current_thread, \
  30. RT_LIST_OBJECT_INIT(((name).list)), \
  31. \
  32. function, \
  33. 0 \
  34. }
  35. #define DEFINE_WAIT(name) DEFINE_WAIT_FUNC(name, __wqueue_default_wake)
  36. #endif