使用 asyncio/aiohttp 未完成响应负载

Question

我编写了一个Python 3.7脚本，该脚本异步(asyncio 3.4.3 and aiohttp 3.5.4)使用单个SOQL语句查询的多个对象创建Salesforce批量 API (v45.0)作业/批处理，等待批处理完成，完成后将结果下载（流式传输）到服务器，进行一些数据转换，然后最终将结果同步上传到SQL Server 2016 SP1 (13.0.4560.0) 。 我已经进行了很多成功的试运行，并认为它运行良好，但是，我最近开始间歇性地收到以下错误，并且对如何修复感到不知所措，因为关于此的报告/解决方案很少在网上：

aiohttp.client_exceptions.ClientPayloadError：响应负载未完成

示例代码片段：

import asyncio,aiohttp,aiofiles
from simple_salesforce import Salesforce
from xml.etree import ElementTree

#Establish a session using the simple_salesforce module
sf = Salesforce(username=username,
                password=password,
                security_token=securityToken,
                organizationId=organizationId)
sfAPIURL = 'https://myinstance.salesforce.com/services/async/45.0/job/'
sfDataPath = 'C:/Salesforce/Data/'

#Dictionary to store information for the object/job/batch while the script is executing
objectDictionary = 
{'Account': {'job':
                {'batch': {'id': '8596P00000ihwpJulI','results': ['8596V00000Bo9iU'],'state': 'Completed'},
             'id': '8752R00000iUjtReqS'},
             'soql': 'select Id,Name from Account'},

 'Contact': {'job':
                {'batch': {'id': '9874G00000iJnBbVgg','results': ['7410t00000Ao9vp'],'state': 'Completed'},
             'id': '8800o00000POIkLlLa'},
             'soql': 'select Id,Name from Contact'}}

async def retrieveResults(jobId, batchId, sfObject):
    headers = {"X-SFDC-Session": sf.session_id, 'Content-Encoding': 'gzip'}
    async with aiohttp.ClientSession() as session:
        async with session.get(url=f'{sfAPIURL}{jobId}/batch/{batchId}/result', headers=headers) as r:
            data = await r.text()
            batchResults = ElementTree.fromstring(data) #list of batch results
            for resultID in batchResults:
                async with session.get(url=f'{sfAPIURL}{jobId}/batch/{batchId}/result/{resultID.text}', headers=headers, timeout=None) as r:
                    async with aiofiles.open(f'{sfDataPath}{sfObject}_TEMP_JOB_{jobId}_BATCH_{batchId}_RESULT_{resultID.text}.csv', 'wb') as outfile: #save in temporary file for manipulation later
                        while True:
                            chunk = await r.content.read(81920)
                            if not chunk:
                                break
                            await outfile.write(chunk)

async def asyncDownload():
    await asyncio.gather(*[retrieveResults(objectDictionary[sfObject]['job']['id'], objectDictionary[sfObject]['job']['batch']['id'], sfObject) for sfObject in objectDictionary])

if __name__ == "__main__":
    asyncio.run(asyncDownload())

回溯（错误行与上面的代码片段不匹配）：

回溯（最近一次通话最后）：

文件“C:\Code\salesforce.py”，第 252 行，在 asyncio.run(asyncDownload())

文件“C:\Program Files\Python37\lib\asyncio\runners.py”，第 43 行，运行返回 loop.run_until_complete(main)

文件“C:\Program Files\Python37\lib\asyncio\base_events.py”，第 584 行，在 run_until_complete 返回 future.result()

文件“C:\Code\salesforce.py”，第 241 行，在 asyncDownload 中等待 asyncio.gather(*[retrieveResults(objectDictionary[sfObject]['job']['id'], objectDictionary[sfObject]['job'] ['batch']['id'], sfObject) for sfObject in objectDictionary])

文件“C:\Code\salesforce.py”，第 183 行，retrieveResults 块 = await r.content.read(81920)

文件“C:\Program Files\Python37\lib\site-packages\aiohttp\streams.py”，第 369 行，读取等待 self._wait('read')

文件“C:\Program Files\Python37\lib\site-packages\aiohttp\streams.py”，第 297 行，在 _wait 等待服务员

aiohttp.client_exceptions.ClientPayloadError：响应负载未完成

问题的根源似乎从r.content.read(81920)开始，它应该是 81920 字节块中的流式数据，但这是我所能得到的。

我认为这不是我的网络问题，因为在此作业运行时，还有其他小型作业连接到此服务器上的外部源，这些作业没有问题地完成。 有谁知道这里发生了什么？

谢谢！

-编辑：

我试过iter_any()而不是read()并且仍然得到同样的错误......

async for data in r.content.iter_any():
    await outfile.write(data)

我试过readline()仍然得到同样的错误......

async for line in r.content.readline():
    await outfile.write(line)

从那以后，我在代码的错误处理部分（不包括在原始问题中）中使用了一些重试功能，最终允许作业完成。 有效负载错误仍在发生，这仍然是主要问题，但重试下载是一种成功的解决方法。 如果有人能够提供更多信息，问题仍然存在。

Answer 1

嗨，您是否尝试将 await asyncio.sleep(0) 插入：

                    ...
                    while True:
                        chunk = await r.content.read(81920)
                        await asyncio.sleep(0)
                        if not chunk:
                            break
                        await outfile.write(chunk)
                    ...

Answer 2

我在 Amazon Lambda 中遇到了这个错误（这是在请求时抛出的）

await asyncio.gather(*tasks) # 类似 asyncio.ensure_future() 的任务

解决方案，修复构建环境：

FROM amazonlinux:2 AS 

至

FROM lambci/lambda:build-python3.8 

我想问题是 .so 文件或较低级别的文件，由库内部用于管理协程，与 lambda 环境不兼容。 因此，在正确的 docker 基础上构建可以解决问题。