60 lines
1.4 KiB
C++
60 lines
1.4 KiB
C++
#include <pthread.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <time.h>
|
|
#include <queue>
|
|
#include <vector>
|
|
#include <atomic>
|
|
|
|
/* 常量定义 */
|
|
#define MESSAGE_QUEUE_SIZE 10000 // 消息队列容量
|
|
|
|
/* 消息结构体 */
|
|
typedef struct {
|
|
int client_index; // 客户端索引
|
|
char* data; // 消息数据
|
|
size_t length; // 消息长度
|
|
} deal_message_t;
|
|
|
|
/* 线程安全的消息队列 */
|
|
class SafeMessageQueue {
|
|
private:
|
|
std::queue<deal_message_t> queue;
|
|
pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
|
|
pthread_cond_t cond = PTHREAD_COND_INITIALIZER;
|
|
std::atomic<int> count{ 0 };
|
|
|
|
public:
|
|
bool push(const deal_message_t& msg) {
|
|
pthread_mutex_lock(&mutex);
|
|
if (queue.size() >= MESSAGE_QUEUE_SIZE) {
|
|
pthread_mutex_unlock(&mutex);
|
|
return false;
|
|
}
|
|
queue.push(msg);
|
|
count++;
|
|
pthread_cond_signal(&cond);
|
|
pthread_mutex_unlock(&mutex);
|
|
return true;
|
|
}
|
|
|
|
bool pop(deal_message_t& msg) {
|
|
pthread_mutex_lock(&mutex);
|
|
while (queue.empty()) {
|
|
pthread_cond_wait(&cond, &mutex);
|
|
}
|
|
msg = queue.front();
|
|
queue.pop();
|
|
count--;
|
|
pthread_mutex_unlock(&mutex);
|
|
return true;
|
|
}
|
|
|
|
size_t size() const {
|
|
return count.load();
|
|
}
|
|
};
|
|
|
|
void process_received_message(int client_index, const char* data, size_t length); |