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.

100 lines
2.7 KiB

/*
* Copyright (C) 2015, The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "line_reader.h"
#include <fstream>
#include <sstream>
using std::istringstream;
using std::ifstream;
using std::string;
using std::unique_ptr;
namespace android {
namespace aidl {
class FileLineReader : public LineReader {
public:
FileLineReader() = default;
~FileLineReader() override {
input_stream_.close();
}
// non-copyable, non-movable
FileLineReader(const FileLineReader&) = delete;
FileLineReader(FileLineReader&&) = delete;
FileLineReader& operator=(const FileLineReader&) = delete;
FileLineReader& operator=(FileLineReader&&) = delete;
bool Init(const std::string& file_path) {
input_stream_.open(file_path, ifstream::in | ifstream::binary);
return input_stream_.is_open() && input_stream_.good();
}
bool ReadLine(string* line) override {
if (!input_stream_.good()) {
return false;
}
line->clear();
std::getline(input_stream_, *line);
return true;
}
private:
ifstream input_stream_;
}; // class FileLineReader
class MemoryLineReader : public LineReader {
public:
explicit MemoryLineReader(const string& contents) : input_stream_(contents) {}
~MemoryLineReader() override = default;
// non-copyable, non-movable
MemoryLineReader(const MemoryLineReader&) = delete;
MemoryLineReader(MemoryLineReader&&) = delete;
MemoryLineReader& operator=(const MemoryLineReader&) = delete;
MemoryLineReader& operator=(MemoryLineReader&&) = delete;
bool ReadLine(string* line) override {
if (!input_stream_.good()) {
return false;
}
line->clear();
std::getline(input_stream_, *line);
return true;
}
private:
istringstream input_stream_;
}; // class MemoryLineReader
unique_ptr<LineReader> LineReader::ReadFromFile(const string& file_path) {
unique_ptr<FileLineReader> file_reader(new FileLineReader());
unique_ptr<LineReader> ret;
if (file_reader->Init(file_path)) {
ret.reset(file_reader.release());
}
return ret;
}
unique_ptr<LineReader> LineReader::ReadFromMemory(const string& contents) {
return unique_ptr<LineReader>(new MemoryLineReader(contents));
}
} // namespace aidl
} // namespace android