/***ACCEPT_FIRST_CONNECTION***/
Receiver::ReceiverState Receiver::acceptFirstConnection(ThreadData &data) {
VLOG(1) << data << " entered ACCEPT_FIRST_CONNECTION state ";
const auto &options = WdtOptions::get();
auto &socket = data.socket_;
auto &threadStats = data.threadStats_;
data.reset();
socket.closeCurrentConnection();
auto timeout = options.accept_timeout_millis;
int acceptAttempts = 0;
while (true) {
{
std::lock_guard<std::mutex> lock(mutex_);
if (hasNewSessionStarted(data)) {
startNewThreadSession(data);
return ACCEPT_WITH_TIMEOUT;
}
}
if (isJoinable_ && acceptAttempts == options.max_accept_retries) {
LOG(ERROR) << "unable to accept after " << acceptAttempts << " attempts";
threadStats.setErrorCode(CONN_ERROR);
incrFailedThreadCountAndCheckForSessionEnd(data);
return FAILED;
}
if (getCurAbortCode() != OK) {
LOG(ERROR) << "Thread marked to abort while trying to accept first"
<< " connection. Num attempts " << acceptAttempts;
// Even though there is a transition FAILED here
// getCurAbortCode() is going to be checked again in the receiveOne.
// So this is pretty much irrelavant
return FAILED;
}
ErrorCode code = socket.acceptNextConnection(timeout);
if (code == OK) {
break;
}
acceptAttempts++;
}
std::lock_guard<std::mutex> lock(mutex_);
if (!hasNewSessionStarted(data)) {
// this thread has the first connection
startNewGlobalSession(data);
}
startNewThreadSession(data);
return READ_NEXT_CMD;
}
