From e2caedd65c2724ae98ed6a7ae97bd798f7fd4345 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9B=BE=E8=81=AA?= Date: Mon, 1 Jan 2024 13:17:59 +0800 Subject: [PATCH] fix bug of FUNASR_RESULT = null --- runtime/websocket/bin/websocket-server.cpp | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/runtime/websocket/bin/websocket-server.cpp b/runtime/websocket/bin/websocket-server.cpp index 42bc60a80..97d6c4718 100644 --- a/runtime/websocket/bin/websocket-server.cpp +++ b/runtime/websocket/bin/websocket-server.cpp @@ -72,21 +72,23 @@ void WebSocketServer::do_decoder(const std::vector& buffer, int num_samples = buffer.size(); // the size of the buf if (!buffer.empty() && hotwords_embedding.size() > 0) { - std::string asr_result; - std::string stamp_res; - std::string stamp_sents; + std::string asr_result=""; + std::string stamp_res=""; + std::string stamp_sents=""; try{ FUNASR_RESULT Result = FunOfflineInferBuffer( asr_handle, buffer.data(), buffer.size(), RASR_NONE, NULL, hotwords_embedding, audio_fs, wav_format, itn, decoder_handle); - - asr_result = ((FUNASR_RECOG_RESULT*)Result)->msg; // get decode result - stamp_res = ((FUNASR_RECOG_RESULT*)Result)->stamp; - stamp_sents = ((FUNASR_RECOG_RESULT*)Result)->stamp_sents; - FunASRFreeResult(Result); + if (Result != NULL){ + asr_result = FunASRGetResult(Result, 0); // get decode result + stamp_res = FunASRGetStamp(Result); + stamp_sents = FunASRGetStampSents(Result); + FunASRFreeResult(Result); + } else{ + LOG(ERROR) << "FUNASR_RESULT is NULL."; + } }catch (std::exception const& e) { LOG(ERROR) << e.what(); - return; } websocketpp::lib::error_code ec;