| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791 |
- /*
- * Copyright: JessMA Open Source (ldcsaa@gmail.com)
- *
- * Author : Bruce Liang
- * Website : https://github.com/ldcsaa
- * Project : https://github.com/ldcsaa/HP-Socket
- * Blog : http://www.cnblogs.com/ldcsaa
- * Wiki : http://www.oschina.net/p/hp-socket
- * QQ Group : 44636872, 75375912
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
- #pragma once
- #include "hpsocket/HPTypeDef.h"
- #ifdef _UDP_SUPPORT
- #include "SocketHelper.h"
- #include "common/FuncHelper.h"
- #include "common/BufferPool.h"
- #include "common/IODispatcher.h"
- #include "common/kcp/ikcp.h"
- #include <memory>
- using namespace std;
- #define DEFAULT_ARQ_NO_DELAY FALSE
- #define DEFAULT_ARQ_TURNOFF_NC FALSE
- #define DEFAULT_ARQ_FLUSH_INTERVAL 60
- #define DEFAULT_ARQ_RESEND_BY_ACKS 0
- #define DEFAULT_ARQ_SEND_WND_SIZE 128
- #define DEFAULT_ARQ_RECV_WND_SIZE 512
- #define DEFAULT_ARQ_MIN_RTO 30
- #define DEFAULT_ARQ_FAST_LIMIT 5
- #define DEFAULT_ARQ_MAX_TRANS_UNIT DEFAULT_UDP_MAX_DATAGRAM_SIZE
- #define DEFAULT_ARQ_MAX_MSG_SIZE DEFAULT_BUFFER_CACHE_CAPACITY
- #define DEFAULT_ARQ_HANND_SHAKE_TIMEOUT 5000
- #define KCP_HEADER_SIZE 24
- #define KCP_MIN_RECV_WND 128
- #define ARQ_MAX_HANDSHAKE_INTERVAL 2000
- using Fn_ArqOutputProc = int (*)(const char* pBuffer, int iLength, IKCPCB* kcp, LPVOID pv);
- DWORD GenerateConversationID();
- /************************************************************************
- 名称:ARQ 握手状态
- 描述:标识当前连接的 ARQ 握手状态
- ************************************************************************/
- enum EnArqHandShakeStatus
- {
- ARQ_HSS_INIT = 0, // 初始状态
- ARQ_HSS_PROC = 1, // 正在握手
- ARQ_HSS_SUCC = 2, // 握手成功
- };
- struct TArqCmd
- {
- public:
- static const UINT16 MAGIC = 0xBB4F;
- static const UINT8 CMD_HANDSHAKE = 0x01;
- static const UINT8 FLAG_COMPLETE = 0x01;
- static const int PACKAGE_LENGTH = 12;
- public:
- UINT16 magic;
- UINT8 cmd;
- UINT8 flag;
- DWORD selfID;
- DWORD peerID;
- public:
- static BYTE* MakePackage(UINT8 cmd, UINT8 flag, DWORD selfID, DWORD peerID, UINT16 magic = MAGIC)
- {
- BYTE* buff = new BYTE[PACKAGE_LENGTH];
- *((UINT16*)(buff + 0)) = magic;
- *((UINT8*)(buff + 2)) = cmd;
- *((UINT8*)(buff + 3)) = flag;
- *((DWORD*)(buff + 4)) = selfID;
- *((DWORD*)(buff + 8)) = peerID;
- return buff;
- }
- BYTE* MakePackage()
- {
- return MakePackage(cmd, flag, selfID, peerID, magic);
- }
- BOOL Parse(const BYTE buff[PACKAGE_LENGTH])
- {
- magic = *((UINT16*)(buff + 0));
- cmd = *((UINT8*) (buff + 2));
- flag = *((UINT8*) (buff + 3));
- selfID = *((DWORD*) (buff + 4));
- peerID = *((DWORD*) (buff + 8));
- return IsValid();
- }
- BOOL IsValid() {return (magic == MAGIC && cmd == CMD_HANDSHAKE && (flag & 0xFE) == 0);}
- public:
- TArqCmd()
- {
- ::ZeroMemory(this, sizeof(TArqCmd));
- }
- TArqCmd(UINT8 c, UINT8 f, DWORD sid, DWORD pid)
- : magic (MAGIC)
- , cmd (c)
- , flag (f)
- , selfID(sid)
- , peerID(pid)
- {
- }
- };
- struct TArqAttr
- {
- BOOL bNoDelay;
- BOOL bTurnoffNc;
- DWORD dwResendByAcks;
- DWORD dwFlushInterval;
- DWORD dwSendWndSize;
- DWORD dwRecvWndSize;
- DWORD dwMinRto;
- DWORD dwMtu;
- DWORD dwFastLimit;
- DWORD dwMaxMessageSize;
- DWORD dwHandShakeTimeout;
- public:
- TArqAttr( BOOL no_delay = DEFAULT_ARQ_NO_DELAY
- , BOOL turnoff_nc = DEFAULT_ARQ_TURNOFF_NC
- , DWORD resend_by_acks = DEFAULT_ARQ_RESEND_BY_ACKS
- , DWORD flush_interval = DEFAULT_ARQ_FLUSH_INTERVAL
- , DWORD send_wnd_size = DEFAULT_ARQ_SEND_WND_SIZE
- , DWORD recv_wnd_size = DEFAULT_ARQ_RECV_WND_SIZE
- , DWORD min_rto = DEFAULT_ARQ_MIN_RTO
- , DWORD mtu = DEFAULT_ARQ_MAX_TRANS_UNIT
- , DWORD fast_limit = DEFAULT_ARQ_FAST_LIMIT
- , DWORD max_msg_size = DEFAULT_ARQ_MAX_MSG_SIZE
- , DWORD hand_shake_timeout = DEFAULT_ARQ_HANND_SHAKE_TIMEOUT
- )
- : bNoDelay (no_delay)
- , bTurnoffNc (turnoff_nc)
- , dwResendByAcks (resend_by_acks)
- , dwFlushInterval (flush_interval)
- , dwSendWndSize (send_wnd_size)
- , dwRecvWndSize (recv_wnd_size)
- , dwMinRto (min_rto)
- , dwMtu (mtu)
- , dwFastLimit (fast_limit)
- , dwMaxMessageSize (max_msg_size)
- , dwHandShakeTimeout(hand_shake_timeout)
- {
- ASSERT(IsValid());
- }
- BOOL IsValid() const
- {
- return ((int)dwResendByAcks >= 0) &&
- ((int)dwFlushInterval > 0) &&
- ((int)dwSendWndSize > 0) &&
- ((int)dwRecvWndSize > 0) &&
- ((int)dwMinRto > 0) &&
- ((int)dwFastLimit >= 0) &&
- ((int)dwHandShakeTimeout > 2 * (int)dwMinRto) &&
- ((int)dwMtu >= 3 * KCP_HEADER_SIZE && dwMtu <= MAXIMUM_UDP_MAX_DATAGRAM_SIZE) &&
- ((int)dwMaxMessageSize > 0 && dwMaxMessageSize < ((KCP_MIN_RECV_WND - 1) * (dwMtu - KCP_HEADER_SIZE))) ;
- }
- };
- template<class T, class S> class CArqSessionT
- {
- public:
- CArqSessionT* Renew(T* pContext, S* pSocket, const TArqAttr& attr, DWORD dwPeerConvID = 0)
- {
- m_pContext = pContext;
- m_pSocket = pSocket;
- m_dwSelfConvID = ::GenerateConversationID();
- DoRenew(attr, dwPeerConvID);
- RenewExtra(attr);
- m_dwCreateTime = ::TimeGetTime();
- m_dwHSNextTime = m_dwCreateTime;
- m_dwHSSndCount = 0;
- m_bHSComplete = FALSE;
- m_enStatus = ARQ_HSS_PROC;
- Check();
- return this;
- }
- BOOL Reset()
- {
- if(!IsValid())
- return FALSE;
- {
- CReentrantCriSecLock recvlock(m_csRecv);
- CReentrantCriSecLock sendlock(m_csSend);
- if(!IsValid())
- return FALSE;
- m_enStatus = ARQ_HSS_INIT;
- DoReset();
- }
- ResetExtra();
- return TRUE;
- }
- BOOL Check()
- {
- if(IsReady())
- {
- if(m_bHSComplete || DoHandShake())
- return Flush();
- else
- return FALSE;
- }
- else if(IsHandShaking())
- return DoHandShake();
- else
- {
- ::SetLastError(ERROR_INVALID_STATE);
- return FALSE;
- }
- }
- BOOL DoHandShake()
- {
- if(!IsValid())
- {
- ::SetLastError(ERROR_INVALID_STATE);
- return FALSE;
- }
- unique_ptr<BYTE[]> bufCmdPtr;
- {
- CReentrantCriSecLock recvlock(m_csRecv);
- if(!IsValid())
- {
- ::SetLastError(ERROR_INVALID_STATE);
- return FALSE;
- }
- DWORD dwCurrent = ::TimeGetTime();
- if(::GetTimeGap32(m_dwCreateTime, dwCurrent) > m_pContext->GetHandShakeTimeout())
- {
- ::SetLastError(ERROR_TIMEOUT);
- return FALSE;
- }
- if((int)(::GetTimeGap32(m_dwHSNextTime, dwCurrent)) < 0)
- return TRUE;
- m_dwHSNextTime = dwCurrent + MIN(m_kcp->interval * (++m_dwHSSndCount), ARQ_MAX_HANDSHAKE_INTERVAL);
- UINT8 iFlag = IsReady() ? TArqCmd::FLAG_COMPLETE : 0;
- bufCmdPtr.reset(TArqCmd::MakePackage(TArqCmd::CMD_HANDSHAKE, iFlag, m_dwSelfConvID, m_dwPeerConvID));
- }
- return m_pContext->DoSend(m_pSocket, bufCmdPtr.get(), TArqCmd::PACKAGE_LENGTH);
- }
- BOOL Flush(BOOL bForce = FALSE)
- {
- if(!IsReady())
- {
- ::SetLastError(ERROR_INVALID_STATE);
- return FALSE;
- }
- {
- CReentrantCriSecTryLock recvlock(m_csRecv);
- if(recvlock.IsValid())
- {
- CReentrantCriSecTryLock sendlock(m_csSend);
- if(sendlock.IsValid())
- {
- if(!IsReady())
- {
- ::SetLastError(ERROR_INVALID_STATE);
- return FALSE;
- }
- if(bForce)
- ::ikcp_flush(m_kcp);
- else
- ::ikcp_update(m_kcp, ::TimeGetTime());
- }
- }
- }
- return TRUE;
- }
- int Send(const BYTE* pBuffer, int iLength)
- {
- if(!IsReady())
- return ERROR_INVALID_STATE;
- int rs = NO_ERROR;
- {
- CReentrantCriSecLock sendlock(m_csSend);
- if(!IsReady())
- return ERROR_INVALID_STATE;
- rs = ::ikcp_send(m_kcp, (const char*)pBuffer, iLength);
- if(rs < 0) rs = ERROR_INCORRECT_SIZE;
- }
- if(rs == NO_ERROR)
- Flush(TRUE);
- return rs;
- }
- int GetWaitingSend()
- {
- if(!IsValid())
- {
- ::SetLastError(ERROR_INVALID_STATE);
- return -1;
- }
- CReentrantCriSecLock sendlock(m_csSend);
- if(!IsValid())
- {
- ::SetLastError(ERROR_INVALID_STATE);
- return -1;
- }
- return ::ikcp_waitsnd(m_kcp);
- }
- EnHandleResult Receive(const BYTE* pData, int iLength, BYTE* pBuffer, int iCapacity)
- {
- if(iLength >= KCP_HEADER_SIZE)
- return ReceiveArq(pData, iLength, pBuffer, iCapacity);
- else if(iLength == TArqCmd::PACKAGE_LENGTH)
- return ReceiveHandShake(pData);
- else
- {
- ::WSASetLastError(ERROR_INVALID_DATA);
- return HR_ERROR;
- }
- }
- EnHandleResult ReceiveHandShake(const BYTE* pBuffer)
- {
- TArqCmd cmd;
- cmd.Parse(pBuffer);
- if(!cmd.IsValid())
- {
- ::WSASetLastError(ERROR_INVALID_DATA);
- return HR_ERROR;
- }
- {
- CReentrantCriSecLock recvlock(m_csRecv);
- if(!IsValid())
- {
- ::WSASetLastError(ERROR_INVALID_STATE);
- return HR_ERROR;
- }
-
- if(IsReady())
- {
- BOOL bReset = FALSE;
- if(cmd.selfID != m_dwPeerConvID)
- bReset = TRUE;
- else if(cmd.peerID != m_dwSelfConvID)
- {
- if(cmd.peerID != 0)
- bReset = TRUE;
- else
- {
- if(::GetTimeGap32(m_dwCreateTime) > 2 * m_pContext->GetHandShakeTimeout())
- bReset = TRUE;
- }
- }
- if(bReset)
- {
- ::WSASetLastError(ERROR_CONNRESET);
- return HR_ERROR;
- }
- }
- else
- {
- if(m_dwPeerConvID == 0)
- {
- m_dwPeerConvID = cmd.selfID;
- m_dwHSNextTime = ::TimeGetTime();
- m_dwHSSndCount = 0;
- }
- else if(cmd.selfID != m_dwPeerConvID)
- {
- ::WSASetLastError(ERROR_CONNRESET);
- return HR_ERROR;
- }
- if(cmd.peerID == m_dwSelfConvID)
- {
- m_enStatus = ARQ_HSS_SUCC;
- return m_pContext->DoFireHandShake(m_pSocket);
- }
- else if(cmd.peerID != 0)
- {
- ::WSASetLastError(ERROR_CONNRESET);
- return HR_ERROR;
- }
- }
- }
- if(!m_bHSComplete && cmd.flag == TArqCmd::FLAG_COMPLETE)
- m_bHSComplete = TRUE;
- DoHandShake();
- return HR_OK;
- }
- EnHandleResult ReceiveArq(const BYTE* pData, int iLength, BYTE* pBuffer, int iCapacity)
- {
- if(!IsReady()) return HR_IGNORE;
- {
- CReentrantCriSecLock recvlock(m_csRecv);
- if(!IsReady())
- {
- ::WSASetLastError(ERROR_INVALID_STATE);
- return HR_ERROR;
- }
- if(iLength < KCP_HEADER_SIZE)
- {
- ::WSASetLastError(ERROR_INVALID_DATA);
- return HR_ERROR;
- }
- int rs = ::ikcp_input(m_kcp, (const char*)pData, iLength);
- if(rs != NO_ERROR)
- {
- ::WSASetLastError(ERROR_INVALID_DATA);
- return HR_ERROR;
- }
- while(TRUE)
- {
- int iRead = ::ikcp_recv(m_kcp, (char*)pBuffer, iCapacity);
- if(iRead >= 0)
- {
- EnHandleResult result = m_pContext->DoFireReceive(m_pSocket, pBuffer, iRead);
- if(result == HR_ERROR)
- return result;
- }
- else if(iRead == -3)
- {
- ::WSASetLastError(ERROR_INCORRECT_SIZE);
- return HR_ERROR;
- }
- else
- break;
- }
- }
- Flush(TRUE);
- return HR_OK;
- }
- private:
- void DoRenew(const TArqAttr& attr, DWORD dwPeerConvID = 0)
- {
- ASSERT(attr.IsValid());
- DoReset();
- m_dwPeerConvID = dwPeerConvID;
- m_kcp = ::ikcp_create(m_dwSelfConvID, m_pSocket);
- ::ikcp_nodelay(m_kcp, attr.bNoDelay ? 1 : 0, (int)attr.dwFlushInterval, (int)attr.dwResendByAcks, attr.bTurnoffNc ? 1 : 0);
- ::ikcp_wndsize(m_kcp, (int)attr.dwSendWndSize, (int)attr.dwRecvWndSize);
- ::ikcp_setmtu(m_kcp, attr.dwMtu);
- m_kcp->rx_minrto = (int)attr.dwMinRto;
- m_kcp->fastlimit = (int)attr.dwFastLimit;
- m_kcp->output = m_pContext->GetArqOutputProc();
- }
- void DoReset()
- {
- if(m_kcp != nullptr)
- {
- ::ikcp_release(m_kcp);
- m_kcp = nullptr;
- }
- }
- public:
- BOOL IsValid() const {return GetStatus() != ARQ_HSS_INIT;}
- BOOL IsHandShaking() const {return GetStatus() == ARQ_HSS_PROC;}
- BOOL IsReady() const {return GetStatus() == ARQ_HSS_SUCC;}
- IKCPCB* GetKcp() {return m_kcp;}
- DWORD GetConvID() const {if(!IsValid()) return 0; return m_kcp->conv;}
- DWORD GetSelfConvID() const {return m_dwSelfConvID;}
- DWORD GetPeerConvID() const {return m_dwPeerConvID;}
-
- EnArqHandShakeStatus GetStatus() const {return m_enStatus;}
- protected:
- virtual void RenewExtra(const TArqAttr& attr) {}
- virtual void ResetExtra() {}
- public:
- CArqSessionT()
- : m_pContext (nullptr)
- , m_pSocket (nullptr)
- , m_kcp (nullptr)
- , m_enStatus (ARQ_HSS_INIT)
- , m_dwSelfConvID(0)
- , m_dwPeerConvID(0)
- , m_dwCreateTime(0)
- , m_dwHSNextTime(0)
- , m_dwHSSndCount(0)
- , m_bHSComplete (FALSE)
- {
- }
- virtual ~CArqSessionT()
- {
- Reset();
- }
- static CArqSessionT* Construct()
- {return new CArqSessionT();}
- static void Destruct(CArqSessionT* pSession)
- {if(pSession) delete pSession;}
- protected:
- T* m_pContext;
- S* m_pSocket;
- private:
- BOOL m_bHSComplete;
- DWORD m_dwHSNextTime;
- DWORD m_dwHSSndCount;
- DWORD m_dwCreateTime;
- DWORD m_dwSelfConvID;
- DWORD m_dwPeerConvID;
- EnArqHandShakeStatus m_enStatus;
- CReentrantCriSec m_csRecv;
- CReentrantCriSec m_csSend;
- IKCPCB* m_kcp;
- };
- template<class T, class S> class CArqSessionPoolT;
- template<class T, class S> class CArqSessionExT : public CArqSessionT<T, S>, public CSafeCounter
- {
- using __super = CArqSessionT<T, S>;
- using __super::Reset;
- friend class CArqSessionPoolT<T, S>;
- public:
- DWORD GetFreeTime () const {return m_dwFreeTime;}
- FD GetTimer () const {return m_fdTimer;}
- protected:
- virtual void RenewExtra(const TArqAttr& attr)
- {
- ResetCount();
- m_fdTimer = m_ioDispatcher.AddTimer(attr.dwFlushInterval, this);
- ASSERT(IS_VALID_FD(m_fdTimer));
- }
- virtual void ResetExtra()
- {
- m_ioDispatcher.DelTimer(m_fdTimer);
- m_dwFreeTime = ::TimeGetTime();
- m_fdTimer = INVALID_FD;
- }
- public:
- CArqSessionExT(CIODispatcher& ioDispatcher)
- : m_ioDispatcher(ioDispatcher)
- , m_fdTimer (INVALID_FD)
- , m_dwFreeTime (0)
- {
- }
- virtual ~CArqSessionExT()
- {
- Reset();
- }
- static CArqSessionExT* Construct(CIODispatcher& ioDispatcher)
- {return new CArqSessionExT(ioDispatcher);}
- static void Destruct(CArqSessionExT* pSession)
- {if(pSession) delete pSession;}
- private:
- CIODispatcher& m_ioDispatcher;
- FD m_fdTimer;
- DWORD m_dwFreeTime;
- };
- template<class T, class S> class CArqSessionPoolT : private CIOHandler
- {
- using CArqSessionEx = CArqSessionExT<T, S>;
- using TArqSessionList = CRingPool<CArqSessionEx>;
- using TArqSessionQueue = CCASQueue<CArqSessionEx>;
- public:
- CArqSessionEx* PickFreeSession(T* pContext, S* pSocket, const TArqAttr& attr)
- {
- DWORD dwIndex;
- CArqSessionEx* pSession = nullptr;
- if(m_lsFreeSession.TryLock(&pSession, dwIndex))
- {
- if(::GetTimeGap32(pSession->GetFreeTime()) >= m_dwSessionLockTime)
- ENSURE(m_lsFreeSession.ReleaseLock(nullptr, dwIndex));
- else
- {
- ENSURE(m_lsFreeSession.ReleaseLock(pSession, dwIndex));
- pSession = nullptr;
- }
- }
- if(!pSession) pSession = CArqSessionEx::Construct(m_ioDispatcher);
- ASSERT(pSession);
- return (CArqSessionEx*)pSession->Renew(pContext, pSocket, attr);
- }
- void PutFreeSession(CArqSessionEx* pSession)
- {
- if(pSession->Reset())
- {
- #ifndef USE_EXTERNAL_GC
- ReleaseGCSession();
- #endif
- if(!m_lsFreeSession.TryPut(pSession))
- m_lsGCSession.PushBack(pSession);
- }
- }
- void Prepare()
- {
- m_lsFreeSession.Reset(m_dwSessionPoolSize);
- m_ioDispatcher.Start(this, m_pContext->GetPostReceiveCount(), m_pContext->GetWorkerThreadCount());
- }
- void Clear()
- {
- m_ioDispatcher.Stop();
- m_lsFreeSession.Clear();
- ReleaseGCSession(TRUE);
- ENSURE(m_lsGCSession.IsEmpty());
- }
- void ReleaseGCSession(BOOL bForce = FALSE)
- {
- ::ReleaseGCObj(m_lsGCSession, m_dwSessionLockTime, bForce);
- }
- virtual BOOL OnReadyRead(const TDispContext* pContext, PVOID pv, UINT events) override
- {
- if(events & _EPOLL_ALL_ERROR_EVENTS)
- return FALSE;
- CArqSessionEx* pSession = (CArqSessionEx*)pv;
- CLocalSafeCounter localcounter(*pSession);
- if(!pSession->Check() && pSession->IsValid() && TUdpSocketObj::IsValid(pSession->m_pSocket))
- pSession->m_pContext->Disconnect(pSession->m_pSocket->connID);
- ::ReadTimer(pSession->GetTimer());
- return TRUE;
- }
- public:
- void SetSessionLockTime (DWORD dwSessionLockTime) {m_dwSessionLockTime = dwSessionLockTime;}
- void SetSessionPoolSize (DWORD dwSessionPoolSize) {m_dwSessionPoolSize = dwSessionPoolSize;}
- void SetSessionPoolHold (DWORD dwSessionPoolHold) {m_dwSessionPoolHold = dwSessionPoolHold;}
- DWORD GetSessionLockTime() {return m_dwSessionLockTime;}
- DWORD GetSessionPoolSize() {return m_dwSessionPoolSize;}
- DWORD GetSessionPoolHold() {return m_dwSessionPoolHold;}
- public:
- CArqSessionPoolT(T* pContext,
- DWORD dwPoolSize = DEFAULT_SESSION_POOL_SIZE,
- DWORD dwPoolHold = DEFAULT_SESSION_POOL_HOLD,
- DWORD dwLockTime = DEFAULT_SESSION_LOCK_TIME)
- : m_pContext(pContext)
- , m_dwSessionPoolSize(dwPoolSize)
- , m_dwSessionPoolHold(dwPoolHold)
- , m_dwSessionLockTime(dwLockTime)
- {
- }
- ~CArqSessionPoolT() {Clear();}
- DECLARE_NO_COPY_CLASS(CArqSessionPoolT)
- public:
- static const DWORD DEFAULT_SESSION_LOCK_TIME;
- static const DWORD DEFAULT_SESSION_POOL_SIZE;
- static const DWORD DEFAULT_SESSION_POOL_HOLD;
- private:
- T* m_pContext;
- DWORD m_dwSessionLockTime;
- DWORD m_dwSessionPoolSize;
- DWORD m_dwSessionPoolHold;
- TArqSessionList m_lsFreeSession;
- TArqSessionQueue m_lsGCSession;
- CIODispatcher m_ioDispatcher;
- };
- template<class T, class S> const DWORD CArqSessionPoolT<T, S>::DEFAULT_SESSION_LOCK_TIME = DEFAULT_OBJECT_CACHE_LOCK_TIME;
- template<class T, class S> const DWORD CArqSessionPoolT<T, S>::DEFAULT_SESSION_POOL_SIZE = DEFAULT_OBJECT_CACHE_POOL_SIZE;
- template<class T, class S> const DWORD CArqSessionPoolT<T, S>::DEFAULT_SESSION_POOL_HOLD = DEFAULT_OBJECT_CACHE_POOL_HOLD;
- #endif
|