You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
2.6 KiB
70 lines
2.6 KiB
/*
|
|
Copyright (c) 2015, 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
|
|
*/
|
|
|
|
#include "client/dump/file_writer.h"
|
|
|
|
#include <functional>
|
|
|
|
using namespace Mysql::Tools::Dump;
|
|
|
|
void File_writer::append(const std::string &data_to_append) {
|
|
fwrite(data_to_append.c_str(), 1, data_to_append.size(), m_file);
|
|
// Check for I/O errors.
|
|
if (ferror(m_file) != 0) {
|
|
this->pass_message(Mysql::Tools::Base::Message_data(
|
|
ferror(m_file), "Error occurred while writing to output.",
|
|
Mysql::Tools::Base::Message_type_error));
|
|
}
|
|
}
|
|
|
|
File_writer::~File_writer() {
|
|
// Check for I/O errors and close file.
|
|
if (m_file && (ferror(m_file) != 0 || fclose(m_file) != 0)) {
|
|
this->pass_message(Mysql::Tools::Base::Message_data(
|
|
ferror(m_file), "Error occurred while finishing writing to output.",
|
|
Mysql::Tools::Base::Message_type_error));
|
|
}
|
|
}
|
|
|
|
File_writer::File_writer(
|
|
std::function<bool(const Mysql::Tools::Base::Message_data &)>
|
|
*message_handler,
|
|
Simple_id_generator *object_id_generator, const std::string &file_name)
|
|
: Abstract_chain_element(message_handler, object_id_generator),
|
|
m_file(nullptr),
|
|
m_file_name(file_name) {}
|
|
|
|
bool File_writer::init() {
|
|
if (!m_file) {
|
|
m_file = fopen(m_file_name.c_str(), "wb");
|
|
if (!m_file) {
|
|
this->pass_message(Mysql::Tools::Base::Message_data(
|
|
errno, "Can't create/open a file " + m_file_name,
|
|
Mysql::Tools::Base::Message_type_error));
|
|
return true;
|
|
} else
|
|
return false;
|
|
}
|
|
return false;
|
|
}
|
|
|