Try writing once on connect thread
This commit is contained in:
@@ -723,10 +723,30 @@ int main(int argc, char *argv[]) {
|
|||||||
auto conn = std::make_unique<Connection>(
|
auto conn = std::make_unique<Connection>(
|
||||||
fd, connectionId.fetch_add(1, std::memory_order_relaxed));
|
fd, connectionId.fetch_add(1, std::memory_order_relaxed));
|
||||||
|
|
||||||
// Add to epoll with proper events matching server pattern
|
// Try to write once in the connect thread before handing off to network
|
||||||
struct epoll_event event{};
|
// threads
|
||||||
assert(conn->hasMessages());
|
assert(conn->hasMessages());
|
||||||
|
bool writeFinished = conn->writeBytes();
|
||||||
|
if (conn->error) {
|
||||||
|
continue; // Connection failed, destructor will clean up
|
||||||
|
}
|
||||||
|
|
||||||
|
// Determine the appropriate epoll events based on write result
|
||||||
|
struct epoll_event event{};
|
||||||
|
if (writeFinished) {
|
||||||
|
// All data was written, wait for response
|
||||||
|
int shutdown_result = shutdown(conn->fd, SHUT_WR);
|
||||||
|
if (shutdown_result == -1) {
|
||||||
|
perror("shutdown");
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
event.events = EPOLLIN | EPOLLONESHOT | EPOLLRDHUP;
|
||||||
|
} else {
|
||||||
|
// Partial write, need to continue writing
|
||||||
event.events = EPOLLOUT | EPOLLONESHOT | EPOLLRDHUP;
|
event.events = EPOLLOUT | EPOLLONESHOT | EPOLLRDHUP;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add to epoll for network threads to handle remaining I/O
|
||||||
conn->tsan_release();
|
conn->tsan_release();
|
||||||
Connection *raw_conn = conn.release();
|
Connection *raw_conn = conn.release();
|
||||||
event.data.ptr = raw_conn;
|
event.data.ptr = raw_conn;
|
||||||
|
|||||||
Reference in New Issue
Block a user