88 lines
3.0 KiB
C++
88 lines
3.0 KiB
C++
#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;//行范围记录器，用于格式化函数统计时取得代码的文件名和行号
|
||
};
|