這是一個建立於 的文章,其中的資訊可能已經有所發展或是發生改變。
操作檔案是任何程式設計語言都繞不過,要掌握一門語言,知道如何操作檔案是必不可少的,今天學習了下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 mainimport "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.bakmanu@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 goif not existed ,please create this fileif existed, Please write appendhello world,hello gothis is test file created by goif not existed ,please create this fileif existed, Please write appendhello world,hello go