微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

GoLang的readLines和writeLines方法

习惯了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)
}

原文地址:https://www.jb51.cc/go/191456.html

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐