JoinThreads.h

Go to the documentation of this file.

// This file is part of CoreLibrary containing useful reusable utility
// classes.
//
// Copyright (C) 2014 to present, Duncan Crutchley
// Contact <dac1976github@outlook.com>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published
// by the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License and GNU Lesser General Public License
// for more details.
//
// You should have received a copy of the GNU General Public License
// and GNU Lesser General Public License along with this program. If
// not, see <http://www.gnu.org/licenses/>.

#ifndef JOINTHREADS
#define JOINTHREADS

#include <thread>

namespace core_lib
{
namespace threads
{

template <template <class, class> class C> class JoinThreads final
{
public:
    using container_type = C<std::thread, std::allocator<std::thread>>;
    explicit JoinThreads(container_type& threads)
        : m_threads(threads)
    {
    }
    JoinThreads(const JoinThreads&) = delete;
    JoinThreads& operator=(const JoinThreads&) = delete;
    JoinThreads(JoinThreads&&) = delete;
    JoinThreads& operator=(JoinThreads&&) = delete;
    ~JoinThreads()
    {
        for (auto& t : m_threads)
        {
            if (t.joinable())
            {
                t.join();
            }
        }
    }

private:
    container_type& m_threads;
};

template <template <class, class> class C> class JoinThreadsP final
{
public:
    using container_type = C<std::thread*, std::allocator<std::thread*>>;
    explicit JoinThreadsP(container_type& threads)
        : m_threads(threads)
    {
    }
    JoinThreadsP(const JoinThreadsP&) = delete;
    JoinThreadsP& operator=(const JoinThreadsP&) = delete;
    JoinThreadsP(JoinThreadsP&&) = delete;
    JoinThreadsP& operator=(JoinThreadsP&&) = delete;
    ~JoinThreadsP()
    {
        for (auto& t : m_threads)
        {
            if (t && t->joinable())
            {
                t->join();
            }
        }
    }

private:
    container_type& m_threads;
};

} // namespace threads
} // namespace core_lib

#endif // JOINTHREADS