golang實戰(zhàn)之truncate日志文件詳解
前言
有時候我們生成的日志文件很大,如果文件過大,或者磁盤空間有限,那么我們就需要把日志文件改小一點。
(當然實際應用場景中,一般采用log rotate的方式實現(xiàn)多個日志文件,定期把舊的日志文件挪走或者刪除)
現(xiàn)在我們考慮如何把日志文件變小的場景。
首先想到的是文件truncate,遺憾的是truncate是把新的內容刪掉,而不是把舊的內容刪掉;因為通常我們的做法是要把舊的日志內容刪除,而保留最新的日志內容。
方法1:拷貝文件,然后把前半部分刪除,保留新的后半部分
$ wc -l logfile $ tail -n {LINENUM}/2 logfile > logfile2 $ mv logfile2 logfile
缺點是需要額外的臨時磁盤空間(logfile2),保留新拷貝的文件
方法2:把日志文件讀入內存,截去前半部分,把后半部分重新寫會文件。
缺點也是要創(chuàng)建一個臨時文件,或者把整個文件內容一次性讀入內存。
下面是一個實現(xiàn)創(chuàng)建臨時文件的例子:
(如果采用整個文件讀入內存,那么可以重復覆蓋使用原來的讀入文件,不用創(chuàng)建臨時文件):
package main import ( "bufio" "log" "io" "os" ) func ReadLine(r *bufio.Reader, w *bufio.Writer) error { for i := 0; ; i++ { line, err := r.ReadString('\n') if i >= 12554364/2 { // this number is hard-codes just for example use if _,err := w.WriteString(line); err != nil { log.Fatal(err) } } if err != nil { if err == io.EOF { return nil } return err } } return nil } func main() { fin, err := os.Open("data.in") if err != nil { log.Fatal(err) } defer fin.Close() fout, err := os.Create("data.dat") if err != nil { log.Fatal(err) } defer fout.Close() reader := bufio.NewReader(fin) writer := bufio.NewWriter(fout) if err := ReadLine(reader, writer); err != nil { log.Fatal(err) } writer.Flush() }
運行:
對于1G大小的日志文件,在我的Linux上執(zhí)行時間大約是:
$ go build && time ./main real 0m6.972s user 0m4.668s sys 0m2.055s
方法3:也是利用文件,把文件mmap映射到內存,然后使用copy函數(shù)覆蓋內存。
缺點是,mmap操作日志文件不能過大。
下面是一個實現(xiàn)的代碼例子:
package main import ( "fmt" "log" "os" "syscall" ) // Truncate file to half size, with line aligned. func truncateText(f *os.File) error { fi, err := f.Stat() if err != nil { return err } mem, err := syscall.Mmap(int(f.Fd()), 0, int(fi.Size()), syscall.PROT_READ|syscall.PROT_WRITE, syscall.MAP_SHARED) if err != nil { return err } size := fi.Size() if size <= 1 { // Don't need to truncate file if it's too small return nil } trun := size/2 - 1 for ; trun < fi.Size(); trun ++ { if mem[trun] == '\n' { break } } //fmt.Printf("size=%d, trun=%d\n", size, trun) if trun >= size - 1 { trun = size/2 } else { trun = trun + 1 } // Overwrite file content copy(mem[0:], mem[trun:]) err = syscall.Munmap(mem) if err != nil { return err } // truncate file f.Truncate(fi.Size() - trun) // reset file offset f.Seek(trun,0) return nil } func main() { path := "data.dat" f, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE, 0644) if err != nil { fmt.Println("Cannot create file") log.Fatal(err) } err = truncateText(f) if err != nil { fmt.Println("Cannot truncateLog file") log.Fatal(err) } f.Write([]byte("ABCD")) f.Close() }
函數(shù)truncateText的功能就是把文件截取到原來的一半大小,并保持按行對齊;分如下幾步:
- 先把文件映射syscall.Mmap 到內存。
- 查找到中間行的位置。
- 然后用內置函數(shù)copy,把中間行以后的挪動到文件開始位置。
- 修改文件的大小file.Truncate
- 重置文件的偏移指針offset
- 取消文件映射syscall.Munmap
運行:
對于1G大小的日志文件,在我的Linux上執(zhí)行時間大約是:
$ go build && time ./main real 0m1.861s user 0m0.335s sys 0m0.650s
總結
以上就是這篇文章的全部內容了,希望本文的內容對大家的學習或者工作具有一定的參考學習價值,如果有疑問大家可以留言交流,謝謝大家對腳本之家的支持。
相關文章
Golang使用CGO與Plugin技術運行加載C動態(tài)庫
這篇文章主要介紹了Golang使用CGO與Plugin技術運行加載C動態(tài)庫,Golang?程序在運行時加載C動態(tài)庫的技術,跳過了Golang項目編譯階段需要鏈接C動態(tài)庫的過程,提高了Golang項目開發(fā)部署的靈活性2022-07-07Go語言crypto包創(chuàng)建自己的密碼加密工具實現(xiàn)示例
Go語言借助它的簡單性和強大的標準庫,實現(xiàn)一個自己的密碼加密工具,本文將會結合代碼示例深入探討如何使用Go語言的crypto包來實現(xiàn)自己的加密工具2023-11-11