TnbDirectoryWatcher.h

[詳解]

#pragma once
#include "TnbThread.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
class CDirectoryWatcher : TNB::CThread::IRunner
{
public:
 
    struct IListener
    {
        virtual ~IListener(void) {}
 
        virtual bool OnDirectoryChange(DWORD dwFlags) = 0;
    };
 
private:
 
    HANDLE          m_hNotify;      
    DWORD           m_dwFlags;      
    IListener*      m_pListener;    
    TNB::CThread    m_thread;       
 
    virtual DWORD Run(void)
    {
        while ( m_hNotify != NULL && IsRunnable() )
        {
            DWORD r = ::WaitForSingleObject(m_hNotify, INFINITE);
            if ( r != WAIT_OBJECT_0 || m_hNotify == NULL )
            {
                break;
            }
            if ( ! m_pListener->OnDirectoryChange(m_dwFlags) )
            {
                break;
            }
            if ( ! ::FindNextChangeNotification(m_hNotify) )
            {
                ASSERT(false);
                break;
            }
        }
        if ( m_hNotify != NULL )
        {
            ::FindCloseChangeNotification(m_hNotify);
            m_hNotify = NULL;
        }
        return 0;
    }
 
public:
 
    CDirectoryWatcher(void) : m_hNotify(NULL), m_pListener(NULL)
    {
    }
 
    virtual ~CDirectoryWatcher(void)
    {
        Stop();
    }
 
    bool Start(LPCTSTR lpszPath, bool boWithSubDir, DWORD dwFlags, IListener* pListener)
    {
        Stop();
        m_hNotify = ::FindFirstChangeNotification(lpszPath, boWithSubDir, dwFlags);
        if ( m_hNotify == NULL )
        {
            return false;
        }
        m_dwFlags = dwFlags;
        m_pListener = pListener;
        m_thread.SetRunner(this);
        if ( ! m_thread.Start(_T("Directory監視")) )
        {
            Stop();
            return false;
        }
        return true;
    }
 
    void Stop(void)
    {
        if ( m_hNotify != NULL )
        {
            HANDLE h = m_hNotify;
            m_hNotify = NULL;
            ::FindCloseChangeNotification(h);
        }
        m_thread.Stop();
    }
 
    CThreadStatus& ReferThreadStatus(void)
    {
        return m_thread;
    }
};
 
 
 
}; //TNB