123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236 |
- /*
- * Copyright 2011-present Facebook, Inc.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- #pragma once
- #include <functional>
- #include <folly/CPortability.h>
- #include <folly/Conv.h>
- #include <folly/Format.h>
- #include <folly/Likely.h>
- #include <folly/detail/Iterators.h>
- #include <folly/lang/Exception.h>
- namespace folly {
- namespace detail {
- struct DynamicHasher {
- using is_transparent = void;
- size_t operator()(dynamic const& d) const {
- return d.hash();
- }
- template <typename T>
- std::enable_if_t<std::is_convertible<T, StringPiece>::value, size_t>
- operator()(T const& val) const {
- // keep consistent with dynamic::hash() for strings
- return Hash()(static_cast<StringPiece>(val));
- }
- };
- struct DynamicKeyEqual {
- using is_transparent = void;
- bool operator()(const dynamic& lhs, const dynamic& rhs) const {
- return std::equal_to<dynamic>()(lhs, rhs);
- }
- // Dynamic objects contains a map<dynamic, dynamic>. At least one of the
- // operands should be a dynamic. Hence, an operator() where both operands are
- // convertible to StringPiece is unnecessary.
- template <typename A, typename B>
- std::enable_if_t<
- std::is_convertible<A, StringPiece>::value &&
- std::is_convertible<B, StringPiece>::value,
- bool>
- operator()(A const& lhs, B const& rhs) const = delete;
- template <typename A>
- std::enable_if_t<std::is_convertible<A, StringPiece>::value, bool> operator()(
- A const& lhs,
- dynamic const& rhs) const {
- return FOLLY_LIKELY(rhs.type() == dynamic::Type::STRING) &&
- std::equal_to<StringPiece>()(lhs, rhs.stringPiece());
- }
- template <typename B>
- std::enable_if_t<std::is_convertible<B, StringPiece>::value, bool> operator()(
- dynamic const& lhs,
- B const& rhs) const {
- return FOLLY_LIKELY(lhs.type() == dynamic::Type::STRING) &&
- std::equal_to<StringPiece>()(lhs.stringPiece(), rhs);
- }
- };
- } // namespace detail
- } // namespace folly
- //////////////////////////////////////////////////////////////////////
- namespace std {
- template <>
- struct hash<::folly::dynamic> {
- size_t operator()(::folly::dynamic const& d) const {
- return d.hash();
- }
- };
- } // namespace std
- //////////////////////////////////////////////////////////////////////
- // This is a higher-order preprocessor macro to aid going from runtime
- // types to the compile time type system.
- #define FB_DYNAMIC_APPLY(type, apply) \
- do { \
- switch ((type)) { \
- case NULLT: \
- apply(std::nullptr_t); \
- break; \
- case ARRAY: \
- apply(Array); \
- break; \
- case BOOL: \
- apply(bool); \
- break; \
- case DOUBLE: \
- apply(double); \
- break; \
- case INT64: \
- apply(int64_t); \
- break; \
- case OBJECT: \
- apply(ObjectImpl); \
- break; \
- case STRING: \
- apply(std::string); \
- break; \
- default: \
- CHECK(0); \
- abort(); \
- } \
- } while (0)
- //////////////////////////////////////////////////////////////////////
- namespace folly {
- struct FOLLY_EXPORT TypeError : std::runtime_error {
- explicit TypeError(const std::string& expected, dynamic::Type actual);
- explicit TypeError(
- const std::string& expected,
- dynamic::Type actual1,
- dynamic::Type actual2);
- // TODO: noexcept calculation required through gcc-v4.9; remove once upgrading
- // to gcc-v5.
- TypeError(const TypeError&) noexcept(
- std::is_nothrow_copy_constructible<std::runtime_error>::value);
- TypeError& operator=(const TypeError&) noexcept(
- std::is_nothrow_copy_assignable<std::runtime_error>::value);
- TypeError(TypeError&&) noexcept(
- std::is_nothrow_move_constructible<std::runtime_error>::value);
- TypeError& operator=(TypeError&&) noexcept(
- std::is_nothrow_move_assignable<std::runtime_error>::value);
- ~TypeError() override;
- };
- //////////////////////////////////////////////////////////////////////
- namespace detail {
- // This helper is used in destroy() to be able to run destructors on
- // types like "int64_t" without a compiler error.
- struct Destroy {
- template <class T>
- static void destroy(T* t) {
- t->~T();
- }
- };
- /*
- * Helper for implementing numeric conversions in operators on
- * numbers. Just promotes to double when one of the arguments is
- * double, or throws if either is not a numeric type.
- */
- template <template <class> class Op>
- dynamic numericOp(dynamic const& a, dynamic const& b) {
- if (!a.isNumber() || !b.isNumber()) {
- throw_exception<TypeError>("numeric", a.type(), b.type());
- }
- if (a.type() != b.type()) {
- auto& integ = a.isInt() ? a : b;
- auto& nonint = a.isInt() ? b : a;
- return Op<double>()(to<double>(integ.asInt()), nonint.asDouble());
- }
- if (a.isDouble()) {
- return Op<double>()(a.asDouble(), b.asDouble());
- }
- return Op<int64_t>()(a.asInt(), b.asInt());
- }
- } // namespace detail
- //////////////////////////////////////////////////////////////////////
- /*
- * We're doing this instead of a simple member typedef to avoid the
- * undefined behavior of parameterizing F14NodeMap<> with an
- * incomplete type.
- *
- * Note: Later we may add separate order tracking here (a multi-index
- * type of thing.)
- */
- struct dynamic::ObjectImpl : F14NodeMap<
- dynamic,
- dynamic,
- detail::DynamicHasher,
- detail::DynamicKeyEqual> {};
- //////////////////////////////////////////////////////////////////////
- // Helper object for creating objects conveniently. See object and
- // the dynamic::dynamic(ObjectMaker&&) ctor.
- struct dynamic::ObjectMaker {
- friend struct dynamic;
- explicit ObjectMaker() : val_(dynamic::object) {}
- explicit ObjectMaker(dynamic key, dynamic val) : val_(dynamic::object) {
- val_.insert(std::move(key), std::move(val));
- }
- // Make sure no one tries to save one of these into an lvalue with
- // auto or anything like that.
- ObjectMaker(ObjectMaker&&) = default;
- ObjectMaker(ObjectMaker const&) = delete;
- ObjectMaker& operator=(ObjectMaker const&) = delete;
- ObjectMaker& operator=(ObjectMaker&&) = delete;
- // This returns an rvalue-reference instead of an lvalue-reference
- // to allow constructs like this to moved instead of copied:
- // dynamic a = dynamic::object("a", "b")("c", "d")
- ObjectMaker&& operator()(dynamic key, dynamic val) {
- val_.insert(std::move(key), std::move(val));
- return std::move(*this);
- }
- private:
- dynamic val_;
- };
- inline void dynamic::array(EmptyArrayTag) {}
- template <class... Args>
- inline dynamic dynamic::array(Args&&... args) {
- return dynamic(Array{std::forward<Args>(args)...});
- }
- inline dynamic::ObjectMaker dynamic::object() {
- return ObjectMaker();
- }
- inline dynamic::ObjectMaker dynamic::object(dynamic a, dynamic b) {
- return ObjectMaker(std::move(a), std::move(b));
- }
- //////////////////////////////////////////////////////////////////////
- struct dynamic::item_iterator : detail::IteratorAdaptor<
- dynamic::item_iterator,
- dynamic::ObjectImpl::iterator,
- std::pair<dynamic const, dynamic>> {
- using Super = detail::IteratorAdaptor<
- dynamic::item_iterator,
- dynamic::ObjectImpl::iterator,
- std::pair<dynamic const, dynamic>>;
- /* implicit */ item_iterator(dynamic::ObjectImpl::iterator b) : Super(b) {}
- using object_type = dynamic::ObjectImpl;
- };
- struct dynamic::value_iterator : detail::IteratorAdaptor<
- dynamic::value_iterator,
- dynamic::ObjectImpl::iterator,
- dynamic> {
- using Super = detail::IteratorAdaptor<
- dynamic::value_iterator,
- dynamic::ObjectImpl::iterator,
- dynamic>;
- /* implicit */ value_iterator(dynamic::ObjectImpl::iterator b) : Super(b) {}
- using object_type = dynamic::ObjectImpl;
- dynamic& dereference() const {
- return base()->second;
- }
- };
- struct dynamic::const_item_iterator
- : detail::IteratorAdaptor<
- dynamic::const_item_iterator,
- dynamic::ObjectImpl::const_iterator,
- std::pair<dynamic const, dynamic> const> {
- using Super = detail::IteratorAdaptor<
- dynamic::const_item_iterator,
- dynamic::ObjectImpl::const_iterator,
- std::pair<dynamic const, dynamic> const>;
- /* implicit */ const_item_iterator(dynamic::ObjectImpl::const_iterator b)
- : Super(b) {}
- /* implicit */ const_item_iterator(const_item_iterator const& i)
- : Super(i.base()) {}
- /* implicit */ const_item_iterator(item_iterator i) : Super(i.base()) {}
- using object_type = dynamic::ObjectImpl const;
- };
- struct dynamic::const_key_iterator : detail::IteratorAdaptor<
- dynamic::const_key_iterator,
- dynamic::ObjectImpl::const_iterator,
- dynamic const> {
- using Super = detail::IteratorAdaptor<
- dynamic::const_key_iterator,
- dynamic::ObjectImpl::const_iterator,
- dynamic const>;
- /* implicit */ const_key_iterator(dynamic::ObjectImpl::const_iterator b)
- : Super(b) {}
- using object_type = dynamic::ObjectImpl const;
- dynamic const& dereference() const {
- return base()->first;
- }
- };
- struct dynamic::const_value_iterator : detail::IteratorAdaptor<
- dynamic::const_value_iterator,
- dynamic::ObjectImpl::const_iterator,
- dynamic const> {
- using Super = detail::IteratorAdaptor<
- dynamic::const_value_iterator,
- dynamic::ObjectImpl::const_iterator,
- dynamic const>;
- /* implicit */ const_value_iterator(dynamic::ObjectImpl::const_iterator b)
- : Super(b) {}
- /* implicit */ const_value_iterator(value_iterator i) : Super(i.base()) {}
- /* implicit */ const_value_iterator(dynamic::ObjectImpl::iterator i)
- : Super(i) {}
- using object_type = dynamic::ObjectImpl const;
- dynamic const& dereference() const {
- return base()->second;
- }
- };
- //////////////////////////////////////////////////////////////////////
- inline dynamic::dynamic() : dynamic(nullptr) {}
- inline dynamic::dynamic(std::nullptr_t) : type_(NULLT) {}
- inline dynamic::dynamic(void (*)(EmptyArrayTag)) : type_(ARRAY) {
- new (&u_.array) Array();
- }
- inline dynamic::dynamic(ObjectMaker (*)()) : type_(OBJECT) {
- new (getAddress<ObjectImpl>()) ObjectImpl();
- }
- inline dynamic::dynamic(StringPiece s) : type_(STRING) {
- new (&u_.string) std::string(s.data(), s.size());
- }
- inline dynamic::dynamic(char const* s) : type_(STRING) {
- new (&u_.string) std::string(s);
- }
- inline dynamic::dynamic(std::string s) : type_(STRING) {
- new (&u_.string) std::string(std::move(s));
- }
- inline dynamic::dynamic(ObjectMaker&& maker) : type_(OBJECT) {
- new (getAddress<ObjectImpl>())
- ObjectImpl(std::move(*maker.val_.getAddress<ObjectImpl>()));
- }
- inline dynamic::dynamic(dynamic const& o) : type_(NULLT) {
- *this = o;
- }
- inline dynamic::dynamic(dynamic&& o) noexcept : type_(NULLT) {
- *this = std::move(o);
- }
- inline dynamic::~dynamic() noexcept {
- destroy();
- }
- // Integral types except bool convert to int64_t, float types to double.
- template <class T>
- struct dynamic::NumericTypeHelper<
- T,
- typename std::enable_if<std::is_integral<T>::value>::type> {
- static_assert(
- !kIsObjC || sizeof(T) > sizeof(char),
- "char-sized types are ambiguous in objc; cast to bool or wider type");
- using type = int64_t;
- };
- template <>
- struct dynamic::NumericTypeHelper<bool> {
- using type = bool;
- };
- template <>
- struct dynamic::NumericTypeHelper<float> {
- using type = double;
- };
- template <>
- struct dynamic::NumericTypeHelper<double> {
- using type = double;
- };
- inline dynamic::dynamic(std::vector<bool>::reference b)
- : dynamic(static_cast<bool>(b)) {}
- inline dynamic::dynamic(VectorBoolConstRefCtorType b)
- : dynamic(static_cast<bool>(b)) {}
- template <
- class T,
- class NumericType /* = typename NumericTypeHelper<T>::type */>
- dynamic::dynamic(T t) {
- type_ = TypeInfo<NumericType>::type;
- new (getAddress<NumericType>()) NumericType(NumericType(t));
- }
- template <class Iterator>
- dynamic::dynamic(Iterator first, Iterator last) : type_(ARRAY) {
- new (&u_.array) Array(first, last);
- }
- //////////////////////////////////////////////////////////////////////
- inline dynamic::const_iterator dynamic::begin() const {
- return get<Array>().begin();
- }
- inline dynamic::const_iterator dynamic::end() const {
- return get<Array>().end();
- }
- inline dynamic::iterator dynamic::begin() {
- return get<Array>().begin();
- }
- inline dynamic::iterator dynamic::end() {
- return get<Array>().end();
- }
- template <class It>
- struct dynamic::IterableProxy {
- typedef It iterator;
- typedef typename It::value_type value_type;
- typedef typename It::object_type object_type;
- /* implicit */ IterableProxy(object_type* o) : o_(o) {}
- It begin() const {
- return o_->begin();
- }
- It end() const {
- return o_->end();
- }
- private:
- object_type* o_;
- };
- inline dynamic::IterableProxy<dynamic::const_key_iterator> dynamic::keys()
- const {
- return &(get<ObjectImpl>());
- }
- inline dynamic::IterableProxy<dynamic::const_value_iterator> dynamic::values()
- const {
- return &(get<ObjectImpl>());
- }
- inline dynamic::IterableProxy<dynamic::const_item_iterator> dynamic::items()
- const {
- return &(get<ObjectImpl>());
- }
- inline dynamic::IterableProxy<dynamic::value_iterator> dynamic::values() {
- return &(get<ObjectImpl>());
- }
- inline dynamic::IterableProxy<dynamic::item_iterator> dynamic::items() {
- return &(get<ObjectImpl>());
- }
- inline bool dynamic::isString() const {
- return get_nothrow<std::string>() != nullptr;
- }
- inline bool dynamic::isObject() const {
- return get_nothrow<ObjectImpl>() != nullptr;
- }
- inline bool dynamic::isBool() const {
- return get_nothrow<bool>() != nullptr;
- }
- inline bool dynamic::isArray() const {
- return get_nothrow<Array>() != nullptr;
- }
- inline bool dynamic::isDouble() const {
- return get_nothrow<double>() != nullptr;
- }
- inline bool dynamic::isInt() const {
- return get_nothrow<int64_t>() != nullptr;
- }
- inline bool dynamic::isNull() const {
- return get_nothrow<std::nullptr_t>() != nullptr;
- }
- inline bool dynamic::isNumber() const {
- return isInt() || isDouble();
- }
- inline dynamic::Type dynamic::type() const {
- return type_;
- }
- inline std::string dynamic::asString() const {
- return asImpl<std::string>();
- }
- inline double dynamic::asDouble() const {
- return asImpl<double>();
- }
- inline int64_t dynamic::asInt() const {
- return asImpl<int64_t>();
- }
- inline bool dynamic::asBool() const {
- return asImpl<bool>();
- }
- inline const std::string& dynamic::getString() const& {
- return get<std::string>();
- }
- inline double dynamic::getDouble() const& {
- return get<double>();
- }
- inline int64_t dynamic::getInt() const& {
- return get<int64_t>();
- }
- inline bool dynamic::getBool() const& {
- return get<bool>();
- }
- inline std::string& dynamic::getString() & {
- return get<std::string>();
- }
- inline double& dynamic::getDouble() & {
- return get<double>();
- }
- inline int64_t& dynamic::getInt() & {
- return get<int64_t>();
- }
- inline bool& dynamic::getBool() & {
- return get<bool>();
- }
- inline std::string&& dynamic::getString() && {
- return std::move(get<std::string>());
- }
- inline double dynamic::getDouble() && {
- return get<double>();
- }
- inline int64_t dynamic::getInt() && {
- return get<int64_t>();
- }
- inline bool dynamic::getBool() && {
- return get<bool>();
- }
- inline const char* dynamic::data() const& {
- return get<std::string>().data();
- }
- inline const char* dynamic::c_str() const& {
- return get<std::string>().c_str();
- }
- inline StringPiece dynamic::stringPiece() const {
- return get<std::string>();
- }
- template <class T>
- struct dynamic::CompareOp {
- static bool comp(T const& a, T const& b) {
- return a < b;
- }
- };
- template <>
- struct dynamic::CompareOp<dynamic::ObjectImpl> {
- static bool comp(ObjectImpl const&, ObjectImpl const&) {
- // This code never executes; it is just here for the compiler.
- return false;
- }
- };
- template <>
- struct dynamic::CompareOp<std::nullptr_t> {
- static bool comp(std::nullptr_t const&, std::nullptr_t const&) {
- return true;
- }
- };
- inline dynamic& dynamic::operator+=(dynamic const& o) {
- if (type() == STRING && o.type() == STRING) {
- *getAddress<std::string>() += *o.getAddress<std::string>();
- return *this;
- }
- *this = detail::numericOp<std::plus>(*this, o);
- return *this;
- }
- inline dynamic& dynamic::operator-=(dynamic const& o) {
- *this = detail::numericOp<std::minus>(*this, o);
- return *this;
- }
- inline dynamic& dynamic::operator*=(dynamic const& o) {
- *this = detail::numericOp<std::multiplies>(*this, o);
- return *this;
- }
- inline dynamic& dynamic::operator/=(dynamic const& o) {
- *this = detail::numericOp<std::divides>(*this, o);
- return *this;
- }
- #define FB_DYNAMIC_INTEGER_OP(op) \
- inline dynamic& dynamic::operator op(dynamic const& o) { \
- if (!isInt() || !o.isInt()) { \
- throw_exception<TypeError>("int64", type(), o.type()); \
- } \
- *getAddress<int64_t>() op o.asInt(); \
- return *this; \
- }
- FB_DYNAMIC_INTEGER_OP(%=)
- FB_DYNAMIC_INTEGER_OP(|=)
- FB_DYNAMIC_INTEGER_OP(&=)
- FB_DYNAMIC_INTEGER_OP(^=)
- #undef FB_DYNAMIC_INTEGER_OP
- inline dynamic& dynamic::operator++() {
- ++get<int64_t>();
- return *this;
- }
- inline dynamic& dynamic::operator--() {
- --get<int64_t>();
- return *this;
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic const&> dynamic::operator[](
- K&& idx) const& {
- return at(std::forward<K>(idx));
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic&> dynamic::operator[](
- K&& idx) & {
- if (!isObject() && !isArray()) {
- throw_exception<TypeError>("object/array", type());
- }
- if (isArray()) {
- return at(std::forward<K>(idx));
- }
- auto& obj = get<ObjectImpl>();
- auto ret = obj.emplace(std::forward<K>(idx), nullptr);
- return ret.first->second;
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic&&> dynamic::operator[](
- K&& idx) && {
- return std::move((*this)[std::forward<K>(idx)]);
- }
- inline dynamic const& dynamic::operator[](StringPiece k) const& {
- return at(k);
- }
- inline dynamic&& dynamic::operator[](StringPiece k) && {
- return std::move((*this)[k]);
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic> dynamic::getDefault(
- K&& k,
- const dynamic& v) const& {
- auto& obj = get<ObjectImpl>();
- auto it = obj.find(std::forward<K>(k));
- return it == obj.end() ? v : it->second;
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic> dynamic::getDefault(
- K&& k,
- dynamic&& v) const& {
- auto& obj = get<ObjectImpl>();
- auto it = obj.find(std::forward<K>(k));
- // Avoid clang bug with ternary
- if (it == obj.end()) {
- return std::move(v);
- } else {
- return it->second;
- }
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic> dynamic::getDefault(
- K&& k,
- const dynamic& v) && {
- auto& obj = get<ObjectImpl>();
- auto it = obj.find(std::forward<K>(k));
- // Avoid clang bug with ternary
- if (it == obj.end()) {
- return v;
- } else {
- return std::move(it->second);
- }
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic> dynamic::getDefault(
- K&& k,
- dynamic&& v) && {
- auto& obj = get<ObjectImpl>();
- auto it = obj.find(std::forward<K>(k));
- return std::move(it == obj.end() ? v : it->second);
- }
- template <typename K, typename V>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic&> dynamic::setDefault(
- K&& k,
- V&& v) {
- auto& obj = get<ObjectImpl>();
- return obj.emplace(std::forward<K>(k), std::forward<V>(v)).first->second;
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic&> dynamic::setDefault(
- K&& k,
- dynamic&& v) {
- auto& obj = get<ObjectImpl>();
- return obj.emplace(std::forward<K>(k), std::move(v)).first->second;
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic&> dynamic::setDefault(
- K&& k,
- const dynamic& v) {
- auto& obj = get<ObjectImpl>();
- return obj.emplace(std::forward<K>(k), v).first->second;
- }
- template <typename V>
- dynamic& dynamic::setDefault(StringPiece k, V&& v) {
- auto& obj = get<ObjectImpl>();
- return obj.emplace(k, std::forward<V>(v)).first->second;
- }
- inline dynamic& dynamic::setDefault(StringPiece k, dynamic&& v) {
- auto& obj = get<ObjectImpl>();
- return obj.emplace(k, std::move(v)).first->second;
- }
- inline dynamic& dynamic::setDefault(StringPiece k, const dynamic& v) {
- auto& obj = get<ObjectImpl>();
- return obj.emplace(k, v).first->second;
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic const*> dynamic::get_ptr(
- K&& k) const& {
- return get_ptrImpl(std::forward<K>(k));
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic*> dynamic::get_ptr(
- K&& idx) & {
- return const_cast<dynamic*>(const_cast<dynamic const*>(this)->get_ptr(idx));
- }
- inline dynamic* dynamic::get_ptr(StringPiece idx) & {
- return const_cast<dynamic*>(const_cast<dynamic const*>(this)->get_ptr(idx));
- }
- inline dynamic* dynamic::get_ptr(json_pointer const& jsonPtr) & {
- return const_cast<dynamic*>(
- const_cast<dynamic const*>(this)->get_ptr(jsonPtr));
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic const&> dynamic::at(
- K&& k) const& {
- return atImpl(std::forward<K>(k));
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic&> dynamic::at(K&& idx) & {
- return const_cast<dynamic&>(const_cast<dynamic const*>(this)->at(idx));
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic&&> dynamic::at(K&& idx) && {
- return std::move(at(idx));
- }
- inline dynamic& dynamic::at(StringPiece idx) & {
- return const_cast<dynamic&>(const_cast<dynamic const*>(this)->at(idx));
- }
- inline dynamic&& dynamic::at(StringPiece idx) && {
- return std::move(at(idx));
- }
- inline bool dynamic::empty() const {
- if (isNull()) {
- return true;
- }
- return !size();
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic::const_item_iterator>
- dynamic::find(K&& key) const {
- return get<ObjectImpl>().find(std::forward<K>(key));
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, dynamic::item_iterator>
- dynamic::find(K&& key) {
- return get<ObjectImpl>().find(std::forward<K>(key));
- }
- inline dynamic::const_item_iterator dynamic::find(StringPiece key) const {
- return get<ObjectImpl>().find(key);
- }
- inline dynamic::item_iterator dynamic::find(StringPiece key) {
- return get<ObjectImpl>().find(key);
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, std::size_t> dynamic::count(
- K&& key) const {
- return find(std::forward<K>(key)) != items().end() ? 1u : 0u;
- }
- inline std::size_t dynamic::count(StringPiece key) const {
- return find(key) != items().end() ? 1u : 0u;
- }
- template <class K, class V>
- inline void dynamic::insert(K&& key, V&& val) {
- auto& obj = get<ObjectImpl>();
- obj[std::forward<K>(key)] = std::forward<V>(val);
- }
- inline void dynamic::update(const dynamic& mergeObj) {
- if (!isObject() || !mergeObj.isObject()) {
- throw_exception<TypeError>("object", type(), mergeObj.type());
- }
- for (const auto& pair : mergeObj.items()) {
- (*this)[pair.first] = pair.second;
- }
- }
- inline void dynamic::update_missing(const dynamic& mergeObj1) {
- if (!isObject() || !mergeObj1.isObject()) {
- throw_exception<TypeError>("object", type(), mergeObj1.type());
- }
- // Only add if not already there
- for (const auto& pair : mergeObj1.items()) {
- if ((*this).find(pair.first) == (*this).items().end()) {
- (*this)[pair.first] = pair.second;
- }
- }
- }
- inline void dynamic::merge_patch(const dynamic& patch) {
- auto& self = *this;
- if (!patch.isObject()) {
- self = patch;
- return;
- }
- // if we are not an object, erase all contents, reset to object
- if (!isObject()) {
- self = object;
- }
- for (const auto& pair : patch.items()) {
- if (pair.second.isNull()) {
- // if name could be found in current object, remove it
- auto it = self.find(pair.first);
- if (it != self.items().end()) {
- self.erase(it);
- }
- } else {
- self[pair.first].merge_patch(pair.second);
- }
- }
- }
- inline dynamic dynamic::merge(
- const dynamic& mergeObj1,
- const dynamic& mergeObj2) {
- // No checks on type needed here because they are done in update_missing
- // Note that we do update_missing here instead of update() because
- // it will prevent the extra writes that would occur with update()
- auto ret = mergeObj2;
- ret.update_missing(mergeObj1);
- return ret;
- }
- template <typename K>
- dynamic::IfIsNonStringDynamicConvertible<K, std::size_t> dynamic::erase(
- K&& key) {
- auto& obj = get<ObjectImpl>();
- return obj.erase(std::forward<K>(key));
- }
- inline std::size_t dynamic::erase(StringPiece key) {
- auto& obj = get<ObjectImpl>();
- return obj.erase(key);
- }
- inline dynamic::iterator dynamic::erase(const_iterator it) {
- auto& arr = get<Array>();
- // std::vector doesn't have an erase method that works on const iterators,
- // even though the standard says it should, so this hack converts to a
- // non-const iterator before calling erase.
- return get<Array>().erase(arr.begin() + (it - arr.begin()));
- }
- inline dynamic::const_key_iterator dynamic::erase(const_key_iterator it) {
- return const_key_iterator(get<ObjectImpl>().erase(it.base()));
- }
- inline dynamic::const_key_iterator dynamic::erase(
- const_key_iterator first,
- const_key_iterator last) {
- return const_key_iterator(get<ObjectImpl>().erase(first.base(), last.base()));
- }
- inline dynamic::value_iterator dynamic::erase(const_value_iterator it) {
- return value_iterator(get<ObjectImpl>().erase(it.base()));
- }
- inline dynamic::value_iterator dynamic::erase(
- const_value_iterator first,
- const_value_iterator last) {
- return value_iterator(get<ObjectImpl>().erase(first.base(), last.base()));
- }
- inline dynamic::item_iterator dynamic::erase(const_item_iterator it) {
- return item_iterator(get<ObjectImpl>().erase(it.base()));
- }
- inline dynamic::item_iterator dynamic::erase(
- const_item_iterator first,
- const_item_iterator last) {
- return item_iterator(get<ObjectImpl>().erase(first.base(), last.base()));
- }
- inline void dynamic::resize(std::size_t sz, dynamic const& c) {
- auto& arr = get<Array>();
- arr.resize(sz, c);
- }
- inline void dynamic::push_back(dynamic const& v) {
- auto& arr = get<Array>();
- arr.push_back(v);
- }
- inline void dynamic::push_back(dynamic&& v) {
- auto& arr = get<Array>();
- arr.push_back(std::move(v));
- }
- inline void dynamic::pop_back() {
- auto& arr = get<Array>();
- arr.pop_back();
- }
- //////////////////////////////////////////////////////////////////////
- inline dynamic::dynamic(Array&& r) : type_(ARRAY) {
- new (&u_.array) Array(std::move(r));
- }
- #define FOLLY_DYNAMIC_DEC_TYPEINFO(T, str, val) \
- template <> \
- struct dynamic::TypeInfo<T> { \
- static constexpr const char* name = str; \
- static constexpr dynamic::Type type = val; \
- }; \
- //
- FOLLY_DYNAMIC_DEC_TYPEINFO(std::nullptr_t, "null", dynamic::NULLT)
- FOLLY_DYNAMIC_DEC_TYPEINFO(bool, "boolean", dynamic::BOOL)
- FOLLY_DYNAMIC_DEC_TYPEINFO(std::string, "string", dynamic::STRING)
- FOLLY_DYNAMIC_DEC_TYPEINFO(dynamic::Array, "array", dynamic::ARRAY)
- FOLLY_DYNAMIC_DEC_TYPEINFO(double, "double", dynamic::DOUBLE)
- FOLLY_DYNAMIC_DEC_TYPEINFO(int64_t, "int64", dynamic::INT64)
- FOLLY_DYNAMIC_DEC_TYPEINFO(dynamic::ObjectImpl, "object", dynamic::OBJECT)
- #undef FOLLY_DYNAMIC_DEC_TYPEINFO
- template <class T>
- T dynamic::asImpl() const {
- switch (type()) {
- case INT64:
- return to<T>(*get_nothrow<int64_t>());
- case DOUBLE:
- return to<T>(*get_nothrow<double>());
- case BOOL:
- return to<T>(*get_nothrow<bool>());
- case STRING:
- return to<T>(*get_nothrow<std::string>());
- default:
- throw_exception<TypeError>("int/double/bool/string", type());
- }
- }
- // Return a T* to our type, or null if we're not that type.
- // clang-format off
- template <class T>
- T* dynamic::get_nothrow() & noexcept {
- if (type_ != TypeInfo<T>::type) {
- return nullptr;
- }
- return getAddress<T>();
- }
- // clang-format on
- template <class T>
- T const* dynamic::get_nothrow() const& noexcept {
- return const_cast<dynamic*>(this)->get_nothrow<T>();
- }
- // Return T* for where we can put a T, without type checking. (Memory
- // might be uninitialized, even.)
- template <class T>
- T* dynamic::getAddress() noexcept {
- return GetAddrImpl<T>::get(u_);
- }
- template <class T>
- T const* dynamic::getAddress() const noexcept {
- return const_cast<dynamic*>(this)->getAddress<T>();
- }
- template <class T>
- struct dynamic::GetAddrImpl {};
- template <>
- struct dynamic::GetAddrImpl<std::nullptr_t> {
- static std::nullptr_t* get(Data& d) noexcept {
- return &d.nul;
- }
- };
- template <>
- struct dynamic::GetAddrImpl<dynamic::Array> {
- static Array* get(Data& d) noexcept {
- return &d.array;
- }
- };
- template <>
- struct dynamic::GetAddrImpl<bool> {
- static bool* get(Data& d) noexcept {
- return &d.boolean;
- }
- };
- template <>
- struct dynamic::GetAddrImpl<int64_t> {
- static int64_t* get(Data& d) noexcept {
- return &d.integer;
- }
- };
- template <>
- struct dynamic::GetAddrImpl<double> {
- static double* get(Data& d) noexcept {
- return &d.doubl;
- }
- };
- template <>
- struct dynamic::GetAddrImpl<std::string> {
- static std::string* get(Data& d) noexcept {
- return &d.string;
- }
- };
- template <>
- struct dynamic::GetAddrImpl<dynamic::ObjectImpl> {
- static_assert(
- sizeof(ObjectImpl) <= sizeof(Data::objectBuffer),
- "In your implementation, F14NodeMap<> apparently takes different"
- " amount of space depending on its template parameters. This is "
- "weird. Make objectBuffer bigger if you want to compile dynamic.");
- static ObjectImpl* get(Data& d) noexcept {
- void* data = &d.objectBuffer;
- return static_cast<ObjectImpl*>(data);
- }
- };
- template <class T>
- T& dynamic::get() {
- if (auto* p = get_nothrow<T>()) {
- return *p;
- }
- throw_exception<TypeError>(TypeInfo<T>::name, type());
- }
- template <class T>
- T const& dynamic::get() const {
- return const_cast<dynamic*>(this)->get<T>();
- }
- //////////////////////////////////////////////////////////////////////
- /*
- * Helper for implementing operator<<. Throws if the type shouldn't
- * support it.
- */
- template <class T>
- struct dynamic::PrintImpl {
- static void print(dynamic const&, std::ostream& out, T const& t) {
- out << t;
- }
- };
- // Otherwise, null, being (void*)0, would print as 0.
- template <>
- struct dynamic::PrintImpl<std::nullptr_t> {
- static void
- print(dynamic const& /* d */, std::ostream& out, std::nullptr_t const&) {
- out << "null";
- }
- };
- template <>
- struct dynamic::PrintImpl<dynamic::ObjectImpl> {
- static void
- print(dynamic const& d, std::ostream& out, dynamic::ObjectImpl const&) {
- d.print_as_pseudo_json(out);
- }
- };
- template <>
- struct dynamic::PrintImpl<dynamic::Array> {
- static void
- print(dynamic const& d, std::ostream& out, dynamic::Array const&) {
- d.print_as_pseudo_json(out);
- }
- };
- inline void dynamic::print(std::ostream& out) const {
- #define FB_X(T) PrintImpl<T>::print(*this, out, *getAddress<T>())
- FB_DYNAMIC_APPLY(type_, FB_X);
- #undef FB_X
- }
- inline std::ostream& operator<<(std::ostream& out, dynamic const& d) {
- d.print(out);
- return out;
- }
- //////////////////////////////////////////////////////////////////////
- // Secialization of FormatValue so dynamic objects can be formatted
- template <>
- class FormatValue<dynamic> {
- public:
- explicit FormatValue(const dynamic& val) : val_(val) {}
- template <class FormatCallback>
- void format(FormatArg& arg, FormatCallback& cb) const {
- switch (val_.type()) {
- case dynamic::NULLT:
- FormatValue<std::nullptr_t>(nullptr).format(arg, cb);
- break;
- case dynamic::BOOL:
- FormatValue<bool>(val_.asBool()).format(arg, cb);
- break;
- case dynamic::INT64:
- FormatValue<int64_t>(val_.asInt()).format(arg, cb);
- break;
- case dynamic::STRING:
- FormatValue<std::string>(val_.asString()).format(arg, cb);
- break;
- case dynamic::DOUBLE:
- FormatValue<double>(val_.asDouble()).format(arg, cb);
- break;
- case dynamic::ARRAY:
- FormatValue(val_.at(arg.splitIntKey())).format(arg, cb);
- break;
- case dynamic::OBJECT:
- FormatValue(val_.at(arg.splitKey().toString())).format(arg, cb);
- break;
- }
- }
- private:
- const dynamic& val_;
- };
- template <class V>
- class FormatValue<detail::DefaultValueWrapper<dynamic, V>> {
- public:
- explicit FormatValue(const detail::DefaultValueWrapper<dynamic, V>& val)
- : val_(val) {}
- template <class FormatCallback>
- void format(FormatArg& arg, FormatCallback& cb) const {
- auto& c = val_.container;
- switch (c.type()) {
- case dynamic::NULLT:
- case dynamic::BOOL:
- case dynamic::INT64:
- case dynamic::STRING:
- case dynamic::DOUBLE:
- FormatValue<dynamic>(c).format(arg, cb);
- break;
- case dynamic::ARRAY: {
- int key = arg.splitIntKey();
- if (key >= 0 && size_t(key) < c.size()) {
- FormatValue<dynamic>(c.at(key)).format(arg, cb);
- } else {
- FormatValue<V>(val_.defaultValue).format(arg, cb);
- }
- break;
- }
- case dynamic::OBJECT: {
- auto pos = c.find(arg.splitKey());
- if (pos != c.items().end()) {
- FormatValue<dynamic>(pos->second).format(arg, cb);
- } else {
- FormatValue<V>(val_.defaultValue).format(arg, cb);
- }
- break;
- }
- }
- }
- private:
- const detail::DefaultValueWrapper<dynamic, V>& val_;
- };
- } // namespace folly
- #undef FB_DYNAMIC_APPLY
|