data_client.hpp

#pragma once
 
#include <cstdint>
 
#include <memory>
#include <string>
#include <vector>
 
#include <boost/optional.hpp>
 
#include <viam/sdk/app/viam_client.hpp>
 
namespace viam {
namespace sdk {
 
class DataClient {
   public:
    enum class TabularDataSourceType { k_standard, k_hot_storage, k_pipeline_sink };
 
    struct tabular_data_by_mql_opts {
        TabularDataSourceType src_type = TabularDataSourceType::k_hot_storage;
 
        boost::optional<std::string> pipeline_id;
 
        boost::optional<std::string> query_prefix;
    };
 
    using BSONBytes = std::vector<uint8_t>;
 
    static DataClient from_viam_client(const ViamClient&);
 
    DataClient(DataClient&&) = default;
 
    DataClient& operator=(DataClient&&) = default;
 
    ~DataClient();
 
    const ViamChannel& channel() const;
 
    std::vector<BSONBytes> tabular_data_by_mql(const std::string& org_id,
                                               const std::vector<BSONBytes>& mql_binary,
                                               const tabular_data_by_mql_opts& opts);
 
    std::vector<BSONBytes> tabular_data_by_mql(const std::string& org_id,
                                               const std::vector<BSONBytes>& mql_binary);
 
   private:
    DataClient(const ViamChannel& channel);
 
    struct impl;
 
    std::unique_ptr<impl> pimpl_;
};
 
}  // namespace sdk
}  // namespace viam