add a close to log, wait for log's goroutinue end.

log/log.go

@@ -35,6 +35,7 @@ type Msg struct {
 type Logger interface {
 	DoPrintf(level int, levelDesc, format string, a ...interface{})
 	SetColored(colored bool)
+	Close()
 }
 
 var glogger Logger
@@ -101,6 +102,15 @@ func Fatal(format string, param ...interface{}) {
 	do(FATAL_LEVEL, FATAL_LEVEL_DESC, format, param...)
 }
 
+func Close() {
+	if glogger == nil {
+		return
+	}
+	gloggerMut.Lock()
+	glogger.Close()
+	gloggerMut.Unlock()
+}
+
 //init log with SimpleLogger
 func Init(path string, fileLevel, shellLevel, maxLine, bufSize int) Logger {
 	logger := CreateLogger(path, fileLevel, shellLevel, maxLine, bufSize)

log/log_logger.go

@@ -5,6 +5,8 @@ import (
 	"log"
 	"os"
 	"path"
+	"runtime"
+	"sync"
 	"time"
 )
 
@@ -20,6 +22,7 @@ type SimpleLogger struct {
 	shellLevel int
 	isColored  bool
 	buffer     chan *Msg
+	wg         sync.WaitGroup
 }
 
 const (
@@ -116,14 +119,26 @@ func (self *SimpleLogger) createLogFile(dir string) (*os.File, error) {
 	return file, nil
 }
 
+func (self *SimpleLogger) Close() {
+	for len(self.buffer) > 0 {
+		runtime.Gosched()
+	}
+	close(self.buffer)
+	self.wg.Wait()
+}
+
 func (self *SimpleLogger) run() {
 	go func() {
+		self.wg.Add(1)
 		for {
 			m, ok := <-self.buffer
 			if ok {
 				self.doPrintf(m.level, m.levelDesc, m.fmt, m.param...)
+			} else {
+				break
 			}
 		}
+		self.wg.Done()
 	}()
 }
 

lotou.go

@@ -77,6 +77,8 @@ func StartWithName(nodeName string, f CloseFunc, customLogger log.Logger, data .
 	}()
 
 	core.Wait()
+
+	log.Close()
 }
 
 //RawStart start lotou, with no wait