golang 文件操作

操作文件是任何编程语言都绕不过,要掌握一门语言,知道如何操作文件是必不可少的,今天学习了下golang对文件操作的支持。

golang对文件的支持是在os package里。我无意将本文写成官方文档的模样,我只是想讨论如何利用这些接口操作文件。

OPEN

func OpenFile(name string, flag int, perm FileMode) (file *File, err error)

我们看到了也有flag,也有FileMode.比如说我要读写打开一个文件,如果不存在就创建,如果存在,就追加写,如何写go 代码?

f,err := os.OpenFile("test.txt",os.O_CREATE|os.O_APPEND|os.O_RDWR,0660)
  if(err != nil){
    panic(err)
  }

CLOSE

这个接口无甚好说。接口如下

func (f *File) Close() error

但说接口没啥说的,但是golang提供了defer,这是一个我认为很赞的特点,就是将不得不做的cleanup放到defer去做。go提供了defer解决这种困境,后面不用时刻惦记close,函数退出前,会执行close

f,err := os.OpenFile("test.txt",os.O_CREATE|os.O_APPEND|os.O_RDWR,0660)
  if(err != nil){
    panic("open file failed")
  }
  defer f.Close()
  ...

READ和WRITE

接口如下

func (f *File) Read(b []byte) (n int, err error)
func (f *File) ReadAt(b []byte, off int64) (n int, err error)
func (f *File) Write(b []byte) (n int, err error)
func (f *File) WriteAt(b []byte, off int64) (n int, err error)
func (f *File) WriteString(s string) (ret int, err error)

看到代码片段,学习使用读写接口:

read_buf := make([]byte,32)
  var pos int64 = 0
  for{
    n,err := f.ReadAt(read_buf,pos)
    if err != nil && err != io.EOF{
      panic(err)
    }
    if n == 0{
      fmt.Printf("\nfinish read\n")
      break
    }
    fmt.Printf("%s",string(read_buf[:n]))
    pos = pos +(int64)(n)
  }

在看一个代码片段:

 var buff = make([]byte,1024)
  for{
    n,err := fi.Read(buff)
    if err != nil && err != io.EOF{
      panic(err)
    }
    
    if n == 0{
      break
    }
    if _,err := fo.Write(buff[:n]); err != nil{
      panic(err)
    }
  }

最后,我写了一个完整的代码,完成简单cp功能,就叫mycp

package main
import "fmt"
import "os"
import "io"
func usage(){
  fmt.Printf("%s %s %s\n",os.Args[0],"filename" , "newfile")
}
func main(){
  
  if len(os.Args) != 3{
    usage()
    return 
  }
  filename_in := os.Args[1]
  fi,err := os.Open(filename_in)
  if err != nil{
    panic(err)
  }
  defer fi.Close()
  filename_out := os.Args[2]
  fo,err := os.Create(filename_out)
  if err != nil{
    panic(err)
  }
  defer fo.Close()
  var buff = make([]byte,1024)
  for{
    n,err := fi.Read(buff)
    if err != nil && err != io.EOF{
      panic(err)
    }
    
    if n == 0{
      break
    }
    if _,err := fo.Write(buff[:n]); err != nil{
      panic(err)
    }
  }
}

执行结果

manu@manu-hacks:~/code/go/self$ ./mycp test.txt test.bak
manu@manu-hacks:~/code/go/self$ diff test.txt test.bak 
manu@manu-hacks:~/code/go/self$ cat test.txt 
this is test file created by go
if not existed ,please create this file
if existed, Please write append
hello world,hello go
this is test file created by go
if not existed ,please create this file
if existed, Please write append
hello world,hello go

本文来自:开源中国博客

感谢作者:冷絮

查看原文:golang 文件操作

• 编辑
• 预览