登录
首页 >  Golang >  Go问答

使用 Go 从文本文件中去除空行

来源:stackoverflow

时间:2024-02-10 11:18:22 172浏览 收藏

目前golang学习网上已经有很多关于Golang的文章了,自己在初次阅读这些文章中,也见识到了很多学习思路;那么本文《使用 Go 从文本文件中去除空行》,也希望能帮助到大家,如果阅读完后真的对你学习Golang有帮助,欢迎动动手指,评论留言并分享~

问题内容

我有一个使用 cobra 用 go 编写的 cli,其中命令之一是从文本文件的用户名列表中删除特定的用户名。例如,我在文件中有一个用户名列表,

user1
user2
user3
user4
user5

这是我的 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

预先感谢您的任何见解。


正确答案


文件以 \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)
            }
        }
    }

当您将该行写入文件时,您将添加一个换行符:

file.WriteString(line + "\n")

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

今天关于《使用 Go 从文本文件中去除空行》的内容介绍就到此结束,如果有什么疑问或者建议,可以在golang学习网公众号下多多回复交流;文中若有不正之处,也希望回复留言以告知!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>