C++ std::optional Specific Encoder and Decoder

Author: philippeVerney

lang/c++/include/avro/Specific.hh

@@ -22,6 +22,7 @@
 #include "array"
 #include <algorithm>
 #include <map>
+#include <optional>
 #include <string>
 #include <vector>
 
@@ -165,6 +166,44 @@ struct codec_traits<double> {
     }
 };
 
+/**
+* codec_traits for Avro optional.
+*/
+template<typename T>
+struct codec_traits<std::optional<T>> {
+    /**
+	* Encodes a given value.
+	*/
+    static void encode(Encoder &e, const std::optional<T> &b) {
+        if (b) {
+            e.encodeUnionIndex(1);
+            avro::encode(e, b.value());
+        } else {
+            e.encodeUnionIndex(0);
+            e.encodeNull();
+        }
+    }
+
+    /**
+	* Decodes into a given value.
+	*/
+    static void decode(Decoder &d, std::optional<T> &s) {
+        size_t n = d.decodeUnionIndex();
+        if (n >= 2) { throw avro::Exception("Union index too big"); }
+        switch (n) {
+            case 0: {
+                d.decodeNull();
+                s = std::nullopt;
+            } break;
+            case 1: {
+                T t;
+                avro::decode(d, t);
+                s.emplace(t);
+            } break;
+        }
+    }
+};
+
 /**
  * codec_traits for Avro string.
  */

lang/c++/test/SpecificTests.cc

@@ -24,6 +24,7 @@
 
 using std::array;
 using std::map;
+using std::optional;
 using std::string;
 using std::unique_ptr;
 using std::vector;
@@ -127,6 +128,18 @@ void testDouble() {
     BOOST_CHECK_CLOSE(b, n, 0.00000001);
 }
 
+void testNonEmptyOptional() {
+    optional<int64_t> n = -109;
+    optional<int64_t> b = encodeAndDecode(n);
+    BOOST_CHECK_EQUAL(b.value(), n.value());
+}
+
+void testEmptyOptional() {
+    optional<int64_t> n;
+    optional<int64_t> b = encodeAndDecode(n);
+    BOOST_CHECK(!b.has_value());
+}
+
 void testString() {
     string n = "abc";
     string b = encodeAndDecode(n);
@@ -191,6 +204,8 @@ init_unit_test_suite(int /*argc*/, char * /*argv*/[]) {
     ts->add(BOOST_TEST_CASE(avro::specific::testLong));
     ts->add(BOOST_TEST_CASE(avro::specific::testFloat));
     ts->add(BOOST_TEST_CASE(avro::specific::testDouble));
+    ts->add(BOOST_TEST_CASE(avro::specific::testNonEmptyOptional));
+    ts->add(BOOST_TEST_CASE(avro::specific::testEmptyOptional));
     ts->add(BOOST_TEST_CASE(avro::specific::testString));
     ts->add(BOOST_TEST_CASE(avro::specific::testBytes));
     ts->add(BOOST_TEST_CASE(avro::specific::testFixed));