You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.9 KiB
56 lines
1.9 KiB
// Copyright (c) 2020 Mobvoi Inc (Binbin Zhang)
|
|
// 2022 PaddlePaddle Authors
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include "base/common.h"
|
|
#include "boost/asio/connect.hpp"
|
|
#include "boost/asio/ip/tcp.hpp"
|
|
#include "boost/beast/core.hpp"
|
|
#include "boost/beast/websocket.hpp"
|
|
|
|
namespace beast = boost::beast; // from <boost/beast.hpp>
|
|
namespace http = beast::http; // from <boost/beast/http.hpp>
|
|
namespace websocket = beast::websocket; // from <boost/beast/websocket.hpp>
|
|
namespace asio = boost::asio; // from <boost/asio.hpp>
|
|
using tcp = boost::asio::ip::tcp; // from <boost/asio/ip/tcp.hpp>
|
|
|
|
namespace ppspeech {
|
|
|
|
class WebSocketClient {
|
|
public:
|
|
WebSocketClient(const std::string& host, int port);
|
|
|
|
void SendTextData(const std::string& data);
|
|
void SendBinaryData(const void* data, size_t size);
|
|
void ReadLoopFunc();
|
|
void Close();
|
|
void Join();
|
|
void SendStartSignal();
|
|
void SendEndSignal();
|
|
void SendDataEnd();
|
|
bool Done() const { return done_; }
|
|
std::string GetResult() const { return result_; }
|
|
std::string GetPartialResult() const { return partial_result_; }
|
|
|
|
private:
|
|
void Connect();
|
|
std::string host_;
|
|
std::string result_;
|
|
std::string partial_result_;
|
|
int port_;
|
|
bool done_ = false;
|
|
asio::io_context ioc_;
|
|
websocket::stream<tcp::socket> ws_{ioc_};
|
|
std::unique_ptr<std::thread> t_{nullptr};
|
|
};
|
|
} // namespace ppspeech
|