add build tags: log_linenumber

Author: lonng

log/log.go

@@ -91,22 +91,6 @@ func (logger *Logger) Close() {
 	logger.baseFile = nil
 }
 
-func (logger *Logger) doPrintf(level int, printLevel string, format string, a ...interface{}) {
-	if level < logger.level {
-		return
-	}
-	if logger.baseLogger == nil {
-		panic("logger closed")
-	}
-
-	format = printLevel + format
-	logger.baseLogger.Printf(format, a...)
-
-	if level == fatalLevel {
-		os.Exit(1)
-	}
-}
-
 func (logger *Logger) Debug(format string, a ...interface{}) {
 	logger.doPrintf(debugLevel, printDebugLevel, format, a...)
 }

log/printf.go

@@ -0,0 +1,21 @@
+// +build !log_linenumber
+
+package log
+
+import "os"
+
+func (logger *Logger) doPrintf(level int, printLevel string, format string, a ...interface{}) {
+	if level < logger.level {
+		return
+	}
+	if logger.baseLogger == nil {
+		panic("logger closed")
+	}
+
+	format = printLevel + format
+	logger.baseLogger.Printf(format, a...)
+
+	if level == fatalLevel {
+		os.Exit(1)
+	}
+}

log/printf_linenumber.go

@@ -0,0 +1,35 @@
+// +build log_linenumber
+
+package log
+
+import (
+	"os"
+	"runtime"
+	"strconv"
+)
+
+func logSite() string {
+	_, file, line, ok := runtime.Caller(4)
+	if !ok {
+		file = "???"
+		line = 0
+	}
+	c := string(file + ":" + strconv.FormatInt(int64(line), 10))
+	return c
+}
+
+func (logger *Logger) doPrintf(level int, printLevel string, format string, a ...interface{}) {
+	if level < logger.level {
+		return
+	}
+	if logger.baseLogger == nil {
+		panic("logger closed")
+	}
+
+	format = printLevel + "["+logSite()+"] " + format
+	logger.baseLogger.Printf(format, a...)
+
+	if level == fatalLevel {
+		os.Exit(1)
+	}
+}