lyfz_repos/经典版源码/Others/lyfzBackupModle/DelPhotoModel/DelFiles.cpp

#include "stdafx.h"
#include "DelFiles.h"
#include "ffscoex.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

using namespace helper_coffs;

unsigned int __stdcall DelThreadPro(IN LPVOID lpParam);

CSC_DelFiles::CSC_DelFiles()
{
	m_dwRefCount = 0;
	m_hThread = NULL;			//线程
	m_hEvent = NULL;			//退出事件
	m_bRunning = FALSE;			//是否正在运行
	m_PFCallBack = NULL;		//回调
}

CSC_DelFiles::~CSC_DelFiles()
{
	m_ListMgr.Release();
	TerminateDel();
}


STDMETHODIMP CSC_DelFiles::QueryInterface(REFIID refiid, void** ppv)
{
	if (IID_IUnknown == refiid)
		*ppv = this;
	else if (IID_ISC_DELFILESINTERFACE == refiid)
		*ppv = (ISC_DelFilesInterface*)this;
	else 
	{
		*ppv = NULL;
		return E_NOINTERFACE;
	}
	((IUnknown*)(*ppv))->AddRef();
	return NOERROR;

}
STDMETHODIMP_(ULONG) CSC_DelFiles::AddRef(void)
{
	InterlockedIncrement((LPLONG)&m_dwRefCount);
	return m_dwRefCount;
}

STDMETHODIMP_(ULONG) CSC_DelFiles::Release(void)
{

	InterlockedDecrement((LPLONG)&m_dwRefCount);
	if (!m_dwRefCount)
		delete this;
	return 0;
}

/************************************************************************/
/*
	函数:	AddDelData
	描述:	添加要删除的数据
	参数:
		IN LPCTSTR lpDomain,					域名
		IN LPCTSTR lpOrderNum,					订单
		IN LPCTSTR lpDir,						目录
		IN const BYTE byPhotoType,				相片类型
		IN const BOOL bPickuped					是否已取件
	返回; 1成功，0失败
*/
/************************************************************************/
int CSC_DelFiles::AddDelData(IN LPCTSTR lpDomain, IN LPCTSTR lpOrderNum, IN LPCTSTR lpDir, IN const BYTE byPhotoType, IN const BOOL bPickuped)
{
	LPSDELINFO lp = new SDELINFO;
	lp->Clone(lpDomain, lpOrderNum, lpDir, byPhotoType, bPickuped);
	m_ListMgr.Push_Back(lp);
	return 1;
}

/************************************************************************/
/*
	函数:	StartDelFiles
	描述:	删除文件
	参数:
	返回; 1成功，0失败
*/
/************************************************************************/
int CSC_DelFiles::StartDelFiles(IN PFCALLBACK callbackfun/*回调函数通知备份完成*/)
{
	if(m_bRunning)
		return 0;

	m_PFCallBack = callbackfun;

	m_hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);

	unsigned int nThreadID = 0;
	m_hThread = (HANDLE)_beginthreadex(NULL, 0, DelThreadPro, (void*)this, 0, &nThreadID);
	m_bRunning = TRUE;
	return 1;
}

/************************************************************************/
/*
	函数:	TerminateDel
	描述:	终止删除操作
	参数:
	返回;
*/
/************************************************************************/
void CSC_DelFiles::TerminateDel()
{
	if(!m_bRunning)
		return;

	if(m_hEvent)
		SetEvent(m_hEvent);

	::WaitForSingleObject(m_hThread, INFINITE);
	m_bRunning = FALSE;

	if(m_hThread)
	{
		CloseHandle(m_hThread);
		m_hThread = NULL;
	}
	if(m_hEvent)
	{
		CloseHandle(m_hEvent);
		m_hEvent = NULL;
	}
}

/************************************************************************/
/*
	函数:	IsDeleting
	描述:	是否正在删除文件
	参数:
	返回;
*/
/************************************************************************/
BOOL CSC_DelFiles::IsDeleting()
{
	return m_bRunning;
}

/************************************************************************/
/*
	函数:	DelWork
	描述:	删除工作
	参数:
	返回;
*/
/************************************************************************/
void CSC_DelFiles::DelWork()
{
	while(m_ListMgr.Size() != 0)
	{
/*
		if(WaitForSingleObject(m_hEvent, 0) == WAIT_OBJECT_0)
			break;
*/
		LPSDELINFO lp = m_ListMgr.Pop_Front();
		if(lp != NULL)
		{
			BOOL bSuccess = DeleteDirectory(lp->szDir);
			m_PFCallBack(&lp->byPhotoType, lp->szDomain, lp->szOrderNum, lp->szDir, &bSuccess, NULL);
			delete lp;
		}
		Sleep(10);
	}
	
	if(m_hThread)
	{
		CloseHandle(m_hThread);
		m_hThread = NULL;
	}
	if(m_hEvent)
	{
		CloseHandle(m_hEvent);
		m_hEvent = NULL;
	}
	m_bRunning = FALSE;
}

/************************************************************************/
/*
	函数:	DelThreadPro
	描述:	线程函数
	参数:
		LPVOID lpParam			任何类型指针
	返回;
*/
/************************************************************************/
unsigned int __stdcall DelThreadPro(IN LPVOID lpParam)
{
	CSC_DelFiles* p = (CSC_DelFiles*)lpParam;
	if(p)
		p->DelWork();
	return 0;
}

/************************************************************************/
/* 
	函数:DeleteDirectory
	描述:删除指定目录;
	参数:
		IN:		lpDirectory	要删除的目录;
	返回: TRUE成功，FALSE失败
*/
/************************************************************************/
BOOL CSC_DelFiles::DeleteDirectory(IN LPCTSTR lpDiretory)// 可删除共享路径;
{
	ffscoex coex;
	return coex.deldir(lpDiretory);
}