解决python封装Logging模块后,log位置显示错误的问题

2022-10-31 15:28:02 浏览数 (1)

引入

今天由于项目需要,要将logging库二次封装成一个类,以实现一些自定义的功能。

我将二次封装了一个logService类,然后在其中同样也实现info, warn, error等日志函数。额外加了一个将日志存入数据库的功能。

大概是像下面这样子:

但是在封装的过程中,出现了一个问题:log中,不能正确显示打日志的地方的代码位置了。

表现如图所示:

我们希望打log的时候显示的代码位置是出错的地方的位置,但是这里显示的是logService类中的代码位置。这该怎么办呢?

解决过程

上网查了一下,排在前面的几条结果都是通过直接将logger返回给其他的类来实现的。但是我必须封装多一层以实现自定义的功能。因此,我尝试去logging库中寻找答案。

首先,我们可以看到,error()函数是这样子的:

它调用了self._log()函数,长这样:

我们可以注意到,参数列表之中有一个叫做stacklevel的参数,且默认值为1.并且这个参数被传递到了一个叫做findCaller的函数里面。这一点值得我们注意,因为这就是解决问题的关键。

要追踪函数的调用位置,那必然与调用栈(callstack)有关,而这里的stacklevel,又传递到了一个“寻找调用者”的函数之中,这是值得我们警觉的。我们猜想:stacklevel也许就是往前追踪的调用栈的层数的意思?因为从打log函数的位置往前追1层,那就是打log的位置了。

带着疑惑,我们继续看findCaller函数,我们可以看到:

根据函数的文档字符串描述,这个函数的作用是“寻找调用者的栈帧,以便于我们定位文件名、代码行数、函数名”。从这一点来看,我们就能明白,这个函数就是逐级的找调用栈的栈帧,然后获取里边的信息。

这个时候,我们可以看到这行代码:

代码语言:javascript复制
f = currentframe()

这行代码字面上写的是“获取当前栈帧”的意思,那么,什么叫做“当前栈帧”呢?我们继续往下看,看这个函数实现:

我们可以根据上面的代码得知,该函数的作用就是返回从当前位置栈帧往前追踪的第3个栈帧。从currentframe()函数算起,往前的第3个栈帧就是error()函数所在的栈帧。

看到这里,小伙伴们肯定很疑惑:为什么是error()函数而不是调用logging.error()的地方呢?原因是,函数只有被压栈了才会形成一个栈帧.举个小例子,函数A调用函数B,那么在调用的时候,A就会被压栈,因此栈中就有了A的栈帧。

可是,这也不是调用error()的地方所在的栈帧呀?别忘了,前面我们看到了那个stacklevel参数,默认为1。从error()函数往上再追踪1层不就是调用logging.error()的地方了吗?这就是stacklevel默认等于1的原因。

我们再回到findCaller()函数:

我们可以看到,在获取到logging.error()所在的栈帧之后,执行了一个f=f_back的操作。从命名上看,这个f_back很显然是指向更深一层栈帧的指针。结合stacklevel默认等于1的情况以及下面的while循环来看,验证了我们的猜想。

然后下方的while循环对于stacklevel>1的情况,不断的往更深层追踪栈帧。

看到这里,我们已经可以回答最初的问题了:如何解决log位置显示错误的问题?答案就是,我们只需要将stacklevel设置为2即可,这样就会再往上一层,追踪到调用logService.error()的地方,而不是调用logging.error()的位置。

我们只需要1行代码即可应用更改:

更改后,情况如下所示,调用logService.error()的位置被正确的显示了出来。

结语

其实本文所述的问题并不是特别的复杂,只是需要熟悉计算机的函数调用的原理:调用函数时,将当前函数的局部变量和返回地址压栈,函数return的时候,将ret_addr填写到指令指针寄存器之中。当你对这一基本原理熟悉之后,就能很顺理成章的解决本文所述的问题了~

0 人点赞