From 65672b471cdcd43daff8fcbbc6ff5d0d54159785 Mon Sep 17 00:00:00 2001
From: littleblack111 <littleblack11111@gmail.com>
Date: Sun, 22 Jun 2025 17:21:46 +0800
Subject: API(server/client:newClient): optional cb function onConnect

---
 src/server/clients.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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

diff --git a/src/server/clients.cpp b/src/server/clients.cpp
index 710f813..58f7bca 100644
--- a/src/server/clients.cpp
+++ b/src/server/clients.cpp
@@ -1,6 +1,7 @@
 #include "libsock/server/clients.hpp"
 #include "libsock/server/client.hpp"
 #include <algorithm>
+#include <functional>
 #include <ranges>
 #include <unistd.h>
 #include <utility>
@@ -58,9 +59,9 @@ Clients::~Clients() {
 	vpClients.erase(std::remove_if(vpClients.begin(), vpClients.end(), [this](const WP<Clients> &wptr) { return !wptr.owner_before(m_self) && !m_self.owner_before(wptr); }), vpClients.end());
 }
 
-SP<Client> Clients::newClient() {
+SP<Client> Clients::newClient(std::function<void(SP<Client> &)> cb) {
 	SP	  client			= SP<Client>(new Client(m_wpServer.lock(), shared_from_this()));
-	auto &instance			= m_vClients.emplace_back(std::jthread([&client]() { client->init(); }), client);
+	auto &instance			= m_vClients.emplace_back(std::jthread([&client, &cb]() { client->init(); cb(client); }), client);
 	instance.second->m_self = std::weak_ptr<Client>(instance.second);
 	return client;
 }
-- 
cgit v1.2.3