操作文件是任何编程语言都绕不过,要掌握一门语言,知道如何操作文件是必不可少的,今天学习了下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
有疑问加站长微信联系(非本文作者)