dockerfile/openalpr-base
1
0
Fork 0
mirror of https://github.com/kerberos-io/openalpr-base.git synced 2025-10-07 04:40:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added exception handling around Beanstalk

Browse Source
This commit is contained in:
Matt Hill
2014-07-02 19:25:07 -04:00
parent e0377c1d4d
commit 1608886cba
1 changed files with 54 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
src/daemon.cpp
View File

@@ -274,20 +274,26 @@ void streamRecognitionThread(void* arg)
bool writeToQueue(std::string jsonResult) bool writeToQueue(std::string jsonResult)
{ {
try
Beanstalk::Client client(BEANSTALK_QUEUE_HOST, BEANSTALK_PORT);
client.use(BEANSTALK_TUBE_NAME);
int id = client.put(jsonResult);
if (id <= 0)
{ {
LOG4CPLUS_ERROR(logger, "Failed to write data to queue"); Beanstalk::Client client(BEANSTALK_QUEUE_HOST, BEANSTALK_PORT);
return false; client.use(BEANSTALK_TUBE_NAME);
}
LOG4CPLUS_INFO(logger, "put job id: " << id );
int id = client.put(jsonResult);
if (id <= 0)
{
LOG4CPLUS_ERROR(logger, "Failed to write data to queue");
return false;
}
LOG4CPLUS_INFO(logger, "put job id: " << id );
}
catch (const std::runtime_error& error)
{
LOG4CPLUS_WARN(logger, "Error writing to Beanstalk. Result has not been saved.");
}
return true; return true;
} }
@@ -302,33 +308,49 @@ void dataUploadThread(void* arg)
UploadThreadData* udata = (UploadThreadData*) arg; UploadThreadData* udata = (UploadThreadData*) arg;
Beanstalk::Client client(BEANSTALK_QUEUE_HOST, BEANSTALK_PORT);
client.watch(BEANSTALK_TUBE_NAME);
while(daemon_active) while(daemon_active)
{ {
Beanstalk::Job job; try
client.reserve(job);
if (job.id() > 0)
{ {
LOG4CPLUS_DEBUG(logger, job.body() ); Beanstalk::Client client(BEANSTALK_QUEUE_HOST, BEANSTALK_PORT);
if (uploadPost(udata->upload_url, job.body()))
{ client.watch(BEANSTALK_TUBE_NAME);
client.del(job.id());
LOG4CPLUS_INFO(logger, "Job: " << job.id() << " successfully uploaded" );
}
else
{
client.release(job);
LOG4CPLUS_WARN(logger, "Job: " << job.id() << " failed to upload. Will retry." );
}
}
usleep(10000); while (daemon_active)
{
Beanstalk::Job job;
client.reserve(job);
if (job.id() > 0)
{
LOG4CPLUS_DEBUG(logger, job.body() );
if (uploadPost(udata->upload_url, job.body()))
{
client.del(job.id());
LOG4CPLUS_INFO(logger, "Job: " << job.id() << " successfully uploaded" );
}
else
{
client.release(job);
LOG4CPLUS_WARN(logger, "Job: " << job.id() << " failed to upload. Will retry." );
}
}
// Wait 10ms
usleep(10000);
}
}
catch (const std::runtime_error& error)
{
LOG4CPLUS_WARN(logger, "Error connecting to Beanstalk. Will retry." );
}
// wait 5 seconds
usleep(5000000);
} }
curl_global_cleanup(); curl_global_cleanup();