110 lines
2.4 KiB
Plaintext
110 lines
2.4 KiB
Plaintext
{# TCP接收器模板 #}
|
||
{% if part == 'header' %}
|
||
#pragma once
|
||
|
||
#include <vector>
|
||
#include <cstdint>
|
||
#include <string>
|
||
|
||
/**
|
||
* TCP数据接收器
|
||
* 用于接收{{ source_message.full_name }}消息
|
||
*/
|
||
class DataReceiver {
|
||
public:
|
||
DataReceiver();
|
||
~DataReceiver();
|
||
|
||
// 初始化TCP连接
|
||
bool initialize(const std::string& host, int port);
|
||
|
||
// 接收数据
|
||
std::vector<uint8_t> receive();
|
||
|
||
// 关闭连接
|
||
void close();
|
||
|
||
private:
|
||
int socket_fd_;
|
||
bool connected_;
|
||
};
|
||
|
||
{% else %}
|
||
#include "receiver.h"
|
||
#include <sys/socket.h>
|
||
#include <netinet/in.h>
|
||
#include <arpa/inet.h>
|
||
#include <unistd.h>
|
||
#include <cstring>
|
||
#include <stdexcept>
|
||
|
||
DataReceiver::DataReceiver()
|
||
: socket_fd_(-1), connected_(false) {
|
||
}
|
||
|
||
DataReceiver::~DataReceiver() {
|
||
close();
|
||
}
|
||
|
||
bool DataReceiver::initialize(const std::string& host, int port) {
|
||
// 创建socket
|
||
socket_fd_ = socket(AF_INET, SOCK_STREAM, 0);
|
||
if (socket_fd_ < 0) {
|
||
return false;
|
||
}
|
||
|
||
// 设置服务器地址
|
||
struct sockaddr_in server_addr;
|
||
memset(&server_addr, 0, sizeof(server_addr));
|
||
server_addr.sin_family = AF_INET;
|
||
server_addr.sin_port = htons(port);
|
||
inet_pton(AF_INET, host.c_str(), &server_addr.sin_addr);
|
||
|
||
// 连接服务器
|
||
if (connect(socket_fd_, (struct sockaddr*)&server_addr, sizeof(server_addr)) < 0) {
|
||
::close(socket_fd_);
|
||
socket_fd_ = -1;
|
||
return false;
|
||
}
|
||
|
||
connected_ = true;
|
||
return true;
|
||
}
|
||
|
||
std::vector<uint8_t> DataReceiver::receive() {
|
||
if (!connected_) {
|
||
throw std::runtime_error("Not connected");
|
||
}
|
||
|
||
// 先接收数据长度（4字节）
|
||
uint32_t data_length = 0;
|
||
ssize_t n = recv(socket_fd_, &data_length, sizeof(data_length), 0);
|
||
if (n != sizeof(data_length)) {
|
||
throw std::runtime_error("Failed to receive data length");
|
||
}
|
||
|
||
// 接收数据
|
||
std::vector<uint8_t> buffer(data_length);
|
||
size_t total_received = 0;
|
||
|
||
while (total_received < data_length) {
|
||
n = recv(socket_fd_, buffer.data() + total_received,
|
||
data_length - total_received, 0);
|
||
if (n <= 0) {
|
||
throw std::runtime_error("Failed to receive data");
|
||
}
|
||
total_received += n;
|
||
}
|
||
|
||
return buffer;
|
||
}
|
||
|
||
void DataReceiver::close() {
|
||
if (socket_fd_ >= 0) {
|
||
::close(socket_fd_);
|
||
socket_fd_ = -1;
|
||
}
|
||
connected_ = false;
|
||
}
|
||
{% endif %}
|