init

sdk/srvlib/include/CustomLuaProfiler.h

@@ -0,0 +1,87 @@
+#pragma once
+
+/************************************************************************
+
+                   LUA 脚本函数调用性能统计类
+
+    函数通过lua_hook的方式实现了记录并统计脚本中函数调用时间消耗的功能。对函数调用时间
+ 消耗的默认计算方式为计算函数调用前与调用好的时间差。因此误差可能比较大，但足以找出调用
+ 消耗的时间较大的函数。同时实现了将统计信息保存为文本流的功能，默认的输出格式为树形统计信
+ 息。
+
+    可以通过派生此类并覆盖HandleDebugHook方法来重新实现对函数调用信息的记录和性能的统
+ 计规则。
+
+    可以通过派生此类并覆盖Save方法来重新实现对统计信息保存为文本流的输出格式。
+
+************************************************************************/
+
+#include "ObjectCounter.h"
+class CCustomLuaProfiler
+{
+public:
+	typedef wylib::misc::CRefObject<CCustomLuaPreProcessor::CLineRangeList> CSourceLineList;
+
+public:
+	CCustomLuaProfiler();
+	~CCustomLuaProfiler();
+
+	/* 
+	* Comments:设置性能统计对象
+	* Param lua_State * L:
+	* Param CSourceLineList * pLRList: 用于取得
+	* @Return bool:如果目标对象已经被安装过性能统计，则函数返回false
+	*/
+	bool SetState(lua_State *L, CSourceLineList *pLRList = NULL);
+	/*
+	* 将函数调用性能统计保存到流中
+	* 函数返回向流中写入多少字节
+	*/
+	size_t Save(wylib::stream::CBaseStream &stm);
+	/*
+	* Comments: 获取调用时间最长的一个函数栈的时间
+	* @Return LONGLONG:
+	*/
+	LONGLONG GetMaxTotalTime();
+
+protected:
+	//在lua中存储当前性能统计调试器对象的值名称
+	static const char szPerStaticDebuggerName[];
+	//静态hook处理函数
+#ifdef WIN32
+	static void __cdecl StaticCallTimeHook(lua_State *L, lua_Debug *ar);
+#else
+	static void  StaticCallTimeHook(lua_State *L, lua_Debug *ar);
+#endif
+
+protected:
+	/** 处理调试HOOK回调 **/
+	virtual void HandleDebugHook(lua_State *L, lua_Debug *ar);
+public:
+	/** Lua函数调用性能记录 **/
+	class CLuaCallRcd : public Counter<CLuaCallRcd>
+	{
+	public:
+		char	 sName[40];		//函数名称
+		LONG	 nLineNo;		//行号
+		LONGLONG dwLastTime;	//上次调用时间
+		LONGLONG dwMaxTime;		//最大调用时间
+		LONGLONG dwMinTime;		//最小调用时间
+		LONGLONG dwCallCount;	//调用次数
+		LONGLONG dwTotalTime;	//本函数总计调用耗时
+		LPVOID	 pSubCallList;	//子级调用函数表，类型为SubCallTable
+	public:
+		~CLuaCallRcd();
+	};
+	//子级调用函数表
+	typedef CCustomHashTable<CLuaCallRcd> CSubCallTable;
+	//当前调用堆栈表
+	typedef wylib::container::CBaseList<CLuaCallRcd*> CCallStack;
+
+protected:
+	lua_State*		m_pLua;
+	CCallStack		m_CallStack;		//当前调用堆栈
+	CSubCallTable	m_RootCallTable;	//根调用记录表
+	LONGLONG		m_dwIgoreRecordTime;//在保存函数调用性能记录时忽略总时间低于此值的记录
+	CSourceLineList m_LRList;//行范围记录器，用于格式化函数统计时取得代码的文件名和行号
+};