mirror of https://gitee.com/bigwinds/arangodb
Merge branch 'sharding' of ssh://github.com/triAGENS/ArangoDB into sharding
This commit is contained in:
Max Neunhoeffer
commit
975f6b19fa
|
|
@ -39,6 +39,59 @@ using namespace triagens::rest;
|
|||
namespace triagens {
|
||||
namespace rest {
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
// --SECTION-- class AsyncCallbackContext
|
||||
// -----------------------------------------------------------------------------
|
||||
|
||||
class AsyncCallbackContext {
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
// --SECTION-- constructors / destructors
|
||||
// -----------------------------------------------------------------------------
|
||||
|
||||
public:
|
||||
AsyncCallbackContext (std::string const& url)
|
||||
: _url(url),
|
||||
_response(0) {
|
||||
|
||||
std::cout << "generated async context " << _url << std::endl;
|
||||
}
|
||||
|
||||
~AsyncCallbackContext () {
|
||||
if (_response != 0) {
|
||||
delete _response;
|
||||
}
|
||||
}
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
// --SECTION-- public functions
|
||||
// -----------------------------------------------------------------------------
|
||||
|
||||
public:
|
||||
|
||||
bool callback (HttpResponse* response) {
|
||||
if (response == 0) {
|
||||
return false;
|
||||
}
|
||||
|
||||
_response = response;
|
||||
|
||||
std::cout << "callback called for async context " << _url << ", BODY: " << _response->body().c_str() << std::endl;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
// --SECTION-- private variables
|
||||
// -----------------------------------------------------------------------------
|
||||
|
||||
private:
|
||||
|
||||
std::string _url;
|
||||
|
||||
HttpResponse* _response;
|
||||
};
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
// --SECTION-- class AsyncJobResult
|
||||
// -----------------------------------------------------------------------------
|
||||
|
|
@ -49,11 +102,6 @@ namespace triagens {
|
|||
// --SECTION-- public types
|
||||
// -----------------------------------------------------------------------------
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @addtogroup HttpServer
|
||||
/// @{
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
public:
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
|
@ -73,19 +121,10 @@ namespace triagens {
|
|||
|
||||
typedef uint64_t IdType;
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @}
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
// --SECTION-- constructors and destructors
|
||||
// -----------------------------------------------------------------------------
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @addtogroup HttpServer
|
||||
/// @{
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief constructor for an unspecified job result
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
|
@ -94,7 +133,8 @@ namespace triagens {
|
|||
_jobId(0),
|
||||
_response(0),
|
||||
_stamp(0.0),
|
||||
_status(JOB_UNDEFINED) {
|
||||
_status(JOB_UNDEFINED),
|
||||
_ctx(0) {
|
||||
}
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
|
@ -104,11 +144,13 @@ namespace triagens {
|
|||
AsyncJobResult (IdType jobId,
|
||||
HttpResponse* response,
|
||||
double stamp,
|
||||
Status status) :
|
||||
Status status,
|
||||
AsyncCallbackContext* ctx) :
|
||||
_jobId(jobId),
|
||||
_response(response),
|
||||
_stamp(stamp),
|
||||
_status(status) {
|
||||
_status(status),
|
||||
_ctx(ctx) {
|
||||
}
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
|
@ -118,19 +160,10 @@ namespace triagens {
|
|||
~AsyncJobResult () {
|
||||
}
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @}
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
// --SECTION-- public variables
|
||||
// -----------------------------------------------------------------------------
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @addtogroup HttpServer
|
||||
/// @{
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief job id
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
|
@ -156,9 +189,11 @@ namespace triagens {
|
|||
Status _status;
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @}
|
||||
/// @brief callback context object (normally 0, used in cluster operations)
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
AsyncCallbackContext* _ctx;
|
||||
|
||||
};
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
|
|
@ -175,11 +210,6 @@ namespace triagens {
|
|||
// --SECTION-- constructors / destructors
|
||||
// -----------------------------------------------------------------------------
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @addtogroup HttpServer
|
||||
/// @{
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
public:
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
|
@ -199,19 +229,10 @@ namespace triagens {
|
|||
~AsyncJobManager () {
|
||||
}
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @}
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
// --SECTION-- public types
|
||||
// -----------------------------------------------------------------------------
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @addtogroup HttpServer
|
||||
/// @{
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
public:
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
|
@ -220,19 +241,10 @@ namespace triagens {
|
|||
|
||||
typedef std::map<AsyncJobResult::IdType, AsyncJobResult> JobList;
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @}
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
// --SECTION-- public methods
|
||||
// -----------------------------------------------------------------------------
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @addtogroup HttpServer
|
||||
/// @{
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief get the result of an async job, and remove it from the list
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
|
@ -393,7 +405,16 @@ namespace triagens {
|
|||
*jobId = (AsyncJobResult::IdType) generate();
|
||||
job->assignId((uint64_t) *jobId);
|
||||
|
||||
AsyncJobResult ajr(*jobId, 0, TRI_microtime(), AsyncJobResult::JOB_PENDING);
|
||||
AsyncCallbackContext* ctx = 0;
|
||||
|
||||
bool found;
|
||||
char const* hdr = job->getHandler()->getRequest()->header("x-arango-coordinator", found);
|
||||
|
||||
if (found) {
|
||||
ctx = new AsyncCallbackContext(std::string(hdr));
|
||||
}
|
||||
|
||||
AsyncJobResult ajr(*jobId, 0, TRI_microtime(), AsyncJobResult::JOB_PENDING, ctx);
|
||||
|
||||
WRITE_LOCKER(_lock);
|
||||
|
||||
|
|
@ -418,7 +439,10 @@ namespace triagens {
|
|||
}
|
||||
|
||||
const double now = TRI_microtime();
|
||||
AsyncCallbackContext* ctx = 0;
|
||||
HttpResponse* response = 0;
|
||||
|
||||
{
|
||||
WRITE_LOCKER(_lock);
|
||||
JobList::iterator it = _jobs.find(jobId);
|
||||
|
||||
|
|
@ -428,25 +452,32 @@ namespace triagens {
|
|||
// will also dispose the response
|
||||
}
|
||||
else {
|
||||
(*it).second._response = handler->stealResponse();
|
||||
response = handler->stealResponse();
|
||||
|
||||
(*it).second._response = response;
|
||||
(*it).second._status = AsyncJobResult::JOB_DONE;
|
||||
(*it).second._stamp = now;
|
||||
|
||||
ctx = (*it).second._ctx;
|
||||
|
||||
if (ctx != 0) {
|
||||
// we have found a context object, so we can immediately remove the job
|
||||
// from the list of "done" jobs
|
||||
_jobs.erase(it);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @}
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
// if callback is set, execute it now (outside of the wr-lock)
|
||||
if (ctx != 0 && response != 0) {
|
||||
ctx->callback(response);
|
||||
}
|
||||
}
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
// --SECTION-- private variables
|
||||
// -----------------------------------------------------------------------------
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @addtogroup HttpServer
|
||||
/// @{
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
private:
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
|
@ -467,10 +498,6 @@ namespace triagens {
|
|||
|
||||
uint64_t (*generate)();
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @}
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
};
|
||||
|
||||
}
|
||||
|
|
|
|||
|
|
@ -141,6 +141,14 @@ namespace triagens {
|
|||
_server = server;
|
||||
}
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// @brief return a pointer to the request
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
HttpRequest const* getRequest () const {
|
||||
return _request;
|
||||
}
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
/// {@inheritDoc}
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
|
|
|||
Loading…
Reference in New Issue