forked from boostorg/beast
119 lines
3.1 KiB
C++
119 lines
3.1 KiB
C++
//
|
|
// Copyright (c) 2013-2017 Vinnie Falco (vinnie dot falco at gmail dot com)
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
|
|
#ifndef BEAST_TEST_BUFFER_TEST_HPP
|
|
#define BEAST_TEST_BUFFER_TEST_HPP
|
|
|
|
#include <beast/core/string_view.hpp>
|
|
#include <beast/core/read_size.hpp>
|
|
#include <beast/core/type_traits.hpp>
|
|
#include <beast/core/detail/type_traits.hpp>
|
|
#include <boost/asio/buffer.hpp>
|
|
#include <algorithm>
|
|
#include <string>
|
|
#include <type_traits>
|
|
|
|
namespace beast {
|
|
namespace test {
|
|
|
|
template<class ConstBufferSequence>
|
|
typename std::enable_if<
|
|
is_const_buffer_sequence<ConstBufferSequence>::value,
|
|
std::string>::type
|
|
to_string(ConstBufferSequence const& bs)
|
|
{
|
|
using boost::asio::buffer_cast;
|
|
using boost::asio::buffer_size;
|
|
std::string s;
|
|
s.reserve(buffer_size(bs));
|
|
for(boost::asio::const_buffer b : bs)
|
|
s.append(buffer_cast<char const*>(b),
|
|
buffer_size(b));
|
|
return s;
|
|
}
|
|
|
|
template<class DynamicBuffer>
|
|
void
|
|
write_buffer(DynamicBuffer& b, string_view s)
|
|
{
|
|
b.commit(boost::asio::buffer_copy(
|
|
b.prepare(s.size()), boost::asio::buffer(
|
|
s.data(), s.size())));
|
|
}
|
|
|
|
template<class ConstBufferSequence>
|
|
typename std::enable_if<
|
|
is_const_buffer_sequence<ConstBufferSequence>::value,
|
|
std::size_t>::type
|
|
buffer_count(ConstBufferSequence const& buffers)
|
|
{
|
|
return std::distance(buffers.begin(), buffers.end());
|
|
}
|
|
|
|
template<class ConstBufferSequence>
|
|
typename std::enable_if<
|
|
is_const_buffer_sequence<ConstBufferSequence>::value,
|
|
std::size_t>::type
|
|
size_pre(ConstBufferSequence const& buffers)
|
|
{
|
|
std::size_t n = 0;
|
|
for(auto it = buffers.begin(); it != buffers.end(); ++it)
|
|
{
|
|
typename ConstBufferSequence::const_iterator it0(std::move(it));
|
|
typename ConstBufferSequence::const_iterator it1(it0);
|
|
typename ConstBufferSequence::const_iterator it2;
|
|
it2 = it1;
|
|
n += boost::asio::buffer_size(*it2);
|
|
it = std::move(it2);
|
|
}
|
|
return n;
|
|
}
|
|
|
|
template<class ConstBufferSequence>
|
|
typename std::enable_if<
|
|
is_const_buffer_sequence<ConstBufferSequence>::value,
|
|
std::size_t>::type
|
|
size_post(ConstBufferSequence const& buffers)
|
|
{
|
|
std::size_t n = 0;
|
|
for(auto it = buffers.begin(); it != buffers.end(); it++)
|
|
n += boost::asio::buffer_size(*it);
|
|
return n;
|
|
}
|
|
|
|
template<class ConstBufferSequence>
|
|
typename std::enable_if<
|
|
is_const_buffer_sequence<ConstBufferSequence>::value,
|
|
std::size_t>::type
|
|
size_rev_pre(ConstBufferSequence const& buffers)
|
|
{
|
|
std::size_t n = 0;
|
|
for(auto it = buffers.end(); it != buffers.begin();)
|
|
n += boost::asio::buffer_size(*--it);
|
|
return n;
|
|
}
|
|
|
|
template<class ConstBufferSequence>
|
|
typename std::enable_if<
|
|
is_const_buffer_sequence<ConstBufferSequence>::value,
|
|
std::size_t>::type
|
|
size_rev_post(ConstBufferSequence const& buffers)
|
|
{
|
|
std::size_t n = 0;
|
|
for(auto it = buffers.end(); it != buffers.begin();)
|
|
{
|
|
it--;
|
|
n += boost::asio::buffer_size(*it);
|
|
}
|
|
return n;
|
|
}
|
|
|
|
} // test
|
|
} // beast
|
|
|
|
#endif
|