diff options
author | MacroFake <falke.marco@gmail.com> | 2022-08-19 19:28:22 +0200 |
---|---|---|
committer | MacroFake <falke.marco@gmail.com> | 2022-08-24 10:15:11 +0200 |
commit | fa1c71695593d5d3b264c2504baca318fc46d580 (patch) | |
tree | 503f4b9a3083d9428dd836e7518b1694334d4300 | |
parent | faf8da3c8da275c104330ef2743c5a062bea2f4b (diff) | |
download | bitcoin-fa1c71695593d5d3b264c2504baca318fc46d580.tar.xz |
Make Join() util work with any container type
Also, remove helper that is only used in tests.
-rw-r--r-- | src/test/util_tests.cpp | 6 | ||||
-rw-r--r-- | src/util/string.h | 29 |
2 files changed, 19 insertions, 16 deletions
diff --git a/src/test/util_tests.cpp b/src/test/util_tests.cpp index a16280f8c6..921cd37327 100644 --- a/src/test/util_tests.cpp +++ b/src/test/util_tests.cpp @@ -244,9 +244,9 @@ BOOST_AUTO_TEST_CASE(util_Join) // Version with unary operator const auto op_upper = [](const std::string& s) { return ToUpper(s); }; - BOOST_CHECK_EQUAL(Join<std::string>({}, ", ", op_upper), ""); - BOOST_CHECK_EQUAL(Join<std::string>({"foo"}, ", ", op_upper), "FOO"); - BOOST_CHECK_EQUAL(Join<std::string>({"foo", "bar"}, ", ", op_upper), "FOO, BAR"); + BOOST_CHECK_EQUAL(Join(std::list<std::string>{}, ", ", op_upper), ""); + BOOST_CHECK_EQUAL(Join(std::list<std::string>{"foo"}, ", ", op_upper), "FOO"); + BOOST_CHECK_EQUAL(Join(std::list<std::string>{"foo", "bar"}, ", ", op_upper), "FOO, BAR"); } BOOST_AUTO_TEST_CASE(util_ReplaceAll) diff --git a/src/util/string.h b/src/util/string.h index f8d36493b8..0bcf7d760f 100644 --- a/src/util/string.h +++ b/src/util/string.h @@ -58,27 +58,30 @@ void ReplaceAll(std::string& in_out, const std::string& search, const std::strin } /** - * Join a list of items + * Join all container items. Typically used to concatenate strings but accepts + * containers with elements of any type. * - * @param list The list to join - * @param separator The separator - * @param unary_op Apply this operator to each item in the list + * @param container The items to join + * @param separator The separator + * @param unary_op Apply this operator to each item */ -template <typename T, typename BaseType, typename UnaryOp> -auto Join(const std::vector<T>& list, const BaseType& separator, UnaryOp unary_op) +template <typename C, typename S, typename UnaryOp> +auto Join(const C& container, const S& separator, UnaryOp unary_op) { - decltype(unary_op(list.at(0))) ret; - for (size_t i = 0; i < list.size(); ++i) { - if (i > 0) ret += separator; - ret += unary_op(list.at(i)); + decltype(unary_op(*container.begin())) ret; + bool first{true}; + for (const auto& item : container) { + if (!first) ret += separator; + ret += unary_op(item); + first = false; } return ret; } -template <typename T, typename T2> -T Join(const std::vector<T>& list, const T2& separator) +template <typename C, typename S> +auto Join(const C& container, const S& separator) { - return Join(list, separator, [](const T& i) { return i; }); + return Join(container, separator, [](const auto& i) { return i; }); } /** |