在函数中返回 ifstream

Question

这对你来说可能是一个非常菜鸟的问题：我如何（如果可能的话）从函数返回 ifstream？

基本上，我需要从用户那里获取数据库的文件名，如果具有该文件名的数据库不存在，那么我需要为用户创建该文件。我知道该怎么做，但只能要求用户在创建文件后重新启动程序。我想尽可能避免给用户带来不便，但是下面的函数不能在 gcc 中编译：

ifstream getFile() {
    string fileName;
    cout << "Please enter in the name of the file you'd like to open: ";
    cin >> fileName;
    ifstream first(fileName.c_str());
    if(first.fail()) {
        cout << "File " << fileName << " not found.\n";
        first.close();
        ofstream second(fileName.c_str());
        cout << "File created.\n";
        second.close();
        ifstream third(fileName.c_str());
        return third; //compiler error here
    }
    else
        return first;
}

编辑：抱歉，忘记告诉您编译器错误在哪里以及是什么：

main.cpp:45: note: synthesized method ‘std::basic_ifstream<char, std::char_traits<char> >::basic_ifstream(const std::basic_ifstream<char, std::char_traits<char> >&)’ first required here 

编辑：我将函数更改为返回指针，而不是Remus 建议，并将 main() 中的行更改为“ifstream database = *getFile()”；现在我再次收到此错误，但这次是在 main() 中的行中：

main.cpp:27: note: synthesized method ‘std::basic_ifstream<char, std::char_traits<char> >::basic_ifstream(const std::basic_ifstream<char, std::char_traits<char> >&)’ first required here

Answer 1

不，不是真的。 ifstream 没有复制构造函数，如果您尝试返回一个复制构造函数，则意味着将函数中的实例复制到需要返回的位置。

通常的解决方法是传递对一个引用的引用，并在函数中修改该引用。

编辑：虽然这将使您的代码正常工作，但它不能解决基本问题。现在，您将两个相当不同的职责混合到一个函数中：1）获取文件名，2）打开或创建该文件。我认为如果将它们分开，代码会更简单，并且更容易消除您所看到的问题的根源。

编辑 2：使用这样的引用在没有 operator= 的情况下效果很好。总体思路是这样的：

int open_file(char const *name, fstream &stream) { 
    stream.open(name);
}

在这种情况下，赋值运算符既不必要也没有用——我们只是通过引用使用现有的 fstream。当且仅当我们必须将参数传递给构造函数时，operator= 才是必需的。对于流，我们可以默认构造一个不连接到文件的流，然后使用 open 来连接到文件。

Answer 2

bool checkFileExistence(const string& filename)
{
    ifstream f(filename.c_str());
    return f.is_open();
}

string getFileName()
{
    string filename;
    cout << "Please enter in the name of the file you'd like to open: ";
    cin >> filename;
    return filename;
}

void getFile(string filename, /*out*/ ifstream& file)
{
    const bool file_exists = checkFileExistence(filename);
    if (!file_exists) {
        cout << "File " << filename << " not found." << endl;
        filename = getFileName();  // poor style to reset input parameter though
        ofstream dummy(filename.c_str();
        if (!dummy.is_open()) {
            cerr << "Could not create file." << endl;
            return;
        }
        cout << "File created." << endl;
    }
    file.open(filename.c_str());
}

int main()
{
    // ...
    ifstream file;
    getFile("filename.ext", file);
    if (file.is_open()) {
        // do any stuff with file
    }
    // ...
}

Answer 3

ifstream 不支持复制构造语义（错误消息基本上所说的内容），因此您无法返回 ifstream。相反，返回 ifstream*，并将删除分配指针的责任传递给调用者。

Answer 4

作为一个选项，可以扩展 ifstream 并将自定义构造函数添加到新类中。

我将其扩展为创建测试资源流，将测试资源查找封装在其中。

// test_utils.h
class TestResourceStream : public std::ifstream {
    public:
        TestResourceStream(const char* file_path);
};

// test_utils.cpp
namespace fs = std::filesystem;
fs::path test_resource_path(const char* file_path) {
    fs::path path{std::string{"tests/resources/"} + file_path};
    if (!fs::exists(path))
        throw std::runtime_error{std::string{"path "} + 
            fs::absolute(path).c_str() + " does not exist"};
    return path;
}
TestResourceStream::TestResourceStream(const char* file_path)
    :std::ifstream{test_resource_path(file_path).c_str()} {}

// usage in test
TEST_CASE("parse") {
    std::list<GosDump::Expertise> expertises;
    TestResourceStream stream("requests/page_response.json");
    GosDump::Json::parse(expertises, stream);
    REQUIRE(10 == expertises.size());
}