150 lines
6.4 KiB
C++
150 lines
6.4 KiB
C++
|
||
#include "Stdafx.h"
|
||
#include "WatcherBase.h"
|
||
#include "Log.h"
|
||
|
||
CWatcherBase::CWatcherBase()
|
||
{
|
||
}
|
||
|
||
CWatcherBase::~CWatcherBase()
|
||
{
|
||
ScopeLock <CSyncLock> sync(m_Sync);
|
||
m_WatchList.clear();
|
||
}
|
||
|
||
bool CWatcherBase::AddWatchSpec(int nVersionID, int nNID, const WCHAR * pType)
|
||
{
|
||
ScopeLock <CSyncLock> sync(m_Sync);
|
||
|
||
_WATCH_SPEC spec;
|
||
std::list <_WATCH_OBJECT>::iterator ii;
|
||
for (ii = m_WatchList.begin(); ii != m_WatchList.end(); ii++)
|
||
{
|
||
if (nVersionID == (*ii).nVersionID)
|
||
{
|
||
std::list <_WATCH_SPEC>::iterator ih;
|
||
for (ih = (*ii).SpecList.begin(); ih != (*ii).SpecList.end(); ih++)
|
||
{
|
||
if (!wcsicmp((*ih).wszType, pType) && (*ih).nNID == nNID)
|
||
{
|
||
g_Log.Log(LogType::_FILELOG, L"AddWatchContent Add Fail\n");
|
||
return false;
|
||
}
|
||
}
|
||
|
||
memset(&spec, 0, sizeof(_WATCH_SPEC));
|
||
|
||
spec.nNID = nNID;
|
||
wcscpy_s(spec.wszType, pType);
|
||
spec.nInsertTick = timeGetTime();
|
||
|
||
(*ii).SpecList.push_back(spec);
|
||
return true;
|
||
}
|
||
}
|
||
|
||
_WATCH_OBJECT object;
|
||
object.nVersionID = nVersionID;
|
||
|
||
memset(&spec, 0, sizeof(_WATCH_SPEC));
|
||
|
||
spec.nNID = nNID;
|
||
wcscpy_s(spec.wszType, pType);
|
||
spec.nInsertTick = timeGetTime();
|
||
|
||
object.SpecList.push_back(spec);
|
||
m_WatchList.push_back(object);
|
||
return true;
|
||
}
|
||
|
||
bool CWatcherBase::DelWatchSpec(int nVersionID, int nNID, const WCHAR * pType, bool &bComplete)
|
||
{
|
||
ScopeLock <CSyncLock> sync(m_Sync);
|
||
|
||
std::list <_WATCH_OBJECT>::iterator ii;
|
||
for (ii = m_WatchList.begin(); ii != m_WatchList.end(); ii++)
|
||
{
|
||
if ((*ii).nVersionID == nVersionID)
|
||
{
|
||
std::list <_WATCH_SPEC>::iterator ih;
|
||
for (ih = (*ii).SpecList.begin(); ih != (*ii).SpecList.end(); )
|
||
{
|
||
if ((*ih).nNID == nNID && !wcsicmp((*ih).wszType, pType))
|
||
{
|
||
ih = (*ii).SpecList.erase(ih);
|
||
}
|
||
else
|
||
ih++;
|
||
}
|
||
|
||
if ((*ii).SpecList.size() <= 0)
|
||
{
|
||
m_WatchList.erase(ii);
|
||
bComplete = true;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
void CWatcherBase::ClearWatchingSpec()
|
||
{
|
||
ScopeLock <CSyncLock> sync(m_Sync);
|
||
std::list <_WATCH_OBJECT>::iterator ii;
|
||
for (ii = m_WatchList.begin(); ii != m_WatchList.end(); ii++)
|
||
{
|
||
(*ii).SpecList.clear();
|
||
}
|
||
|
||
m_WatchList.clear();
|
||
g_Log.Log(LogType::_NORMAL, L"All Watching Spec Deleted\n");
|
||
}
|
||
|
||
bool CWatcherBase::HasWatchingSpec()
|
||
{
|
||
ScopeLock <CSyncLock> sync(m_Sync);
|
||
|
||
if (m_WatchList.size() > 0)
|
||
{
|
||
std::list <_WATCH_OBJECT>::iterator ii;
|
||
for (ii = m_WatchList.begin(); ii != m_WatchList.end(); ii++)
|
||
{
|
||
std::list <_WATCH_SPEC>::iterator ih;
|
||
for (ih = (*ii).SpecList.begin(); ih != (*ii).SpecList.end(); ih++)
|
||
{
|
||
g_Log.Log(LogType::_ERROR, L"Still Running [NID:%d][Type:%s]\n", (*ih).nNID, (*ih).wszType);
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
return false;
|
||
}
|
||
|
||
void CWatcherBase::DelWatchSpec(int nNID)
|
||
{
|
||
ScopeLock <CSyncLock> sync(m_Sync);
|
||
|
||
std::list <_WATCH_OBJECT>::iterator ii;
|
||
for (ii = m_WatchList.begin(); ii != m_WatchList.end();)
|
||
{
|
||
std::list <_WATCH_SPEC>::iterator ih;
|
||
for (ih = (*ii).SpecList.begin(); ih != (*ii).SpecList.end(); )
|
||
{
|
||
if ((*ih).nNID == nNID)
|
||
{
|
||
ih = (*ii).SpecList.erase(ih);
|
||
}
|
||
else
|
||
ih++;
|
||
}
|
||
|
||
if ((*ii).SpecList.size() <= 0)
|
||
ii = m_WatchList.erase(ii);
|
||
else
|
||
ii++;
|
||
}
|
||
}
|
||
|