获取从 Go 中的文本文件中删除行的空行

Question

我有一个使用 cobra 用 Go 编写的 CLI，其中一个命令是从文本文件的用户名列表中删除特定用户名。例如，我在文件中有一个用户名列表，user1user2user3user4user5这是我的 cobra 命令中的 Go 代码，Run: func(cmd *cobra.Command, args []string) {    userBlacklist, _ := cmd.Flags().GetString("blacklist-file-path")     userName, _ := cmd.Flags().GetString("user-name")    if info, err := os.Stat(userBlacklist); err == nil && !info.IsDir() {        data, err := ioutil.ReadFile(userBlacklist)        if err != nil {                logger.Fatal("Failed to read in blacklist: %s", err)        }        lines := strings.Split(string(data), "\n")        file, err := os.Create(userBlacklist)        if err != nil {                logger.Fatal("Could not overwrite blacklist file %s\n", err)        }        defer file.Close()        for _, line := range lines {            if line != userName{                if _, err := file.WriteString(line + "\n"); err != nil {                    logger.Fatal("Failed to write user to open blacklist file %s\n", err)                                       }            }         }    } else {        if err != nil {            logger.Fatal("Error stating blacklist %s\n", err )        } else {            logger.Fatal("--blacklist-file-path is a directory, not a file\n")        }    }},当我运行它时，例如，将 userName 设置为“user3”，生成的输出文件中确实没有 user3。但是它在文件末尾确实有一个空行“”。如果我然后运行要求它删除另一个用户的命令，它也会删除该用户，但生成的文件现在将在末尾有两个空行，等等。我可以通过将第 22 行更改为来防止空行，if !(line == userName || line == "") {                我不明白这是怎么回事。似乎不太明智？空白行来自哪里？我在 Ubuntu 20.04、Go 版本 go1.16.5 linux/amd64 和 github.com/spf13/cobra v1.5.0提前感谢您的任何见解。

Answer 1

该文件以\n. 的最后一个元素strings.Split(string(data), "\n")是空字符串。程序在处理最后一个元素时将空字符串添加到文件末尾。

通过使用bufio.Scanner解析行来修复：

    scanner := bufio.NewScanner(bytes.NewReader(data))

    for scanner.Scan() {

        line := scanner.Text()

        if line != userName {

            if _, err := file.WriteString(line + "\n"); err != nil {

                logger.Fatal("Failed to write user to open blacklist file %s\n", err)

            }

        }

    }

Answer 2

当您将行写入文件时，您正在添加一个换行符：

file.WriteString(line + "\n")

这就是文件末尾的空换行符的来源。