added simple log rotator

app/server/server.go

@@ -3,11 +3,11 @@ package server
 import (
 	"os"
 	"os/signal"
-	"syscall"
-
 	"os/user"
 	"path/filepath"
 	"strconv"
+	"syscall"
+	"time"
 
 	"helmet/app/config"
 	"helmet/app/handler"
@@ -88,13 +88,13 @@ func (srv *Server) Build() error {
 	if err != nil {
 		return err
 	}
-	// Create ghandler
+	// Create handler
 	handlerConfig := &handler.HandlerConfig{
 		Operator: srv.oper,
 	}
 	srv.hand = handler.NewHandler(handlerConfig)
 
-	// Create gservice
+	// Create service
 	serviceConfig := &service.ServiceConfig{
 		PortNum:  srv.conf.Service.Port,
 		Hostname: srv.conf.Hostname,
@@ -220,6 +220,27 @@ func (srv *Server) Daemonize() error {
 		if err != nil {
 			return err
 		}
+		// Log file rotator
+		logFunc := func() {
+			for {
+				stat, err := os.Stat(srv.conf.LogPath)
+				if err == nil && stat.Size() > srv.conf.LogLimit {
+					os.Rename(srv.conf.LogPath+".2", srv.conf.LogPath+".3")
+					os.Rename(srv.conf.LogPath+".1", srv.conf.LogPath+".2")
+					os.Rename(srv.conf.LogPath, srv.conf.LogPath+".1")
+					prevLogFile := logFile
+					logFile, err = os.OpenFile(srv.conf.LogPath, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0640)
+					if err == nil {
+						syscall.Dup2(int(logFile.Fd()), int(os.Stdout.Fd()))
+						syscall.Dup2(int(logFile.Fd()), int(os.Stderr.Fd()))
+						prevLogFile.Close()
+					}
+					time.Sleep(10 * time.Second)
+				}
+			}
+		}
+		go logFunc()
+
 		// Write process ID
 		rundir := filepath.Dir(srv.conf.RunPath)
 		err = os.MkdirAll(rundir, 0750)