#ifndef __SHM_QUEUE_WRAPPER_H__ #define __SHM_QUEUE_WRAPPER_H__ #include "usg_common.h" #include "usg_typedef.h" #include "shm_queue.h" #include "shm_allocator.h" #ifdef __cplusplus extern "C" { #endif #define MAX_ELE_SIZE 512 typedef struct ele_t { char buf[MAX_ELE_SIZE]; } ele_t; typedef struct shmqueue_t { size_t ele_size; void *mqueue; } shmqueue_t; /** * 初始化 * @ shmqueue * @ key 标识共享队列的唯一key * @ queue_size 队列大小 * @ ele_size 队列中元素大小 */ inline void shmqueue_init(shmqueue_t *shmqueue, int key, int queue_size, int ele_size) { if(ele_size > MAX_ELE_SIZE) { err_exit(0, "shmqueue_init 元素大小超过设计的最大大小"); } SHMQueue *queue = new SHMQueue(key, queue_size); shmqueue->mqueue = (void *)queue; shmqueue->ele_size = ele_size; } /** * 销毁 */ inline void shmqueue_destroy(shmqueue_t *shmqueue) { delete (SHMQueue *)shmqueue->mqueue; } /** * 队列元素的个数 */ inline uint32_t shmqueue_size(shmqueue_t *shmqueue) { return ((SHMQueue *)(shmqueue->mqueue))->size(); } /** * 是否已满 */ inline int shmqueue_full(shmqueue_t *shmqueue) { return ((SHMQueue *)(shmqueue->mqueue))->full(); } /** * 是否为空 */ inline int shmqueue_empty(shmqueue_t *shmqueue) { return ((SHMQueue *)(shmqueue->mqueue))->empty(); } /** * 入队, 队列满时等待 */ inline int shmqueue_push(shmqueue_t *shmqueue, void *src_ele) { ele_t dest_ele; memcpy(&dest_ele.buf, src_ele, shmqueue->ele_size); return ((SHMQueue *)(shmqueue->mqueue))->push(dest_ele); } /** * 入队, 队列满时立即返回 */ inline int shmqueue_push_nowait(shmqueue_t *shmqueue, void *src_ele) { ele_t dest_ele; memcpy(&dest_ele.buf, src_ele, shmqueue->ele_size); return ((SHMQueue *)(shmqueue->mqueue))->push_nowait(dest_ele); } /** * 入队, 指定时间内入对不成功就返回 */ inline int shmqueue_push_timeout(shmqueue_t *shmqueue, void *src_ele, struct timespec * timeout) { ele_t dest_ele; memcpy(&dest_ele.buf, src_ele, shmqueue->ele_size); return ((SHMQueue *)(shmqueue->mqueue))->push_timeout(dest_ele, timeout); } /** * 出队, 队列空时等待 */ inline int shmqueue_pop(shmqueue_t *shmqueue, void *dest_ele) { ele_t src_ele; bool rv = ((SHMQueue *)(shmqueue->mqueue))->pop(src_ele); if (rv) { memcpy(dest_ele, &src_ele.buf, shmqueue->ele_size); return 1; } else { return 0; } } /** * 出队, 队列空时立即返回 */ inline int shmqueue_pop_nowait(shmqueue_t *shmqueue, void *dest_ele) { ele_t src_ele; bool rv = ((SHMQueue *)(shmqueue->mqueue))->pop_nowait(src_ele); if (rv) { memcpy(dest_ele, &src_ele.buf, shmqueue->ele_size); return 1; } else { return 0; } } /** * 出队, 指定时间内出对不成功就返回 */ inline int shmqueue_pop_timeout(shmqueue_t *shmqueue, void *dest_ele, struct timespec * timeout) { ele_t src_ele; bool rv = ((SHMQueue *)(shmqueue->mqueue))->pop_timeout(src_ele, timeout); if (rv) { memcpy(dest_ele, &src_ele.buf, shmqueue->ele_size); return 1; } else { return 0; } } #ifdef __cplusplus } #endif #endif