星期日, 26 7 月, 2026
Latest:
i007.cc

i007.cc

优先队列-降维打击

C++监测文件夹变化

Will

C++监测文件夹变化

转载 2016年11月30日 11:05:44

转自：http://blog.csdn.net/zhihu008/article/details/7995829

Barry注：只是学习ReadDirectoryChangesW的时候发现的一个例子，用于理解ReadDirectoryChangesW的用法，目前发现的不足之处有：

1. ReadDirectoryChangesW的退出不优雅，直接TerminateThread实属下下策，如果close时无限等待，很容易死锁

2. 监控目录不为空时，直接删除见空目录，ReadDirectoryChangesW也不会感知到，线程不会退出，除非手动再次start（会死锁，因为start中的close是无限等待）

FileSystemWatcher.h

[cpp] view plain copy

  1. #ifndef __FILESYSTEMWATCHER_HPP__
  2. #define __FILESYSTEMWATCHER_HPP__
  4. #if(_WIN32_WINNT < 0x0400)
  5. #define _WIN32_WINNT 0x0400
  6. #endif
  7. #include <windows.h>
  9. class FileSystemWatcher
  10. {
  11. public:
  12.  enum Filter
  13.  {
  14.   FILTER_FILE_NAME        = 0x00000001, // add/remove/rename
  15.   FILTER_DIR_NAME         = 0x00000002, // add/remove/rename
  16.   FILTER_ATTR_NAME        = 0x00000004,
  17.   FILTER_SIZE_NAME        = 0x00000008,
  18.   FILTER_LAST_WRITE_NAME  = 0x00000010, // timestamp
  19.   FILTER_LAST_ACCESS_NAME = 0x00000020, // timestamp
  20.   FILTER_CREATION_NAME    = 0x00000040, // timestamp
  21.   FILTER_SECURITY_NAME    = 0x00000100
  22.  };
  23.  enum ACTION
  24.  {
  25.   ACTION_ERRSTOP          = -1,
  26.   ACTION_ADDED            = 0x00000001,
  27.   ACTION_REMOVED          = 0x00000002,
  28.   ACTION_MODIFIED         = 0x00000003,
  29.   ACTION_RENAMED_OLD      = 0x00000004,
  30.   ACTION_RENAMED_NEW      = 0x00000005
  31.  };
  33.  typedef void (__stdcall *LPDEALFUNCTION)( ACTION act, LPCWSTR filename, LPVOID lParam );
  35.  FileSystemWatcher();
  36.  ~FileSystemWatcher();
  38.  // LPCTSTR dir: dont end-with “\\”
  39.  bool Run( LPCTSTR dir, bool bWatchSubtree, DWORD dwNotifyFilter, LPDEALFUNCTION dealfun, LPVOID lParam );
  40.  void Close( DWORD dwMilliseconds=INFINITE );
  42. private: // no-impl
  43.  FileSystemWatcher( const FileSystemWatcher& );
  44.  FileSystemWatcher operator=( const FileSystemWatcher );
  46. private:
  47.  HANDLE m_hDir;
  48.  DWORD m_dwNotifyFilter;
  49.  bool m_bWatchSubtree;
  50.  HANDLE m_hThread;
  51.  volatile bool m_bRequestStop;
  52.  LPDEALFUNCTION m_DealFun;
  53.  LPVOID m_DealFunParam;
  54.  static DWORD WINAPI Routine( LPVOID lParam );
  55. };
  57. #endif // __FILESYSTEMWATCHER_HPP__

FileSystemWatcher.cpp

[cpp] view plain copy

  1. #include “FileSystemWatcher.h”
  2. #include <cassert>
  4. FileSystemWatcher::FileSystemWatcher() : m_hDir(INVALID_HANDLE_VALUE), m_hThread(NULL)
  5. {
  6.  assert( FILTER_FILE_NAME        == FILE_NOTIFY_CHANGE_FILE_NAME   );
  7.  assert( FILTER_DIR_NAME         == FILE_NOTIFY_CHANGE_DIR_NAME    );
  8.  assert( FILTER_ATTR_NAME        == FILE_NOTIFY_CHANGE_ATTRIBUTES  );
  9.  assert( FILTER_SIZE_NAME        == FILE_NOTIFY_CHANGE_SIZE        );
  10.  assert( FILTER_LAST_WRITE_NAME  == FILE_NOTIFY_CHANGE_LAST_WRITE  );
  11.  assert( FILTER_LAST_ACCESS_NAME == FILE_NOTIFY_CHANGE_LAST_ACCESS );
  12.  assert( FILTER_CREATION_NAME    == FILE_NOTIFY_CHANGE_CREATION    );
  13.  assert( FILTER_SECURITY_NAME    == FILE_NOTIFY_CHANGE_SECURITY    );
  15.  assert( ACTION_ADDED            == FILE_ACTION_ADDED              );
  16.  assert( ACTION_REMOVED          == FILE_ACTION_REMOVED            );
  17.  assert( ACTION_MODIFIED         == FILE_ACTION_MODIFIED           );
  18.  assert( ACTION_RENAMED_OLD      == FILE_ACTION_RENAMED_OLD_NAME   );
  19.  assert( ACTION_RENAMED_NEW      == FILE_ACTION_RENAMED_NEW_NAME   );
  20. }
  22. FileSystemWatcher::~FileSystemWatcher()
  23. {
  24.  Close();
  25. }
  27. bool FileSystemWatcher::Run( LPCTSTR dir, bool bWatchSubtree, DWORD dwNotifyFilter, LPDEALFUNCTION dealfun, LPVOID lParam )
  28. {
  29.  Close();
  31.  m_hDir = CreateFile(//dir目录不能以’\’结尾，否则监测不到dir目录被删除，不以\结尾，可以检测到（仅限于空目录时）
  32.   dir,
  33.   GENERIC_READ,
  34.   FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE,
  35.   NULL,
  36.   OPEN_EXISTING,
  37.   FILE_FLAG_BACKUP_SEMANTICS,
  38.   NULL
  39.   );
  40.  if( INVALID_HANDLE_VALUE == m_hDir ) return false;
  42.  m_bWatchSubtree = bWatchSubtree;
  43.  m_dwNotifyFilter = dwNotifyFilter;
  44.  m_DealFun = dealfun;
  45.  m_DealFunParam = lParam;
  46.  m_bRequestStop = false;
  48.  DWORD ThreadId;
  49.  m_hThread = CreateThread( NULL,0,Routine,this,0,&ThreadId );
  50.  if( NULL == m_hThread )
  51.  {
  52.   CloseHandle( m_hDir );
  53.   m_hDir = INVALID_HANDLE_VALUE;
  54.  }
  56.  return NULL!=m_hThread;
  57. }
  59. void FileSystemWatcher::Close( DWORD dwMilliseconds )
  60. {
  61.  if( NULL != m_hThread )
  62.  {
  63.   m_bRequestStop = true;
  64.   if( WAIT_TIMEOUT == WaitForSingleObject(m_hThread,dwMilliseconds) )
  65.    TerminateThread( m_hThread, 0 );
  66.   CloseHandle( m_hThread );
  67.   m_hThread = NULL;
  68.  }
  69.  if( INVALID_HANDLE_VALUE != m_hDir )
  70.  {
  71.   CloseHandle( m_hDir );
  72.   m_hDir = INVALID_HANDLE_VALUE;
  73.  }
  74. }
  76. DWORD WINAPI FileSystemWatcher::Routine( LPVOID lParam )
  77. {
  78.  FileSystemWatcher& obj = *(FileSystemWatcher*)lParam;
  80.  BYTE buf[ 2*(sizeof(FILE_NOTIFY_INFORMATION)+2*MAX_PATH)+2 ];
  81.  FILE_NOTIFY_INFORMATION* pNotify=(FILE_NOTIFY_INFORMATION *)buf;
  82.  DWORD BytesReturned;
  83.  while( !obj.m_bRequestStop )
  84.  {
  85.   if( ReadDirectoryChangesW( obj.m_hDir,
  86.    pNotify,
  87.    sizeof(buf)-2,
  88.    obj.m_bWatchSubtree,
  89.    obj.m_dwNotifyFilter,
  90.    &BytesReturned,
  91.    NULL,
  92.    NULL ) ) // 无限等待，应当使用异步方式
  93.   {
  94.    for( FILE_NOTIFY_INFORMATION* p=pNotify; p; )
  95.    {
  96.     WCHAR c = p->FileName[p->FileNameLength/2];
  97.     p->FileName[p->FileNameLength/2] = L‘\0’;
  99.     obj.m_DealFun( (ACTION)p->Action, p->FileName, obj.m_DealFunParam );
  101.     p->FileName[p->FileNameLength/2] = c;
  103.     if( p->NextEntryOffset )
  104.      p  = (PFILE_NOTIFY_INFORMATION)( (BYTE*)p + p->NextEntryOffset );
  105.     else
  106.      p = 0;
  107.    }
  108.   }
  109.   else
  110.   {
  111.    obj.m_DealFun( (ACTION)ACTION_ERRSTOP, 0, obj.m_DealFunParam );
  112.    break;
  113.   }
  114.  }
  116.  return 0;
  117. }

main函数

[cpp] view plain copy

  1. #include “stdafx.h”
  2. #include “FileSystemWatcher.h”
  3. #include <stdio.h>
  4. #include <locale.h>
  5. #include <assert.h>
  6. #include <tchar.h>
  7. #include <conio.h>
  9. void __stdcall MyDeal( FileSystemWatcher::ACTION act, LPCWSTR filename, LPVOID lParam )
  10. {
  11.  static FileSystemWatcher::ACTION pre = FileSystemWatcher::ACTION_ERRSTOP;
  12.  switch( act )
  13.  {
  14.  case FileSystemWatcher::ACTION_ADDED:
  15.   wprintf_s( L“Added     – %s\n”, filename );
  16.   break;
  17.  case FileSystemWatcher::ACTION_REMOVED:
  18.   wprintf_s( L“Removed   – %s\n”, filename );
  19.   break;
  20.  case FileSystemWatcher::ACTION_MODIFIED:
  21.   wprintf_s( L“Modified  – %s\n”, filename );
  22.   break;
  23.  case FileSystemWatcher::ACTION_RENAMED_OLD:
  24.   wprintf_s( L“Rename(O) – %s\n”, filename );
  25.   break;
  26.  case FileSystemWatcher::ACTION_RENAMED_NEW:
  27.   assert( pre == FileSystemWatcher::ACTION_RENAMED_OLD );
  28.   wprintf_s( L“Rename(N) – %s\n”, filename );
  29.   break;
  30.  case FileSystemWatcher::ACTION_ERRSTOP:
  31.  default:
  32.   wprintf_s( L“—ERROR—%s\n”, filename );
  33.   break;
  34.  }
  35.  pre = act;
  36. }
  38. int main()
  39. {
  40.  LPCTSTR sDir= TEXT(“D:\\监测文件夹”);
  41.  DWORD dwNotifyFilter = FileSystemWatcher::FILTER_FILE_NAME|FileSystemWatcher::FILTER_DIR_NAME;
  43.  FileSystemWatcher fsw;
  44.  bool r = fsw.Run( sDir, true, dwNotifyFilter, &MyDeal, 0 );
  45.  if( !r ) return -1;
  47.  _tsetlocale( LC_CTYPE, TEXT(“chs”) );
  48.  _tprintf_s( TEXT(“Watch%s\n”), sDir );
  49.  _tprintf_s( TEXT(“Press <q> to quit.\n”) );
  51.  while( _getch() != ‘q’ );
  53.  fsw.Close(1000);
  55.  return 0;
  56. }

You May Also Like

C++对象计数

Will

IOCP模型与EPOLL模型的比较

Will

稳定排序和不稳定排序

Will

One thought on “C++监测文件夹变化”

  • willPost author
    2018年2月11日 at 下午6:39
    Permalink

    可以实现配置文件更新自动reload，不用再手动敲reload指令了，reload指令太难记了

发表回复