gevent 使用踩坑

简单介绍

1. gevent 基本概念：
      调度器： hub
             上下文切换管理： switch
             主循环： loop
      协程： greenlet
2. gevent 特性：
       优点：
               高效，实现简单，易维护
       缺点：
               和go不一样，并不是python原生支持的功能，所以使用起来难免会踩一些坑，但是由于并不是实现方式有问题，所以存在着点弊病并没有什么问题。

开始使用

1. 什么情况下可以使用：
       足够了解自己服务所使用的io,  写的代码足够规范，否则会出现出了问题都不知道那里有问题。
       需要并行化的代码不能阻塞时间过长，否则没有意义。
       存在大量的io, 前提是可以变成非阻塞的，否则不能满足条件2。

2. 由于发现api的代码满足以上三条，所以开始进行优化：
   第一版代码如下：
          

   class ParallelTask(object):     def __init__(self, timeout=5):         # 覆盖python原生的socket         gevent.monkey.patch_socket()         self.timeout = timeout         self.task = []       def taskAppend(self, task, *args):         self.task.append(gevent.spawn(task, *args))       def run(self):         errno, err_msg = get_error(AwemeStatus.SUCCESS)         try:             gevent.joinall(self.task, timeout=self.timeout)             return errno, err_msg         except Exception, ex:             logger.exception("[ParallelTask] run task fail error=%s" % ex)             errno, err_msg = get_error(AwemeStatus.REE_PARALLEL_TASK)             return errno, err_msg

   # handler def __getAwemeList(self, category_id, req_type):     pass    # 创建一个并行处理的任务 paralle_obj = ParallelTask() paralle_obj.taskAppend(self.__getAwemeList, category_id, req_type) errno, err_msg = paralle_obj.run()

   代码看似没有问题，自测也没发现什么问题。

3. 上线观察：
   果不其然，出问题了。。。
   
   错误发生在 challengeDetail 接口， 也是奇怪，上的明明是category 接口。
   不过看错误知道应该是掉RPC没有成功， 查看RPC的日志可以看到全是fail。
4. 分析问题：
    首先RPC失败肯定是由于使用gevent 引起的。但是我在这category使用gevent 为啥会影响challengeDetail 呢。
    观察最上面的 ParallelTask 代码，里面使用了monkey patch , 没错就是这个东西，重写了python原生socket ，使其支持了非阻塞io。
    所以在没有并行化的代码里使用RPC调用，里面使用的socket变成了非阻塞的，而这些调用不受hup控制，所以肯定会出现调用失败的情况。

5. 解决：
   解决办法很简单，在用完monkey patch 之后恢复原生的socket不久好了。 gevent 没有提供接口恢复，所以自己实现了下。

   def repatching(item):     try:         # 需要重新写回的module         module = __import__(item)         # 需要重写的属性         saved = gevent.monkey.saved         mapper = saved.get(item, {})         for attr in mapper:             old_value = mapper.get(attr)             if not old_value:                 continue             setattr(module, attr, old_value)     except Exception, ex:         logger.exception("[gevent] repatching fail error=%s" % ex)   class ParallelTask(object):     def __init__(self, timeout=5):         # 覆盖python原生的socket, 使用完记得repatching 不然会有未知错误         gevent.monkey.patch_socket()         self.timeout = timeout         self.task = []       def taskAppend(self, task, *args):         self.task.append(gevent.spawn(task, *args))       def run(self):         errno, err_msg = get_error(AwemeStatus.SUCCESS)         try:             gevent.joinall(self.task, timeout=self.timeout)             return errno, err_msg         except Exception, ex:             logger.exception("[ParallelTask] run task fail error=%s" % ex)             errno, err_msg = get_error(AwemeStatus.REE_PARALLEL_TASK)             return errno, err_msg         finally:             # 使用完非阻塞的网络io之后一定要改回来             repatching(‘socket‘)

6. 线上观察：
   bug 修复之后线上没有新的问题爆出来，至今稳定运行着， 本来打算用go优化的接口看样子也不需要了， 并行化之后接口的时延由3s 降低到了300ms, 降低的幅度也符合预期 （30个rpc并行）

gevent 使用踩坑

原文：http://www.cnblogs.com/acvc/p/6307412.html