From 9b61c218b665f94f5cc383a5e607b951ce6af121 Mon Sep 17 00:00:00 2001
From: littleblack111 <littleblack11111@gmail.com>
Date: Sun, 22 Jun 2025 15:09:42 +0800
Subject: refactor(server): properly use weak pointer

API!(server/client:create): require manually specifying server
API!(server/client:newClient): now returns a SP instead of WP
---
 src/server/client.cpp | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

(limited to 'src/server/client.cpp')

diff --git a/src/server/client.cpp b/src/server/client.cpp
index 6e42b69..e84a654 100644
--- a/src/server/client.cpp
+++ b/src/server/client.cpp
@@ -16,16 +16,16 @@ using namespace LibSock::Server;
 Client::Client(SP<Server> server, SP<Clients> clients, bool track, bool oneShot)
 	: m_track(track)
 	, m_oneShot(oneShot) {
-	pServer	 = std::move(server);
-	pClients = std::move(clients);
+	m_wpServer	= std::move(server);
+	m_wpClients = std::move(clients);
 
-	if (!pClients)
+	if (m_wpClients.expired())
 		throw std::runtime_error("server:client: ClientManager doesn't exist");
-	if (!pServer)
+	if (m_wpServer.expired())
 		throw std::runtime_error("server:client: Server doesn't exist");
 
-	m_sockfd = std::make_shared<LibSock::CFileDescriptor>(accept(pServer->getSocket()->get(), reinterpret_cast<sockaddr *>(&m_addr), &m_addrLen)); // if this is in the init list, it will run before
-																																				   // m_addrLen, so it won't work :/
+	m_sockfd = std::make_shared<LibSock::CFileDescriptor>(accept(m_wpServer.lock()->getSocket()->get(), reinterpret_cast<sockaddr *>(&m_addr), &m_addrLen)); // if this is in the init list, it will run before
+																																							 // m_addrLen, so it won't work :/
 
 	if (!m_sockfd->isValid())
 		throw std::runtime_error("server:client: Failed to create socket");
@@ -91,7 +91,7 @@ UP<SRecvData> Client::read(std::optional<std::function<bool(const SRecvData &)>>
 	}
 
 	if (m_track)
-		pClients->m_vDatas.push_back({recvData->data, self});
+		m_wpClients.lock()->m_vDatas.push_back({recvData->data, m_self});
 
 	return recvData;
 }
@@ -108,11 +108,11 @@ void Client::runLoop(bool resumeHist, std::optional<std::function<bool(const SRe
 
 	recvLoop(cb);
 
-	pClients->kick(self);
+	m_wpClients.lock()->kick(m_self);
 }
 
 void Client::resumeHistory() {
-	for (const auto &chat : pClients->getDatas())
+	for (const auto &chat : m_wpClients.lock()->getDatas())
 		write(chat.msg);
 }
 
-- 
cgit v1.2.3