123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152 |
- #ifndef __UPGRADERUN_SERVER_G_H
- #define __UPGRADERUN_SERVER_G_H
- #pragma once
- // This code is generated by spgen tool!
- #include "UpgradeRun_def_g.h"
- namespace UpgradeRun {
- class UpgradeRunService_ServerSessionBase : public CServerSessionBase
- {
- public:
- UpgradeRunService_ServerSessionBase() { }
- virtual ~UpgradeRunService_ServerSessionBase() { }
- virtual bool IsExclusive() { return false; }
- virtual bool IsSessionOverlap() { return true; }
- virtual ErrorCodeEnum GetMessageAttr(DWORD dwMessageID, DWORD dwSignature, bool &bOverlap)
- {
- ErrorCodeEnum Error = Error_Succeed;
- switch (dwMessageID) {
- case UpgradeRunService_Method_PushUpdateTask:
- if (dwSignature == UpgradeRunService_MethodSignature_PushUpdateTask) {
- bOverlap = true;
- } else {
- Error = Error_MethodSignatureFailed;
- }
- break;
- case UpgradeRunService_Method_GetRunState:
- if (dwSignature == UpgradeRunService_MethodSignature_GetRunState) {
- bOverlap = true;
- } else {
- Error = Error_MethodSignatureFailed;
- }
- break;
- case UpgradeRunService_Method_UpdateTaskCancel:
- if (dwSignature == UpgradeRunService_MethodSignature_UpdateTaskCancel) {
- bOverlap = true;
- } else {
- Error = Error_MethodSignatureFailed;
- }
- break;
- default:
- Error = Error_MethodNotFound;
- break;
- }
- return Error;
- }
- int CheckMessageSignature(DWORD dwMessageID, DWORD dwSignature)
- {
- ErrorCodeEnum Error = Error_Succeed;
- switch (dwMessageID) {
- case UpgradeRunService_Method_PushUpdateTask:
- if (dwSignature != UpgradeRunService_MethodSignature_PushUpdateTask) {
- Error = Error_MethodSignatureFailed;
- }
- break;
- case UpgradeRunService_Method_GetRunState:
- if (dwSignature != UpgradeRunService_MethodSignature_GetRunState) {
- Error = Error_MethodSignatureFailed;
- }
- break;
- case UpgradeRunService_Method_UpdateTaskCancel:
- if (dwSignature != UpgradeRunService_MethodSignature_UpdateTaskCancel) {
- Error = Error_MethodSignatureFailed;
- }
- break;
- default:
- Error = Error_MethodNotFound;
- break;
- }
- return Error;
- }
- virtual void Handle_PushUpdateTask(SpOnewayCallContext<UpgradeRunService_PushUpdateTask_Info>::Pointer ctx)
- {
- /// override by user
- }
- virtual void Handle_GetRunState(SpReqAnsContext<UpgradeRunService_GetRunState_Req, UpgradeRunService_GetRunState_Ans>::Pointer ctx)
- {
- /// override by user
- }
- virtual void Handle_UpdateTaskCancel(SpOnewayCallContext<UpgradeRunService_UpdateTaskCancel_Info>::Pointer ctx)
- {
- /// override by user
- }
- virtual void OnRequest(CSmartPointer<ITransactionContext> pTransactionContext)
- {
- CAutoBuffer Buf;
- DWORD dwMessageID;
- DWORD dwMessageSignature;
- ErrorCodeEnum Error = pTransactionContext->GetReceiveBuffer(dwMessageID, dwMessageSignature, Buf);
- if (Error == Error_Succeed) {
- #ifdef DEBUG
- assert(CheckMessageSignature(dwMessageID, dwMessageSignature) == Error_Succeed);
- #else
- if (CheckMessageSignature(dwMessageID, dwMessageSignature) != Error_Succeed) {
- pTransactionContext->SendAnswer(Error_MethodSignatureFailed);
- return;
- }
- #endif
- switch (dwMessageID) {
- case UpgradeRunService_Method_PushUpdateTask:
- {
- SpOnewayCallContext<UpgradeRunService_PushUpdateTask_Info>::Pointer ctx;
- ctx.Attach(new SpOnewayCallContext<UpgradeRunService_PushUpdateTask_Info>());
- SpBuffer2Object(Buf, ctx->Info);
- Handle_PushUpdateTask(ctx);
- }
- break;
- case UpgradeRunService_Method_GetRunState:
- {
- SpReqAnsContext<UpgradeRunService_GetRunState_Req,UpgradeRunService_GetRunState_Ans>::Pointer ctx;
- ctx.Attach(new SpReqAnsContext<UpgradeRunService_GetRunState_Req,UpgradeRunService_GetRunState_Ans>(pTransactionContext));
- SpBuffer2Object(Buf, ctx->Req);
- Handle_GetRunState(ctx);
- }
- break;
- case UpgradeRunService_Method_UpdateTaskCancel:
- {
- SpOnewayCallContext<UpgradeRunService_UpdateTaskCancel_Info>::Pointer ctx;
- ctx.Attach(new SpOnewayCallContext<UpgradeRunService_UpdateTaskCancel_Info>());
- SpBuffer2Object(Buf, ctx->Info);
- Handle_UpdateTaskCancel(ctx);
- }
- break;
- default:
- assert(0);
- break;
- }
-
- } else {
- pTransactionContext->SendAnswer(Error);
- }
- }
- };
- ///////////////////////////
- } // namespace UpgradeRun
- #endif // __UPGRADERUN_SERVER_G_H
|