1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107
|
/*
* Copyright (c) 2017, 2025, Oracle and/or its affiliates.
*
* 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 designed to work 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 either included with
* the program or referenced in the documentation.
*
* 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_TESTS_DRIVER_PROCESSOR_VARIABLE_CONTAINER_H_
#define PLUGIN_X_TESTS_DRIVER_PROCESSOR_VARIABLE_CONTAINER_H_
#include <list>
#include <map>
#include <memory>
#include <string>
#include "plugin/x/tests/driver/common/utils_string_parsing.h"
#include "plugin/x/tests/driver/processor/variable.h"
class Variable_container {
public:
using Container_initialization = std::map<std::string, std::string>;
Variable_container() = default;
explicit Variable_container(const Container_initialization &variables) {
for (const auto &kv : variables) {
set(kv.first, kv.second);
}
}
void replace(std::string *s) {
bool replaced;
do {
replaced = false;
for (Container::const_iterator sub = m_variables.begin();
sub != m_variables.end(); ++sub) {
std::string tmp(sub->second->get_value());
replaced |= aux::replace_all(*s, sub->first, tmp) > 0;
}
} while (replaced);
}
void make_special_variable(const std::string &key,
Variable_interface *value) {
m_variables[key].reset(value);
}
bool set(const std::string &key, const std::string &value) {
auto &variable = m_variables[key];
if (!variable) variable.reset(new Variable_string());
return variable->set_value(value);
}
std::string get(const std::string &key) const {
if (!is_present(key)) return "";
return m_variables.at(key)->get_value();
}
bool is_present(const std::string &key) const {
return m_variables.count(key);
}
std::string unreplace(const std::string &in, const bool clear = false) {
std::string s = in;
for (std::list<std::string>::const_iterator sub = m_to_unreplace.begin();
sub != m_to_unreplace.end(); ++sub) {
aux::replace_all(s, m_variables[*sub]->get_value(), *sub);
}
if (clear) m_to_unreplace.clear();
return s;
}
void clear_unreplace() { m_to_unreplace.clear(); }
void push_unreplace(const std::string &value) {
m_to_unreplace.push_back(value);
}
private:
using Container = std::map<std::string, std::unique_ptr<Variable_interface>>;
Container m_variables;
std::list<std::string> m_to_unreplace;
};
#endif // PLUGIN_X_TESTS_DRIVER_PROCESSOR_VARIABLE_CONTAINER_H_
|