| /* |
| * 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. |
| */ |
| |
| /** |
| * @file |
| * Declares ignite::commom::SharedState class template. |
| */ |
| |
| #ifndef _IGNITE_SHARED_STATE |
| #define _IGNITE_SHARED_STATE |
| |
| #include <memory> |
| |
| #include <ignite/common/common.h> |
| #include <ignite/common/concurrent.h> |
| #include <ignite/common/cancelable.h> |
| #include <ignite/ignite_error.h> |
| |
| namespace ignite |
| { |
| namespace common |
| { |
| template<typename T> |
| class SharedState |
| { |
| public: |
| /** Template value type */ |
| typedef T ValueType; |
| |
| /** |
| * Default constructor. |
| * Constructs non-set SharedState instance. |
| */ |
| SharedState() : |
| value(), |
| error() |
| { |
| // No-op. |
| } |
| |
| /** |
| * Destructor. |
| */ |
| ~SharedState() |
| { |
| // No-op. |
| } |
| |
| /** |
| * Checks if the value or error set for the state. |
| * @return True if the value or error set for the state. |
| */ |
| bool IsSet() const |
| { |
| return value.get() || error.GetCode() != IgniteError::IGNITE_SUCCESS; |
| } |
| |
| /** |
| * Set value. |
| * |
| * @throw IgniteError with IgniteError::IGNITE_ERR_FUTURE_STATE if error or value has been set already. |
| * @param val Value to set. |
| */ |
| void SetValue(std::auto_ptr<ValueType> val) |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| if (IsSet()) |
| { |
| if (value.get()) |
| throw IgniteError(IgniteError::IGNITE_ERR_FUTURE_STATE, "Future value already set"); |
| |
| if (error.GetCode() != IgniteError::IGNITE_SUCCESS) |
| throw IgniteError(IgniteError::IGNITE_ERR_FUTURE_STATE, "Future error already set"); |
| } |
| |
| value = val; |
| |
| cond.NotifyAll(); |
| } |
| |
| /** |
| * Set error. |
| * |
| * @throw IgniteError with IgniteError::IGNITE_ERR_FUTURE_STATE if error or value has been set already. |
| * @param err Error to set. |
| */ |
| void SetError(const IgniteError& err) |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| if (IsSet()) |
| { |
| if (value.get()) |
| throw IgniteError(IgniteError::IGNITE_ERR_FUTURE_STATE, "Future value already set"); |
| |
| if (error.GetCode() != IgniteError::IGNITE_SUCCESS) |
| throw IgniteError(IgniteError::IGNITE_ERR_FUTURE_STATE, "Future error already set"); |
| } |
| |
| error = err; |
| |
| cond.NotifyAll(); |
| } |
| |
| /** |
| * Wait for value to be set. |
| * Active thread will be blocked until value or error will be set. |
| */ |
| void Wait() const |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| while (!IsSet()) |
| cond.Wait(mutex); |
| } |
| |
| /** |
| * Wait for value to be set for specified time. |
| * Active thread will be blocked until value or error will be set or timeout will end. |
| * |
| * @param msTimeout Timeout in milliseconds. |
| * @return True if the object has been triggered and false in case of timeout. |
| */ |
| bool WaitFor(int32_t msTimeout) const |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| if (IsSet()) |
| return true; |
| |
| return cond.WaitFor(mutex, msTimeout); |
| } |
| |
| /** |
| * Get the set value. |
| * Active thread will be blocked until value or error will be set. |
| * |
| * @throw IgniteError if error has been set. |
| * @return Value that has been set on success. |
| */ |
| const ValueType& GetValue() const |
| { |
| Wait(); |
| |
| if (value.get()) |
| return *value; |
| |
| assert(error.GetCode() != IgniteError::IGNITE_SUCCESS); |
| |
| throw error; |
| } |
| |
| /** |
| * Set cancel target. |
| */ |
| void SetCancelTarget(std::auto_ptr<Cancelable>& target) |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| cancelTarget = target; |
| } |
| |
| /** |
| * Cancel related operation. |
| */ |
| void Cancel() |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| if (cancelTarget.get()) |
| cancelTarget->Cancel(); |
| } |
| |
| private: |
| IGNITE_NO_COPY_ASSIGNMENT(SharedState); |
| |
| /** Cancel target. */ |
| std::auto_ptr<Cancelable> cancelTarget; |
| |
| /** Value. */ |
| std::auto_ptr<ValueType> value; |
| |
| /** Error. */ |
| IgniteError error; |
| |
| /** Condition variable which serves to signal that value is set. */ |
| mutable concurrent::ConditionVariable cond; |
| |
| /** Lock that used to prevent double-set of the value. */ |
| mutable concurrent::CriticalSection mutex; |
| }; |
| |
| /** |
| * Specialization for void type. |
| */ |
| template<> |
| class SharedState<void> |
| { |
| public: |
| /** Template value type */ |
| typedef void ValueType; |
| |
| /** |
| * Default constructor. |
| * Constructs non-set SharedState instance. |
| */ |
| SharedState() : |
| done(false), |
| error() |
| { |
| // No-op. |
| } |
| |
| /** |
| * Destructor. |
| */ |
| ~SharedState() |
| { |
| // No-op. |
| } |
| |
| /** |
| * Checks if the value or error set for the state. |
| * @return True if the value or error set for the state. |
| */ |
| bool IsSet() const |
| { |
| return done || error.GetCode() != IgniteError::IGNITE_SUCCESS; |
| } |
| |
| /** |
| * Set value. |
| * |
| * @throw IgniteError with IgniteError::IGNITE_ERR_FUTURE_STATE if error or value has been set already. |
| */ |
| void SetValue() |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| if (IsSet()) |
| { |
| if (done) |
| throw IgniteError(IgniteError::IGNITE_ERR_FUTURE_STATE, "Future value already set"); |
| |
| if (error.GetCode() != IgniteError::IGNITE_SUCCESS) |
| throw IgniteError(IgniteError::IGNITE_ERR_FUTURE_STATE, "Future error already set"); |
| } |
| |
| done = true; |
| |
| cond.NotifyAll(); |
| } |
| |
| /** |
| * Set error. |
| * |
| * @throw IgniteError with IgniteError::IGNITE_ERR_FUTURE_STATE if error or value has been set already. |
| * @param err Error to set. |
| */ |
| void SetError(const IgniteError& err) |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| if (IsSet()) |
| { |
| if (done) |
| throw IgniteError(IgniteError::IGNITE_ERR_FUTURE_STATE, "Future value already set"); |
| |
| if (error.GetCode() != IgniteError::IGNITE_SUCCESS) |
| throw IgniteError(IgniteError::IGNITE_ERR_FUTURE_STATE, "Future error already set"); |
| } |
| |
| error = err; |
| |
| cond.NotifyAll(); |
| } |
| |
| /** |
| * Wait for value to be set. |
| * Active thread will be blocked until value or error will be set. |
| */ |
| void Wait() const |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| while (!IsSet()) |
| cond.Wait(mutex); |
| } |
| |
| /** |
| * Wait for value to be set for specified time. |
| * Active thread will be blocked until value or error will be set or timeout will end. |
| * |
| * @param msTimeout Timeout in milliseconds. |
| * @return True if the object has been triggered and false in case of timeout. |
| */ |
| bool WaitFor(int32_t msTimeout) const |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| if (IsSet()) |
| return true; |
| |
| return cond.WaitFor(mutex, msTimeout); |
| } |
| |
| /** |
| * Get the set value. |
| * Active thread will be blocked until value or error will be set. |
| * |
| * @throw IgniteError if error has been set. |
| */ |
| void GetValue() const |
| { |
| Wait(); |
| |
| if (done) |
| return; |
| |
| assert(error.GetCode() != IgniteError::IGNITE_SUCCESS); |
| |
| throw error; |
| } |
| |
| /** |
| * Set cancel target. |
| */ |
| void SetCancelTarget(std::auto_ptr<Cancelable>& target) |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| cancelTarget = target; |
| } |
| |
| /** |
| * Cancel related operation. |
| */ |
| void Cancel() |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| if (cancelTarget.get()) |
| cancelTarget->Cancel(); |
| } |
| |
| private: |
| IGNITE_NO_COPY_ASSIGNMENT(SharedState); |
| |
| /** Cancel target. */ |
| std::auto_ptr<Cancelable> cancelTarget; |
| |
| /** Marker. */ |
| bool done; |
| |
| /** Error. */ |
| IgniteError error; |
| |
| /** Condition variable which serves to signal that value is set. */ |
| mutable concurrent::ConditionVariable cond; |
| |
| /** Lock that used to prevent double-set of the value. */ |
| mutable concurrent::CriticalSection mutex; |
| }; |
| |
| /** |
| * Specialization for shared pointer type. |
| */ |
| template<typename T> |
| class SharedState< concurrent::SharedPointer<T> > |
| { |
| public: |
| /** Template value type */ |
| typedef T ValueType; |
| |
| /** |
| * Default constructor. |
| * Constructs non-set SharedState instance. |
| */ |
| SharedState() : |
| value(), |
| error() |
| { |
| // No-op. |
| } |
| |
| /** |
| * Destructor. |
| */ |
| ~SharedState() |
| { |
| // No-op. |
| } |
| |
| /** |
| * Checks if the value or error set for the state. |
| * @return True if the value or error set for the state. |
| */ |
| bool IsSet() const |
| { |
| return value.IsValid() || error.GetCode() != IgniteError::IGNITE_SUCCESS; |
| } |
| |
| /** |
| * Set value. |
| * |
| * @throw IgniteError with IgniteError::IGNITE_ERR_FUTURE_STATE if error or value has been set already. |
| * @param val Value to set. |
| */ |
| void SetValue(const concurrent::SharedPointer<ValueType>& val) |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| if (IsSet()) |
| { |
| if (value.IsValid()) |
| throw IgniteError(IgniteError::IGNITE_ERR_FUTURE_STATE, "Future value already set"); |
| |
| if (error.GetCode() != IgniteError::IGNITE_SUCCESS) |
| throw IgniteError(IgniteError::IGNITE_ERR_FUTURE_STATE, "Future error already set"); |
| } |
| |
| value = val; |
| |
| cond.NotifyAll(); |
| } |
| |
| /** |
| * Set error. |
| * |
| * @throw IgniteError with IgniteError::IGNITE_ERR_FUTURE_STATE if error or value has been set already. |
| * @param err Error to set. |
| */ |
| void SetError(const IgniteError& err) |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| if (IsSet()) |
| { |
| if (value.IsValid()) |
| throw IgniteError(IgniteError::IGNITE_ERR_FUTURE_STATE, "Future value already set"); |
| |
| if (error.GetCode() != IgniteError::IGNITE_SUCCESS) |
| throw IgniteError(IgniteError::IGNITE_ERR_FUTURE_STATE, "Future error already set"); |
| } |
| |
| error = err; |
| |
| cond.NotifyAll(); |
| } |
| |
| /** |
| * Wait for value to be set. |
| * Active thread will be blocked until value or error will be set. |
| */ |
| void Wait() const |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| while (!IsSet()) |
| cond.Wait(mutex); |
| } |
| |
| /** |
| * Wait for value to be set for specified time. |
| * Active thread will be blocked until value or error will be set or timeout will end. |
| * |
| * @param msTimeout Timeout in milliseconds. |
| * @return True if the object has been triggered and false in case of timeout. |
| */ |
| bool WaitFor(int32_t msTimeout) const |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| if (IsSet()) |
| return true; |
| |
| return cond.WaitFor(mutex, msTimeout); |
| } |
| |
| /** |
| * Get the set value. |
| * Active thread will be blocked until value or error will be set. |
| * |
| * @throw IgniteError if error has been set. |
| * @return Value that has been set on success. |
| */ |
| concurrent::SharedPointer<ValueType> GetValue() const |
| { |
| Wait(); |
| |
| if (value.IsValid()) |
| return value; |
| |
| assert(error.GetCode() != IgniteError::IGNITE_SUCCESS); |
| |
| throw error; |
| } |
| |
| /** |
| * Set cancel target. |
| */ |
| void SetCancelTarget(std::auto_ptr<Cancelable>& target) |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| cancelTarget = target; |
| } |
| |
| /** |
| * Cancel related operation. |
| */ |
| void Cancel() |
| { |
| concurrent::CsLockGuard guard(mutex); |
| |
| if (cancelTarget.get()) |
| cancelTarget->Cancel(); |
| } |
| |
| private: |
| IGNITE_NO_COPY_ASSIGNMENT(SharedState); |
| |
| /** Cancel target. */ |
| std::auto_ptr<Cancelable> cancelTarget; |
| |
| /** Value. */ |
| concurrent::SharedPointer<ValueType> value; |
| |
| /** Error. */ |
| IgniteError error; |
| |
| /** Condition variable which serves to signal that value is set. */ |
| mutable concurrent::ConditionVariable cond; |
| |
| /** Lock that used to prevent double-set of the value. */ |
| mutable concurrent::CriticalSection mutex; |
| }; |
| } |
| } |
| |
| #endif //_IGNITE_SHARED_STATE |