tcp_client.h

Go to the documentation of this file.

// Copyright 2019-2021:
//   GobySoft, LLC (2013-)
//   Community contributors (see AUTHORS file)
// File authors:
//   Toby Schneider <toby@gobysoft.org>
//
//
// This file is part of the Goby Underwater Autonomy Project Libraries
// ("The Goby Libraries").
//
// The Goby Libraries are free software: you can redistribute them and/or modify
// them under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 2.1 of the License, or
// (at your option) any later version.
//
// The Goby Libraries are distributed in the hope that they will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with Goby.  If not, see <http://www.gnu.org/licenses/>.
 
#ifndef GOBY_MIDDLEWARE_IO_LINE_BASED_TCP_CLIENT_H
#define GOBY_MIDDLEWARE_IO_LINE_BASED_TCP_CLIENT_H
 
#include <istream> // for istream
#include <memory>  // for make_shared
#include <string>  // for basic_st...
 
#include <boost/asio/read_until.hpp>   // for async_re...
#include <boost/asio/streambuf.hpp>    // for streambuf
#include <boost/system/error_code.hpp> // for error_code
 
#include "goby/middleware/io/detail/io_interface.h"         // for PubSubLayer
#include "goby/middleware/io/detail/tcp_client_interface.h" // for TCPClien...
#include "goby/middleware/io/line_based/common.h"           // for match_regex
#include "goby/middleware/protobuf/io.pb.h"                 // for IOData
 
namespace goby
{
namespace middleware
{
class Group;
}
} // namespace goby
namespace goby
{
namespace middleware
{
namespace protobuf
{
class TCPClientConfig;
}
} // namespace middleware
} // namespace goby
 
namespace goby
{
namespace middleware
{
namespace io
{
template <const goby::middleware::Group& line_in_group,
          const goby::middleware::Group& line_out_group,
          PubSubLayer publish_layer = PubSubLayer::INTERPROCESS,
          PubSubLayer subscribe_layer = PubSubLayer::INTERTHREAD,
          typename Config = goby::middleware::protobuf::TCPClientConfig,
          template <class> class ThreadType = goby::middleware::SimpleThread,
          bool use_indexed_groups = false>
class TCPClientThreadLineBased
    : public detail::TCPClientThread<line_in_group, line_out_group, publish_layer, subscribe_layer,
                                     Config, ThreadType, use_indexed_groups>
{
    using Base = detail::TCPClientThread<line_in_group, line_out_group, publish_layer,
                                         subscribe_layer, Config, ThreadType, use_indexed_groups>;
 
  public:
    TCPClientThreadLineBased(const goby::middleware::protobuf::TCPClientConfig& config,
                             int index = -1)
        : Base(config, index), eol_matcher_(this->cfg().end_of_line())
    {
    }
 
    ~TCPClientThreadLineBased() {}
 
  private:
    void async_read() override;
 
  private:
    match_regex eol_matcher_;
    boost::asio::streambuf buffer_;
};
} // namespace io
} // namespace middleware
} // namespace goby
 
template <const goby::middleware::Group& line_in_group,
          const goby::middleware::Group& line_out_group,
          goby::middleware::io::PubSubLayer publish_layer,
          goby::middleware::io::PubSubLayer subscribe_layer, typename Config,
          template <class> class ThreadType, bool use_indexed_groups>
void goby::middleware::io::TCPClientThreadLineBased<line_in_group, line_out_group, publish_layer,
                                                    subscribe_layer, Config, ThreadType,
                                                    use_indexed_groups>::async_read()
{
    boost::asio::async_read_until(
        this->mutable_socket(), buffer_, eol_matcher_,
        [this](const boost::system::error_code& ec, std::size_t bytes_transferred) {
            if (!ec && bytes_transferred > 0)
            {
                auto io_msg = std::make_shared<goby::middleware::protobuf::IOData>();
                auto& bytes = *io_msg->mutable_data();
                bytes = std::string(bytes_transferred, 0);
                std::istream is(&buffer_);
                is.read(&bytes[0], bytes_transferred);
                this->insert_endpoints(io_msg);
                this->handle_read_success(bytes_transferred, io_msg);
                this->async_read();
            }
            else
            {
                this->handle_read_error(ec);
            }
        });
}
 
#endif