fs/io/BufferedReader: new class to replace class TextFile

The new class is pluggable, to prepare for gzipped database files.

For now, the TextFile class remains, and will be refactored away
later.
This commit is contained in:
Max Kellermann 2014-08-07 18:54:06 +02:00
parent 0ea66a1275
commit aa2e4d92e0
9 changed files with 210 additions and 68 deletions

View File

@ -508,6 +508,7 @@ FS_LIBS = libfs.a
libfs_a_SOURCES = \
src/fs/io/Reader.hxx \
src/fs/io/FileReader.cxx src/fs/io/FileReader.hxx \
src/fs/io/BufferedReader.cxx src/fs/io/BufferedReader.hxx \
src/fs/io/TextFile.cxx src/fs/io/TextFile.hxx \
src/fs/io/OutputStream.hxx \
src/fs/io/StdoutOutputStream.hxx \

View File

@ -116,6 +116,29 @@ spl_map_to_fs(const char *name_utf8, Error &error)
return path_fs;
}
gcc_pure
static bool
IsNotFoundError(const Error &error)
{
#ifdef WIN32
return error.IsDomain(win32_domain) &&
error.GetCode() == ERROR_FILE_NOT_FOUND;
#else
return error.IsDomain(errno_domain) &&
error.GetCode() == ENOENT;
#endif
}
static void
TranslatePlaylistError(Error &error)
{
if (IsNotFoundError(error)) {
error.Clear();
error.Set(playlist_domain, int(PlaylistResult::NO_SUCH_LIST),
"No such playlist");
}
}
/**
* Create an #Error for the current errno.
*/
@ -228,9 +251,9 @@ LoadPlaylistFile(const char *utf8path, Error &error)
if (path_fs.IsNull())
return contents;
TextFile file(path_fs);
TextFile file(path_fs, error);
if (file.HasFailed()) {
playlist_errno(error);
TranslatePlaylistError(error);
return contents;
}

View File

@ -103,10 +103,10 @@ StateFile::Read()
FormatDebug(state_file_domain, "Loading state file %s", path_utf8.c_str());
TextFile file(path);
Error error;
TextFile file(path, error);
if (file.HasFailed()) {
FormatErrno(state_file_domain, "failed to open %s",
path_utf8.c_str());
LogError(error);
return;
}

View File

@ -168,12 +168,9 @@ SimpleDatabase::Load(Error &error)
assert(!path.IsNull());
assert(root != nullptr);
TextFile file(path);
if (file.HasFailed()) {
error.FormatErrno("Failed to open database file \"%s\"",
path_utf8.c_str());
TextFile file(path, error);
if (file.HasFailed())
return false;
}
if (!db_load_internal(file, *root, error))
return false;

View File

@ -40,6 +40,7 @@
#endif
#ifdef USE_XDG
#include "util/Error.hxx"
#include "util/StringUtil.hxx"
#include "io/TextFile.hxx"
#include <string.h>
@ -204,7 +205,7 @@ static AllocatedPath GetUserDir(const char *name)
if (config_dir.IsNull())
return result;
auto dirs_file = AllocatedPath::Build(config_dir, "user-dirs.dirs");
TextFile input(dirs_file);
TextFile input(dirs_file, IgnoreError());
if (input.HasFailed())
return result;
char *line;

View File

@ -0,0 +1,82 @@
/*
* Copyright (C) 2003-2014 The Music Player Daemon Project
* http://www.musicpd.org
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* 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 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 Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#include "config.h"
#include "BufferedReader.hxx"
#include "Reader.hxx"
#include "util/TextFile.hxx"
bool
BufferedReader::Fill(bool need_more)
{
if (gcc_unlikely(last_error.IsDefined()))
return false;
if (eof)
return !need_more;
auto w = buffer.Write();
if (w.IsEmpty()) {
if (buffer.GetCapacity() >= MAX_SIZE)
return !need_more;
buffer.Grow(buffer.GetCapacity() * 2);
w = buffer.Write();
assert(!w.IsEmpty());
}
size_t nbytes = reader.Read(w.data, w.size, last_error);
if (nbytes == 0) {
if (gcc_unlikely(last_error.IsDefined()))
return false;
eof = true;
return !need_more;
}
buffer.Append(nbytes);
return true;
}
char *
BufferedReader::ReadLine()
{
do {
char *line = ReadBufferedLine(buffer);
if (line != nullptr)
return line;
} while (Fill(true));
if (last_error.IsDefined() || !eof || buffer.IsEmpty())
return nullptr;
auto w = buffer.Write();
if (w.IsEmpty()) {
buffer.Grow(buffer.GetCapacity() + 1);
w = buffer.Write();
assert(!w.IsEmpty());
}
/* terminate the last line */
w[0] = 0;
char *line = buffer.Read().data;
buffer.Clear();
return line;
}

View File

@ -0,0 +1,75 @@
/*
* Copyright (C) 2003-2014 The Music Player Daemon Project
* http://www.musicpd.org
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* 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 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 Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#ifndef MPD_BUFFERED_READER_HXX
#define MPD_BUFFERED_READER_HXX
#include "check.h"
#include "Compiler.h"
#include "util/DynamicFifoBuffer.hxx"
#include "util/Error.hxx"
#include <stddef.h>
class Reader;
class Error;
class BufferedReader {
static constexpr size_t MAX_SIZE = 512 * 1024;
Reader &reader;
DynamicFifoBuffer<char> buffer;
Error last_error;
bool eof;
public:
BufferedReader(Reader &_reader)
:reader(_reader), buffer(4096), eof(false) {}
gcc_pure
bool Check() const {
return !last_error.IsDefined();
}
bool Check(Error &error) const {
if (last_error.IsDefined()) {
error.Set(last_error);
return false;
} else
return true;
}
bool Fill(bool need_more);
gcc_pure
WritableBuffer<void> Read() const {
return buffer.Read().ToVoid();
}
void Consume(size_t n) {
buffer.Consume(n);
}
char *ReadLine();
};
#endif

View File

@ -19,63 +19,30 @@
#include "config.h"
#include "TextFile.hxx"
#include "util/Alloc.hxx"
#include "FileReader.hxx"
#include "BufferedReader.hxx"
#include "fs/Path.hxx"
#include "fs/FileSystem.hxx"
#include <assert.h>
#include <string.h>
#include <stdlib.h>
TextFile::TextFile(Path path_fs)
:file(FOpen(path_fs, FOpenMode::ReadText)),
buffer((char *)xalloc(step)), capacity(step), length(0) {}
TextFile::TextFile(Path path_fs, Error &error)
:file_reader(new FileReader(path_fs, error)),
buffered_reader(file_reader->IsDefined()
? new BufferedReader(*file_reader)
: nullptr)
{
}
TextFile::~TextFile()
{
free(buffer);
if (file != nullptr)
fclose(file);
delete buffered_reader;
delete file_reader;
}
char *
TextFile::ReadLine()
{
assert(file != nullptr);
assert(buffered_reader != nullptr);
while (true) {
if (length >= capacity) {
if (capacity >= max_length)
/* too large already - bail out */
return nullptr;
capacity <<= 1;
char *new_buffer = (char *)realloc(buffer, capacity);
if (new_buffer == nullptr)
/* out of memory - bail out */
return nullptr;
}
char *p = fgets(buffer + length, capacity - length, file);
if (p == nullptr) {
if (length == 0 || ferror(file))
return nullptr;
break;
}
length += strlen(buffer + length);
if (buffer[length - 1] == '\n')
break;
}
/* remove the newline characters */
if (buffer[length - 1] == '\n')
--length;
if (buffer[length - 1] == '\r')
--length;
buffer[length] = 0;
length = 0;
return buffer;
return buffered_reader->ReadLine();
}

View File

@ -22,29 +22,26 @@
#include "Compiler.h"
#include <stdio.h>
#include <stddef.h>
class Path;
class Error;
class FileReader;
class BufferedReader;
class TextFile {
static constexpr size_t max_length = 512 * 1024;
static constexpr size_t step = 1024;
FILE *const file;
char *buffer;
size_t capacity, length;
FileReader *const file_reader;
BufferedReader *const buffered_reader;
public:
TextFile(Path path_fs);
TextFile(Path path_fs, Error &error);
TextFile(const TextFile &other) = delete;
~TextFile();
bool HasFailed() const {
return gcc_unlikely(file == nullptr);
return gcc_unlikely(buffered_reader == nullptr);
}
/**
@ -53,7 +50,6 @@ public:
* prevent denial of service.
*
* @param file the source file, opened in text mode
* @param buffer an allocator for the buffer
* @return a pointer to the line, or nullptr on end-of-file or error
*/
char *ReadLine();