polardbxengine/plugin/x/client/validator/option_context_validator.h

98 lines
3.3 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_VALIDATOR_OPTION_CONTEXT_VALIDATOR_H_
#define PLUGIN_X_CLIENT_VALIDATOR_OPTION_CONTEXT_VALIDATOR_H_
#include <set>
#include <string>
#include <vector>
#include "mysqlxclient/xcompression.h"
#include "plugin/x/client/context/xcontext.h"
#include "plugin/x/client/validator/translation_validator.h"
namespace xcl {
class Contex_ip_validator
: public Translate_validator<Internet_protocol, Context, false> {
public:
Contex_ip_validator()
: Translate_validator({{"ANY", Internet_protocol::Any},
{"IP4", Internet_protocol::V4},
{"IP6", Internet_protocol::V6}}) {}
void visit_translate(const Internet_protocol mode) override {
get_ctxt()->m_internet_protocol = mode;
}
};
class Contex_auth_validator
: public Translate_array_validator<Auth, Context, false> {
public:
Contex_auth_validator()
: Translate_array_validator(
{{"AUTO", Auth::k_auto},
{"FROM_CAPABILITIES", Auth::k_auto_from_capabilities},
{"FALLBACK", Auth::k_auto_fallback},
{"MYSQL41", Auth::k_mysql41},
{"PLAIN", Auth::k_plain},
{"SHA256_MEMORY", Auth::k_sha256_memory}}) {}
bool valid_array_value(const Array_of_enums &values) override {
const std::set<Auth> scalar_only_values{Auth::k_auto,
Auth::k_auto_from_capabilities};
if (values.size() == 1) return true;
for (const auto v : values) {
if (scalar_only_values.count(v)) return false;
}
return true;
}
void visit_translate(const std::vector<Auth> &auth) override {
get_ctxt()->m_use_auth_methods = auth;
}
};
template <uint32_t(Context::*member)>
class Ctxt_uint32_store : public Value_validator<Context, Integer_validator> {
public:
void visit_integer(const int64_t value) override {
get_ctxt()->*member = static_cast<uint32_t>(value);
}
};
template <bool(Context::*member)> // NOLINT
class Ctxt_bool_store : public Value_validator<Context, Bool_validator> {
public:
void visit_bool(const bool value) override { get_ctxt()->*member = value; }
};
} // namespace xcl
#endif // PLUGIN_X_CLIENT_VALIDATOR_OPTION_CONTEXT_VALIDATOR_H_