speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update websocket-server-2pass

Browse Source
This commit is contained in:
雾聪 2023-08-10 16:13:47 +08:00
parent 6a31591716
commit 800b28152b
1 changed files with 30 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
funasr/runtime/websocket/websocket-server-2pass.cpp
View File

@ -320,14 +320,20 @@ void WebSocketServer::on_message(websocketpp::connection_hdl hdl,
LOG(INFO) << "client done";
// if it is in final message, post the sample_data to decode
asio::post(
io_decoder_,
std::bind(&WebSocketServer::do_decoder, this,
std::move(*(sample_data_p.get())), std::move(hdl),
std::ref(msg_data->msg), std::ref(*(punc_cache_p.get())),
std::ref(*thread_lock_p), std::move(true),
msg_data->msg["wav_name"],
std::ref(msg_data->tpass_online_handle)));
try{
asio::post(
io_decoder_,
std::bind(&WebSocketServer::do_decoder, this,
std::move(*(sample_data_p.get())), std::move(hdl),
std::ref(msg_data->msg), std::ref(*(punc_cache_p.get())),
std::ref(*thread_lock_p), std::move(true),
msg_data->msg["wav_name"],
std::ref(msg_data->tpass_online_handle)));
}
catch (std::exception const &e)
{
LOG(ERROR)<<e.what();
}
}
break;
}
@ -351,15 +357,22 @@ void WebSocketServer::on_message(websocketpp::connection_hdl hdl,
// keep remain in sample_data
sample_data_p->erase(sample_data_p->begin(),
sample_data_p->begin() + chunksize * setpsize);
// post to decode
asio::post(io_decoder_,
std::bind(&WebSocketServer::do_decoder, this,
std::move(subvector), std::move(hdl),
std::ref(msg_data->msg),
std::ref(*(punc_cache_p.get())),
std::ref(*thread_lock_p), std::move(false),
msg_data->msg["wav_name"],
std::ref(msg_data->tpass_online_handle)));
try{
// post to decode
asio::post(io_decoder_,
std::bind(&WebSocketServer::do_decoder, this,
std::move(subvector), std::move(hdl),
std::ref(msg_data->msg),
std::ref(*(punc_cache_p.get())),
std::ref(*thread_lock_p), std::move(false),
msg_data->msg["wav_name"],
std::ref(msg_data->tpass_online_handle)));
}
catch (std::exception const &e)
{
LOG(ERROR)<<e.what();
}
}
} else {
sample_data_p->insert(sample_data_p->end(), pcm_data,