scbc_repos/SATService/SATService/PythonExecutor.h

#ifndef __SCRIPT_EXECUTOR__
#define __SCRIPT_EXECUTOR__

#pragma once

#define BUFSIZE 4096

class CPythonExecutor
{
public:
	CPythonExecutor(void);
	~CPythonExecutor(void);

protected:
	SATHTTP::STCase *m_pCaseObj;						// 用例对象;
	HANDLE			m_hWorkThread;						// 线程句柄;
	HANDLE			m_hStdoutLogThread;					// 线程句柄;
	HANDLE			m_hStderrLogThread;					// 线程句柄;
	TCHAR			m_szScriptPath[MAX_PATH];			// 脚本路径;
	TCHAR			m_szLogPath[MAX_PATH];				// 日志路径;	
	TCHAR			m_szExtraSentence[MAX_PATH];		// 脚本命令行参数或额外要执行的语句;
	// 重定向标准输出和标准错误;
	HANDLE			m_hStdOutRead;
	HANDLE			m_hStdOutWrite;
	HANDLE			m_hStdErrorRead;
	HANDLE			m_hStdErrorWrite;
	// 子进程信息;
	STARTUPINFO			m_si;
	PROCESS_INFORMATION	m_pi;

	// 停止日志导出线程;
	BOOL			m_bStopLogExport;
	// 执行器执行时间;
	ULONGLONG		m_ulStartTickCount;
protected:
	// 工作者线程函数;
	static DWORD WINAPI _WorkerThread(LPVOID lpParam);
	// 日志导出线程;
	static DWORD WINAPI _StdoutLogExportThread(LPVOID lpParam);
	static DWORD WINAPI _StderrLogExportThread(LPVOID lpParam);

	// 重定子进程向标准输出;
	int RedirectSubprocessStdout(LPSTARTUPINFO si = NULL);	

	// 运行脚本进程;
	int RunScriptProcess();
	// 在服务进程中创建脚本进程;
	int ServiceRunScriptProcess();
	// 开启线程;
	bool StartThread();
	// 结束线程;
	void EndWorkThread();
	void EndLogThread();
	void EndThread();
	// 从管道读取日志;
	void ReadFromPipe();
	// 结束重定向句柄,恢复默认;
	void EndSubprocessStdOut();
	// 获取当前活动的会话ID;
	DWORD GetActiveSessionID();
public:
	// 初始脚本运行参数;
	bool InitScript(
		std::string strScript, 
		std::string strLogPath, 
		std::string strScriptCmd);
	// 运行脚本;
	bool StartScript();
	// 停止脚本;
	void StopScript();
	// 脚本是否结束;
	bool IsScriptOver();
	// 设置用例对象;
	void SetCaseObje(SATHTTP::STCase *pObj){ m_pCaseObj = pObj;}
	// 结束子进程;
	BOOL EndSubprocess();
	// 获取活动的CPU时间;
	ULONGLONG GetActiveTickCount() const { return m_ulStartTickCount;}
};

#endif // __SCRIPT_EXECUTOR__