HappyTanuki/Np_Term
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 5 Wiki Activity

모델 업로드 구조 완성

Browse Source
This commit is contained in:
HappyTanuki
2025-06-15 03:42:25 +09:00
parent 93fda88d75
commit 06a1e41dc8
13 changed files with 149 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
Server/src/server.cpp
View File

@@ -1,35 +1,50 @@
#include <chrono>
#include "socket/iocp.h"
#include "socket/packet.h"
#include "socket/tcp_socket.h"
#include "socket/udp_socket.h"
#include "socket/wsa_manager.h"
#include "utils/log.h"
#include "utils/snowflake.h"
#include "socket/packet.h"
#include "vulkan_engine/asset/object/model.h"
#define LISTENIP "::"
#define LISTENPORT 9010
std::uint8_t CLIENTID = 0;
std::unordered_map<utils::Snowflake, veng::Model> MODELS;
std::mutex MODELS_MUTEX;
std::vector<Network::TCPSocket> CLIENTS;
std::mutex CLIENTS_MUTEX;
void EchoClient(utils::ThreadPool* tp, Network::IOCP* iocp,
Network::TCPSocket NewSock, Network::Address NewAddr) {
Network::IOCPPASSINDATA* recv_data = new Network::IOCPPASSINDATA(16 * 1024);
recv_data->socket = std::make_shared<Network::Socket>(NewSock);
recv_data->IOCPInstance = iocp;
auto timestamp = std::chrono::system_clock::now();
while (!iocp->recv(recv_data)) {
if (std::chrono::duration_cast<std::chrono::milliseconds>(
std::chrono::system_clock::now() - timestamp)
.count() == 1000) {
tp->enqueueJob(EchoClient, iocp, NewSock, NewAddr);
return;
while (iocp->GetRecvedBytes(NewSock.sock) < 6);
std::vector<char> recv_data(6);
iocp->recv(NewSock, recv_data);
Packet::Header header;
header.Deserialize(recv_data);
recv_data.resize(header.body_length);
while (iocp->GetRecvedBytes(NewSock.sock) < header.body_length);
iocp->recv(NewSock, recv_data);
switch (header.opcode) {
case Packet::Opcode::UPDATEMODEL: {
veng::Model model;
model.Deserialize(recv_data);
std::lock_guard lock(MODELS_MUTEX);
if (MODELS.find(model.ID) == MODELS.end())
spdlog::info("model received: {}", model.ID.snowflake);
MODELS[model.ID] = std::move(model);
}
break;
default:
spdlog::error("unknown data type");
}
std::vector<Network::IOCPPASSINDATA*> send_data;
recv_data->event = Network::IOCPEVENT::WRITE;
send_data.push_back(recv_data);
iocp->send(NewSock.sock, &send_data);
// iocp->send(NewSock, recv_data);
tp->enqueueJob(EchoClient, iocp, NewSock, NewAddr);
}
@@ -67,10 +82,18 @@ int main(int argc, char* argv[]) {
spdlog::info("Waiting for connection");
TCPSock.accept(NewSock, NewAddr);
Network::IOCPPASSINDATA* data = new Network::IOCPPASSINDATA(16 * 1024);
data->socket = std::make_shared<Network::Socket>(NewSock);
data->IOCPInstance = &iocp;
iocp.registerTCPSocket(data);
iocp.registerTCPSocket(NewSock, 16 * 1024);
Packet::Header header;
header.opcode = Packet::Opcode::CLIENTID;
header.body_length = 8;
auto packet = header.Serialize();
packet.resize(packet.size() + 8);
std::lock_guard lock(CLIENTS_MUTEX);
CLIENTS.push_back(NewSock);
std::uint8_t client_id = CLIENTS.size() - 1;
::memcpy(packet.data() + 6, &client_id, 1);
iocp.send(NewSock, packet);
tp.enqueueJob(EchoClient, &iocp, NewSock, NewAddr);
}