装饰器概念,有什么功能,及业务场景?
程序员文章站
2022-04-06 16:42:59
...
概念:
1.装饰器的实现是由闭包支撑的;
2.装饰器本质上是⼀个python函数,它可以在让其他函数在不需要做任何代码的变动的前提下增加额外的功能;
3.装饰器的返回值也是⼀个函数的对象,它经常用于有切面需求的场景,实现路由传参,flask的路由传参依赖于装饰器,浏览器通过url访问到装饰器的路由,从而访问视图函数获得返回的HTML页面;
应用场景:
1.可以在外层函数加上时间计算函数,计算函数运行时间;
2.计算函数运行次数;
3.可以用在框架的路由传参上;
4.插入日志,作为函数的运行日志;
5.事务处理,可以让函数实现事务的一致性,让函数要么一起运行成功,要么一起运行失败;
6.缓存,实现缓存处理;
7.权限的校验,在函数外层套上权限校验的代码,实现权限校验;
写出一个单例的装饰器(使一个本来不是单例类的类变成单例类))
- def set_func(func):
- __singleton = None
- def call_func(*args, **kwargs):
- nonlocal __singleton
- if not __singleton:
- __singleton = func(*args, **kwargs)
- return __singleton
- else:
- return __singleton
- return call_func
- @set_func
- class Std(object):
- def __init__(self, name, age):
- self.name = name
- self.age = age
- s2 = Std('jack',18)
- print(id(s2),s2.name,s2.age)
- s1 = Std('leo',23)
- print(id(s1),s1.name,s1.age)
运行结果:
- 139727292442832 jack 18
- 139727292442832 jack 18
上下文提供者(Context Provider)
上下文装饰器被用来确保函数在正确的上下文环境运行,或者确保在函数运行前和运行后执行一些代码。换言之,该装饰器进行设置或取消设置某个特定的运行环境。例如,当一条数据要在多个线程之间共享时,就需要使用锁(lock)来确保数据被保护,以防出现并发访问。锁能够像下面这样放在装饰器
上下文装饰器,通常被上下文管理器(with 语句)取代。from threading import RLock
lock = RLock()
def synchronized(function):
def _synchronized(*args, **kw):
lock.acquire()
try:
return function(*args, **kw)
finally:
lock.release()
return _synchronized
@synchronized
def thread_safe(): # make sure it locks the resource
pass
登录判断装饰器:
之前做过的一个用flask框架实现的移动app项目,里面大量用到是否已经登录的判断,如果这个业务逻辑大量重复地写在视图函数,代码的复用性很差,因此我将登录判断封装成装饰器,然后用这个装饰器装饰每一个需要验证是否登录的视图函数,代码如下:
- def login_required(view_func):
- """自定义装饰器判断用户是否登录"""
- @wraps(view_func)
- def wrapper(*args, **kwargs):
- """具体实现判断用户是否登录的逻辑"""
- user_id = session.get('user_id')
- if not user_id:
- return jsonify(errno=RET.SESSIONERR, errmsg='用户未登录')
- else:
- g.user_id = user_id
- return view_func(*args, **kwargs)
- return wrapper
推荐阅读