{# 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 %}