void readCB(bool success) {
if(unlikely(!success)) {
destruct();
return;
}
//if((sp=thr._stringPoolPool.tryGet())==nullptr) sp=new StringPool();
if((resp=thr._responsePool.tryGet())) resp->init(this->s,&sp);
else resp=new Response(this->s,&sp);
//keepAlive=true;
auto it=req.headers.find("connection");
if(it!=req.headers.end() && (*it).value=="close")keepAlive=false;
else keepAlive=true;
resp->headers.insert({"Connection", keepAlive?"keep-alive":"close"});
/*char* date=sp.beginAdd(32);
tm time;
gmtime_r(&thr.curClockTime.tv_sec,&time);
int l=rfctime(time,date);
if(l>32)l=32;
sp.endAdd(l);
*/
resp->headers.insert({"Date", sp.addString(thr.curRFCTime)});
thr.performanceCounters.totalRequestsReceived++;
try {
thr.handleRequest(req,*resp,{&handler::finalize,this});
} catch(exception& ex) {
thr.handleError(req,*resp,ex,{&handler::finalize,this});
}
}
void handleStatic(staticPage* Sp) {
Response& resp(*this->resp);
(_staticPage=Sp)->retain();
try {
int bufferL = resp.buffer.length();
if(Sp->mime.length()>0)resp.headers["Content-Type"]=Sp->mime;
{
char* tmps = sp.beginAdd(22);
int l = itoa64(Sp->fileLen, tmps);
sp.endAdd(l);
resp.headers.insert({"Content-Length", { tmps, l }});
StreamWriter sw(resp.buffer);
resp.serializeHeaders(sw);
}
if(Sp->fileLen>=CPPSP_SENDFILE_MIN_SIZE) {
_sendFileOffset=0;
s.sendAll(resp.buffer.data()+bufferL,resp.buffer.length()-bufferL,
MSG_MORE, { &handler::sendHeadersCB, this });
} else {
String data=Sp->data;
iov[0]= {resp.buffer.data()+bufferL, (size_t)(resp.buffer.length()-bufferL)};
iov[1]= {data.data(), (size_t)data.length()};
resp.outputStream->writevAll(iov, data.length()<=0?1:2, { &handler::writevCB, this });
}
} catch(exception& ex) {
Sp->release();
server->handleError(req,resp,ex,{&handler::finalize,this});
}
}
//deallocate resources after a request has been completed
void cleanup() {
server->performanceCounters.totalRequestsFinished++;
thr.performanceCounters.totalRequestsFinished++;
req.reset();
resp->reset();
thr._responsePool.put(resp);
resp=nullptr;
sp.clear();
}
void finalize() {
thr.performanceCounters.totalRequestsFinished++;
if(resp->closed) {
destruct(); return;
}
req.reset();
resp->reset();
thr._responsePool.put(resp);
resp=nullptr;
sp.clear();
if(keepAlive) {
req.init(s,&sp);
if(req.readRequest({&handler::readCB,this})) readCB(true);
} else {
s.shutdown(SHUT_WR);
buf=(uint8_t*)malloc(4096);
s.repeatRead(buf,4096,{&handler::sockReadCB,this});
}
}
void handleStatic(staticPage* Sp) {
Response& resp(*this->resp);
try {
String data=Sp->data;
int bufferL = resp.buffer.length();
if(Sp->mime.length()>0)resp.headers["Content-Type"]=Sp->mime;
{
char* tmps = sp.beginAdd(16);
int l = itoa(data.length(), tmps);
sp.endAdd(l);
resp.headers.insert({"Content-Length", { tmps, l }});
StreamWriter sw(resp.buffer);
resp.serializeHeaders(sw);
}
iov[0]= {resp.buffer.data()+bufferL, (size_t)(resp.buffer.length()-bufferL)};
iov[1]= {data.data(), (size_t)data.length()};
resp.outputStream->writevAll(iov, 2, { &handler::writevCB, this });
} catch(exception& ex) {
thr.handleError(req,resp,ex,{&handler::finalize,this});
}
}