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
100 lines
2.7 KiB
4 months ago
|
/*
|
||
|
|
* 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
|