polardbxengine/plugin/x/tests/driver/json_to_any_handler.cc

116 lines
3.7 KiB
C++

/*
* Copyright (c) 2017, 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
*/
#include "plugin/x/tests/driver/json_to_any_handler.h"
bool Json_to_any_handler::Key(const char *str, rapidjson::SizeType length,
bool copy) {
typedef ::Mysqlx::Datatypes::Object_ObjectField Field;
Field *f = m_stack.top()->mutable_obj()->add_fld();
f->set_key(str, length);
m_stack.push(f->mutable_value());
return true;
}
bool Json_to_any_handler::Null() {
get_scalar(::Mysqlx::Datatypes::Scalar_Type_V_NULL);
return true;
}
bool Json_to_any_handler::Bool(bool b) {
get_scalar(::Mysqlx::Datatypes::Scalar_Type_V_BOOL)->set_v_bool(b);
return true;
}
bool Json_to_any_handler::Int(int i) {
get_scalar(::Mysqlx::Datatypes::Scalar_Type_V_SINT)->set_v_signed_int(i);
return true;
}
bool Json_to_any_handler::Uint(unsigned u) {
get_scalar(::Mysqlx::Datatypes::Scalar_Type_V_UINT)->set_v_unsigned_int(u);
return true;
}
bool Json_to_any_handler::Int64(int64_t i) {
get_scalar(::Mysqlx::Datatypes::Scalar_Type_V_SINT)->set_v_signed_int(i);
return true;
}
bool Json_to_any_handler::Uint64(uint64_t u) {
get_scalar(::Mysqlx::Datatypes::Scalar_Type_V_UINT)->set_v_unsigned_int(u);
return true;
}
bool Json_to_any_handler::Double(double d, bool) {
get_scalar(::Mysqlx::Datatypes::Scalar_Type_V_DOUBLE)->set_v_double(d);
return true;
}
bool Json_to_any_handler::String(const char *str, rapidjson::SizeType length,
bool) {
get_scalar(::Mysqlx::Datatypes::Scalar_Type_V_STRING)
->mutable_v_string()
->set_value(str, length);
return true;
}
bool Json_to_any_handler::StartObject() {
Any *any = m_stack.top();
if (any->has_type() && any->type() == ::Mysqlx::Datatypes::Any_Type_ARRAY)
m_stack.push(any->mutable_array()->add_value());
m_stack.top()->set_type(::Mysqlx::Datatypes::Any_Type_OBJECT);
m_stack.top()->mutable_obj();
return true;
}
bool Json_to_any_handler::EndObject(rapidjson::SizeType /*member_count*/) {
m_stack.pop();
return true;
}
bool Json_to_any_handler::StartArray() {
m_stack.top()->set_type(::Mysqlx::Datatypes::Any_Type_ARRAY);
m_stack.top()->mutable_array();
return true;
}
bool Json_to_any_handler::EndArray(rapidjson::SizeType /*element_count*/) {
m_stack.pop();
return true;
}
Json_to_any_handler::Scalar *Json_to_any_handler::get_scalar(
Scalar::Type scalar_type) {
Any *any = m_stack.top();
if (any->has_type() && any->type() == ::Mysqlx::Datatypes::Any_Type_ARRAY)
any = any->mutable_array()->add_value();
else
m_stack.pop();
any->set_type(::Mysqlx::Datatypes::Any_Type_SCALAR);
Scalar *s = any->mutable_scalar();
s->set_type(scalar_type);
return s;
}