IoCompletionThreadPool.h

Go to the documentation of this file.

/*  IoCompletionThreadPool.h - Io Completion Port thread pool
    Copyright (C) 2001-2004 Mark Weaver
    Written by Mark Weaver <mark@npsl.co.uk>

    Part of the Open-Win32 library.
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Library General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.

    This library 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
    Library General Public License for more details.

    You should have received a copy of the GNU Library General Public
    License along with this library; if not, write to the
    Free Software Foundation, Inc., 59 Temple Place - Suite 330,
    Boston, MA  02111-1307, USA.
*/

#ifndef OW32_IoCompletionThreadPool_h
#define OW32_IoCompletionThreadPool_h

#include <OW32/OW32Libs.h>
#include <OW32/IoCompletion.h>
#include <OW32/Thread.h>
#include <OW32/auto_array_ptr.h>

namespace OW32
{

#ifdef _MSC_VER
#pragma warning(disable: 4251)
#endif

class OW32_LIB_EXPORT CIoCompletionThread : 
    public CThread
{
public:
    CIoCompletionThread() :
        CThread(false)
    {
    }

    void SetCompletionPort(HANDLE hCompletionPort) { m_hCompletionPort = hCompletionPort; }

    virtual unsigned int Run();

private:
    HANDLE m_hCompletionPort;
};

class OW32_LIB_EXPORT CIoCompletionThreadPool
{
public:
    CIoCompletionThreadPool(HANDLE hExistingCompletionPort = NULL, bool fCloseCompletionPort = false) :
        m_hCompletionPort(hExistingCompletionPort),
        m_fCloseCompletionPort(fCloseCompletionPort),
        m_bCreated(FALSE),
        m_nWorkers(0)
    {
    }

    BOOL Create(DWORD threads, DWORD concurrentThreads = 0, int priority = THREAD_PRIORITY_NORMAL);

    BOOL Destroy();

    operator HANDLE() const { return m_hCompletionPort; }
    
    HANDLE GetCompletionPort() const { return m_hCompletionPort; }

    void SetCompletionPort(HANDLE hCompletionPort, bool fCloseCompletionPort = false) { 
        m_hCompletionPort = hCompletionPort; 
        m_fCloseCompletionPort = fCloseCompletionPort;
    }

private:
    CIoCompletionThreadPool operator= (const CIoCompletionThreadPool &rhs);
    CIoCompletionThreadPool(const CIoCompletionThreadPool &other);

    bool m_fCloseCompletionPort;

    HANDLE m_hCompletionPort;

    BOOL m_bCreated;

    DWORD m_nWorkers;
    auto_array_ptr<CIoCompletionThread> m_Workers;
};

#ifdef _MSC_VER
#pragma warning(default: 4251)
#endif

} // namespace OW32

#endif // OW32_IoCompletionThreadPool_h

---