Goで時間でファイルをローテートする

単純に時間で分割してほしいだけだったんですが、ピンポイントのものがなかったので作ってみました

package trfile

import (
    "os"
    "path"
    "time"
)

type TimeRotateFile struct {
    fileNameFormat string
    file           *os.File
}

var (
    FileMode    os.FileMode = 0644
    DirFileMode os.FileMode = 0755
)

func NewFormat(fileNameFormat string) (*TimeRotateFile, error) {
    trf := &TimeRotateFile{
        fileNameFormat: fileNameFormat,
    }

    if err := trf.fileSet(); err != nil {
        return nil, err
    }

    return trf, nil
}

func (trf *TimeRotateFile) fileNameNow() string {
    return time.Now().Format(trf.fileNameFormat)
}

func (trf *TimeRotateFile) fileSet() error {
    fileNameNow := trf.fileNameNow()

    if dir, _ := path.Split(fileNameNow); len(dir) > 0 {
        if err := os.MkdirAll(dir, DirFileMode); err != nil {
            return err
        }
    }

    file, err := os.OpenFile(fileNameNow, os.O_RDWR|os.O_CREATE|os.O_APPEND, FileMode)

    if err != nil {
        return err
    }

    trf.file = file

    return nil
}

func (trf *TimeRotateFile) Write(b []byte) (int, error) {
    if fileNameNow := trf.fileNameNow(); fileNameNow != trf.file.Name() {
        if err := trf.file.Close(); err != nil {
            return 0, err
        }

        if err := trf.fileSet(); err != nil {
            return 0, err
        }
    }

    return trf.file.Write(b)
}

func (trf *TimeRotateFile) Name() string {
    return trf.file.Name()
}

func (trf *TimeRotateFile) Close() error {
    return trf.file.Close()
}

Golangの "2006-01-02 Mon 15:04:05" みたいなタイムフォーマット用いてファイル名やディレクトリを指定します

package main

import (
    "log"
    "os"

    "github.com/ngc224/trfile"
)

var fileNameFormat = "logs/2006/01/02/halo.log"

func main() {
    w, err := trfile.NewFormat(fileNameFormat)

    if err != nil {
        os.Exit(1)
    }

    log.SetOutput(w)
    log.Println("Halo World!")
}