-
Notifications
You must be signed in to change notification settings - Fork 3
/
lperf.h
72 lines (61 loc) · 1.66 KB
/
lperf.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#pragma once
#include <stdio.h>
#include <stdint.h>
#include <stdbool.h>
#include <string.h>
#include <string>
#include <map>
#include <unordered_map>
#include <fcntl.h>
#include <unistd.h>
#include <sys/time.h>
#include <signal.h>
extern "C" {
#include "lua.h"
#include "lualib.h"
#include "lauxlib.h"
}
struct skynet_context;
extern "C" void skynet_error(struct skynet_context* context, const char* msg, ...);
// Notice: same with 'snlua' in skynet/service-src/service_snlua.c
struct snlua {
lua_State* L;
struct skynet_context* ctx;
size_t mem;
size_t mem_report;
size_t mem_limit;
lua_State* activeL;
#if LUA_VERSION_NUM >= 504
int trap; // ATOM_INT trap
#endif
};
static const uint8_t MAX_FUNC_NAME_SIZE = 80;
static const char* const IGNORE_NAME[] = {"?", "function 'xpcall'", "function 'pcall'"};
static const int VALID_MIN_ID = sizeof(IGNORE_NAME) / sizeof(const char*);
static const int PROF_HZ = 50; // hard code here
static const int MAX_STACK_SIZE = 64;
static const int MAX_BUCKET_SIZE = 1 << 12;
struct sample {
int count;
int depth;
uint32_t stack[MAX_STACK_SIZE];
struct sample* next;
};
class lperf_app {
public:
lperf_app(const char* prof_file);
~lperf_app();
uint32_t get_func_Id(std::string name);
void add_record(uint32_t* stack, int depth);
void save_records();
void write_file(const char* buf, size_t len);
public:
std::unordered_map<std::string, uint32_t> m_func_name2Id;
std::unordered_map<uint32_t, std::string> m_func_Id2name;
std::string m_filename;
int m_fd;
lua_State* m_gL;
int m_records_num;
int m_samples_num;
struct sample* m_buckets[MAX_BUCKET_SIZE];
};