84 lines
3.3 KiB
C++
84 lines
3.3 KiB
C++
// Copyright (c) 2023, 2024, Oracle and/or its affiliates.
|
|
//
|
|
// This program is free software; you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License, version 2.0,
|
|
// as published by the Free Software Foundation.
|
|
//
|
|
// This program is designed to work with certain software (including
|
|
// but not limited to OpenSSL) that is licensed under separate terms,
|
|
// as designated in a particular file or component or in included license
|
|
// documentation. The authors of MySQL hereby grant you an additional
|
|
// permission to link the program and your derivative works with the
|
|
// separately licensed software that they have either included with
|
|
// the program or referenced in the documentation.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License, version 2.0, for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program; if not, write to the Free Software
|
|
// Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
|
|
/// @file
|
|
/// Experimental API header
|
|
|
|
namespace mysql::serialization {
|
|
|
|
template <typename Field_type>
|
|
Archive_binary &Archive_binary::operator<<(Field_type &&arg) {
|
|
static constexpr auto value_size = Field_type::value_size;
|
|
using value_type = typename std::decay_t<Field_type>::value_type;
|
|
|
|
if (is_good() == true) {
|
|
try {
|
|
size_t current_size = m_stream.size();
|
|
size_t field_written_size = Primitive_type_codec<
|
|
value_type>::template count_write_bytes<value_size>(arg.get());
|
|
m_stream.resize(m_stream.size() + field_written_size);
|
|
// and overwrite it with correct, formatted data
|
|
if (Primitive_type_codec<value_type>::template write_bytes<value_size>(
|
|
m_stream.data() + current_size, arg.get()) == 0) {
|
|
m_error = Serialization_error(
|
|
__FILE__, __LINE__, "Unable to write data to the stream",
|
|
Serialization_error_type::archive_write_error);
|
|
}
|
|
} catch (std::bad_alloc &) {
|
|
m_error = Serialization_error(__FILE__, __LINE__,
|
|
"Got bad alloc, unable to insert field",
|
|
Serialization_error_type::out_of_memory);
|
|
}
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
template <typename Field_type>
|
|
Archive_binary &Archive_binary::operator>>(Field_type &&arg) {
|
|
static constexpr auto value_size = Field_type::value_size;
|
|
using value_type = typename std::decay_t<Field_type>::value_type;
|
|
|
|
if (is_good() == true) {
|
|
auto bytes_read =
|
|
Primitive_type_codec<std::decay_t<value_type>>::template read_bytes<
|
|
value_size>(m_stream.data() + read_pos, m_stream.size() - read_pos,
|
|
arg.get());
|
|
if (bytes_read == 0) {
|
|
m_error = Serialization_error(
|
|
__FILE__, __LINE__, "Unable to read data from the stream",
|
|
Serialization_error_type::archive_read_error);
|
|
return *this;
|
|
}
|
|
read_pos += bytes_read;
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
template <class Field_type>
|
|
void Archive_binary::peek(Field_type &&field) {
|
|
auto read_pos_saved = read_pos;
|
|
this->operator>>(std::forward<Field_type>(field));
|
|
read_pos = read_pos_saved;
|
|
}
|
|
|
|
} // namespace mysql::serialization
|