/* | |
* Licensed to the Apache Software Foundation (ASF) under one or more | |
* contributor license agreements. See the NOTICE file distributed with | |
* this work for additional information regarding copyright ownership. | |
* The ASF licenses this file to You under the Apache License, Version 2.0 | |
* (the "License"); you may not use this file except in compliance with | |
* the License. You may obtain a copy of the License at | |
* | |
* http://www.apache.org/licenses/LICENSE-2.0 | |
* | |
* Unless required by applicable law or agreed to in writing, software | |
* distributed under the License is distributed on an "AS IS" BASIS, | |
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
* See the License for the specific language governing permissions and | |
* limitations under the License. | |
*/ | |
#include "CountDownLatch.h" | |
#include <decaf/lang/Integer.h> | |
#include <decaf/lang/exceptions/IllegalArgumentException.h> | |
#include <decaf/util/concurrent/TimeUnit.h> | |
#include <decaf/util/concurrent/locks/AbstractQueuedSynchronizer.h> | |
using namespace decaf; | |
using namespace decaf::lang; | |
using namespace decaf::lang::exceptions; | |
using namespace decaf::util; | |
using namespace decaf::util::concurrent; | |
using namespace decaf::util::concurrent::locks; | |
//////////////////////////////////////////////////////////////////////////////// | |
namespace decaf{ | |
namespace util{ | |
namespace concurrent{ | |
class LatchSync : public AbstractQueuedSynchronizer { | |
private: | |
LatchSync(const LatchSync&); | |
LatchSync& operator= (const LatchSync&); | |
public: | |
LatchSync(int count) : AbstractQueuedSynchronizer() { | |
this->setState(count); | |
} | |
virtual ~LatchSync() {} | |
int getCount() const { | |
return getState(); | |
} | |
protected: | |
virtual int tryAcquireShared(int acquires DECAF_UNUSED) { | |
return getState() == 0 ? 1 : -1; | |
} | |
virtual bool tryReleaseShared(int releases DECAF_UNUSED) { | |
for (;;) { | |
int current = getState(); | |
if (current == 0) { | |
return false; | |
} | |
int next = current - 1; | |
if (compareAndSetState(current, next)) { | |
return next == 0; | |
} | |
} | |
} | |
}; | |
}}} | |
//////////////////////////////////////////////////////////////////////////////// | |
CountDownLatch::CountDownLatch(int count) : sync(NULL) { | |
if (count < 0) { | |
throw IllegalArgumentException(__FILE__, __LINE__, "Count must be non-negative."); | |
} | |
this->sync = new LatchSync(count); | |
} | |
//////////////////////////////////////////////////////////////////////////////// | |
CountDownLatch::~CountDownLatch() { | |
try { | |
delete sync; | |
} | |
DECAF_CATCHALL_NOTHROW() | |
} | |
//////////////////////////////////////////////////////////////////////////////// | |
void CountDownLatch::await() { | |
try { | |
this->sync->acquireSharedInterruptibly(1); | |
} | |
DECAF_CATCH_RETHROW( decaf::lang::exceptions::InterruptedException ) | |
DECAF_CATCH_RETHROW( decaf::lang::Exception ) | |
DECAF_CATCHALL_THROW( decaf::lang::Exception ) | |
} | |
//////////////////////////////////////////////////////////////////////////////// | |
bool CountDownLatch::await( long long timeOut ) { | |
try { | |
return this->sync->tryAcquireSharedNanos(1, TimeUnit::MILLISECONDS.toNanos(timeOut)); | |
} | |
DECAF_CATCH_RETHROW( decaf::lang::exceptions::InterruptedException ) | |
DECAF_CATCH_RETHROW( decaf::lang::Exception ) | |
DECAF_CATCHALL_THROW( decaf::lang::Exception ) | |
} | |
//////////////////////////////////////////////////////////////////////////////// | |
bool CountDownLatch::await( long long timeout, const TimeUnit& unit ) { | |
try{ | |
return this->sync->tryAcquireSharedNanos(1, unit.toNanos(timeout)); | |
} | |
DECAF_CATCH_RETHROW( decaf::lang::exceptions::InterruptedException ) | |
DECAF_CATCH_RETHROW( decaf::lang::Exception ) | |
DECAF_CATCHALL_THROW( decaf::lang::Exception ) | |
} | |
//////////////////////////////////////////////////////////////////////////////// | |
void CountDownLatch::countDown() { | |
try { | |
this->sync->releaseShared(1); | |
} | |
DECAF_CATCHALL_NOTHROW() | |
} | |
//////////////////////////////////////////////////////////////////////////////// | |
int CountDownLatch::getCount() const { | |
return this->sync->getCount(); | |
} | |
//////////////////////////////////////////////////////////////////////////////// | |
std::string CountDownLatch::toString() const { | |
return std::string("CountDownLatch[count = ") + | |
Integer::toString(this->sync->getCount()) + "]"; | |
} |