myvtm/Module/mod_UpgradeRun/mod_UpgradeRun.h

#ifndef RVC_MOD_UPGRADE_RUN_H_
#define RVC_MOD_UPGRADE_RUN_H_

#include "SpBase.h"
#include "UpgradeRunFSM.h"
#include "UpgradeRun_server_g.h"
#include "UpgradeRun_msg_g.h"

#include "SpTest.h"
#include "modVer.h"

using namespace UpgradeRun;

// Éý¼¶Ö´ÐÐ UpgradeRun	0x507
class CUpgradeRunEntity : public CEntityBase, public ISysVarListener
{
public:
	CUpgradeRunEntity(){}
	virtual ~CUpgradeRunEntity() {}
	virtual const char *GetEntityName() const { return "UpgradeRun"; }
	virtual bool IsService()const{return true;}

	const char* GetEntityVersion() const override
	{
		return MODULE_VERSION_FULL;
	}

	ON_ENTITYT_TEST()

	virtual void OnStarted();
	virtual void OnPreClose(EntityCloseCauseEnum eCloseCause,CSmartPointer<ITransactionContext> pTransactionContext);

	virtual void OnSysVarEvent(const char *pszKey,  const char *pszValue,const char *pszOldValue,const char *pszEntityName);
	virtual CServerSessionBase *OnNewSession(const char* /*pszRemoteEntityName*/, const char * /*pszParam*/);

	ErrorCodeEnum PushUpdateTask(const CSimpleStringA &strPackName, const int nPackType);
	ErrorCodeEnum GetRunState(CSimpleStringA &strPackName, char &cState);
	ErrorCodeEnum UpdateTaskCancel(const CSimpleStringA &strPackName);
	ErrorCodeEnum BroadcastUpgradeCheckEvent(UpgradeCheckEvent &e);
	ErrorCodeEnum BroadcastUpgradeDoneEvent(UpgradeDoneEvent &e);
	bool GetServerPackUpgradeResult(const char *pszPackageName, CSimpleStringA &strVerison, CSimpleStringA &strErrMsg);
	bool GetFirmwarePackUpgradeResult(const char *pszPackageName, CSimpleStringA &strVerison, CSimpleStringA &strErrMsg);

private:
	CUpgradeRunFSM m_FSM;
};


class CUpgradeRunSession : public UpgradeRunService_ServerSessionBase
{
public:
	CUpgradeRunSession(CUpgradeRunEntity *pEntity) : m_pEntity(pEntity) {}
	virtual ~CUpgradeRunSession() {}

	virtual void Handle_PushUpdateTask(SpOnewayCallContext<UpgradeRunService_PushUpdateTask_Info>::Pointer ctx)
	{
		m_pEntity->PushUpdateTask(ctx->Info.strPackName, ctx->Info.nTaskType);
	}

	virtual void Handle_GetRunState(SpReqAnsContext<UpgradeRunService_GetRunState_Req, UpgradeRunService_GetRunState_Ans>::Pointer ctx)
	{
		ErrorCodeEnum rc = m_pEntity->GetRunState(ctx->Ans.strPackName, ctx->Ans.state);
		ctx->Answer(rc);
	}

	virtual void Handle_UpdateTaskCancel(SpOnewayCallContext<UpgradeRunService_UpdateTaskCancel_Info>::Pointer ctx)
	{
		m_pEntity->UpdateTaskCancel(ctx->Info.strPackName);
	}

private:
	CUpgradeRunEntity *m_pEntity;
};
#endif //RVC_MOD_UPGRADE_RUN_H_