easycz / logrotate Goto Github PK

package main

import (
	"bufio"
	"fmt"
	"os"
	"path/filepath"
	"time"

	"github.com/easyCZ/logrotate"
	jww "github.com/spf13/jwalterweatherman"
)

func main() {
	jww.SetLogThreshold(jww.LevelTrace)
	jww.SetStdoutThreshold(jww.LevelTrace)

	writer, err := logrotate.New(jww.TRACE, logrotate.Options{
		// Where should the writer be outputting files?
		// If the directory does not exist, it will be created.
		// Required.
		Directory: getExecutableDir(),
		// What is the maximum size of each file?
		// Optional. Use 0 for unlimited.
		MaximumFileSize: 1024 * 1024 * 1024,
		// How often should a new file be created, based on time?
		// Optional. Use 0 to disable time based log rotation.
		MaximumLifetime: time.Hour,
		// How would you like to name your files?
		// Invoked each time a new file is being created.
		FileNameFunc: logrotate.DefaultFilenameFunc,
	})
	if err != nil {
		fatal(err)
	}
	jww.SetLogOutput(writer)

	jww.INFO.Println("This is a test info log.")
	jww.TRACE.Println("This is a test trace log.")
	jww.ERROR.Println("This is a test error log.")
	jww.FATAL.Println("This is a test fatal log.")

	// Ensure all messages are flushed to files before exiting
	err = writer.Close()
	if err != nil {
		fatal(err)
	}
}

func fatal(err error) {
	jww.FATAL.Println(err)
	fmt.Print("Press the Enter key to continue...")
	bufio.NewReader(os.Stdin).ReadBytes('\n')
	os.Exit(1)
}

func getExecutableDir() string {
	ex, err := os.Executable()
	if err != nil {
		jww.FATAL.Println(err)
		panic("Failed to find the executable directory!")
	}
	dir := filepath.Dir(ex)
	return dir
}