python 实现简单的PerformanceCountCallHandler装饰器

Python的functools模块， 提供了3个有趣函数， partial, update_wrapper 和wraps 。

partial函数，它可以重新绑定函数的可选参数，生成一个callable的partial对象。

update_wrapper函数，把被封装函数的__name__、__module__、__doc__和 __dict__都复制到封装函数去。

wraps函数，对update_wrapper更进一步封装。

可以利用wraps函数，实现简单的方法拦截机制，来实现自己的PerformanceCountCallHandler,

具体实现：

#-*- coding: UTF-8 -*-
#-------------------------------------------------------------------------------
# Name:        模块2
# Purpose:
#
# Author:      ankier
#
# Created:     22-12-2012
# Copyright:   (c) Ankier 2012
# Licence:     <2012~2020>
#-------------------------------------------------------------------------------
import time

from functools import wraps

_Cache ={}

def PerformanceCountCallHandler():
    def _PerformanceCountCallHandler(fun):
        @wraps(fun)
        def wrap(args, kw):
            glos = fun.func_globals
            package = glos['__package__']
            model = glos['__name__']
            methodName = fun.func_name       
            
            timeStart = time.time()
            
            result = fun(args, kw)               
            
            timeEnd = time.time()
            
            print 'package:',package,' , model:', model, ' , methodName:',methodName,'. ', timeEnd - timeStart, 's'
            
            return result 
        return wrap
    return _PerformanceCountCallHandler

 

import time
from cacheCallHandler import CacheCallHandler
from performanceCountCallHandler import PerformanceCountCallHandler


@CacheCallHandler()
@PerformanceCountCallHandler()
def Sum(xx , yy ):
    sum = xx + yy
    print '------sum----- '
    time.sleep(10)            
    return sum

print Sum(4, 5)

print Sum(4, 5)

import time
from cacheCallHandler import CacheCallHandler
from performanceCountCallHandler import PerformanceCountCallHandler


@PerformanceCountCallHandler()
@CacheCallHandler()
def Sum(xx , yy ):
    sum = xx + yy
    print '------sum----- '
    time.sleep(10)            
    return sum

print Sum(4, 5)

print Sum(4, 5)

运行结果

------sum----- 
file =  /root/workspace/study/source/cacheCallHandler/cacheCallHandler.py , method =  Sum ,  performance count (s): 10.0078930855 s
9
file =  /root/workspace/study/source/cacheCallHandler/cacheCallHandler.py , method =  Sum ,  performance count (s): 6.8187713623e-05 s
9