看一下shark_init.lua这个文件（省去版权信息）：
local uv = require("uv")
local ffi = require("ffi")
package.path = package.path .. ";./deps/?.lua"
package.cpath = package.cpath .. ";./deps/?.so"
ffi.cdef[[
typedef long time_t;
typedef struct timeval {
time_t tv_sec;
time_t tv_usec;
} timeval;
int gettimeofday(struct timeval *t, void *tzp);
]]
local gettimeofday_struct = ffi.new("timeval")
shark.gettimeofday = function()
ffi.C.gettimeofday(gettimeofday_struct, nil)
return tonumber(gettimeofday_struct.tv_sec) * 1000000 +
tonumber(gettimeofday_struct.tv_usec)
end
set_interval = function(callback, interval)
local timer = uv.new_timer()
local function ontimeout()
callback(timer)
end
uv.timer_start(timer, interval, interval, ontimeout)
return timer
end
set_timeout = function(callback, timeout)
local timer = uv.new_timer()
local function ontimeout()
uv.timer_stop(timer)
uv.close(timer)
callback(timer)
end
uv.timer_start(timer, timeout, 0, ontimeout)
return timer
end
local shark_end_notify_list = {}
shark.add_end_notify = function(callback)
table.insert(shark_end_notify_list, callback)
end
shark.on_end = function(callback)
local function call_end()
for _, cb in pairs(shark_end_notify_list) do
cb()
end
callback()
os.exit(0)
end
local sigint = uv.new_signal()
uv.signal_start(sigint, "sigint", function()
call_end()
end)
local sigterm = uv.new_signal()
uv.signal_start(sigterm, "sigterm", function()
call_end()
end)
end
local function fill_line(n, max)
for i = 1, max do
if i < n then
io.write("*")
else
io.write(" ")
end
end
end
local __print_hist = function(t, cmp_func, mode)
local stdSum = 0
local array = {}
for k, v in pairs(t) do
stdSum = stdSum + v
if tostring(k) ~= "" then
array[#array + 1] = {k = k, v = v}
end
end
table.sort(array, function(v1, v2)
if cmp_func ~= nil then
return cmp_func(v1.v, v2.v)
else
if v1.v > v2.v then return true end
end
end)
if mode == "default" then
io.write(" value ---------- Distribution ---------- count\n")
end
for k, v in pairs(array) do
if mode == "default" then
io.write(string.format("%33s |", tostring(v.k)))
fill_line(v.v * 34 / stdSum, 34)
io.write(string.format("| %d\n", v.v))
else
io.write(string.format("%s\n%d\n", tostring(v.k), v.v))
end
end
end
function print_hist(t, cmp_func)
__print_hist(t, cmp_func, "default")
end
function print_hist_raw(t, cmp_func)
__print_hist(t, cmp_func, "raw")
end
shark.print_hist = print_hist
shark.print_hist_raw = print_hist_raw
（1）
local uv = require("uv")
local ffi = require("ffi")
package.path = package.path .. ";./deps/?.lua"
package.cpath = package.cpath .. ";./deps/?.so"
require("uv")加载luv模块（在main函数中已经将luaopen_luv注册进了package.preload table），require("ffi")加载luajit中的ffi模块。
然后修改package.path和package.cpath，这样可以找到shark依赖的lua文件和库。
（2）
-- microsecond precision
ffi.cdef[[
typedef long time_t;
typedef struct timeval {
time_t tv_sec;
time_t tv_usec;
} timeval;
int gettimeofday(struct timeval *t, void *tzp);
]]
local gettimeofday_struct = ffi.new("timeval")
shark.gettimeofday = function()
ffi.C.gettimeofday(gettimeofday_struct, nil)
return tonumber(gettimeofday_struct.tv_sec) * 1000000 +
tonumber(gettimeofday_struct.tv_usec)
end
ffi.new()返回一个cdata类型的值。shark.gettimeofday返回的是当前时间的微秒（us）值。
（3）
set_interval = function(callback, interval)
local timer = uv.new_timer()
local function ontimeout()
callback(timer)
end
uv.timer_start(timer, interval, interval, ontimeout)
return timer
end
set_timeout = function(callback, timeout)
local timer = uv.new_timer()
local function ontimeout()
uv.timer_stop(timer)
uv.close(timer)
callback(timer)
end
uv.timer_start(timer, timeout, 0, ontimeout)
return timer
end
set_interval和set_timeout利用了luv模块中定时器相关函数。set_interval函数会让callback函数间隔interval时间执行。而set_timeout函数会让callback函数在timeout后执行，且执行一次。要注意这两个函数的时间单位是毫秒(ms)。
（4）
local shark_end_notify_list = {}
shark.add_end_notify = function(callback)
table.insert(shark_end_notify_list, callback)
end
创建一个shark_end_notify_list的table，并且定义一个shark.add_end_notify的函数。这个函数的作用是向shark_end_notify_list添加回调函数（callback），这个table中的函数会在shark.on_end这个函数中调用，也就是在脚本退出时执行收尾工作。
（5）
shark.on_end = function(callback)
local function call_end()
--notify registered on_end function
for _, cb in pairs(shark_end_notify_list) do
cb()
end
callback()
os.exit(0)
end
local sigint = uv.new_signal()
uv.signal_start(sigint, "sigint", function()
call_end()
end)
local sigterm = uv.new_signal()
uv.signal_start(sigterm, "sigterm", function()
call_end()
end)
end
shark.on_end函数的参数是一个回调函数（callback）。在shark.on_end里定义了一个local函数：call_end，call_end首先会遍历shark_end_notify_list，并把其中的函数都执行一遍，最后调用shark.on_end函数传入的回调函数（callback）。 call_end函数就是shark.on_end函数为sigint和sigterm信号注册的信号处理函数。
（6）
local function fill_line(n, max)
for i = 1, max do
if i < n then
io.write("*")
else
io.write(" ")
end
end
end
fill_line函数用来输出*，用在接下来的__print_hist函数中。
（7）
-- standard histogram print function
-- all type keys and number value
local __print_hist = function(t, cmp_func, mode)
local stdSum = 0
local array = {}
for k, v in pairs(t) do
stdSum = stdSum + v
if tostring(k) ~= "" then
array[#array + 1] = {k = k, v = v}
end
end
table.sort(array, function(v1, v2)
if cmp_func ~= nil then
return cmp_func(v1.v, v2.v)
else
if v1.v > v2.v then return true end
end
end)
if mode == "default" then
io.write(" value ---------- Distribution ---------- count\n")
end
for k, v in pairs(array) do
if mode == "default" then
io.write(string.format("%33s |", tostring(v.k)))
fill_line(v.v * 34 / stdSum, 34)
io.write(string.format("| %d\n", v.v))
else
io.write(string.format("%s\n%d\n", tostring(v.k), v.v))
end
end
end
__print_hist函数用来打印柱状图。
local stdSum = 0
local array = {}
for k, v in pairs(t) do
stdSum = stdSum + v
if tostring(k) ~= "" then
array[#array + 1] = {k = k, v = v}
end
end
输入参数t是一个table。以上代码是遍历这个table，并且生成一个新的“array table”。这个array的值又是一个table：包含输入table的key和value。
table.sort(array, function(v1, v2)
if cmp_func ~= nil then
return cmp_func(v1.v, v2.v)
else
if v1.v > v2.v then return true end
end
end)
以上代码是对array进行排序。如果没有输入cmp_func函数参数，就用默认的比较方式。
if mode == "default" then
io.write(" value ---------- Distribution ---------- count\n")
end
for k, v in pairs(array) do
if mode == "default" then
io.write(string.format("%33s |", tostring(v.k)))
fill_line(v.v * 34 / stdSum, 34)
io.write(string.format("| %d\n", v.v))
else
io.write(string.format("%s\n%d\n", tostring(v.k), v.v))
end
end
以上代码就是打印最后的柱状图了。输出结果类似：
value ---------- Distribution ---------- count
syscalls:sys_enter_gettimeofday |******** | 25940
syscalls:sys_exit_gettimeofday |******** | 25940
syscalls:sys_enter_epoll_wait |**** | 12977
syscalls:sys_exit_epoll_wait |**** | 12977
syscalls:sys_exit_alarm |* | 3917
（8）
function print_hist(t, cmp_func)
__print_hist(t, cmp_func, "default")
end
function print_hist_raw(t, cmp_func)
__print_hist(t, cmp_func, "raw")
end
shark.print_hist = print_hist
shark.print_hist_raw = print_hist_raw
最后就是把__print_hist封装出shark.print_hist和shark.print_hist_raw两个函数供其它程序调用。
