add support for whenAll to waitWithSemaphore
[folly.git] / folly / dynamic-inl.h
1 /*
2  * Copyright 2014 Facebook, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *   http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #ifndef FOLLY_DYNAMIC_INL_H_
18 #define FOLLY_DYNAMIC_INL_H_
19
20 #include <functional>
21 #include <boost/iterator/iterator_adaptor.hpp>
22 #include <boost/iterator/iterator_facade.hpp>
23 #include "folly/Likely.h"
24 #include "folly/Conv.h"
25 #include "folly/Format.h"
26
27 //////////////////////////////////////////////////////////////////////
28
29 namespace std {
30
31 template<>
32 struct hash< ::folly::dynamic> {
33   size_t operator()(::folly::dynamic const& d) const {
34     return d.hash();
35   }
36 };
37
38 }
39
40 //////////////////////////////////////////////////////////////////////
41
42 // This is a higher-order preprocessor macro to aid going from runtime
43 // types to the compile time type system.
44 #define FB_DYNAMIC_APPLY(type, apply) do {         \
45   switch ((type)) {                             \
46   case NULLT:   apply(void*);          break;   \
47   case ARRAY:   apply(Array);          break;   \
48   case BOOL:    apply(bool);           break;   \
49   case DOUBLE:  apply(double);         break;   \
50   case INT64:   apply(int64_t);        break;   \
51   case OBJECT:  apply(ObjectImpl);     break;   \
52   case STRING:  apply(fbstring);       break;   \
53   default:      CHECK(0); abort();              \
54   }                                             \
55 } while (0)
56
57 //////////////////////////////////////////////////////////////////////
58
59 namespace folly {
60
61 struct TypeError : std::runtime_error {
62   explicit TypeError(const std::string& expected, dynamic::Type actual)
63     : std::runtime_error(to<std::string>("TypeError: expected dynamic "
64         "type `", expected, '\'', ", but had type `",
65         dynamic::typeName(actual), '\''))
66   {}
67   explicit TypeError(const std::string& expected,
68       dynamic::Type actual1, dynamic::Type actual2)
69     : std::runtime_error(to<std::string>("TypeError: expected dynamic "
70         "types `", expected, '\'', ", but had types `",
71         dynamic::typeName(actual1), "' and `", dynamic::typeName(actual2),
72         '\''))
73   {}
74 };
75
76
77 //////////////////////////////////////////////////////////////////////
78
79 namespace detail {
80
81   // This helper is used in destroy() to be able to run destructors on
82   // types like "int64_t" without a compiler error.
83   struct Destroy {
84     template<class T> static void destroy(T* t) { t->~T(); }
85   };
86
87   /*
88    * The enable_if junk here is necessary to avoid ambiguous
89    * conversions relating to bool and double when you implicitly
90    * convert an int or long to a dynamic.
91    */
92   template<class T, class Enable = void> struct ConversionHelper;
93   template<class T>
94   struct ConversionHelper<
95     T,
96     typename std::enable_if<
97       std::is_integral<T>::value && !std::is_same<T,bool>::value
98     >::type
99   > {
100     typedef int64_t type;
101   };
102   template<class T>
103   struct ConversionHelper<
104     T,
105     typename std::enable_if<
106       (!std::is_integral<T>::value || std::is_same<T,bool>::value) &&
107       !std::is_same<T,std::nullptr_t>::value
108     >::type
109   > {
110     typedef T type;
111   };
112   template<class T>
113   struct ConversionHelper<
114     T,
115     typename std::enable_if<
116       std::is_same<T,std::nullptr_t>::value
117     >::type
118   > {
119     typedef void* type;
120   };
121
122   /*
123    * Helper for implementing numeric conversions in operators on
124    * numbers.  Just promotes to double when one of the arguments is
125    * double, or throws if either is not a numeric type.
126    */
127   template<template<class> class Op>
128   dynamic numericOp(dynamic const& a, dynamic const& b) {
129     if (!a.isNumber() || !b.isNumber()) {
130       throw TypeError("numeric", a.type(), b.type());
131     }
132     if (a.type() != b.type()) {
133       auto& integ  = a.isInt() ? a : b;
134       auto& nonint = a.isInt() ? b : a;
135       return Op<double>()(to<double>(integ.asInt()), nonint.asDouble());
136     }
137     if (a.isDouble()) {
138       return Op<double>()(a.asDouble(), b.asDouble());
139     }
140     return Op<int64_t>()(a.asInt(), b.asInt());
141   }
142
143 }
144
145 //////////////////////////////////////////////////////////////////////
146
147 /*
148  * We're doing this instead of a simple member typedef to avoid the
149  * undefined behavior of parameterizing std::unordered_map<> with an
150  * incomplete type.
151  *
152  * Note: Later we may add separate order tracking here (a multi-index
153  * type of thing.)
154  */
155 struct dynamic::ObjectImpl : std::unordered_map<dynamic, dynamic> {};
156
157 //////////////////////////////////////////////////////////////////////
158
159 // Helper object for creating objects conveniently.  See object and
160 // the dynamic::dynamic(ObjectMaker&&) ctor.
161 struct dynamic::ObjectMaker {
162   friend struct dynamic;
163
164   explicit ObjectMaker() : val_(dynamic::object) {}
165   explicit ObjectMaker(dynamic const& key, dynamic val)
166     : val_(dynamic::object)
167   {
168     val_.insert(key, std::move(val));
169   }
170   explicit ObjectMaker(dynamic&& key, dynamic val)
171     : val_(dynamic::object)
172   {
173     val_.insert(std::move(key), std::move(val));
174   }
175
176   // Make sure no one tries to save one of these into an lvalue with
177   // auto or anything like that.
178   ObjectMaker(ObjectMaker&&) = default;
179   ObjectMaker(ObjectMaker const&) = delete;
180   ObjectMaker& operator=(ObjectMaker const&) = delete;
181   ObjectMaker& operator=(ObjectMaker&&) = delete;
182
183   // These return rvalue-references instead of lvalue-references to allow
184   // constructs like this to moved instead of copied:
185   //  dynamic a = dynamic::object("a", "b")("c", "d")
186   ObjectMaker&& operator()(dynamic const& key, dynamic val) {
187     val_.insert(key, std::move(val));
188     return std::move(*this);
189   }
190
191   ObjectMaker&& operator()(dynamic&& key, dynamic val) {
192     val_.insert(std::move(key), std::move(val));
193     return std::move(*this);
194   }
195
196 private:
197   dynamic val_;
198 };
199
200 // This looks like a case for perfect forwarding, but our use of
201 // std::initializer_list for constructing dynamic arrays makes it less
202 // functional than doing this manually.
203 inline dynamic::ObjectMaker dynamic::object() { return ObjectMaker(); }
204 inline dynamic::ObjectMaker dynamic::object(dynamic&& a, dynamic&& b) {
205   return ObjectMaker(std::move(a), std::move(b));
206 }
207 inline dynamic::ObjectMaker dynamic::object(dynamic const& a, dynamic&& b) {
208   return ObjectMaker(a, std::move(b));
209 }
210 inline dynamic::ObjectMaker dynamic::object(dynamic&& a, dynamic const& b) {
211   return ObjectMaker(std::move(a), b);
212 }
213 inline dynamic::ObjectMaker
214 dynamic::object(dynamic const& a, dynamic const& b) {
215   return ObjectMaker(a, b);
216 }
217
218 //////////////////////////////////////////////////////////////////////
219
220 struct dynamic::const_item_iterator
221   : boost::iterator_adaptor<dynamic::const_item_iterator,
222                             dynamic::ObjectImpl::const_iterator> {
223   /* implicit */ const_item_iterator(base_type b) : iterator_adaptor_(b) { }
224
225  private:
226   friend class boost::iterator_core_access;
227 };
228
229 struct dynamic::const_key_iterator
230   : boost::iterator_adaptor<dynamic::const_key_iterator,
231                             dynamic::ObjectImpl::const_iterator,
232                             dynamic const> {
233   /* implicit */ const_key_iterator(base_type b) : iterator_adaptor_(b) { }
234
235  private:
236   dynamic const& dereference() const {
237     return base_reference()->first;
238   }
239   friend class boost::iterator_core_access;
240 };
241
242 struct dynamic::const_value_iterator
243   : boost::iterator_adaptor<dynamic::const_value_iterator,
244                             dynamic::ObjectImpl::const_iterator,
245                             dynamic const> {
246   /* implicit */ const_value_iterator(base_type b) : iterator_adaptor_(b) { }
247
248  private:
249   dynamic const& dereference() const {
250     return base_reference()->second;
251   }
252   friend class boost::iterator_core_access;
253 };
254
255 //////////////////////////////////////////////////////////////////////
256
257 inline dynamic::dynamic(ObjectMaker (*)())
258   : type_(OBJECT)
259 {
260   new (getAddress<ObjectImpl>()) ObjectImpl();
261 }
262
263 inline dynamic::dynamic(StringPiece s)
264   : type_(STRING)
265 {
266   new (&u_.string) fbstring(s.data(), s.size());
267 }
268
269 inline dynamic::dynamic(char const* s)
270   : type_(STRING)
271 {
272   new (&u_.string) fbstring(s);
273 }
274
275 inline dynamic::dynamic(std::string const& s)
276   : type_(STRING)
277 {
278   new (&u_.string) fbstring(s);
279 }
280
281 inline dynamic::dynamic(fbstring const& s)
282   : type_(STRING)
283 {
284   new (&u_.string) fbstring(s);
285 }
286
287 inline dynamic::dynamic(fbstring&& s)
288   : type_(STRING)
289 {
290   new (&u_.string) fbstring(std::move(s));
291 }
292
293 inline dynamic::dynamic(std::initializer_list<dynamic> il)
294   : type_(ARRAY)
295 {
296   new (&u_.array) Array(il.begin(), il.end());
297 }
298
299 inline dynamic::dynamic(ObjectMaker&& maker)
300   : type_(OBJECT)
301 {
302   new (getAddress<ObjectImpl>())
303     ObjectImpl(std::move(*maker.val_.getAddress<ObjectImpl>()));
304 }
305
306 inline dynamic::dynamic(dynamic const& o)
307   : type_(NULLT)
308 {
309   *this = o;
310 }
311
312 inline dynamic::dynamic(dynamic&& o)
313   : type_(NULLT)
314 {
315   *this = std::move(o);
316 }
317
318 inline dynamic::~dynamic() { destroy(); }
319
320 template<class T>
321 dynamic::dynamic(T t) {
322   typedef typename detail::ConversionHelper<T>::type U;
323   type_ = TypeInfo<U>::type;
324   new (getAddress<U>()) U(std::move(t));
325 }
326
327 template<class Iterator>
328 dynamic::dynamic(Iterator first, Iterator last)
329   : type_(ARRAY)
330 {
331   new (&u_.array) Array(first, last);
332 }
333
334 //////////////////////////////////////////////////////////////////////
335
336 inline dynamic::const_iterator dynamic::begin() const {
337   return get<Array>().begin();
338 }
339 inline dynamic::const_iterator dynamic::end() const {
340   return get<Array>().end();
341 }
342
343 template <class It>
344 struct dynamic::IterableProxy {
345   typedef It const_iterator;
346   typedef typename It::value_type value_type;
347
348   /* implicit */ IterableProxy(const dynamic::ObjectImpl* o) : o_(o) { }
349
350   It begin() const {
351     return o_->begin();
352   }
353
354   It end() const {
355     return o_->end();
356   }
357
358  private:
359   const dynamic::ObjectImpl* o_;
360 };
361
362 inline dynamic::IterableProxy<dynamic::const_key_iterator> dynamic::keys()
363   const {
364   return &(get<ObjectImpl>());
365 }
366
367 inline dynamic::IterableProxy<dynamic::const_value_iterator> dynamic::values()
368   const {
369   return &(get<ObjectImpl>());
370 }
371
372 inline dynamic::IterableProxy<dynamic::const_item_iterator> dynamic::items()
373   const {
374   return &(get<ObjectImpl>());
375 }
376
377 inline bool dynamic::isString() const { return get_nothrow<fbstring>(); }
378 inline bool dynamic::isObject() const { return get_nothrow<ObjectImpl>(); }
379 inline bool dynamic::isBool()   const { return get_nothrow<bool>(); }
380 inline bool dynamic::isArray()  const { return get_nothrow<Array>(); }
381 inline bool dynamic::isDouble() const { return get_nothrow<double>(); }
382 inline bool dynamic::isInt()    const { return get_nothrow<int64_t>(); }
383 inline bool dynamic::isNull()   const { return get_nothrow<void*>(); }
384 inline bool dynamic::isNumber() const { return isInt() || isDouble(); }
385
386 inline dynamic::Type dynamic::type() const {
387   return type_;
388 }
389
390 inline fbstring dynamic::asString() const { return asImpl<fbstring>(); }
391 inline double   dynamic::asDouble() const { return asImpl<double>(); }
392 inline int64_t  dynamic::asInt()    const { return asImpl<int64_t>(); }
393 inline bool     dynamic::asBool()   const { return asImpl<bool>(); }
394
395 inline const char* dynamic::data()  const { return get<fbstring>().data();  }
396 inline const char* dynamic::c_str() const { return get<fbstring>().c_str(); }
397
398 template<class T>
399 struct dynamic::CompareOp {
400   static bool comp(T const& a, T const& b) { return a < b; }
401 };
402 template<>
403 struct dynamic::CompareOp<dynamic::ObjectImpl> {
404   static bool comp(ObjectImpl const& a, ObjectImpl const& b) {
405     // This code never executes; it is just here for the compiler.
406     return false;
407   }
408 };
409
410 inline bool dynamic::operator<(dynamic const& o) const {
411   if (UNLIKELY(type_ == OBJECT || o.type_ == OBJECT)) {
412     throw TypeError("object", type_);
413   }
414   if (type_ != o.type_) {
415     return type_ < o.type_;
416   }
417
418 #define FB_X(T) return CompareOp<T>::comp(*getAddress<T>(),   \
419                                           *o.getAddress<T>())
420   FB_DYNAMIC_APPLY(type_, FB_X);
421 #undef FB_X
422 }
423
424 inline bool dynamic::operator==(dynamic const& o) const {
425   if (type() != o.type()) {
426     if (isNumber() && o.isNumber()) {
427       auto& integ = isInt() ? *this : o;
428       auto& doubl = isInt() ? o     : *this;
429       return integ.asInt() == doubl.asDouble();
430     }
431     return false;
432   }
433
434 #define FB_X(T) return *getAddress<T>() == *o.getAddress<T>();
435   FB_DYNAMIC_APPLY(type_, FB_X);
436 #undef FB_X
437 }
438
439 inline dynamic& dynamic::operator+=(dynamic const& o) {
440   if (type() == STRING && o.type() == STRING) {
441     *getAddress<fbstring>() += *o.getAddress<fbstring>();
442     return *this;
443   }
444   *this = detail::numericOp<std::plus>(*this, o);
445   return *this;
446 }
447
448 inline dynamic& dynamic::operator-=(dynamic const& o) {
449   *this = detail::numericOp<std::minus>(*this, o);
450   return *this;
451 }
452
453 inline dynamic& dynamic::operator*=(dynamic const& o) {
454   *this = detail::numericOp<std::multiplies>(*this, o);
455   return *this;
456 }
457
458 inline dynamic& dynamic::operator/=(dynamic const& o) {
459   *this = detail::numericOp<std::divides>(*this, o);
460   return *this;
461 }
462
463 #define FB_DYNAMIC_INTEGER_OP(op)                           \
464   inline dynamic& dynamic::operator op(dynamic const& o) {  \
465     if (!isInt() || !o.isInt()) {                           \
466       throw TypeError("int64", type(), o.type());           \
467     }                                                       \
468     *getAddress<int64_t>() op o.asInt();                    \
469     return *this;                                           \
470   }
471
472 FB_DYNAMIC_INTEGER_OP(%=)
473 FB_DYNAMIC_INTEGER_OP(|=)
474 FB_DYNAMIC_INTEGER_OP(&=)
475 FB_DYNAMIC_INTEGER_OP(^=)
476
477 #undef FB_DYNAMIC_INTEGER_OP
478
479 inline dynamic& dynamic::operator++() {
480   ++get<int64_t>();
481   return *this;
482 }
483
484 inline dynamic& dynamic::operator--() {
485   --get<int64_t>();
486   return *this;
487 }
488
489 inline dynamic& dynamic::operator=(dynamic const& o) {
490   if (&o != this) {
491     destroy();
492 #define FB_X(T) new (getAddress<T>()) T(*o.getAddress<T>())
493     FB_DYNAMIC_APPLY(o.type_, FB_X);
494 #undef FB_X
495     type_ = o.type_;
496   }
497   return *this;
498 }
499
500 inline dynamic& dynamic::operator=(dynamic&& o) {
501   if (&o != this) {
502     destroy();
503 #define FB_X(T) new (getAddress<T>()) T(std::move(*o.getAddress<T>()))
504     FB_DYNAMIC_APPLY(o.type_, FB_X);
505 #undef FB_X
506     type_ = o.type_;
507   }
508   return *this;
509 }
510
511 inline dynamic& dynamic::operator[](dynamic const& k) {
512   if (!isObject() && !isArray()) {
513     throw TypeError("object/array", type());
514   }
515   if (isArray()) {
516     return at(k);
517   }
518   auto& obj = get<ObjectImpl>();
519   auto ret = obj.insert({k, nullptr});
520   return ret.first->second;
521 }
522
523 inline dynamic const& dynamic::operator[](dynamic const& idx) const {
524   return at(idx);
525 }
526
527 inline dynamic dynamic::getDefault(const dynamic& k, const dynamic& v) const {
528   auto& obj = get<ObjectImpl>();
529   auto it = obj.find(k);
530   return it == obj.end() ? v : it->second;
531 }
532
533 inline dynamic&& dynamic::getDefault(const dynamic& k, dynamic&& v) const {
534   auto& obj = get<ObjectImpl>();
535   auto it = obj.find(k);
536   if (it != obj.end()) {
537     v = it->second;
538   }
539
540   return std::move(v);
541 }
542
543 template<class K, class V> inline dynamic& dynamic::setDefault(K&& k, V&& v) {
544   auto& obj = get<ObjectImpl>();
545   return obj.insert(std::make_pair(std::forward<K>(k),
546                                    std::forward<V>(v))).first->second;
547 }
548
549 inline dynamic* dynamic::get_ptr(dynamic const& idx) {
550   return const_cast<dynamic*>(const_cast<dynamic const*>(this)->get_ptr(idx));
551 }
552
553 inline const dynamic* dynamic::get_ptr(dynamic const& idx) const {
554   if (auto* parray = get_nothrow<Array>()) {
555     if (!idx.isInt()) {
556       throw TypeError("int64", idx.type());
557     }
558     if (idx >= parray->size()) {
559       return nullptr;
560     }
561     return &(*parray)[idx.asInt()];
562   } else if (auto* pobject = get_nothrow<ObjectImpl>()) {
563     auto it = pobject->find(idx);
564     if (it == pobject->end()) {
565       return nullptr;
566     }
567     return &it->second;
568   } else {
569     throw TypeError("object/array", type());
570   }
571 }
572
573 inline dynamic& dynamic::at(dynamic const& idx) {
574   return const_cast<dynamic&>(const_cast<dynamic const*>(this)->at(idx));
575 }
576
577 inline dynamic const& dynamic::at(dynamic const& idx) const {
578   if (auto* parray = get_nothrow<Array>()) {
579     if (!idx.isInt()) {
580       throw TypeError("int64", idx.type());
581     }
582     if (idx >= parray->size()) {
583       throw std::out_of_range("out of range in dynamic array");
584     }
585     return (*parray)[idx.asInt()];
586   } else if (auto* pobject = get_nothrow<ObjectImpl>()) {
587     auto it = pobject->find(idx);
588     if (it == pobject->end()) {
589       throw std::out_of_range(to<std::string>(
590           "couldn't find key ", idx.asString(), " in dynamic object"));
591     }
592     return it->second;
593   } else {
594     throw TypeError("object/array", type());
595   }
596 }
597
598 inline bool dynamic::empty() const {
599   if (isNull()) {
600     return true;
601   }
602   return !size();
603 }
604
605 inline std::size_t dynamic::size() const {
606   if (auto* ar = get_nothrow<Array>()) {
607     return ar->size();
608   }
609   if (auto* obj = get_nothrow<ObjectImpl>()) {
610     return obj->size();
611   }
612   if (auto* str = get_nothrow<fbstring>()) {
613     return str->size();
614   }
615   throw TypeError("array/object", type());
616 }
617
618 inline std::size_t dynamic::count(dynamic const& key) const {
619   return find(key) != items().end();
620 }
621
622 inline dynamic::const_item_iterator dynamic::find(dynamic const& key) const {
623   return get<ObjectImpl>().find(key);
624 }
625
626 template<class K, class V> inline void dynamic::insert(K&& key, V&& val) {
627   auto& obj = get<ObjectImpl>();
628   auto rv = obj.insert(std::make_pair(std::forward<K>(key),
629                                       std::forward<V>(val)));
630   if (!rv.second) {
631     // note, the second use of std:forward<V>(val) is only correct
632     // if the first one did not result in a move. obj[key] = val
633     // would be preferrable but doesn't compile because dynamic
634     // is (intentionally) not default constructable
635     rv.first->second = std::forward<V>(val);
636   }
637 }
638
639 inline std::size_t dynamic::erase(dynamic const& key) {
640   auto& obj = get<ObjectImpl>();
641   return obj.erase(key);
642 }
643
644 inline dynamic::const_iterator dynamic::erase(const_iterator it) {
645   auto& arr = get<Array>();
646   // std::vector doesn't have an erase method that works on const iterators,
647   // even though the standard says it should, so this hack converts to a
648   // non-const iterator before calling erase.
649   return get<Array>().erase(arr.begin() + (it - arr.begin()));
650 }
651
652 inline dynamic::const_iterator
653 dynamic::erase(const_iterator first, const_iterator last) {
654   auto& arr = get<Array>();
655   return get<Array>().erase(
656     arr.begin() + (first - arr.begin()),
657     arr.begin() + (last - arr.begin()));
658 }
659
660 inline dynamic::const_key_iterator dynamic::erase(const_key_iterator it) {
661   return const_key_iterator(get<ObjectImpl>().erase(it.base()));
662 }
663
664 inline dynamic::const_key_iterator dynamic::erase(const_key_iterator first,
665                                                   const_key_iterator last) {
666   return const_key_iterator(get<ObjectImpl>().erase(first.base(),
667                                                     last.base()));
668 }
669
670 inline dynamic::const_value_iterator dynamic::erase(const_value_iterator it) {
671   return const_value_iterator(get<ObjectImpl>().erase(it.base()));
672 }
673
674 inline dynamic::const_value_iterator dynamic::erase(const_value_iterator first,
675                                                     const_value_iterator last) {
676   return const_value_iterator(get<ObjectImpl>().erase(first.base(),
677                                                       last.base()));
678 }
679
680 inline dynamic::const_item_iterator dynamic::erase(const_item_iterator it) {
681   return const_item_iterator(get<ObjectImpl>().erase(it.base()));
682 }
683
684 inline dynamic::const_item_iterator dynamic::erase(const_item_iterator first,
685                                                    const_item_iterator last) {
686   return const_item_iterator(get<ObjectImpl>().erase(first.base(),
687                                                      last.base()));
688 }
689
690 inline void dynamic::resize(std::size_t sz, dynamic const& c) {
691   auto& array = get<Array>();
692   array.resize(sz, c);
693 }
694
695 inline void dynamic::push_back(dynamic const& v) {
696   auto& array = get<Array>();
697   array.push_back(v);
698 }
699
700 inline void dynamic::push_back(dynamic&& v) {
701   auto& array = get<Array>();
702   array.push_back(std::move(v));
703 }
704
705 inline void dynamic::pop_back() {
706   auto& array = get<Array>();
707   array.pop_back();
708 }
709
710 inline std::size_t dynamic::hash() const {
711   switch (type()) {
712   case OBJECT:
713   case ARRAY:
714   case NULLT:
715     throw TypeError("not null/object/array", type());
716   case INT64:
717     return std::hash<int64_t>()(asInt());
718   case DOUBLE:
719     return std::hash<double>()(asDouble());
720   case BOOL:
721     return std::hash<bool>()(asBool());
722   case STRING:
723     return std::hash<fbstring>()(asString());
724   default:
725     CHECK(0); abort();
726   }
727 }
728
729 //////////////////////////////////////////////////////////////////////
730
731 template<class T> struct dynamic::TypeInfo {
732   static char const name[];
733   static Type const type;
734 };
735
736 #define FB_DEC_TYPE(T)                                      \
737   template<> char const dynamic::TypeInfo<T>::name[];       \
738   template<> dynamic::Type const dynamic::TypeInfo<T>::type
739
740 FB_DEC_TYPE(void*);
741 FB_DEC_TYPE(bool);
742 FB_DEC_TYPE(fbstring);
743 FB_DEC_TYPE(dynamic::Array);
744 FB_DEC_TYPE(double);
745 FB_DEC_TYPE(int64_t);
746 FB_DEC_TYPE(dynamic::ObjectImpl);
747
748 #undef FB_DEC_TYPE
749
750 template<class T>
751 T dynamic::asImpl() const {
752   switch (type()) {
753   case INT64:    return to<T>(*get_nothrow<int64_t>());
754   case DOUBLE:   return to<T>(*get_nothrow<double>());
755   case BOOL:     return to<T>(*get_nothrow<bool>());
756   case STRING:   return to<T>(*get_nothrow<fbstring>());
757   default:
758     throw TypeError("int/double/bool/string", type());
759   }
760 }
761
762 // Return a T* to our type, or null if we're not that type.
763 template<class T>
764 T* dynamic::get_nothrow() {
765   if (type_ != TypeInfo<T>::type) {
766     return nullptr;
767   }
768   return getAddress<T>();
769 }
770
771 template<class T>
772 T const* dynamic::get_nothrow() const {
773   return const_cast<dynamic*>(this)->get_nothrow<T>();
774 }
775
776 // Return T* for where we can put a T, without type checking.  (Memory
777 // might be uninitialized, even.)
778 template<class T>
779 T* dynamic::getAddress() {
780   return GetAddrImpl<T>::get(u_);
781 }
782
783 template<class T>
784 T const* dynamic::getAddress() const {
785   return const_cast<dynamic*>(this)->getAddress<T>();
786 }
787
788 template<class T> struct dynamic::GetAddrImpl {};
789 template<> struct dynamic::GetAddrImpl<void*> {
790   static void** get(Data& d) { return &d.nul; }
791 };
792 template<> struct dynamic::GetAddrImpl<dynamic::Array> {
793   static Array* get(Data& d) { return &d.array; }
794 };
795 template<> struct dynamic::GetAddrImpl<bool> {
796   static bool* get(Data& d) { return &d.boolean; }
797 };
798 template<> struct dynamic::GetAddrImpl<int64_t> {
799   static int64_t* get(Data& d) { return &d.integer; }
800 };
801 template<> struct dynamic::GetAddrImpl<double> {
802   static double* get(Data& d) { return &d.doubl; }
803 };
804 template<> struct dynamic::GetAddrImpl<fbstring> {
805   static fbstring* get(Data& d) { return &d.string; }
806 };
807 template<> struct dynamic::GetAddrImpl<dynamic::ObjectImpl> {
808   static_assert(sizeof(ObjectImpl) <= sizeof(Data::objectBuffer),
809     "In your implementation, std::unordered_map<> apparently takes different"
810     " amount of space depending on its template parameters.  This is "
811     "weird.  Make objectBuffer bigger if you want to compile dynamic.");
812
813   static ObjectImpl* get(Data& d) {
814     void* data = &d.objectBuffer;
815     return static_cast<ObjectImpl*>(data);
816   }
817 };
818
819 template<class T>
820 T& dynamic::get() {
821   if (auto* p = get_nothrow<T>()) {
822     return *p;
823   }
824   throw TypeError(TypeInfo<T>::name, type());
825 }
826
827 template<class T>
828 T const& dynamic::get() const {
829   return const_cast<dynamic*>(this)->get<T>();
830 }
831
832 inline char const* dynamic::typeName(Type t) {
833 #define FB_X(T) return TypeInfo<T>::name
834   FB_DYNAMIC_APPLY(t, FB_X);
835 #undef FB_X
836 }
837
838 inline void dynamic::destroy() {
839   // This short-circuit speeds up some microbenchmarks.
840   if (type_ == NULLT) return;
841
842 #define FB_X(T) detail::Destroy::destroy(getAddress<T>())
843   FB_DYNAMIC_APPLY(type_, FB_X);
844 #undef FB_X
845   type_ = NULLT;
846   u_.nul = nullptr;
847 }
848
849 //////////////////////////////////////////////////////////////////////
850
851 /*
852  * Helper for implementing operator<<.  Throws if the type shouldn't
853  * support it.
854  */
855 template<class T>
856 struct dynamic::PrintImpl {
857   static void print(dynamic const&, std::ostream& out, T const& t) {
858     out << t;
859   }
860 };
861 template<>
862 struct dynamic::PrintImpl<dynamic::ObjectImpl> {
863   static void print(dynamic const& d,
864                     std::ostream& out,
865                     dynamic::ObjectImpl const&) {
866     d.print_as_pseudo_json(out);
867   }
868 };
869 template<>
870 struct dynamic::PrintImpl<dynamic::Array> {
871   static void print(dynamic const& d,
872                     std::ostream& out,
873                     dynamic::Array const&) {
874     d.print_as_pseudo_json(out);
875   }
876 };
877
878 inline void dynamic::print(std::ostream& out) const {
879 #define FB_X(T) PrintImpl<T>::print(*this, out, *getAddress<T>())
880   FB_DYNAMIC_APPLY(type_, FB_X);
881 #undef FB_X
882 }
883
884 inline std::ostream& operator<<(std::ostream& out, dynamic const& d) {
885   d.print(out);
886   return out;
887 }
888
889 //////////////////////////////////////////////////////////////////////
890
891 // Secialization of FormatValue so dynamic objects can be formatted
892 template <>
893 class FormatValue<dynamic> {
894  public:
895   explicit FormatValue(const dynamic& val) : val_(val) { }
896
897   template <class FormatCallback>
898   void format(FormatArg& arg, FormatCallback& cb) const {
899     switch (val_.type()) {
900     case dynamic::NULLT:
901       FormatValue<std::nullptr_t>(nullptr).format(arg, cb);
902       break;
903     case dynamic::BOOL:
904       FormatValue<bool>(val_.asBool()).format(arg, cb);
905       break;
906     case dynamic::INT64:
907       FormatValue<int64_t>(val_.asInt()).format(arg, cb);
908       break;
909     case dynamic::STRING:
910       FormatValue<fbstring>(val_.asString()).format(arg, cb);
911       break;
912     case dynamic::DOUBLE:
913       FormatValue<double>(val_.asDouble()).format(arg, cb);
914       break;
915     case dynamic::ARRAY:
916       FormatValue(val_.at(arg.splitIntKey())).format(arg, cb);
917       break;
918     case dynamic::OBJECT:
919       FormatValue(val_.at(arg.splitKey().toFbstring())).format(arg, cb);
920       break;
921     }
922   }
923
924  private:
925   const dynamic& val_;
926 };
927
928 template <class V>
929 class FormatValue<detail::DefaultValueWrapper<dynamic, V>> {
930  public:
931   explicit FormatValue(
932       const detail::DefaultValueWrapper<dynamic, V>& val)
933     : val_(val) { }
934
935   template <class FormatCallback>
936   void format(FormatArg& arg, FormatCallback& cb) const {
937     auto& c = val_.container;
938     switch (c.type()) {
939     case dynamic::NULLT:
940     case dynamic::BOOL:
941     case dynamic::INT64:
942     case dynamic::STRING:
943     case dynamic::DOUBLE:
944       FormatValue<dynamic>(c).format(arg, cb);
945       break;
946     case dynamic::ARRAY:
947       {
948         int key = arg.splitIntKey();
949         if (key >= 0 && key < c.size()) {
950           FormatValue<dynamic>(c.at(key)).format(arg, cb);
951         } else{
952           FormatValue<V>(val_.defaultValue).format(arg, cb);
953         }
954       }
955       break;
956     case dynamic::OBJECT:
957       {
958         auto pos = c.find(arg.splitKey());
959         if (pos != c.items().end()) {
960           FormatValue<dynamic>(pos->second).format(arg, cb);
961         } else {
962           FormatValue<V>(val_.defaultValue).format(arg, cb);
963         }
964       }
965       break;
966     }
967   }
968
969  private:
970   const detail::DefaultValueWrapper<dynamic, V>& val_;
971 };
972
973 }  // namespaces
974
975 #undef FB_DYNAMIC_APPLY
976
977 #endif