UDP接收大量高速数据并处理

原创已于 2023-07-13 22:57:36 修改 · 1.2k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#c++

于 2023-07-13 22:56:18 首次发布

本文展示了一个C++程序，用于接收通过UDP协议传输的大量高速数据。程序使用多线程，一个线程负责接收数据，将其放入队列，另一个线程池负责从队列中取出数据并进行处理。程序中包含了接收数据、处理数据的逻辑以及套接字创建、绑定和关闭等操作。

#include <iostream>
#include <thread>
#include <cstring>
#include <arpa/inet.h>
#include <queue>
#include <mutex>
#include <condition_variable>

// 定义数据类型
struct Data {
char buffer[1024];
struct sockaddr_in client_address;
socklen_t client_address_len;
};

std::queue<Data> dataQueue;
std::mutex queueMutex;
std::condition_variable queueCondition;

void receiveData(int sock) {
while (true) {
Data data;

// 接收数据
data.client_address_len = sizeof(data.client_address);
ssize_t num_bytes = recvfrom(sock, data.buffer, sizeof(data.buffer), 0, (struct sockaddr*)&data.client_address, &data.client_address_len);
if (num_bytes > 0) {
std::lock_guard<std::mutex> lock(queueMutex);
dataQueue.push(data);
queueCondition.notify_one();
}
}
}

void processData() {
while (true) {
std::unique_lock<std::mutex> lock(queueMutex);
queueCondition.wait(lock, [] { return !dataQueue.empty(); });

Data data = dataQueue.front();
dataQueue.pop();

lock.unlock();

// 处理数据，这里可以使用线程池进行处理

std::cout << "Received data from " << inet_ntoa(data.client_address.sin_addr) << ": " << data.buffer << std::endl;
}
}

int main() {
int sock;

// 创建UDP套接字
if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
std::cerr << "Failed to create socket" << std::endl;
return 1;
}

// 绑定地址和端口
struct sockaddr_in server_address;
server_address.sin_family = AF_INET;
server_address.sin_addr.s_addr = INADDR_ANY;
server_address.sin_port = htons(8000);

if (bind(sock, (struct sockaddr*)&server_address, sizeof(server_address)) < 0) {
std::cerr << "Failed to bind socket" << std::endl;
return 1;
}

// 创建接收数据的线程
std::thread receiverThread(receiveData, sock);

// 创建线程池，用于处理数据
int numThreads = 4;
std::vector<std::thread> threadPool;
for (int i = 0; i < numThreads; ++i) {
threadPool.emplace_back(processData);
}

// 等待接收线程和线程池中的线程结束
receiverThread.join();
for (auto& thread : threadPool) {
thread.join();
}

// 关闭套接字
close(sock);

return 0;
}