scsample.h

Vc.Net入门与提高源码

// SCSample.h : Defines the ATL Server request handler class
//
// (c) 2000 Microsoft Corporation
// Copyright (c) Microsoft Corporation.  All rights reserved.
//
// This source code is only intended as a supplement to the
// Microsoft Classes Reference and related electronic
// documentation provided with the library.
// See these sources for detailed information regarding the
// Microsoft C++ Libraries products.

#pragma once
#define		DEFAULT_LIFESPAN	30
#include	<time.h>
#include	<fstream>
#include	<atlfile.h>
using namespace std;

// The default replacement handler.
[ request_handler("Default") ]
class CSCSampleHandler
{
private:
	// Put private members here

protected:
	// Put protected members here
	
public:
	// Put public members here

	HTTP_CODE ValidateAndExchange()
	{
		// TODO: Put all initialization and validation code here
		
		// Set the content-type
		m_HttpResponse.SetContentType("text/html");
		
		return HTTP_SUCCESS;
	}
	// Here is an example of how to use a replacement tag with the stencil processor
	// This replacement method will inject the local time in the response stream, whenever 
	// the "getLocalTime" replacement tag is found
	[ tag_name(name="getLocalTime") ]
	HTTP_CODE OnGetLocalTime(void)
	{
		struct tm	*stNow;
		time_t		t;	

		time( &t );

		stNow	=	localtime( &t );


		m_HttpResponse << asctime( stNow );
		return HTTP_SUCCESS;
	}
}; // class CSCSampleHandler




// Special request handler for creating, rendering and removing a temporary file that 
// is customized for the request
[ request_handler("Frame") ]
class CSCSampleFrameHandler
{
private:
	// Put private members here

protected:
	// Put protected members here
	CStringA			_szFileName;
	CStringA			_szStartTime;
	CStringA			_szLifeSpan;
	CStringA				_szUserName;
	
public:
	// Put public members here
	HTTP_CODE ValidateAndExchange()
	{
		// TODO: Put all initialization and validation code here
		POSITION	pos	=	NULL;
		DWORD		dwRet	=	VALIDATION_E_PARAMNOTFOUND;
		long		lSpan;
		LPCSTR		szUserName;



		
		// Set the content-type
		m_HttpResponse.SetContentType("text/html");

		// Use Exchange to transfer between a form variable and a string local variable
		dwRet	=	m_HttpRequest.FormVars.Exchange("name", &szUserName);

		// When the frame is loaded from the HTML, the name field is not empty, 
		// so the validation will return VALIDATION_S_OK. If the validation fails,
		// then this is a re-load
		if(  dwRet	==	VALIDATION_S_OK )
		{
			// first loading of the frame
			time_t				tNow;					//	to hold the current time
			int					iLifespan = 30;			//	the lifespan indicated by the user (Default 30 secs)
			unsigned __int64	newLifespan;			//	the lifespan indicated by the user in 100 nanoseconds units
			CComPtr<IStencilCacheControl>	stCache;	//	wrapper for the cache control interface

			
			// Create a unique, new temporary file that generates the lower frame of the response
			_szFileName.Format("t%d.srf", clock() );

			time(&tNow);
			_szStartTime.Format("%d", tNow );

			_szUserName	=	szUserName;

			// Use Exchange to transfer between a form variable and a numeric local variable
			dwRet	=	m_HttpRequest.FormVars.Exchange("lifespan", &iLifespan);
			if( dwRet != S_OK )
			{
				iLifespan	=	DEFAULT_LIFESPAN;
			}
			_szLifeSpan.Format("%d", iLifespan);

			
			// Render the temporary file
			renderTemporaryFile();


			// convert from seconds (the unit used by the user) to 100 nanosecond units, used by the cache
			newLifespan	=	iLifespan * CFileTime::Second;

			
			
			// Query the IStencilCacheControl interface
			m_spServiceProvider->QueryService(__uuidof(IStencilCache), __uuidof(IStencilCacheControl), (void**)&stCache);

			// set the new lifespan
			stCache->SetDefaultLifespan(newLifespan);


			// Append the temporary file name, the new timespan 
			// and the request processing time as cookies to the response
			// They will be used in refresh requests
			m_HttpResponse.AppendCookie( "filename", _szFileName );
			m_HttpResponse.AppendCookie( "lifespan", _szLifeSpan );
			m_HttpResponse.AppendCookie( "starttime", _szStartTime);
		}
		else
		{
			//	This is an attempt to reload the page.
			LPCSTR			szVal;
			CStringA		strVal;
			
			// Use Exchange to transfer between a frm variable and a local variable
			// "mode" is supposed to describe the required reload ("touch" or "refresh")
			dwRet	=	m_HttpRequest.FormVars.Exchange("mode", &szVal);

			// look for the "filename" cookie, that has been set when the page was initialized
			CCookie	cookie	=	m_HttpRequest.Cookies("filename");
			
			// Get the cookie value
			cookie.GetValue( strVal );

			_szFileName	=	strVal;

			if( stricmp( szVal, "touch" ) == 0 )
			{
				// add a new line to the temporary file, to force reloading from the disk
				touchTemporaryFile();
			}

			if( stricmp( szVal, "reset" ) == 0 )
			{
				// reset the cache settings

				// destroy the temporary file
				reset();
				
				// redirect the client to the start page
				m_HttpResponse.Redirect("SCSample.htm");
				
				// clear the statistics for cache activity
				CComPtr<IMemoryCacheStats>	stCache;
				m_spServiceProvider->QueryService(__uuidof(IStencilCache), __uuidof(IMemoryCacheStats), (void**)&stCache);
				stCache->ClearStats();
			}


		}
		
		return HTTP_SUCCESS;
	}


	// Adds a new line to the temporary file that generates the lower frame of the response,
	// to force reloading from the hard-drive
	void touchTemporaryFile()
	{
		char		path[MAX_PATH], modPath[MAX_PATH];
		char		file[MAX_PATH];
		char		*pCh;
		ofstream	out;
		struct tm	*stNow;
		time_t		t;	
		CStringA	szLocalTime;

		time( &t );

		stNow	=	localtime( &t );

		


		::GetModuleFileNameA( this->m_hInstHandler, modPath, MAX_PATH);

		pCh	=	strchr( modPath, '\\');

		if( pCh == NULL )
		{
			return;
		}
		
		strncpy( path, modPath, pCh - modPath);
		path[pCh - modPath] = 0;

		sprintf( file, "%s\\%s", path, _szFileName);


		out.open(file, ios::ate);
		out	<<	"Last modified: " << asctime( stNow ) <<	"<br>"	<<	endl;
		out.close();
		
	}
	

	
	// Removes the temporary file
	void reset()
	{
		char		path[MAX_PATH], modPath[MAX_PATH];
		char		file[MAX_PATH];
		char		*pCh;
		ofstream	out;
		struct tm	*stNow;
		time_t		t;	
		CStringA	szLocalTime;

		time( &t );

		stNow	=	localtime( &t );

		


		::GetModuleFileNameA( this->m_hInstHandler, modPath, MAX_PATH);

		pCh	=	strchr( modPath, '\\');

		if( pCh == NULL )
		{
			return;
		}
		
		strncpy( path, modPath, pCh - modPath);
		path[pCh - modPath] = 0;

		sprintf( file, "%s\\%s", path, _szFileName);

		::DeleteFileA( file );

	}

	
	// Renders the SRF content of the temporary file that generates the lower frame of the response
	// The temp file is customized to the user name and the current server time
	void	renderTemporaryFile()
	{
		char		path[MAX_PATH], modPath[MAX_PATH];
		char		file[MAX_PATH];
		char		*pCh;
		ofstream	out;


		::GetModuleFileNameA( this->m_hInstHandler, modPath, MAX_PATH);

		pCh	=	strrchr( modPath, '\\');

		if( pCh == NULL )
		{
			return;
		}
		
		strncpy( path, modPath, pCh - modPath);
		path[pCh - modPath] = 0;

		sprintf( file, "%s\\%s", path, _szFileName);

		out.open( file );

		if( out.is_open() )
		{
			struct tm	*stNow;
			time_t		t;	
			CStringA	szLocalTime;

			time( &t );

			stNow	=	localtime( &t );

			szLocalTime	=	asctime( stNow );

			out	<<	"{{handler SCSample.dll/Default}}"	<<	endl;
			out	<<	"<html>"	<<	endl;
			out	<<	"<title>SRF File to be cached</title>"	<<	endl;
			out	<<	"<body>"	<<	endl;
			out	<<	"Hello, "	<< (LPCSTR)_szUserName	<<	"<br>"	<<	endl;
			out	<<	"It was generated and cached first time at : "	<<	(LPCSTR)szLocalTime	<<	"(server time)<br>"	<<	endl;
			out	<<	"Now, it is : {{getLocalTime}} (server time) <br>"	<<	endl;
			out	<<	"</body>";
			out	<<	"</html>";


			out.flush();

			out.close();
		}
	}



	// Here is an example of how to use a replacement tag with the stencil processor
	[ tag_name(name="getFileName") ]
	HTTP_CODE OnGetFileName(void)
	{
		m_HttpResponse << _szFileName;
		return HTTP_SUCCESS;
	}
}; // class CSCSampleHandler








// Special request handler used in rendering statistical information related to the stencil cache
[ request_handler("Stats") ]
class CSCSampleStatsHandler
{
private:
	// Put private members here

protected:
	// Put protected members here
	
public:
	// Put public members here

	HTTP_CODE ValidateAndExchange()
	{
		// TODO: Put all initialization and validation code here
		
		// Set the content-type
		m_HttpResponse.SetContentType("text/html");
		
		return HTTP_SUCCESS;
	}
	// Here is an example of how to use a replacement tag with the stencil processor
	[ tag_name(name="getDefaultLifespan") ]
	HTTP_CODE OnGetDefaultLifespan(void)
	{
		CComPtr<IStencilCacheControl>	stCache;
		unsigned __int64		dwLifeSpan;
		

		m_spServiceProvider->QueryService(__uuidof(IStencilCache), __uuidof(IStencilCacheControl), (void**)&stCache);
		stCache->GetDefaultLifespan(&dwLifeSpan);

		dwLifeSpan	=	dwLifeSpan / CFileTime::Second;

		m_HttpResponse	<<	dwLifeSpan;

		return HTTP_SUCCESS;
	}


	[ tag_name(name="getStats") ]
	HTTP_CODE OnGetStats(void)
	{
		CComPtr<IMemoryCacheStats>	stStats;
		DWORD		dwData;
		CStringA	szLine;

		// First, get a pointer to the IMemiryCacheStats interface implemented by the Stencil Cache service
		m_spServiceProvider->QueryService(__uuidof(IStencilCache), __uuidof(IMemoryCacheStats), (void**)&stStats);

		// Use GetHitCount method to get the number of successfull cache lookups
		stStats->GetHitCount(&dwData);
		szLine.Format("Hits count : %d<br>", dwData);
		m_HttpResponse	<<	szLine;

		// Use GetMissCount method to get the number of insuccessfull cache lookups
		stStats->GetMissCount(&dwData);
		szLine.Format("Miss count : %d<br>", dwData);
		m_HttpResponse	<<	szLine;

		// Use GetCurrentAllocSize method to get the current cache allocated size
		stStats->GetCurrentAllocSize(&dwData);
		szLine.Format("Current Alloc Size : %d<br>", dwData);
		m_HttpResponse	<<	szLine;

		// Use GetMaxAllocSize method to get the maximum cache allocated size during this session
		stStats->GetMaxAllocSize(&dwData);
		szLine.Format("Max Alloc Size : %d<br>", dwData);
		m_HttpResponse	<<	szLine;

		// Use GetCurrentEntryCount method to get the current number of entries in the cache
		stStats->GetCurrentEntryCount(&dwData);
		szLine.Format("Current Entry Count : %d<br>", dwData);
		m_HttpResponse	<<	szLine;

		
		// Use GetCurrentAllocSize method to get the maximum number of entries in the cache during this session
		stStats->GetMaxEntryCount(&dwData);
		szLine.Format("Max Entry Count : %d<br>", dwData);
		m_HttpResponse	<<	szLine;

		return HTTP_SUCCESS;
	}

}; // class CSCSampleHandler