C ++:搜索和写入文件以替换字符串

时间:2012-08-02 19:48:43

标签: c++ file io

所以我创建了一个简单的测试应用程序。它的目的是读取或写入文本文件。它正在寻找字符串“Firmware:”。这个想法是设备将一个包转移到它并存储,然后激活。我想存储该软件包的名称(例如:Update1),当我需要时,查询并获取软件包名称(从文件中读取)。

问题1:它似乎没有写入文件。 问题2:当我将其写入文件时,如果存在“Firmware:pkgName”行,则如何将pkgName替换为新的包名?

这个程序运行并编译,所以随意把它扔到IDE或文本编辑器中,它只是没有完全做我想要的。我使用log.txt作为我的示例文件来读取。我确信有一种更简单的方法可以做到这一点。我已经有一段时间了,我已经难过了。

感谢您的帮助!

#include <cstdlib>
#include <iostream>
#include <string>
#include <fstream>

using namespace std;

int writeFile(string filename, string pkgName);
int readFile(string filename);

int main(void) {
    string filename = "log.txt", userInput, pkgName;
    system("touch log.txt");

    while (true) {
        cout << "Write or Read?\n>>>";
        cin >> userInput;

        if (userInput == "Write") {
            cout << "What's your package name?\n>>>";
            cin >> pkgName;
            writeFile(filename, pkgName);
        } else if (userInput == "Read") {
            readFile(filename);
        } else {
            cout << "Invalid Input";
        }
    }
}

int writeFile(string filename, string pkgName) {
    bool found, exists = true;
    string line, word;
    fstream firmwareFile;
    int pos, size, num = 1;
    firmwareFile.open(filename.c_str(), ios::in | ios::out | ios::app);
    //firmwareFile << "This is a test.";
    if (firmwareFile.is_open()) {
        while (!found) {
            getline(firmwareFile, line);
            pos = line.find("Firmware: ");
            if (pos != string::npos) {
                //Found It
                found = true;
                exists = true;
                size = line.size();
                word = line.substr(10, ((size - pos) + 10));
                cout << word << " is the old package name" << endl;
                //TODO - Replace that with the new package name
                firmwareFile.close();
            } else if (firmwareFile.eof()) {
                //Doesn't Exist
                found = true;
                exists = false;
                cout << "No last firmware package available.\n";
            } else {
                //Still Looking
                found = false;
                cout << "Searching line #" << num <<  endl;
            }
            num++;
            if (!exists) {
                firmwareFile << endl << "Firmware: " << pkgName << endl;
                cout << "Wrote package name to file\n" << endl;
            }
            firmwareFile.close();
        }   
    } else {
        cout << "Unable to open file";
    }
}

int readFile(string filename) {
    bool found;
    string line, word;
    fstream firmwareFile;
    int pos, size, num = 1;
    firmwareFile.open(filename.c_str(), ios::in | ios::out | ios::app);
    if (firmwareFile.is_open()) {
        while (!found) {
            getline(firmwareFile, line);
            pos = line.find("Firmware: ");
            if (pos != string::npos) {
                found = true;
                size = line.size();
                word = line.substr(10, ((size - pos) + 10));
                cout << word << endl;
            } else if (firmwareFile.eof()) {
                found = true;
                cout << "No last firmware package available.\n";
                firmwareFile << endl << "Firmware: ";
            } else {
                cout << "Searching line #" << num << endl;
            }
            num++;
        }
        firmwareFile.close();
    } else {
        cout << "Unable to open file";
        firmwareFile.close();
    }
    firmwareFile.close();
}

这是我的log.txt文件的内容(除了包含“固件:”的行之外的所有内容都是微不足道的)

You
Think
It's
Here
But
It's
Not
So
That
Sucks
Doesn't
It
?
Firmware: Update1
I
Hope
You
Caught
It
!

4 个答案:

答案 0 :(得分:2)

好。第一个问题,在函数写入文件中,您有一个变量found,您可以在while while (!found) { ... }条件下进行测试。问题是,当您第一次测试此变量时,您没有将其设置为任何值。所以你有不确定的行为。看看你们其余的代码我认为你应该使用do while循环do { ... } while (!found);而不是while循环。看起来你在readFile中也有同样的问题。

第二个问题是我害怕的坏消息。除非要替换为长度完全相同的字符串,否则不能替换文件中的字符串。这只是文件在磁盘上组织方式的限制。实现替换的唯一方法是将整个文件读入内存,在内存中进行替换,然后将整个文件写回磁盘。

答案 1 :(得分:2)

@jrok回答了你的问题2,但我怀疑你的代码每次都没有写,因为找不到初始化,因此未定义,因此你的while循环的一半时间(大约)根本不会执行。您需要初始化找到的和存在的内容,如:bool found = false, exists = true;

答案 2 :(得分:0)

如果必须替换文件的内容(不删除并将其他文件重命名为其名称),则需要读入整个文件,对内存中的文件进行必要的更改,然后将内存中文件的全部内容写出到磁盘文件中,覆盖它。

如果允许删除源文件,则执行这些转换要容易得多。您只需要读取源文件,打开临时文件并写出来,随时进行所需的更改。处理完源文件后,将其删除,并将新文件重命名为与原始源文件相同的名称。

答案 3 :(得分:0)

如果您编写另一个只更改了相关行的文件,则可以简化操作:

std::ifstream cfg("log.txt");
std::ofstream tmp("log.tmp");
std::string line;
std::string pkgName;

while(std::getline(cfg, line)) {
    if (line.find("Firmware:") != std::string::npos) {
        line = "Firmware: " + pkgName; 
    }
    tmp << line << '\n';
}

现在,重命名和删除文件是特定于平台的。如果你想要一个可移植的解决方案,我建议你看一下Boost Filesystem库。