AOSContext *AOSContextManager::allocate(AFile_Socket *pSocket)
{
AASSERT(this, pSocket);
AOSContext *p = _newContext(pSocket);
//a_Set logging level
switch(m_DefaultEventLogLevel)
{
case AEventVisitor::EL_DEBUG: p->useEventVisitor().setEventThresholdLevel(AEventVisitor::EL_DEBUG); break;
case AEventVisitor::EL_INFO: p->useEventVisitor().setEventThresholdLevel(AEventVisitor::EL_INFO); break;
case AEventVisitor::EL_WARN: p->useEventVisitor().setEventThresholdLevel(AEventVisitor::EL_WARN); break;
case AEventVisitor::EL_ERROR: p->useEventVisitor().setEventThresholdLevel(AEventVisitor::EL_ERROR); break;
case AEventVisitor::EL_NONE: p->useEventVisitor().setEventThresholdLevel(AEventVisitor::EL_NONE); break;
default:
p->useEventVisitor().setEventThresholdLevel(AEventVisitor::EL_EVENT);
}
ARope rope("AOSContextManager::allocate[",28);
rope.append(AString::fromPointer(p));
rope.append(", pFile=",8);
rope.append(AString::fromPointer(pSocket));
rope.append("] new create",12);
p->useEventVisitor().startEvent(rope);
{
ALock lock(m_InUseSync);
AASSERT(this, p && m_InUse.end() == m_InUse.find(p));
m_InUse.insert(p);
}
AASSERT(this, &p->useSocket());
return p;
}
u4 AOSContextQueue_IsAvailable::_threadprocWorker(AThread& thread)
{
AOSContextQueue_IsAvailable::QueueWithThread *pThis = dynamic_cast<AOSContextQueue_IsAvailable::QueueWithThread *>(thread.getThis());
AOSContextQueue_IsAvailable *pOwner = dynamic_cast<AOSContextQueue_IsAvailable *>(thread.getParameter());
AASSERT(&thread, pThis);
AASSERT(&thread, pOwner);
timeval timeout;
timeout.tv_sec = 0;
timeout.tv_usec = 0;
thread.setRunning(true);
fd_set sockSet;
while (thread.isRun())
{
try
{
if (pThis->queue.size() > 0)
{
FD_ZERO(&sockSet);
int count = 0;
// Convert to ABasePtrQueue
ABase *p = pThis->queue.useHead();
while (count < FD_SETSIZE && p)
{
AOSContext *pContext = (AOSContext *)p;
FD_SET((SOCKET)pContext->useSocket().getSocketInfo().m_handle, &sockSet);
++count;
p = p->useNext();
}
if (count > 0)
{
int availCount = ::select(count, &sockSet, NULL, NULL, &timeout);
if (availCount > 0)
{
int count2 = 0;
p = pThis->queue.useHead();
while (availCount > 0 && count2 <= count && p)
{
AOSContext *pContext = (AOSContext *)p;
p = p->useNext();
if (FD_ISSET(pContext->useSocket().getSocketInfo().m_handle, &sockSet))
{
pContext->useConnectionFlags().setBit(AOSContext::CONFLAG_ISAVAILABLE_SELECT_SET);
//a_Attempt to read some data to check if it actually has data, if not then connection is closed
size_t bytesRead = AConstant::npos;
try
{
bytesRead = pContext->useSocket().readBlockIntoLookahead();
if (!bytesRead || AConstant::npos == bytesRead)
{
//a_No data to read, socket is closed
pContext->useConnectionFlags().setBit(AOSContext::CONFLAG_IS_SOCKET_CLOSED);
pContext->useEventVisitor().startEvent(ASW("AOSContextQueue_IsAvailable: Detected closed remote socket",58), AEventVisitor::EL_WARN);
pContext->useSocket().close();
{
ALock lock(pThis->sync);
pThis->queue.remove(pContext);
}
pOwner->m_Services.useContextManager().changeQueueState(AOSContextManager::STATE_TERMINATE, &pContext);
}
else if (AConstant::unavail == bytesRead)
{
// Would block, keep waiting
if (pContext->useTimeoutTimer().getInterval() > pOwner->m_NoDataTimeout)
{
AString str;
str.append("AOSContextQueue_IsAvailable: No data (non-blocking) after timeout: ",67);
pContext->useTimeoutTimer().emit(str);
//a_We are done with this request, still no data
pContext->useEventVisitor().startEvent(
str,
(pContext->useConnectionFlags().isSet(AOSContext::CONFLAG_IS_HTTP11_PIPELINING) ? AEventVisitor::EL_EVENT : AEventVisitor::EL_ERROR)
);
{
ALock lock(pThis->sync);
pThis->queue.remove(pContext);
}
pOwner->m_Services.useContextManager().changeQueueState(AOSContextManager::STATE_TERMINATE, &pContext);
}
}
else
{
//a_Add to next queue
pContext->useConnectionFlags().setBit(AOSContext::CONFLAG_ISAVAILABLE_PENDING);
if (pContext->useEventVisitor().isLogging(AEventVisitor::EL_DEBUG))
pContext->useEventVisitor().startEvent(ASW("AOSContextQueue_IsAvailable: HTTP header has more data",54), AEventVisitor::EL_DEBUG);
{
ALock lock(pThis->sync);
pThis->queue.remove(pContext);
}
pOwner->m_Services.useContextManager().changeQueueState(AOSContextManager::STATE_PRE_EXECUTE, &pContext);
}
}
catch(AException e)
{
bytesRead = AConstant::npos; //a_Following switch will handle this
ARope rope("AOSContextQueue_IsAvailable: Exception reading from socket",58);
rope.append(e);
pContext->useEventVisitor().addEvent(rope, AEventVisitor::EL_ERROR);
}
--availCount;
}
else
{
//a_Select called on this AOSContext and no data available
if (pContext->useTimeoutTimer().getInterval() > pOwner->m_NoDataTimeout)
{
AString str;
str.append("AOSContextQueue_IsAvailable: No data after timeout: ",52);
pContext->useTimeoutTimer().emit(str);
//a_We are done with this request, still no data
pContext->useEventVisitor().startEvent(
str,
(pContext->useConnectionFlags().isSet(AOSContext::CONFLAG_IS_HTTP11_PIPELINING) ? AEventVisitor::EL_EVENT : AEventVisitor::EL_ERROR)
);
{
ALock lock(pThis->sync);
pThis->queue.remove(pContext);
}
pOwner->m_Services.useContextManager().changeQueueState(AOSContextManager::STATE_TERMINATE, &pContext);
}
}
++count2;
}
AASSERT(pOwner, !availCount);
}
else
{
//a_Nothing has data, flag them all
p = pThis->queue.useHead();
while (p)
{
AOSContext *pContext = (AOSContext *)p;
p = p->useNext();
if (pContext->useTimeoutTimer().getInterval() > pOwner->m_NoDataTimeout)
{
AString str("AOSContextQueue_IsAvailable: No data after timeout: ",52);
pContext->useTimeoutTimer().emit(str);
//a_We are done with this request, still no data
ALock lock(pThis->sync);
pContext->useEventVisitor().startEvent(
str,
(pContext->useConnectionFlags().isSet(AOSContext::CONFLAG_IS_HTTP11_PIPELINING) ? AEventVisitor::EL_EVENT : AEventVisitor::EL_ERROR)
);
pThis->queue.remove(pContext);
pOwner->m_Services.useContextManager().changeQueueState(AOSContextManager::STATE_TERMINATE, &pContext);
}
}
}
}
//a_Sleep between selects, requests in this queue are already slow
AThread::sleep(pOwner->m_LoopDelay);
}
else
AThread::sleep(pOwner->m_EmptyQueueDelay); //a_Nothing in queue, deep sleep
}
catch(AException& e)
{
pOwner->m_Services.useLog().addException(e);
}
catch(std::exception& e)
{
pOwner->m_Services.useLog().addException(e);
}
catch(...)
{
pOwner->m_Services.useLog().add(ASWNL("Unknown exception caught in AOSContextQueue_IsAvailable::threadproc"), ALog::EVENT_EXCEPTION);
}
}
thread.setRunning(false);
return 0;
}
u4 AOSContextQueue_ErrorExecutor::_threadproc(AThread& thread)
{
AOSContextQueue_ErrorExecutor *pThis = dynamic_cast<AOSContextQueue_ErrorExecutor *>(thread.getThis());
AASSERT(&thread, pThis);
AOSContext *pContext = NULL;
thread.setRunning(true);
while(thread.isRun())
{
try
{
while (pContext = pThis->_nextContext())
{
#ifndef NDEBUG
if (!ADebugDumpable::isPointerValid(pContext))
{
AString error("AOSContext pointer is invalid: ");
error.append(AString::fromPointer(pContext));
AASSERT_EX(NULL, false, error);
continue;
}
#endif
pContext->useEventVisitor().startEvent(ASW("AOSContextQueue_ErrorExecutor: Processing error condition", 57));
//a_Should only be here if an error occured, if status not set >200, then assume 500
if (pContext->useResponseHeader().getStatusCode() == AHTTPResponseHeader::SC_200_Ok)
pContext->useResponseHeader().setStatusCode(AHTTPResponseHeader::SC_500_Internal_Server_Error);
//a_Check is socket was closed, if so do nothing else, we are done
if (pContext->useConnectionFlags().isSet(AOSContext::CONFLAG_IS_SOCKET_ERROR))
{
//a_Proceed
m_Services.useContextManager().changeQueueState(AOSContextManager::STATE_TERMINATE, &pContext);
continue;
}
if (
pContext->useContextFlags().isSet(AOSContext::CTXFLAG_IS_RESPONSE_HEADER_SENT)
|| pContext->useContextFlags().isSet(AOSContext::CTXFLAG_IS_OUTPUT_SENT)
)
{
//a_Log to file, output is done already
m_Services.useLog().add(pContext->useEventVisitor(), ALog::EVENT_FAILURE);
}
else
{
//a_Add XSLT stylesheet for the error XML
//if (m_Services.useConfiguration().exists(ASW("/config/server/error-stylesheet",31)))
//{
// AString errorStylesheet;
// m_Services.useConfiguration().emitString(ASW("/config/server/error-stylesheet",31), errorStylesheet);
// pContext->useModelXmlDocument().addInstruction(AXmlInstruction::XML_STYLESHEET)
// .addAttribute(ASW("type",4), ASW("text/xsl",8))
// .addAttribute(ASW("href",4), errorStylesheet);
//}
//a_Add request header to result XML
if (!pContext->useModel().exists(ASW("REQUEST",7)))
pContext->useRequestHeader().emitXml(pContext->useModel().overwriteElement(ASW("REQUEST",7)));
pContext->useResponseHeader().emitXml(pContext->useModel().overwriteElement(ASW("RESPONSE",8)));
//a_Check if dumpContext is specified to override and emit XML
int dumpContextLevel = pContext->getDumpContextLevel();
pContext->dumpContext(dumpContextLevel);
if (dumpContextLevel > 0)
{
//a_Write contents of the output XML instead of output buffer
m_Services.useConfiguration().setMimeTypeFromExt(ASW("xml",3), *pContext);
pContext->useResponseHeader().setStatusCode(AHTTPResponseHeader::SC_200_Ok);
pContext->writeOutputBuffer(true);
}
else
{
//a_Set the current content type as text/html
pContext->useResponseHeader().set(AHTTPHeader::HT_ENT_Content_Type, ASW("text/html; charset=utf-8",24));
int statusCode = pContext->useResponseHeader().getStatusCode();
AAutoPtr<ATemplate> pTemplate(NULL, false); //a_Call to cache manager will set a template
pContext->clearOutputBuffer();
if (m_Services.useCacheManager().getStatusTemplate(statusCode, pTemplate))
{
//a_Template for this status code is found, so process and emit into output buffer
pTemplate->process(pContext->useLuaTemplateContext(), pContext->useOutputBuffer());
if (pContext->useEventVisitor().isLogging(AEventVisitor::EL_DEBUG))
{
ARope rope("Using error template for status ");
rope.append(AString::fromInt(statusCode));
pContext->useEventVisitor().startEvent(rope, AEventVisitor::EL_DEBUG);
}
}
else
{
if (pContext->useEventVisitor().isLogging(AEventVisitor::EL_WARN))
{
ARope rope("Did not find error template for status ");
rope.append(AString::fromInt(statusCode));
pContext->useEventVisitor().startEvent(rope, AEventVisitor::EL_WARN);
}
}
if (pContext->isOutputBufferEmpty())
{
AString strError(1024, 256);
strError.assign("Error ",6);
strError.append(AString::fromInt(pContext->useResponseHeader().getStatusCode()));
strError.append(": ", 2);
strError.append(pContext->useResponseHeader().getStatusCodeReasonPhrase(pContext->useResponseHeader().getStatusCode()));
//a_Put some generic stuff since there is no error template
pContext->useOutputBuffer().append("<html><head><title>",19);
pContext->useOutputBuffer().append(strError);
pContext->useOutputBuffer().append("</title></head>",15);
pContext->useOutputBuffer().append("<body>",6);
pContext->useOutputBuffer().append(strError);
pContext->useOutputBuffer().append("</body></html>",14);
}
try
{
pContext->writeOutputBuffer();
}
catch(ASocketException& ex)
{
pContext->useEventVisitor().addEvent(ex, AEventVisitor::EL_ERROR);
pContext->useConnectionFlags().setBit(AOSContext::CONFLAG_IS_SOCKET_ERROR);
m_Services.useContextManager().changeQueueState(AOSContextManager::STATE_TERMINATE, &pContext);
continue;
}
}
}
//a_Close connection
pContext->useSocket().close();
//a_Proceed
m_Services.useContextManager().changeQueueState(AOSContextManager::STATE_TERMINATE, &pContext);
continue;
}
AThread::sleep(pThis->m_SleepDelay); //a_Empty queue, avoid thrashing
}
catch(AException& e)
{
pContext->useEventVisitor().addEvent(e, AEventVisitor::EL_ERROR);
m_Services.useLog().add(pContext->useEventVisitor(), ALog::EVENT_FAILURE);
m_Services.useContextManager().changeQueueState(AOSContextManager::STATE_TERMINATE, &pContext);
}
catch(std::exception& e)
{
pContext->useEventVisitor().addEvent(ASWNL(e.what()), AEventVisitor::EL_ERROR);
m_Services.useLog().add(pContext->useEventVisitor(), ALog::EVENT_FAILURE);
m_Services.useContextManager().changeQueueState(AOSContextManager::STATE_TERMINATE, &pContext);
}
catch(...)
{
m_Services.useLog().add(pContext->useEventVisitor(), ALog::EVENT_FAILURE);
pContext->useEventVisitor().addEvent(ASWNL("Unknown exception caught in AOSContextQueue_ErrorExecutor::_threadproc"), AEventVisitor::EL_ERROR);
m_Services.useContextManager().changeQueueState(AOSContextManager::STATE_TERMINATE, &pContext);
break;
}
}
thread.setRunning(false);
return 0;
}