Files
scylladb/to_string.hh
Vladimir Krivopalov 7b1d4915a1 to_string: Add operator<< overload for std::optional.
Signed-off-by: Vladimir Krivopalov <vladimir@scylladb.com>
2018-08-17 18:20:05 -07:00

157 lines
3.8 KiB
C++

/*
* Copyright (C) 2015 ScyllaDB
*/
/*
* This file is part of Scylla.
*
* Scylla is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Scylla 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 for more details.
*
* You should have received a copy of the GNU General Public License
* along with Scylla. If not, see <http://www.gnu.org/licenses/>.
*/
#pragma once
#include "core/sstring.hh"
#include <vector>
#include <sstream>
#include <unordered_set>
#include <set>
#include <experimental/optional>
#include "seastarx.hh"
#include "utils/chunked_vector.hh"
template<typename Iterator>
static inline
sstring join(sstring delimiter, Iterator begin, Iterator end) {
std::ostringstream oss;
while (begin != end) {
oss << *begin;
++begin;
if (begin != end) {
oss << delimiter;
}
}
return oss.str();
}
template<typename PrintableRange>
static inline
sstring join(sstring delimiter, const PrintableRange& items) {
return join(delimiter, items.begin(), items.end());
}
namespace std {
template<typename Printable>
static inline
sstring
to_string(const std::vector<Printable>& items) {
return "[" + join(", ", items) + "]";
}
template<typename Printable>
static inline
sstring
to_string(const std::set<Printable>& items) {
return "{" + join(", ", items) + "}";
}
template<typename Printable>
static inline
sstring
to_string(const std::unordered_set<Printable>& items) {
return "{" + join(", ", items) + "}";
}
template<typename Printable>
static inline
sstring
to_string(std::initializer_list<Printable> items) {
return "[" + join(", ", std::begin(items), std::end(items)) + "]";
}
template <typename K, typename V>
std::ostream& operator<<(std::ostream& os, const std::pair<K, V>& p) {
os << "{" << p.first << ", " << p.second << "}";
return os;
}
template <typename T>
std::ostream& operator<<(std::ostream& os, const std::unordered_set<T>& items) {
os << "{" << join(", ", items) << "}";
return os;
}
template <typename T>
std::ostream& operator<<(std::ostream& os, const std::set<T>& items) {
os << "{" << join(", ", items) << "}";
return os;
}
template<typename T, size_t N>
std::ostream& operator<<(std::ostream& os, const std::array<T, N>& items) {
os << "{" << join(", ", items) << "}";
return os;
}
template <typename K, typename V, typename... Args>
std::ostream& operator<<(std::ostream& os, const std::unordered_map<K, V, Args...>& items) {
os << "{" << join(", ", items) << "}";
return os;
}
template <typename K, typename V, typename... Args>
std::ostream& operator<<(std::ostream& os, const std::map<K, V, Args...>& items) {
os << "{" << join(", ", items) << "}";
return os;
}
template <typename T>
std::ostream& operator<<(std::ostream& os, const utils::chunked_vector<T>& items) {
os << "[" << join(", ", items) << "]";
return os;
}
template <typename T>
std::ostream& operator<<(std::ostream& os, const std::list<T>& items) {
os << "[" << join(", ", items) << "]";
return os;
}
template <typename T>
std::ostream& operator<<(std::ostream& os, const std::optional<T>& opt) {
if (opt) {
os << "{" << *opt << "}";
} else {
os << "{}";
}
return os;
}
namespace experimental {
template <typename T>
std::ostream& operator<<(std::ostream& os, const std::experimental::optional<T>& opt) {
if (opt) {
os << "{" << *opt << "}";
} else {
os << "{}";
}
return os;
}
}
}