如何在Sanic中使用aiohttp ClientSession?

时间:2018-08-01 16:38:13

标签: python python-3.x asynchronous aiohttp sanic

我试图了解在Sanic中使用aiohttp的正确方法。

在aiohttp documentation中,我发现了以下内容:

  

不要为每个请求创建会话。每个应用程序很可能需要一个会话来执行所有请求。   更复杂的情况可能需要每个站点进行一次会话,例如一个用于Github,另一个用于Facebook API。无论如何,为每个请求建立会话是一个非常糟糕的主意。   会话内部包含一个连接池。连接重用和保持活动状态(默认情况下均处于启用状态)可能会提高整体性能。

当我去Sanic文档时,我会找到一个像这样的例子:

这是一个示例:

from sanic import Sanic
from sanic.response import json

import asyncio
import aiohttp

app = Sanic(__name__)

sem = None

@app.route("/")
async def test(request):
    """
    Download and serve example JSON
    """
    url = "https://api.github.com/repos/channelcat/sanic"

    async with aiohttp.ClientSession() as session:
         async with sem, session.get(url) as response:
         return await response.json()

app.run(host="0.0.0.0", port=8000, workers=2)

这不是管理aiohttp会话的正确方法...

那么正确的方法是什么?
我应该在应用中启动一个会话并将该会话注入所有层中的所有方法吗?

我发现的唯一问题是this,但这无济于事,因为我需要创建自己的类来使用该会话,而不是sanic。
还可以在Sanic文档中找到this,该文档指出您不应在eventloop之外创建会话。

我有点困惑:( 正确的做法是什么?

2 个答案:

答案 0 :(得分:4)

为了使用单个aiohttp.ClientSession,我们只需要实例化一次会话,并在应用程序的其余部分中使用该特定实例。

要实现这一点,我们可以使用before_server_start listener,它允许我们在应用程序提供第一个字节之前创建实例。

from sanic import Sanic 
from sanic.response import json

import aiohttp

app = Sanic(__name__)

@app.listener('before_server_start')
def init(app, loop):
    app.aiohttp_session = aiohttp.ClientSession(loop=loop)

@app.listener('after_server_stop')
def finish(app, loop):
    loop.run_until_complete(app.session.close())
    loop.close()

@app.route("/")
async def test(request):
    """
    Download and serve example JSON
    """
    url = "https://api.github.com/repos/channelcat/sanic"

    async with app.aiohttp_session.get(url) as response:
        return await response.json()


app.run(host="0.0.0.0", port=8000, workers=2)

代码分解:

  • 我们正在创建aiohttp.ClientSession,将Sanic应用在开始时创建的循环作为参数传递,从而避免了this pitfall的使用过程。
  • 我们将该会话存储在Sanic app中。
  • 最后,我们正在使用此会话发出请求。

答案 1 :(得分:2)

从本质上讲,这就是我在做什么。

我创建了一个模块(interactions.py),该模块具有以下功能:

async def get(url, headers=None, **kwargs):
    async with aiohttp.ClientSession() as session:
        log.debug(f'Fetching {url}')
        async with session.get(url, headers=headers, ssl=ssl) as response:
            try:
                return await response.json()
            except Exception as e:
                log.error(f'Unable to complete interaction: {e}')
                return await response.text()

然后我就这样await

results = await interactions.get(url)

我不确定为什么这不是“正确方法”。一旦我的请求完成,就可以关闭该会话(至少出于我的需要)。