1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96
|
/****************************************************************************
* Copyright (C) 2009-2015 EPAM Systems
*
* This file is part of Indigo toolkit.
*
* This file may be distributed and/or modified under the terms of the
* GNU General Public License version 3 as published by the Free Software
* Foundation and appearing in the file LICENSE.GPL included in the
* packaging of this file.
*
* This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
* WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
***************************************************************************/
#include "base_cpp/cancellation_handler.h"
#include "base_c/nano.h"
#include "base_cpp/output.h"
#include "base_cpp/tlscont.h"
namespace indigo
{
//
// TimeoutCancellationHandler
//
TimeoutCancellationHandler::TimeoutCancellationHandler (int mseconds)
{
reset(mseconds);
}
TimeoutCancellationHandler::~TimeoutCancellationHandler ()
{
}
bool TimeoutCancellationHandler::isCancelled ()
{
qword dif_time = nanoClock() - _currentTime;
if (_mseconds > 0 && nanoHowManySeconds(dif_time) * 1000 > _mseconds)
{
ArrayOutput mes_out(_message);
mes_out.printf("The operation timed out: %d ms", _mseconds);
mes_out.writeChar(0);
return true;
}
return false;
}
const char* TimeoutCancellationHandler::cancelledRequestMessage ()
{
return _message.ptr();
}
void TimeoutCancellationHandler::reset (int mseconds)
{
_mseconds = mseconds;
_currentTime = nanoClock();
}
//
// Global thread-local cancellation handler
//
class CancellationHandlerWrapper
{
public:
CancellationHandlerWrapper () : handler(0) {}
CancellationHandler* handler;
};
static _SessionLocalContainer<CancellationHandlerWrapper> cancellation_handler;
CancellationHandler* getCancellationHandler ()
{
CancellationHandlerWrapper &wrapper = cancellation_handler.getLocalCopy();
return wrapper.handler;
}
CancellationHandler* setCancellationHandler (CancellationHandler* handler)
{
CancellationHandlerWrapper &wrapper = cancellation_handler.getLocalCopy();
CancellationHandler* prev = wrapper.handler;
wrapper.handler = handler;
return prev;
}
AutoCancellationHandler::AutoCancellationHandler(CancellationHandler& hand) {
_prev = setCancellationHandler(&hand);
}
AutoCancellationHandler::~AutoCancellationHandler() {
setCancellationHandler(_prev);
}
}
|