polardbxengine/plugin/polarx_rpc/tests/client/visitor/any_filler.h

141 lines
4.8 KiB
C++

/*
* Copyright (c) 2019, Oracle and/or its affiliates. All rights reserved.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License, version 2.0,
* as published by the Free Software Foundation.
*
* This program is also distributed with certain software (including
* but not limited to OpenSSL) that is licensed under separate terms,
* as designated in a particular file or component or in included license
* documentation. The authors of MySQL hereby grant you an additional
* permission to link the program and your derivative works with the
* separately licensed software that they have included with MySQL.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License, version 2.0, for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
#ifndef PLUGIN_X_CLIENT_ANY_FILLER_H_
#define PLUGIN_X_CLIENT_ANY_FILLER_H_
#include <string>
#include "../mysqlxclient/xargument.h"
#include "../mysqlxclient/xmessage.h"
namespace xcl {
class Any_filler : public Argument_visitor {
public:
explicit Any_filler(::Polarx::Datatypes::Any *any) : m_any(any) {}
private:
::Polarx::Datatypes::Any *m_any;
void visit_null() override {
m_any->set_type(::Polarx::Datatypes::Any_Type_SCALAR);
m_any->mutable_scalar()->set_type(::Polarx::Datatypes::Scalar_Type_V_NULL);
}
void visit_integer(const int64_t value) override {
m_any->set_type(::Polarx::Datatypes::Any_Type_SCALAR);
m_any->mutable_scalar()->set_type(::Polarx::Datatypes::Scalar_Type_V_SINT);
m_any->mutable_scalar()->set_v_signed_int(value);
}
void visit_uinteger(const uint64_t value) override {
m_any->set_type(::Polarx::Datatypes::Any_Type_SCALAR);
m_any->mutable_scalar()->set_type(::Polarx::Datatypes::Scalar_Type_V_UINT);
m_any->mutable_scalar()->set_v_unsigned_int(value);
}
void visit_double(const double value) override {
m_any->set_type(::Polarx::Datatypes::Any_Type_SCALAR);
m_any->mutable_scalar()->set_type(
::Polarx::Datatypes::Scalar_Type_V_DOUBLE);
m_any->mutable_scalar()->set_v_double(value);
}
void visit_float(const float value) override {
m_any->set_type(::Polarx::Datatypes::Any_Type_SCALAR);
m_any->mutable_scalar()->set_type(::Polarx::Datatypes::Scalar_Type_V_FLOAT);
m_any->mutable_scalar()->set_v_float(value);
}
void visit_bool(const bool value) override {
m_any->set_type(::Polarx::Datatypes::Any_Type_SCALAR);
m_any->mutable_scalar()->set_type(::Polarx::Datatypes::Scalar_Type_V_BOOL);
m_any->mutable_scalar()->set_v_bool(value);
}
void visit_object(const Argument_object &obj) override {
m_any->set_type(::Polarx::Datatypes::Any_Type_OBJECT);
auto any_object = m_any->mutable_obj();
for (const auto &key_value : obj) {
auto fld = any_object->add_fld();
Any_filler filler(fld->mutable_value());
fld->set_key(key_value.first);
key_value.second.accept(&filler);
}
}
void visit_uobject(const Argument_uobject &obj) override {
m_any->set_type(::Polarx::Datatypes::Any_Type_OBJECT);
auto any_object = m_any->mutable_obj();
for (const auto &key_value : obj) {
auto fld = any_object->add_fld();
Any_filler filler(fld->mutable_value());
fld->set_key(key_value.first);
key_value.second.accept(&filler);
}
}
void visit_array(const Argument_array &values) override {
m_any->set_type(::Polarx::Datatypes::Any_Type_ARRAY);
auto any_array = m_any->mutable_array();
for (const auto &value : values) {
Any_filler filler(any_array->add_value());
value.accept(&filler);
}
}
void visit_string(const std::string &value) override {
fill_string(value, ::Polarx::Datatypes::Scalar_Type_V_STRING);
}
void visit_decimal(const std::string &value) override {
fill_string(value, ::Polarx::Datatypes::Scalar_Type_V_STRING);
}
void visit_octets(const std::string &value) override {
fill_string(value, ::Polarx::Datatypes::Scalar_Type_V_OCTETS);
}
void fill_string(const std::string &value,
const Polarx::Datatypes::Scalar_Type st) {
m_any->set_type(::Polarx::Datatypes::Any_Type_SCALAR);
m_any->mutable_scalar()->set_type(st);
if (::Polarx::Datatypes::Scalar_Type_V_OCTETS == st)
m_any->mutable_scalar()->mutable_v_octets()->set_value(value);
else
m_any->mutable_scalar()->mutable_v_string()->set_value(value);
}
};
} // namespace xcl
#endif // PLUGIN_X_CLIENT_ANY_FILLER_H_