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