171 lines
4.8 KiB
C++
171 lines
4.8 KiB
C++
|
// (C) Copyright 2012 Vicente J. Botet Escriba
|
||
|
// Distributed under the Boost Software License, Version 1.0. (See
|
||
|
// accompanying file LICENSE_1_0.txt or copy at
|
||
|
// http://www.boost.org/LICENSE_1_0.txt)
|
||
|
|
||
|
|
||
|
#ifndef BOOST_THREAD_EXTERNALLY_LOCKED_STREAM_HPP
|
||
|
#define BOOST_THREAD_EXTERNALLY_LOCKED_STREAM_HPP
|
||
|
|
||
|
#include <boost/thread/detail/config.hpp>
|
||
|
#include <boost/thread/detail/move.hpp>
|
||
|
#include <boost/thread/detail/delete.hpp>
|
||
|
|
||
|
#include <boost/thread/externally_locked.hpp>
|
||
|
#include <boost/thread/lock_traits.hpp>
|
||
|
#include <boost/thread/recursive_mutex.hpp>
|
||
|
#include <boost/thread/strict_lock.hpp>
|
||
|
|
||
|
#include <boost/config/abi_prefix.hpp>
|
||
|
|
||
|
namespace boost
|
||
|
{
|
||
|
|
||
|
template <typename Stream, typename RecursiveMutex=recursive_mutex>
|
||
|
class externally_locked_stream;
|
||
|
|
||
|
template <class Stream, typename RecursiveMutex=recursive_mutex>
|
||
|
class stream_guard
|
||
|
{
|
||
|
|
||
|
friend class externally_locked_stream<Stream, RecursiveMutex> ;
|
||
|
public:
|
||
|
typedef typename externally_locked_stream<Stream, RecursiveMutex>::mutex_type mutex_type;
|
||
|
|
||
|
BOOST_THREAD_MOVABLE_ONLY( stream_guard)
|
||
|
|
||
|
stream_guard(externally_locked_stream<Stream, RecursiveMutex>& mtx) :
|
||
|
mtx_(&mtx)
|
||
|
{
|
||
|
mtx.lock();
|
||
|
}
|
||
|
|
||
|
stream_guard(externally_locked_stream<Stream, RecursiveMutex>& mtx, adopt_lock_t) :
|
||
|
mtx_(&mtx)
|
||
|
{
|
||
|
}
|
||
|
|
||
|
stream_guard(BOOST_THREAD_RV_REF(stream_guard) rhs) BOOST_NOEXCEPT
|
||
|
: mtx_(rhs.mtx_)
|
||
|
{
|
||
|
rhs.mtx_= 0;
|
||
|
}
|
||
|
|
||
|
~stream_guard()
|
||
|
{
|
||
|
if (mtx_ != 0) mtx_->unlock();
|
||
|
}
|
||
|
|
||
|
bool owns_lock(const mutex_type * l) const BOOST_NOEXCEPT
|
||
|
{
|
||
|
return l == mtx_->mutex();
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* @Requires mtx_
|
||
|
*/
|
||
|
Stream& get() const
|
||
|
{
|
||
|
BOOST_THREAD_ASSERT_PRECONDITION( mtx_, lock_error() );
|
||
|
return mtx_->get(*this);
|
||
|
}
|
||
|
Stream& bypass() const
|
||
|
{
|
||
|
return get();
|
||
|
}
|
||
|
|
||
|
|
||
|
private:
|
||
|
externally_locked_stream<Stream, RecursiveMutex>* mtx_;
|
||
|
};
|
||
|
|
||
|
template <typename Stream, typename RecursiveMutex>
|
||
|
struct is_strict_lock_sur_parole<stream_guard<Stream, RecursiveMutex> > : true_type
|
||
|
{
|
||
|
};
|
||
|
|
||
|
/**
|
||
|
* externally_locked_stream cloaks a reference to an stream of type Stream, and actually
|
||
|
* provides full access to that object through the get and set member functions, provided you
|
||
|
* pass a reference to a strict lock object.
|
||
|
*/
|
||
|
|
||
|
//[externally_locked_stream
|
||
|
template <typename Stream, typename RecursiveMutex>
|
||
|
class externally_locked_stream: public externally_locked<Stream&, RecursiveMutex>
|
||
|
{
|
||
|
typedef externally_locked<Stream&, RecursiveMutex> base_type;
|
||
|
public:
|
||
|
BOOST_THREAD_NO_COPYABLE( externally_locked_stream)
|
||
|
|
||
|
/**
|
||
|
* Effects: Constructs an externally locked object storing the cloaked reference object.
|
||
|
*/
|
||
|
externally_locked_stream(Stream& stream, RecursiveMutex& mtx) BOOST_NOEXCEPT :
|
||
|
base_type(stream, mtx)
|
||
|
{
|
||
|
}
|
||
|
|
||
|
stream_guard<Stream, RecursiveMutex> hold() BOOST_NOEXCEPT
|
||
|
{
|
||
|
return stream_guard<Stream, RecursiveMutex> (*this);
|
||
|
}
|
||
|
Stream& bypass() const
|
||
|
{
|
||
|
stream_guard<Stream, RecursiveMutex> lk(*this);
|
||
|
return get(lk);
|
||
|
}
|
||
|
};
|
||
|
//]
|
||
|
|
||
|
template <typename Stream, typename RecursiveMutex, typename T>
|
||
|
inline const stream_guard<Stream, RecursiveMutex>& operator<<(const stream_guard<Stream, RecursiveMutex>& lck, T arg)
|
||
|
{
|
||
|
lck.get() << arg;
|
||
|
return lck;
|
||
|
}
|
||
|
|
||
|
template <typename Stream, typename RecursiveMutex>
|
||
|
inline const stream_guard<Stream, RecursiveMutex>& operator<<(const stream_guard<Stream, RecursiveMutex>& lck, Stream& (*arg)(Stream&))
|
||
|
{
|
||
|
lck.get() << arg;
|
||
|
return lck;
|
||
|
}
|
||
|
|
||
|
template <typename Stream, typename RecursiveMutex, typename T>
|
||
|
inline const stream_guard<Stream, RecursiveMutex>& operator>>(const stream_guard<Stream, RecursiveMutex>& lck, T& arg)
|
||
|
{
|
||
|
lck.get() >> arg;
|
||
|
return lck;
|
||
|
}
|
||
|
|
||
|
template <typename Stream, typename RecursiveMutex, typename T>
|
||
|
inline stream_guard<Stream, RecursiveMutex> operator<<(externally_locked_stream<Stream, RecursiveMutex>& mtx, T arg)
|
||
|
{
|
||
|
stream_guard<Stream, RecursiveMutex> lk(mtx);
|
||
|
mtx.get(lk) << arg;
|
||
|
return boost::move(lk);
|
||
|
}
|
||
|
|
||
|
template <typename Stream, typename RecursiveMutex>
|
||
|
inline stream_guard<Stream, RecursiveMutex> operator<<(externally_locked_stream<Stream, RecursiveMutex>& mtx, Stream& (*arg)(Stream&))
|
||
|
{
|
||
|
stream_guard<Stream, RecursiveMutex> lk(mtx);
|
||
|
mtx.get(lk) << arg;
|
||
|
return boost::move(lk);
|
||
|
}
|
||
|
|
||
|
template <typename Stream, typename RecursiveMutex, typename T>
|
||
|
inline stream_guard<Stream, RecursiveMutex> operator>>(externally_locked_stream<Stream, RecursiveMutex>& mtx, T& arg)
|
||
|
{
|
||
|
stream_guard<Stream, RecursiveMutex> lk(mtx);
|
||
|
mtx.get(lk) >> arg;
|
||
|
return boost::move(lk);
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
#include <boost/config/abi_suffix.hpp>
|
||
|
|
||
|
#endif // header
|