Enable log rotation

logger.go

@@ -6,13 +6,15 @@ import (
 	"log"
 	"os"
 	"runtime"
+	"time"
 
 	"github.com/fatih/color"
 )
 
 var (
 	currentLogLevel LogLevel = LevelError
 	logFilePath     string   = "logfile.txt"
+	maxSizeInBytes  int64    = 5 * 1024 * 1024
 )
 
 // Init initializes a logger
@@ -24,13 +26,46 @@ func Init(opts *LoggerOptions) (func(), error) {
 		if opts.LogFilePath != "" {
 			logFilePath = opts.LogFilePath
 		}
+		if opts.LogFileMaxSize > 0 {
+			maxSizeInBytes = opts.LogFileMaxSize
+		}
 	}
 	f, err := os.OpenFile(logFilePath, os.O_APPEND|os.O_RDWR|os.O_CREATE, 0755)
 	if err != nil {
 		return func() {}, err
 	}
 	mw := io.MultiWriter(os.Stdout, f)
 	log.SetOutput(mw)
+
+	go func() {
+		for {
+			size, err := GetFileSize()
+			if err != nil {
+				panic(err)
+			}
+			if size >= maxSizeInBytes {
+				source, err := os.Open(logFilePath)
+				if err != nil {
+					panic(err)
+				}
+				destination, err := os.Create(logFilePath + "-" + time.Now().Format("20060102150405"))
+				if err != nil {
+					panic(err)
+				}
+				_, err = io.Copy(destination, source)
+				if err != nil {
+					panic(err)
+				}
+				source.Close()
+				destination.Close()
+				if err := os.Truncate(logFilePath, 0); err != nil {
+					panic(err)
+				}
+			}
+			time.Sleep(10 * time.Second)
+		}
+	}()
+
 	return func() {
 		defer f.Close()
 	}, nil
@@ -98,3 +133,16 @@ func Error(s string) {
 	str := color.RedString("%s", s)
 	log.Print("[" + filename + ":" + fmt.Sprint(line) + "] " + str)
 }
+
+// GetFileSize returns the log file size
+func GetFileSize() (int64, error) {
+	info, err := os.Stat(logFilePath)
+	if err != nil {
+		return 0, err
+	}
+	if !info.Mode().IsRegular() {
+		return 0, fmt.Errorf("%s is not a regular file", logFilePath)
+	}
+	bytes := info.Size()
+	return bytes, nil
+}

types.go

@@ -5,6 +5,7 @@ type LogLevel string
 
 // LoggerOptions to pass to the logger instance
 type LoggerOptions struct {
-	LogLevel    LogLevel
-	LogFilePath string
+	LogLevel       LogLevel
+	LogFilePath    string
+	LogFileMaxSize int64
 }