client_demo_multi_thread.py

# -*-coding:utf-8-*-
from baidu_acu_asr.asr_client import AsrClient
import threading
import sys
import logging
import time
from baidu_acu_asr.asr_product import AsrProduct
import baidu_acu_asr.audio_streaming_pb2


class demo:

    def run(self, repeat_num=1):
        i = 0
        # ip和端口可根据需要修改
        url = "180.76.107.131"
        port = "8050"
        product_id = AsrProduct.CUSTOMER_SERVICE_FINANCE
        enable_flush_data = True
        log_level = 0
        product_id = "888"
        sample_rate = 16000
        user_name = "abc"
        password = "123"
        client = AsrClient(url, port, None, enable_flush_data,
                           log_level=log_level,
                           product_id=product_id,
                           sample_rate=sample_rate,
                           user_name=user_name,
                           password=password)
        name = threading.currentThread().getName()
        audio_path = "/Users/lijialong02/code/client/data/10s.wav"
        while i < repeat_num:
            try:
                response = client.get_result(audio_path)
                for res in response:
                    if res.type == baidu_acu_asr.audio_streaming_pb2.FRAGMENT_DATA:
                        logging.info("%s\t%s\t%s\t%s\t%s\t%s\t%s\t%s",
                                     name,
                                     str(res.error_code),
                                     res.error_message,
                                     res.audio_fragment.start_time,
                                     res.audio_fragment.end_time,
                                     res.audio_fragment.result,
                                     res.audio_fragment.serial_num,
                                     str(res.audio_fragment.completed))
                    else:
                        logging.warning("type is: %d", res.type)
                i += 1
            except Exception as ex:
                # 如果出现异常，此处需要重试当前音频
                logging.error("encounter an error: %s, will create a new channel and retry audio!", ex.message)
                time.sleep(0.5)
                client = AsrClient(url, port, product_id, enable_flush_data,
                                   log_level=log_level,
                                   user_name=user_name,
                                   password=password)


if __name__ == '__main__':
    logging.basicConfig(filename="asr_result.log", level=logging.DEBUG)
    demo1 = demo()
    thread_num = sys.argv[1]
    repeat_num = sys.argv[2]
    threads = []
    start_time = time.time()
    for i in range(int(thread_num)):
        t = threading.Thread(target=demo1.run, args=[int(repeat_num)])
        time.sleep(0.5)
        t.start()
        threads.append(t)

    for thread in threads:
        thread.join()
    logging.info("complete!")
    logging.info("use time: %s", str(time.time() - start_time))