1

我想使用 asyncio 将 UDP 消息发送到我的 InfluxDB 服务器。这是我的同步代码：

import socket
import requests
from logger import get_logger

log = get_logger(__name__)


class InfluxDB(object):
    def __init__(self, username, password, host, port, udp=True, udp_port=4444):
        log.info("InfluxDBClient()")
        self.udp = udp
        if self.udp:
            self.udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
            self.udp_port_tuple = (host, udp_port)
        self.session = requests.session()
        self.session.auth = (username, password)
        self.base_url = "http://{}:{}".format(host, port)

    def _query(self, payload, endpoint="/query"):
        log.debug(payload)
        return self.session.post(self.base_url + endpoint, params=payload)

    def _write(self, db, data, endpoint="/write"):
        log.debug("{} \t {}".format(db, data))
        return self.session.post(self.base_url + endpoint, params={"db": db}, data=data)

    def write_udp(self, data_str):
        if self.udp:
            return self.udp_socket.sendto(data_str.encode("utf-8"), self.udp_port_tuple)
        else:
            raise Exception("UDP disabled")

如何使用 async/await 语法重写“write_udp”函数？

-- 编辑 2018 年 7 月 5 日 --

更具体地说，我不确定如何引用“socket”的异步等价物。大概有一个基于异步套接字的版本，我将其引用为 self.udp_socket = socket.socket_asyncio，然后我将通过 await socket.socket_asyncio.sendto(....) 触发消息我需要了解的是如何引用这个基于异步的套接字对象。

4

0 回答 0