登录
首页 >  Golang >  Go教程

golang实战之truncate日志文件详解

来源:脚本之家

时间:2023-01-07 12:04:55 404浏览 收藏

怎么入门Golang编程?需要学习哪些知识点?这是新手们刚接触编程时常见的问题;下面golang学习网就来给大家整理分享一些知识点,希望能够给初学者一些帮助。本篇文章就来介绍《golang实战之truncate日志文件详解》,涉及到truncate、日志文件,有需要的可以收藏一下

前言

有时候我们生成的日志文件很大,如果文件过大,或者磁盘空间有限,那么我们就需要把日志文件改小一点。

(当然实际应用场景中,一般采用log rotate的方式实现多个日志文件,定期把旧的日志文件挪走或者删除)

现在我们考虑如何把日志文件变小的场景。

首先想到的是文件truncate,遗憾的是truncate是把新的内容删掉,而不是把旧的内容删掉;因为通常我们的做法是要把旧的日志内容删除,而保留最新的日志内容。

方法1:拷贝文件,然后把前半部分删除,保留新的后半部分

$ wc -l logfile
$ tail -n {LINENUM}/2 logfile > logfile2
$ mv logfile2 logfile

缺点是需要额外的临时磁盘空间(logfile2),保留新拷贝的文件

方法2:把日志文件读入内存,截去前半部分,把后半部分重新写会文件。

缺点也是要创建一个临时文件,或者把整个文件内容一次性读入内存。

下面是一个实现创建临时文件的例子:

(如果采用整个文件读入内存,那么可以重复覆盖使用原来的读入文件,不用创建临时文件):

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上执行时间大约是:

$ go build && time ./main
real 0m6.972s
user 0m4.668s
sys  0m2.055s

方法3:也是利用文件,把文件mmap映射到内存,然后使用copy函数覆盖内存。

缺点是,mmap操作日志文件不能过大。

下面是一个实现的代码例子:

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 = 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()
}

函数truncateText的功能就是把文件截取到原来的一半大小,并保持按行对齐;分如下几步:

  • 先把文件映射syscall.Mmap 到内存。
  • 查找到中间行的位置。
  • 然后用内置函数copy,把中间行以后的挪动到文件开始位置。
  • 修改文件的大小file.Truncate
  • 重置文件的偏移指针offset
  • 取消文件映射syscall.Munmap

运行:

对于1G大小的日志文件,在我的Linux上执行时间大约是:

$ go build && time ./main
real 0m1.861s
user 0m0.335s
sys  0m0.650s

总结

到这里,我们也就讲完了《golang实战之truncate日志文件详解》的内容了。个人认为,基础知识的学习和巩固,是为了更好的将其运用到项目中,欢迎关注golang学习网公众号,带你了解更多关于golang的知识点!

声明:本文转载于:脚本之家 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>
评论列表