86 lines
1.9 KiB
C++
86 lines
1.9 KiB
C++
// Copyright (c) 2017 Computer Vision Center (CVC) at the Universitat Autonoma
|
|
// de Barcelona (UAB).
|
|
//
|
|
// This work is licensed under the terms of the MIT license.
|
|
// For a copy, see <https://opensource.org/licenses/MIT>.
|
|
|
|
#pragma once
|
|
|
|
#include "carla/Buffer.h"
|
|
#include "carla/NonCopyable.h"
|
|
#include "carla/profiler/LifetimeProfiled.h"
|
|
#include "carla/streaming/detail/Token.h"
|
|
#include "carla/streaming/detail/Types.h"
|
|
|
|
#include <boost/asio/deadline_timer.hpp>
|
|
#include <boost/asio/io_context.hpp>
|
|
#include <boost/asio/ip/tcp.hpp>
|
|
#include <boost/asio/strand.hpp>
|
|
|
|
#include <atomic>
|
|
#include <functional>
|
|
#include <memory>
|
|
|
|
namespace carla {
|
|
|
|
class BufferPool;
|
|
|
|
namespace streaming {
|
|
namespace detail {
|
|
namespace tcp {
|
|
|
|
/// A client that connects to a single stream.
|
|
///
|
|
/// @warning This client should be stopped before releasing the shared pointer
|
|
/// or won't be destroyed.
|
|
class Client
|
|
: public std::enable_shared_from_this<Client>,
|
|
private profiler::LifetimeProfiled,
|
|
private NonCopyable {
|
|
public:
|
|
|
|
using endpoint = boost::asio::ip::tcp::endpoint;
|
|
using protocol_type = endpoint::protocol_type;
|
|
using callback_function_type = std::function<void (Buffer)>;
|
|
|
|
Client(
|
|
boost::asio::io_context &io_context,
|
|
const token_type &token,
|
|
callback_function_type callback);
|
|
|
|
~Client();
|
|
|
|
void Connect();
|
|
|
|
stream_id_type GetStreamId() const {
|
|
return _token.get_stream_id();
|
|
}
|
|
|
|
void Stop();
|
|
|
|
private:
|
|
|
|
void Reconnect();
|
|
|
|
void ReadData();
|
|
|
|
const token_type _token;
|
|
|
|
callback_function_type _callback;
|
|
|
|
boost::asio::ip::tcp::socket _socket;
|
|
|
|
boost::asio::io_context::strand _strand;
|
|
|
|
boost::asio::deadline_timer _connection_timer;
|
|
|
|
std::shared_ptr<BufferPool> _buffer_pool;
|
|
|
|
std::atomic_bool _done{false};
|
|
};
|
|
|
|
} // namespace tcp
|
|
} // namespace detail
|
|
} // namespace streaming
|
|
} // namespace carla
|