Function ws_client

runtime/python/websocket/funasr_wss_client.py:474–510

(id, chunk_begin, chunk_size)

async def ws_client(id, chunk_begin, chunk_size):
    if args.audio_in is None:
        chunk_begin = 0
        chunk_size = 1
    global websocket, voices, offline_msg_done

    for i in range(chunk_begin, chunk_begin + chunk_size):
        offline_msg_done = False
        voices = Queue()

        if args.ssl == 1:
            ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
            ssl_context.check_hostname = False
            ssl_context.verify_mode = ssl.CERT_NONE
            uri = "wss://{}:{}".format(args.host, args.port)
        else:
            uri = "ws://{}:{}".format(args.host, args.port)
            ssl_context = None

        print("connect to", uri)

        async with websockets.connect(
            uri, subprotocols=["binary"], ping_interval=None, ssl=ssl_context
        ) as websocket:
            meeting_tag = f"{id}_{i}"
            writer = MeetingWriter(args.log_dir, meeting_id=meeting_tag, flush_every=args.log_flush_every)
            try:
                if args.audio_in is not None:
                    task = asyncio.create_task(record_from_scp(i, 1))
                else:
                    task = asyncio.create_task(record_microphone())
                task3 = asyncio.create_task(message(str(id) + "_" + str(i), writer))  # processid+fileid
                await asyncio.gather(task, task3)
            finally:
                writer.close()

    return


def one_thread(id, chunk_begin, chunk_size):