一日一技：在Python中实现延迟调用

[[413590]]

成都创新互联公司网站建设提供从项目策划、软件开发，软件安全维护、网站优化(SEO)、网站分析、效果评估等整套的建站服务，主营业务为网站设计、网站制作，成都App定制开发以传统方式定制建设网站，并提供域名空间备案等一条龙服务，秉承以专业、用心的态度为用户提供真诚的服务。成都创新互联公司深信只要达到每一位用户的要求，就会得到认可，从而选择与我们长期合作。这样，我们也可以走得更远！

熟悉 Golang 的同学都知道，Golang 里面有一个关键词叫做defer，它可以实现延迟调用。

实际上在 Python 里面也有相关的语法，那就是contextlib.ExitStack。

我们来看这样一个场景：

我有一个函数parse，它的作用是从 Redis 中持续读入数据，并写入到MongoDB 中。示例代码如下：

 
 
 
 
  
  
  
  import json   
  
  
  import redis   
  
  
  import pymongo   
  
  
     
  
  
  client = redis.Redis()   
  
  
  handler = pymongo.MongoClient().test.data   
  
  
     
  
  
  def parse():   
  
  
      data = client.lpop('test')   
  
  
      if not data:   
  
  
          return   
  
  
      handler.insert_one(json.loads(data))

但现在我想增加一个需求，当Redis 读取结束或者读取数据报错的时候，能把当前的时间也写入到MongoDB 中。

那么代码可能变成下面这样：

 
 
 
 
  
  
  
  import json   
  
  
  import redis   
  
  
  import datetime   
  
  
  import pymongo   
  
  
     
  
  
  client = redis.Redis()   
  
  
  handler = pymongo.MongoClient().test.data   
  
  
     
  
  
  def parse():   
  
  
      while True:   
  
  
          try:   
  
  
              data = client.lpop('test')   
  
  
              if not data:   
  
  
                  handler.insert_one({'finished': True, 'ts': datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')})   
  
  
                  return   
  
  
              handler.insert_one(json.loads(data))   
  
  
          except Exception:   
  
  
              handler.insert_one({'finished': True, 'ts': datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S'})

可以看到，代码变得很难看了。

现在，我们可以使用延迟调用来让代码变得更好看。

要实现这个目的，就可以开始使用ExitStack了。它可以注册多个回调函数，在退出上下文缩进时执行。

我们先来看一个简单的例子：

 
 
 
 
  
  
  
  import contextlib   
  
  
     
  
  
  def callback_1():   
  
  
      print('我是第一个回调函数')   
  
  
     
  
  
  def callback_2(x):   
  
  
      print(f'我是第二个回调函数，传入参数：{x}')   
  
  
     
  
  
     
  
  
  with contextlib.ExitStack() as stack:   
  
  
      stack.callback(callback_1)   
  
  
      stack.callback(callback_2, 100)   
  
  
      print(12345)   
  
  
      print('xxxx')   
  
  
  print('退出缩进')

运行效果如下图所示：

可以看出以下特点：

被添加的回调函数进入了一个栈，所以后添加的回调函数先调用
回调函数会在结束缩进的时候被调用

现在我们来人工构造一个异常：

可以看到，即使缩进里面出现了报错，回调函数仍然可以正常运行。等所有回调函数运行完成以后，Python 才会退出。

基于以上特点，我们就可以来重构最开始的代码了：

 
 
 
 
  
  
  
  import json   
  
  
  import redis   
  
  
  import datetime   
  
  
  import pymongo   
  
  
  import contextlib   
  
  
     
  
  
  client = redis.Redis()   
  
  
  handler = pymongo.MongoClient().test.data   
  
  
     
  
  
  def add_ts():   
  
  
      handler.insert_one({'finished': True, 'ts': datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')})   
  
  
     
  
  
  def parse():   
  
  
      with contextlib.ExitStack() as stack:   
  
  
          stack.callback(add_ts)   
  
  
          while True:   
  
  
              data = client.lpop('test')   
  
  
              if not data:   
  
  
                  return   
  
  
              handler.insert_one(json.loads(data))

无论是正常运行结束还是运行过程中报错，add_ts函数都会正常运行，确保始终增加一条日期数据。

本文转载自微信公众号「未闻Code」，可以通过以下二维码关注。转载本文请联系未闻Code公众号。

分享标题：一日一技：在Python中实现延迟调用
转载注明：http://www.shufengxianlan.com/qtweb/news18/126168.html

网站建设、网络推广公司-创新互联，是专注品牌与效果的网站制作，网络营销seo公司；服务项目有等

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源：创新互联

猜你还喜欢下面的内容