GoLang的readLines和writeLines方法

最新推荐文章于 2023-06-07 16:23:48 发布

mercury137

最新推荐文章于 2023-06-07 16:23:48 发布

阅读量2.2k

点赞数

分类专栏： go 文章标签： golang python c/c++

本文链接：https://blog.csdn.net/mercury137/article/details/84255035

版权

go 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

习惯了Python里面对文件的操作简单随意，刚上手Go语言这样类似C语言的文件操作方法，一时还不习惯使用buffer来读取和吸入文件。在python中文件对象可以使用readLines和writeLines来遍历文件的行，十分的方便。
如果在Go语言里面也能有这样的方法就好了，找了半天API，还是没有收获。求人不如求己，自己写一个。
代码如下：


package main

import (
   "io"
   "os"
   "bufio"
   "bytes"
   "fmt"
   "strings"
)


func readLines(path string)(lines [] string,err error){
   var (
       file *os.File
       part [] byte
       prefix bool
   )

   if file, err = os.Open(path); err != nil {
       return
   }

   reader := bufio.NewReader(file)
   buffer := bytes.NewBuffer(make([]byte,1024))

   for {
      if part, prefix, err = reader.ReadLine();err != nil {
          break
      }
      buffer.Write(part)
      if !prefix {
         lines = append(lines,buffer.String())
         buffer.Reset()
      }
   }
   if err == io.EOF {
      err = nil
   }
   return
}


func writeLines(lines [] string,path string)(err error){
    var file *os.File

    if file,err = os.Create(path); err != nil{
         return
    }

    defer file.Close()

    for _,elem := range lines {
       _,err := file.WriteString(strings.TrimSpace(elem)+"\n")
       if err != nil {
           fmt.Println(err)
           break
       }
    }
    return
}

func main(){
   lines, err := readLines("5906376")
   if err != nil{
      fmt.Println("Error:%s\n",err)
      return
   }
   for _,line := range lines {
      fmt.Println(line)
   }

   err= writeLines(lines,"a.txt")
   fmt.Println(err)
}