84 lines
2.2 KiB
C++
84 lines
2.2 KiB
C++
#ifndef _CCM_H_
|
|
#define _CCM_H_
|
|
|
|
|
|
// ccm.h : Declaration of CCM, the Central Management Object
|
|
#include "resource.h" // main symbols
|
|
#include "page.h"
|
|
#include "Monitor.h"
|
|
|
|
// Registry Keys
|
|
#define HITCNT_HKEY HKEY_CLASSES_ROOT
|
|
#define HITCNT_KEYNAME _T("MSWC.PageCounter")
|
|
#define HITCNT_FILELOCN _T("File_Location")
|
|
#define HITCNT_DFLT_FILE _T("%windir%\\system32\\inetsrv\\data\\HitCnt.cnt")
|
|
#define HITCNT_SAVECNT _T("Save_Count")
|
|
#define HITCNT_DFLT_SAVE 25
|
|
|
|
// Registry notification class.
|
|
class CRegNotify : public CMonitorNotify
|
|
{
|
|
public:
|
|
CRegNotify();
|
|
virtual void Notify();
|
|
String FileName();
|
|
DWORD SaveCount();
|
|
|
|
private:
|
|
//Looks in Registry to get the Save Count
|
|
bool GetSaveCount();
|
|
//Looks in Registry to get the file path and name
|
|
bool GetFileName();
|
|
|
|
String m_strFileName; //path and filename for persisting
|
|
DWORD m_dwSaveCount; //threshold to trigger persisting
|
|
CComAutoCriticalSection m_cs;
|
|
};
|
|
|
|
DECLARE_REFPTR( CRegNotify,CMonitorNotify )
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// Central Counter Manager
|
|
|
|
class CCM
|
|
{
|
|
public:
|
|
CCM();
|
|
~CCM();
|
|
|
|
//---- Object Methods ----
|
|
|
|
//Increments hit count for this page
|
|
DWORD IncrementAndGetHits(const BSTR bstrURL);
|
|
|
|
// Get the hit count for this page
|
|
LONG GetHits(const BSTR bstrURL);
|
|
|
|
// Reset the hit count for this page to zero
|
|
void Reset(const BSTR bstrURL);
|
|
|
|
//---- Helper Functions ----
|
|
|
|
//Initializes Page Counter
|
|
BOOL Initialize();
|
|
|
|
//Writes the data to disk
|
|
BOOL Persist();
|
|
|
|
private:
|
|
//Loads the persisted data into an array
|
|
BOOL LoadData();
|
|
|
|
CRegNotifyPtr m_pRegKey;
|
|
UINT m_cUnsavedHits; //hits since we last persisted
|
|
CPageArray m_pages; //array of pages
|
|
volatile bool m_bInitialized;
|
|
CComAutoCriticalSection m_critsec; //Critical Section to protect data
|
|
|
|
};
|
|
|
|
extern CCM CountManager;
|
|
|
|
|
|
#endif
|