flush bufferred log on exit

gollog.go

@@ -2,6 +2,7 @@ package gol
 
 import (
 	"fmt"
+	"os"
 	"runtime"
 
 	"bytes"
@@ -13,21 +14,27 @@ import (
 )
 
 type gollog struct {
-	level    LogLevel
-	option   LogOption
-	adapters []adapter.Adapter
-	logChan  chan string
+	level      LogLevel
+	option     LogOption
+	adapters   []adapter.Adapter
+	logChan    chan string
+	doneChan   chan struct{}
+	signalChan chan os.Signal
 }
 
 func (l *gollog) msgPump() {
 	for {
 		select {
-		case msg := <-l.logChan:
+		case msg, ok := <-l.logChan:
+			if !ok {
+				break
+			}
 			for _, adap := range l.adapters {
 				adap.Write(internal.Str2bytes(msg))
 			}
 		}
 	}
+	close(l.doneChan)
 }
 
 func (l *gollog) put(msg string) {
@@ -209,3 +216,8 @@ func (l *gollog) AddLogAdapter(a adapter.Adapter) {
 		l.adapters = append(l.adapters, a)
 	}
 }
+
+func (l *gollog) flush() {
+	close(l.logChan)
+	<-l.doneChan
+}

logger.go

@@ -1,6 +1,10 @@
 package gol
 
 import (
+	"os"
+
+	"os/signal"
+
 	"github.com/philchia/gol/adapter"
 	"github.com/philchia/gol/adapter/console"
 )
@@ -30,11 +34,21 @@ type Logger interface {
 // NewLogger create a Logger
 func NewLogger(level LogLevel) Logger {
 	logger := &gollog{
-		level:   level,
-		option:  LstdFlags,
-		logChan: make(chan string, 1024),
+		level:      level,
+		option:     LstdFlags,
+		logChan:    make(chan string, 1024),
+		doneChan:   make(chan struct{}),
+		signalChan: make(chan os.Signal, 1),
 	}
+
+	signal.Notify(logger.signalChan, os.Interrupt, os.Kill)
 	logger.AddLogAdapter(console.NewAdapter())
+	go func() {
+		select {
+		case <-logger.signalChan:
+			logger.flush()
+		}
+	}()
 
 	go logger.msgPump()
 	return logger