雾聪 2 лет назад
Родитель
Сommit
263802e20d
1 измененных файлов с 7 добавлено и 5 удалено
  1. 7 5
      funasr/runtime/websocket/funasr-wss-client.cpp

+ 7 - 5
funasr/runtime/websocket/funasr-wss-client.cpp

@@ -106,10 +106,10 @@ class WebsocketClient {
         const std::string& payload = msg->get_payload();
         switch (msg->get_opcode()) {
             case websocketpp::frame::opcode::text:
-				total_num=total_num+1;
+				total_recv=total_recv+1;
                 LOG(INFO)<< "Thread: " << this_thread::get_id() <<",on_message = " << payload;
-                LOG(INFO) << "total_num=" << total_num << " wav_index=" <<wav_index;
-				if((total_num+1)==wav_index)
+                LOG(INFO) << "total_recv=" << total_recv << " total_send=" <<total_send;
+				if(total_recv==total_send)
 				{
                     LOG(INFO) << "close client";
 					websocketpp::lib::error_code ec;
@@ -149,6 +149,7 @@ class WebsocketClient {
             if (i >= wav_list.size()) {
                 break;
             }
+            total_send += 1;
             send_wav_data(wav_list[i], wav_ids[i], hotwords, send_hotword);
             if(send_hotword){
                 send_hotword = false;
@@ -321,7 +322,7 @@ class WebsocketClient {
         jsonresult["is_speaking"] = false;
         m_client.send(m_hdl, jsonresult.dump(), websocketpp::frame::opcode::text,
                       ec);
-        std::this_thread::sleep_for(std::chrono::milliseconds(10));
+        std::this_thread::sleep_for(std::chrono::milliseconds(20));
     }
     websocketpp::client<T> m_client;
 
@@ -330,7 +331,8 @@ class WebsocketClient {
     websocketpp::lib::mutex m_lock;
     bool m_open;
     bool m_done;
-	int total_num=0;
+	int total_send=0;
+    int total_recv=0;
 };
 
 int main(int argc, char* argv[]) {