Python功能点实现：函数级/代码块级计时器

本文原作者“赵澈”，通过维权骑士士值品牌馆授权转载。

工程中我们常常需要对某一个函数或者一块代码计时，从而监测系统关键位置的性能。计时方法是在代码块前后分别记录当前系统时间，然后两者相减得到代码块的耗时。最简单原始的实现类似：

这种方式缺点明显：假如系统内有很多地方都需要计时，那么每个地方都需要插入这样的计时代码，首先是重复性工作很麻烦，其次这样会降低代码的可读性，干扰对业务逻辑的理解。本文将给出一些更好的实现，主要涉及的技术是装饰器（Decorator）和运行时上下文（runtime context）。

基于装饰器的函数级计时器

第一种计时器是比较常见的函数级计时器，通过装饰器完成，将原函数改装成拥有计时功能的新函数，使其可以完成运行原来函数和计时两件事。在使用时，只用在需要计时功能的函数代码前加上类似的语法糖，这样每次调用原函数时，运行的将会是新函数。这样就大大减少了重复性劳动。

具体实现如下：

基于上下文的代码块级计时器

装饰器实现的计时器可以为函数添加计时功能，可以满足大部分情况的需要，但是假如我们想要更灵活一些，对任意一段连续的代码块做计时，怎样做？使用原始的插计时代码的方法显然不是我们想要的；也可以将代码块重构成一个函数，再在上面加装饰器，然而这就显得不够优雅。因此我做出了下面的实现。

首先了解上下文管理的概念。大致是说Python中允许创建一种叫上下文管理器（Context Manager）的对象，它可以管理一个代码块执行时的上下文信息。具体的方法是创建一个类，并为其实现和方法，前者在进入代码块时自动执行，后者在完成代码块执行时自动执行。

在使用时，通过和关键字，将的返回值绑定到某一个变量名，这个返回值里可以储存代码块运行过程中得到的一些信息，在这里就是运行时间啦。具体的实现是创建一个计时器类，在时记录代码块运行的开始时间，时记录完成时间、计算并储存耗时到实例中。在使用时，将加到要计时的代码块前面，中将会储存代码块耗时，可以任意使用。

在这个基础上，我们还可以做出一个装饰器来实现基于上下文的函数级计时器。

具体实现如下：

更灵活的实现

更优雅地，我们还可以使用自带的，参考官方示例，做出既可以又可以作为的计时器：

本文仅代表作者观点，与本公众号立场无关。