forked from StoneAtom/StoneDB
82 lines
2.6 KiB
C++
82 lines
2.6 KiB
C++
/* Copyright (c) 2015, 2021, 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 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 BOOTSTRAP_IMPL_H
|
|
#define BOOTSTRAP_IMPL_H 1
|
|
#include <string>
|
|
#include "sql_bootstrap.h" // MAX_BOOTSTRAP_QUERY_SIZE
|
|
|
|
|
|
/** abstract interface to reading bootstrap commands */
|
|
class Command_iterator
|
|
{
|
|
public:
|
|
/** start processing the iterator */
|
|
virtual void begin(void) {}
|
|
|
|
/**
|
|
Get the next query string.
|
|
|
|
@param[out] query return the query
|
|
@param[out] read_error return the read error code
|
|
@param[out] query_source return the source of the query
|
|
(if it is from a file or a compiled one).
|
|
@return one of the READ_BOOTSTRAP
|
|
*/
|
|
virtual int next(std::string &query, int *read_error, int *query_source)= 0;
|
|
|
|
/** end processing the iterator */
|
|
virtual void end(void) {}
|
|
|
|
/** The current bootstrap command reader */
|
|
static Command_iterator *current_iterator;
|
|
|
|
protected:
|
|
/** needed because of the virtual functions */
|
|
virtual ~Command_iterator() {}
|
|
};
|
|
|
|
/** File bootstrap command reader */
|
|
class File_command_iterator : public Command_iterator
|
|
{
|
|
public:
|
|
File_command_iterator(fgets_input_t input, fgets_fn_t fgets_fn)
|
|
: m_input(input), m_fgets_fn(fgets_fn), is_allocated(false)
|
|
{}
|
|
File_command_iterator(const char *file_name);
|
|
virtual ~File_command_iterator();
|
|
|
|
int next(std::string &query, int *read_error, int *query_source);
|
|
void end(void);
|
|
bool has_file()
|
|
{
|
|
return m_input != 0;
|
|
}
|
|
protected:
|
|
fgets_input_t m_input;
|
|
fgets_fn_t m_fgets_fn;
|
|
bool is_allocated;
|
|
};
|
|
|
|
|
|
#endif /* BOOTSTRAP_IMPL_H */
|