add log hooks

Author: zhufuyi

pkg/ggorm/test_sqlite.db

@@ -34,4 +34,16 @@ Logger library wrapped in [zap](https://github.com/uber-go/zap).
     logger.Info("this is info")
     logger.Warn("this is warn", logger.String("foo","bar"), logger.Int("size",10), logger.Any("obj",obj))
     logger.Error("this is error", logger.Err(err), logger.String("foo","bar"))
+
+// (3) with hooks
+    logger.Init(
+        logger.WithLevel("info"),
+        logger.WithHooks(func(entry zapcore.Entry) error {
+            if strings.Contains(entry.Message, "error") {
+                fmt.Println("it contains error message")
+            }
+            return nil
+        }),
+    )
+    logger.Error("this is error", logger.Err(err), logger.String("foo","bar"))
 ```

pkg/logger/README.md

@@ -38,6 +38,7 @@ func getLogger() *zap.Logger {
 // print the debug level log in the terminal, example: Init()
 // print the info level log in the terminal, example: Init(WithLevel("info"))
 // print the json format, debug level log in the terminal, example: Init(WithFormat("json"))
+// log with hooks, example: Init(WithHooks(func(zapcore.Entry) error{return nil}))
 // output the log to the file out.log, using the default cut log-related parameters, debug-level log, example: Init(WithSave())
 // output the log to the specified file, custom set the log file cut log parameters, json format, debug level log, example:
 // Init(
@@ -72,6 +73,10 @@ func Init(opts ...Option) (*zap.Logger, error) {
 		str = fmt.Sprintf("initialize logger finish, config is output to 'file', format=%s, level=%s, file=%s", encoding, levelName, o.fileConfig.filename)
 	}
 
+	if len(o.hooks) > 0 {
+		zapLog = zapLog.WithOptions(zap.Hooks(o.hooks...))
+	}
+
 	defaultLogger = zapLog
 	Info(str)
 

pkg/logger/logger.go

@@ -3,8 +3,11 @@ package logger
 import (
 	"fmt"
 	"os"
+	"strings"
 	"testing"
 	"time"
+
+	"go.uber.org/zap/zapcore"
 )
 
 func printInfo() {
@@ -52,7 +55,7 @@ func TestInit(t *testing.T) {
 		{
 			name: "terminal json info",
 			args: args{[]Option{
-				WithFormat("json"), WithLevel(levelInfo),
+				WithFormat("json"), WithLevel("info"),
 			}},
 			wantErr: false,
 		},
@@ -63,6 +66,20 @@ func TestInit(t *testing.T) {
 			}},
 			wantErr: false,
 		},
+		{
+			name: "with hooks info",
+			args: args{[]Option{
+				WithFormat("json"),
+				WithLevel("info"),
+				WithHooks(func(entry zapcore.Entry) error {
+					if strings.Contains(entry.Message, "this is error") {
+						fmt.Println("it contains error message")
+					}
+					return nil
+				}),
+			}},
+			wantErr: false,
+		},
 		{
 			name: "file json debug",
 			args: args{[]Option{

pkg/logger/logger_test.go

@@ -1,6 +1,10 @@
 package logger
 
-import "strings"
+import (
+	"strings"
+
+	"go.uber.org/zap/zapcore"
+)
 
 var (
 	defaultLevel    = "debug" // output log levels debug, info, warn, error, default is debug
@@ -20,6 +24,8 @@ type options struct {
 	isSave   bool
 
 	fileConfig *fileOptions
+
+	hooks []func(zapcore.Entry) error
 }
 
 func defaultOptions() *options {
@@ -73,6 +79,13 @@ func WithSave(isSave bool, opts ...FileOption) Option {
 	}
 }
 
+// WithHooks set the log hooks
+func WithHooks(hooks ...func(zapcore.Entry) error) Option {
+	return func(o *options) {
+		o.hooks = hooks
+	}
+}
+
 // ------------------------------------------------------------------------------------------
 
 type fileOptions struct {

pkg/logger/option.go

@@ -234,7 +234,7 @@ option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_swagger) = {
         type: TYPE_API_KEY;
         in: IN_HEADER;
         name: "Authorization";
-        description: "Input a \"Bearer your-jwt-token\" to Value";
+        description: "Type Bearer your-jwt-token to Value";
       }
     }
   }