| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418 |
- //
- // Copyright (c) 2019-2023 Ruben Perez Hidalgo (rubenperez038 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 BOOST_MYSQL_IMPL_FIELD_VIEW_HPP
- #define BOOST_MYSQL_IMPL_FIELD_VIEW_HPP
- #pragma once
- #include <boost/mysql/bad_field_access.hpp>
- #include <boost/mysql/blob.hpp>
- #include <boost/mysql/blob_view.hpp>
- #include <boost/mysql/field_kind.hpp>
- #include <boost/mysql/field_view.hpp>
- #include <boost/mysql/detail/auxiliar/access_fwd.hpp>
- #include <cstdio>
- #include <cstring>
- #include <limits>
- #include <ostream>
- namespace boost {
- namespace mysql {
- namespace detail {
- inline std::ostream& print_blob(std::ostream& os, blob_view value)
- {
- if (value.empty())
- return os << "{}";
- char buffer[16]{};
- os << "{ ";
- for (std::size_t i = 0; i < value.size(); ++i)
- {
- if (i != 0)
- os << ", ";
- unsigned byte = value[i];
- std::snprintf(buffer, sizeof(buffer), "0x%02x", byte);
- os << buffer;
- }
- os << " }";
- return os;
- }
- inline std::ostream& print_time(std::ostream& os, const time& value)
- {
- // Worst-case output is 26 chars, extra space just in case
- char buffer[64]{};
- using namespace std::chrono;
- const char* sign = value < microseconds(0) ? "-" : "";
- auto num_micros = value % seconds(1);
- auto num_secs = duration_cast<seconds>(value % minutes(1) - num_micros);
- auto num_mins = duration_cast<minutes>(value % hours(1) - num_secs);
- auto num_hours = duration_cast<hours>(value - num_mins);
- snprintf(
- buffer,
- sizeof(buffer),
- "%s%02d:%02u:%02u.%06u",
- sign,
- static_cast<int>(std::abs(num_hours.count())),
- static_cast<unsigned>(std::abs(num_mins.count())),
- static_cast<unsigned>(std::abs(num_secs.count())),
- static_cast<unsigned>(std::abs(num_micros.count()))
- );
- os << buffer;
- return os;
- }
- inline bool blobs_equal(blob_view b1, blob_view b2)
- {
- if (b1.size() != b2.size())
- return false;
- return b1.empty() || std::memcmp(b1.data(), b2.data(), b2.size()) == 0;
- }
- } // namespace detail
- } // namespace mysql
- } // namespace boost
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(signed char v) noexcept
- : ikind_(internal_kind::int64), repr_(std::int64_t(v))
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(short v) noexcept
- : ikind_(internal_kind::int64), repr_(std::int64_t(v))
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(int v) noexcept
- : ikind_(internal_kind::int64), repr_(std::int64_t(v))
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(long v) noexcept
- : ikind_(internal_kind::int64), repr_(std::int64_t(v))
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(long long v) noexcept
- : ikind_(internal_kind::int64), repr_(std::int64_t(v))
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(unsigned char v) noexcept
- : ikind_(internal_kind::uint64), repr_(std::uint64_t(v))
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(unsigned short v) noexcept
- : ikind_(internal_kind::uint64), repr_(std::uint64_t(v))
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(unsigned int v) noexcept
- : ikind_(internal_kind::uint64), repr_(std::uint64_t(v))
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(unsigned long v) noexcept
- : ikind_(internal_kind::uint64), repr_(std::uint64_t(v))
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(unsigned long long v) noexcept
- : ikind_(internal_kind::uint64), repr_(std::uint64_t(v))
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(string_view v) noexcept
- : ikind_(internal_kind::string), repr_(v)
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(blob_view v) noexcept
- : ikind_(internal_kind::blob), repr_(v)
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(float v) noexcept
- : ikind_(internal_kind::float_), repr_(v)
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(double v) noexcept
- : ikind_(internal_kind::double_), repr_(v)
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(const date& v) noexcept
- : ikind_(internal_kind::date), repr_(v)
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(const datetime& v) noexcept
- : ikind_(internal_kind::datetime), repr_(v)
- {
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::field_view::field_view(const time& v) noexcept
- : ikind_(internal_kind::time), repr_(v)
- {
- }
- BOOST_CXX14_CONSTEXPR inline boost::mysql::field_kind boost::mysql::field_view::kind() const noexcept
- {
- switch (ikind_)
- {
- case internal_kind::null: return field_kind::null;
- case internal_kind::int64: return field_kind::int64;
- case internal_kind::uint64: return field_kind::uint64;
- case internal_kind::string: return field_kind::string;
- case internal_kind::blob: return field_kind::blob;
- case internal_kind::float_: return field_kind::float_;
- case internal_kind::double_: return field_kind::double_;
- case internal_kind::date: return field_kind::date;
- case internal_kind::datetime: return field_kind::datetime;
- case internal_kind::time: return field_kind::time;
- case internal_kind::field_ptr: return repr_.field_ptr->kind();
- // sv_offset values must be converted via offset_to_string_view before calling any other fn
- default: return field_kind::null;
- }
- }
- BOOST_CXX14_CONSTEXPR std::int64_t boost::mysql::field_view::as_int64() const
- {
- if (is_field_ptr())
- return repr_.field_ptr->as<std::int64_t>();
- check_kind(internal_kind::int64);
- return repr_.int64;
- }
- BOOST_CXX14_CONSTEXPR std::uint64_t boost::mysql::field_view::as_uint64() const
- {
- if (is_field_ptr())
- return repr_.field_ptr->as<std::uint64_t>();
- check_kind(internal_kind::uint64);
- return repr_.uint64;
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::string_view boost::mysql::field_view::as_string() const
- {
- if (is_field_ptr())
- return repr_.field_ptr->as<std::string>();
- check_kind(internal_kind::string);
- return repr_.string;
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::blob_view boost::mysql::field_view::as_blob() const
- {
- if (is_field_ptr())
- return repr_.field_ptr->as<blob>();
- check_kind(internal_kind::blob);
- return repr_.blob;
- }
- BOOST_CXX14_CONSTEXPR float boost::mysql::field_view::as_float() const
- {
- if (is_field_ptr())
- return repr_.field_ptr->as<float>();
- check_kind(internal_kind::float_);
- return repr_.float_;
- }
- BOOST_CXX14_CONSTEXPR double boost::mysql::field_view::as_double() const
- {
- if (is_field_ptr())
- return repr_.field_ptr->as<double>();
- check_kind(internal_kind::double_);
- return repr_.double_;
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::date boost::mysql::field_view::as_date() const
- {
- if (is_field_ptr())
- return repr_.field_ptr->as<date>();
- check_kind(internal_kind::date);
- return repr_.date_;
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::datetime boost::mysql::field_view::as_datetime() const
- {
- if (is_field_ptr())
- return repr_.field_ptr->as<datetime>();
- check_kind(internal_kind::datetime);
- return repr_.datetime_;
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::time boost::mysql::field_view::as_time() const
- {
- if (is_field_ptr())
- return repr_.field_ptr->as<time>();
- check_kind(internal_kind::time);
- return repr_.time_;
- }
- BOOST_CXX14_CONSTEXPR std::int64_t boost::mysql::field_view::get_int64() const noexcept
- {
- return is_field_ptr() ? repr_.field_ptr->get<std::int64_t>() : repr_.int64;
- }
- BOOST_CXX14_CONSTEXPR std::uint64_t boost::mysql::field_view::get_uint64() const noexcept
- {
- return is_field_ptr() ? repr_.field_ptr->get<std::uint64_t>() : repr_.uint64;
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::string_view boost::mysql::field_view::get_string() const noexcept
- {
- return is_field_ptr() ? string_view(repr_.field_ptr->get<std::string>()) : repr_.string;
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::blob_view boost::mysql::field_view::get_blob() const noexcept
- {
- return is_field_ptr() ? repr_.field_ptr->get<blob>() : repr_.blob;
- }
- BOOST_CXX14_CONSTEXPR float boost::mysql::field_view::get_float() const noexcept
- {
- return is_field_ptr() ? repr_.field_ptr->get<float>() : repr_.float_;
- }
- BOOST_CXX14_CONSTEXPR double boost::mysql::field_view::get_double() const noexcept
- {
- return is_field_ptr() ? repr_.field_ptr->get<double>() : repr_.double_;
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::date boost::mysql::field_view::get_date() const noexcept
- {
- return is_field_ptr() ? repr_.field_ptr->get<date>() : repr_.date_;
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::datetime boost::mysql::field_view::get_datetime() const noexcept
- {
- return is_field_ptr() ? repr_.field_ptr->get<datetime>() : repr_.datetime_;
- }
- BOOST_CXX14_CONSTEXPR boost::mysql::time boost::mysql::field_view::get_time() const noexcept
- {
- return is_field_ptr() ? repr_.field_ptr->get<time>() : repr_.time_;
- }
- BOOST_CXX14_CONSTEXPR void boost::mysql::field_view::check_kind(internal_kind expected) const
- {
- if (ikind_ != expected)
- throw bad_field_access();
- }
- BOOST_CXX14_CONSTEXPR bool boost::mysql::field_view::operator==(const field_view& rhs) const noexcept
- {
- // Make operator== work for types not representable by field_kind
- if (ikind_ == internal_kind::sv_offset_string || ikind_ == internal_kind::sv_offset_blob)
- {
- return rhs.ikind_ == ikind_ && repr_.sv_offset_ == rhs.repr_.sv_offset_;
- }
- auto k = kind(), rhs_k = rhs.kind();
- switch (k)
- {
- case field_kind::null: return rhs_k == field_kind::null;
- case field_kind::int64:
- if (rhs_k == field_kind::int64)
- return get_int64() == rhs.get_int64();
- else if (rhs_k == field_kind::uint64)
- {
- std::int64_t this_val = get_int64();
- if (this_val < 0)
- return false;
- else
- return static_cast<std::uint64_t>(this_val) == rhs.get_uint64();
- }
- else
- return false;
- case field_kind::uint64:
- if (rhs_k == field_kind::uint64)
- return get_uint64() == rhs.get_uint64();
- else if (rhs_k == field_kind::int64)
- {
- std::int64_t rhs_val = rhs.get_int64();
- if (rhs_val < 0)
- return false;
- else
- return static_cast<std::uint64_t>(rhs_val) == get_uint64();
- }
- else
- return false;
- case field_kind::string: return rhs_k == field_kind::string && get_string() == rhs.get_string();
- case field_kind::blob:
- return rhs_k == field_kind::blob && detail::blobs_equal(get_blob(), rhs.get_blob());
- case field_kind::float_: return rhs_k == field_kind::float_ && get_float() == rhs.get_float();
- case field_kind::double_: return rhs_k == field_kind::double_ && get_double() == rhs.get_double();
- case field_kind::date: return rhs_k == field_kind::date && get_date() == rhs.get_date();
- case field_kind::datetime: return rhs_k == field_kind::datetime && get_datetime() == rhs.get_datetime();
- case field_kind::time: return rhs_k == field_kind::time && get_time() == rhs.get_time();
- default: assert(false); return false;
- }
- }
- inline std::ostream& boost::mysql::operator<<(std::ostream& os, const field_view& value)
- {
- // Make operator<< work for detail::string_view_offset types
- if (value.ikind_ == field_view::internal_kind::sv_offset_string ||
- value.ikind_ == field_view::internal_kind::sv_offset_blob)
- {
- return os << "<sv_offset>";
- }
- switch (value.kind())
- {
- case field_kind::null: return os << "<NULL>";
- case field_kind::int64: return os << value.get_int64();
- case field_kind::uint64: return os << value.get_uint64();
- case field_kind::string: return os << value.get_string();
- case field_kind::blob: return detail::print_blob(os, value.get_blob());
- case field_kind::float_: return os << value.get_float();
- case field_kind::double_: return os << value.get_double();
- case field_kind::date: return os << value.get_date();
- case field_kind::datetime: return os << value.get_datetime();
- case field_kind::time: return detail::print_time(os, value.get_time());
- default: assert(false); return os;
- }
- }
- struct boost::mysql::detail::field_view_access
- {
- static field_view construct(detail::string_view_offset v, bool is_blob) noexcept
- {
- return field_view(v, is_blob);
- }
- static void offset_to_string_view(field_view& fv, const std::uint8_t* buffer_first) noexcept
- {
- if (fv.ikind_ == field_view::internal_kind::sv_offset_string)
- {
- fv.ikind_ = field_view::internal_kind::string;
- fv.repr_.string = {
- reinterpret_cast<const char*>(buffer_first) + fv.repr_.sv_offset_.offset(),
- fv.repr_.sv_offset_.size()};
- }
- else if (fv.ikind_ == field_view::internal_kind::sv_offset_blob)
- {
- fv.ikind_ = field_view::internal_kind::blob;
- fv.repr_.blob = blob_view(
- buffer_first + fv.repr_.sv_offset_.offset(),
- fv.repr_.sv_offset_.size()
- );
- }
- }
- };
- #endif
|