update

嘉渊

2023-07-11 f62d6314b93416b301ca7bb0b62ccf9c4e6933e2

 funasr/runtime/websocket/funasr-wss-client.cpp

@@ -20,6 +20,7 @@
#include <websocketpp/config/asio_client.hpp>
#include <fstream>
#include <atomic>
#include <thread>
#include <glog/logging.h>

#include "audio.h"
@@ -106,7 +107,7 @@
        switch (msg->get_opcode()) {
            case websocketpp::frame::opcode::text:
            total_num=total_num+1;
                LOG(INFO)<<total_num<<",on_message = " << payload;
                LOG(INFO)<< "Thread: " << this_thread::get_id() <<",on_message = " << payload;
            if((total_num+1)==wav_index)
            {
               websocketpp::lib::error_code ec;
@@ -238,15 +239,26 @@

        // fetch wav data use asr engine api
        while (audio.Fetch(buff, len, flag) > 0) {
            short iArray[len];

            // convert float -1,1 to short -32768,32767
            short* iArray = new short[len];
            for (size_t i = 0; i < len; ++i) {
              iArray[i] = (short)(buff[i] * 32767);
              iArray[i] = (short)(buff[i]*32768);
            }

            // send data to server
            m_client.send(m_hdl, iArray, len * sizeof(short),
                          websocketpp::frame::opcode::binary, ec);
            int offset = 0;
            int block_size = 102400;
            while(offset < len){
                int send_block = 0;
                if (offset + block_size <= len){
                    send_block = block_size;
                }else{
                    send_block = len - offset;
                }
                m_client.send(m_hdl, iArray+offset, send_block * sizeof(short),
                    websocketpp::frame::opcode::binary, ec);
                offset += send_block;
            }

            LOG(INFO) << "sended data len=" << len * sizeof(short);
            // The most likely error that we will get is that the connection is
            // not in the right state. Usually this means we tried to send a
@@ -258,6 +270,7 @@
                                        "Send Error: " + ec.message());
              break;
            }
            delete[] iArray;
            // WaitABit();
        }
        nlohmann::json jsonresult;
@@ -363,4 +376,4 @@
    for (auto& t : client_threads) {
        t.join();
    }
}
}