Gevent首先取得成绩

编程问答 2022-06-11

问题描述

我目前有一些代码可以向主要的主要数据库集发出请求。每个请求所花费的时间可能超过超时时间。我想阅读第一个结果，然后决定是否等待第二个结果。 代码看起来像这样：

GREENLET_TIMEOUT = 1  # Seconds
greenlet_a = gevent.spawn(start_request,redis_a)
greenlet_b = gevent.spawn(start_request,redis_b)
gevent.joinall([greenlet_a,greenlet_b],timeout=GREENLET_TIMEOUT)

result_a = greenlet_a.value
result_b = greenlet_b.value

我看到了类似的帖子here 我尝试做这里建议的事情：

import time
import gevent
from gevent.event import Event

def first():
  time.sleep(1)
  return

def second():
  time.sleep(5)
  return

event = gevent.event.Event()
event.clear()

def callback(value):
    event.set()

lst = [gevent.spawn(first),gevent.spawn(second)]
for g in lst:
    g.link(callback)

start_time = time.time()
res = event.wait()
print("--- %s seconds ---" % (time.time() - start_time))
print(res)

我希望它花一秒钟多一点的时间才能返回，但显然正在等待两者完成。

我还尝试了使用AsyncResult的另一种方法

import time
import gevent
from gevent.event import AsyncResult

result = AsyncResult()

def first():
  result.set(1)

def second():
  time.sleep(5)
  result.set(2)

first_event = gevent.spawn(first)
second_event = gevent.spawn(second)

res = result.get()
print(res)

任何帮助将不胜感激！

解决方法

我缺少猴子补丁部分。这按预期工作！

import time
import gevent
from gevent.event import AsyncResult
from gevent import monkey
monkey.patch_all(subprocess=True)

result = AsyncResult()

def first():
  result.set(1)

def second():
  time.sleep(5)
  result.set(2)

first_event = gevent.spawn(first)
second_event = gevent.spawn(second)

res = result.get()
print(res)
asynchronouscoroutinecoroutinegeventpython