mirror of
https://github.com/xfgryujk/blivechat.git
synced 2024-12-27 05:10:21 +08:00
103 lines
3.5 KiB
Python
103 lines
3.5 KiB
Python
# -*- coding: utf-8 -*-
|
||
import asyncio
|
||
import dataclasses
|
||
import datetime
|
||
import logging
|
||
from typing import *
|
||
|
||
import api.open_live
|
||
import config
|
||
import utils.async_io
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# 正在等待发送的心跳任务，game_id -> HeartbeatTask
|
||
_game_id_heart_task_map: Dict[str, 'HeartbeatTask'] = {}
|
||
|
||
|
||
@dataclasses.dataclass
|
||
class HeartbeatTask:
|
||
game_id: str
|
||
future: 'asyncio.Future[dict]'
|
||
|
||
|
||
def init():
|
||
cfg = config.get_config()
|
||
# 批量心跳只支持配置了开放平台的公共服务器，私有服务器用的人少，意义不大
|
||
if cfg.is_open_live_configured:
|
||
utils.async_io.create_task_with_ref(_game_heartbeat_consumer())
|
||
|
||
|
||
async def send_game_heartbeat(game_id) -> dict:
|
||
"""发送项目心跳。成功则返回符合开放平台格式的结果，失败则抛出异常"""
|
||
assert config.get_config().is_open_live_configured
|
||
if game_id in (None, ''):
|
||
raise api.open_live.BusinessError({'code': 4000, 'message': '参数错误', 'request_id': '0', 'data': None})
|
||
|
||
task = _game_id_heart_task_map.get(game_id, None)
|
||
if task is None:
|
||
task = HeartbeatTask(
|
||
game_id=game_id,
|
||
future=asyncio.get_running_loop().create_future(),
|
||
)
|
||
|
||
_game_id_heart_task_map[game_id] = task
|
||
# 限制一次发送的数量，数量太多了就立即发送
|
||
if len(_game_id_heart_task_map) >= 200:
|
||
await _flush_game_heartbeat_tasks()
|
||
|
||
return await task.future
|
||
|
||
|
||
async def _game_heartbeat_consumer():
|
||
while True:
|
||
try:
|
||
start_time = datetime.datetime.now()
|
||
await _flush_game_heartbeat_tasks()
|
||
cost_time = (datetime.datetime.now() - start_time).total_seconds()
|
||
|
||
# 如果等待时间太短，请求频率会太高；如果等待时间太长，前端请求、项目心跳会超时
|
||
await asyncio.sleep(4 - cost_time)
|
||
except Exception: # noqa
|
||
logger.exception('_heartbeat_consumer error:')
|
||
|
||
|
||
async def _flush_game_heartbeat_tasks():
|
||
global _game_id_heart_task_map
|
||
if not _game_id_heart_task_map:
|
||
return
|
||
game_id_task_map = _game_id_heart_task_map
|
||
_game_id_heart_task_map = {}
|
||
|
||
game_ids = list(game_id_task_map.keys())
|
||
logger.info('Sending game batch heartbeat for %d games', len(game_ids))
|
||
try:
|
||
res = await api.open_live.request_open_live(
|
||
api.open_live.GAME_BATCH_HEARTBEAT_OPEN_LIVE_URL,
|
||
{'game_ids': game_ids},
|
||
ignore_rate_limit=True
|
||
)
|
||
failed_game_ids = res['data']['failed_game_ids']
|
||
if failed_game_ids is None: # 哪个SB后端给数组传null的
|
||
failed_game_ids = set()
|
||
else:
|
||
failed_game_ids = set(failed_game_ids)
|
||
request_id = res['request_id']
|
||
except Exception as e:
|
||
for task in game_id_task_map.values():
|
||
task.future.set_exception(e)
|
||
return
|
||
if failed_game_ids:
|
||
logger.info(
|
||
'Game batch heartbeat res: %d succeeded, %d failed, request_id=%s',
|
||
len(game_ids) - len(failed_game_ids), len(failed_game_ids), request_id
|
||
)
|
||
|
||
for task in game_id_task_map.values():
|
||
if task.game_id in failed_game_ids:
|
||
task.future.set_exception(api.open_live.BusinessError(
|
||
{'code': 7003, 'message': '心跳过期或GameId错误', 'request_id': request_id, 'data': None}
|
||
))
|
||
else:
|
||
task.future.set_result({'code': 0, 'message': '0', 'request_id': request_id, 'data': None})
|