Add wrap_and_manage function for arrow::DataType

Author: sgilmore10

matlab/src/cpp/arrow/matlab/proxy/wrap.cc

@@ -189,4 +189,17 @@ namespace arrow::matlab::proxy {
         WrapTypeFunctor functor;
         return wrap(datatype, functor);
     }
+
+    arrow::Result<::matlab::data::StructArray> wrap_and_manage(const std::shared_ptr<arrow::DataType>& datatype) {
+        namespace mda = ::matlab::data;
+        mda::ArrayFactory factory;
+
+        ARROW_ASSIGN_OR_RAISE(auto proxy, wrap(datatype));
+        const auto proxy_id = libmexclass::proxy::ProxyManager::manageProxy(proxy);
+
+        mda::StructArray output = factory.createStructArray({1, 1}, {"ProxyID", "TypeID"});
+        output[0]["ProxyID"] = factory.createScalar(proxy_id);
+        output[0]["TypeID"] = factory.createScalar( static_cast<int32_t>(datatype->id()));
+        return output;
+    }
 }

matlab/src/cpp/arrow/matlab/proxy/wrap.h

@@ -40,4 +40,8 @@ namespace arrow::matlab::proxy {
     /// \return arrow::result<std::shared_ptr<arrow::matlab::type::proxy::Type>>
     arrow::Result<std::shared_ptr<arrow::matlab::type::proxy::Type>> wrap(const std::shared_ptr<arrow::DataType>& datatype);
 
-}
+    /// \brief Wraps an DataType within a proxy::DataType and adds the proxy to the ProxyManager..
+    ///
+    /// \return arrow::Result<mda::StructArray>. The mda::StructArray has two fields: ProxyID (uint64) and TypeID (int32).
+    arrow::Result<std::shared_ptr<arrow::matlab::type::proxy::Type>> wrap(const std::shared_ptr<arrow::DataType>& datatype);
+}