warsocketcpslimit.cpp

来自「ftpserver very good sample」· C++ 代码 · 共 92 行

#include "StdAfx.h"

#define WAR_CPS_LIMIT_WANT_PLUGINS 1
#include "WarSocketCpsLimit.h"   // class implemented


/////////////////////////////// PUBLIC ///////////////////////////////////////

WarSocketCpsLimit *WarSocketCpsLimit::spThis;

//============================= LIFECYCLE ====================================

WarSocketCpsLimit::WarSocketCpsLimit() :
WarPluginModule("WarSocketCpsLimit")
{
    if (NULL != spThis)
        WarThrow(WarError(WAR_ERR_ALREADY_INITIALIZED), NULL);
    spThis = this;

    // Register what classes we have interest in
    NotifyOnClassInstance("WarSocketIo");

}// WarSocketCpsLimit

WarSocketCpsLimit::~WarSocketCpsLimit()
{
    if (this == spThis)
        spThis = NULL;
}// ~WarSocketCpsLimit


//============================= OPERATORS  ====================================
//============================= OPERATIONS ===================================

void WarSocketCpsLimit::EnableCps(
    WarSocket *pSocket,
    const unsigned delayMs,
    const unsigned holdMs,
    const unsigned minCnt,
    const unsigned maxCnt,
    const unsigned minCps,
    const unsigned maxCps,
    const WarSocketCpsLimit::DirectionE forDirection)
{
    switch(forDirection)
    {
    case RECEIVING:
        pSocket->GetSocketIo().AddPlugin(
            WarSocketIo::PLUGIN_ON_RECEIVED,
            new WarSocketCpsLimit_WarSocketIo_OnReceived(pSocket,
            delayMs,
            holdMs,
            minCnt,
            maxCnt,
            minCps,
            maxCps));
        break;

    case SENDING:
        pSocket->GetSocketIo().AddPlugin(
            WarSocketIo::PLUGIN_ON_RECEIVED,
            new WarSocketCpsLimit_WarSocketIo_OnSent(pSocket,
            delayMs,
            holdMs,
            minCnt,
            maxCnt,
            minCps,
            maxCps));
        break;
        
    }
}


//============================= CALLBACK   ===================================

void WarSocketCpsLimit::OnClassInstance(
    WarPluginBaseSupport *pClassInstance,
    const NotificationE notificationType)
    throw(WarException) // Should not throw
{
   ; // We are initialized manually, on demand.
}


//============================= ACCESS     ===================================
//============================= INQUIRY    ===================================
/////////////////////////////// PROTECTED  ///////////////////////////////////

/////////////////////////////// PRIVATE    ///////////////////////////////////