u

code/base-project/base-go-v3/log/log.go

@@ -32,7 +32,7 @@ func InitLog() {
 		}
 	}
 
-	logger := hertzzap.NewLogger(hertzzap.WithZapOptions(zap.AddCaller(), zap.AddCallerSkip(3)))
+	logger := hertzzap.NewLogger(hertzzap.WithCores([]hertzzap.CoreConfig{}))
 
 	lumberjackLogger := &lumberjack.Logger{
 		Filename:   fileName,

code/base-project/base-go-v3/log/option.go

@@ -0,0 +1,115 @@
+package log
+
+import (
+	"os"
+
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+)
+
+type Option interface {
+	apply(cfg *config)
+}
+
+type ExtraKey string
+
+type option func(cfg *config)
+
+func (fn option) apply(cfg *config) {
+	fn(cfg)
+}
+
+type CoreConfig struct {
+	Enc zapcore.Encoder
+	Ws  zapcore.WriteSyncer
+	Lvl zapcore.LevelEnabler
+}
+
+type config struct {
+	extraKeys     []ExtraKey
+	coreConfigs   []CoreConfig
+	zapOpts       []zap.Option
+	extraKeyAsStr bool
+}
+
+// defaultCoreConfig default zapcore config: json encoder, atomic level, stdout write syncer
+func defaultCoreConfig() *CoreConfig {
+	// default log encoder
+	enc := zapcore.NewJSONEncoder(zap.NewProductionEncoderConfig())
+	// default log level
+	lvl := zap.NewAtomicLevelAt(zap.InfoLevel)
+	// default write syncer stdout
+	ws := zapcore.AddSync(os.Stdout)
+
+	return &CoreConfig{
+		Enc: enc,
+		Ws:  ws,
+		Lvl: lvl,
+	}
+}
+
+// defaultConfig default config
+func defaultConfig() *config {
+	return &config{
+		coreConfigs:   []CoreConfig{*defaultCoreConfig()},
+		zapOpts:       []zap.Option{},
+		extraKeyAsStr: false,
+	}
+}
+
+// WithCoreEnc zapcore encoder
+func WithCoreEnc(enc zapcore.Encoder) Option {
+	return option(func(cfg *config) {
+		cfg.coreConfigs[0].Enc = enc
+	})
+}
+
+// WithCoreWs zapcore write syncer
+func WithCoreWs(ws zapcore.WriteSyncer) Option {
+	return option(func(cfg *config) {
+		cfg.coreConfigs[0].Ws = ws
+	})
+}
+
+// WithCoreLevel zapcore log level
+func WithCoreLevel(lvl zap.AtomicLevel) Option {
+	return option(func(cfg *config) {
+		cfg.coreConfigs[0].Lvl = lvl
+	})
+}
+
+// WithCores zapcore
+func WithCores(coreConfigs ...CoreConfig) Option {
+	return option(func(cfg *config) {
+		cfg.coreConfigs = coreConfigs
+	})
+}
+
+// WithZapOptions add origin zap option
+func WithZapOptions(opts ...zap.Option) Option {
+	return option(func(cfg *config) {
+		cfg.zapOpts = append(cfg.zapOpts, opts...)
+	})
+}
+
+// WithExtraKeys allow you log extra values from context
+func WithExtraKeys(keys []ExtraKey) Option {
+	return option(func(cfg *config) {
+		for _, k := range keys {
+			if !InArray(k, cfg.extraKeys) {
+				cfg.extraKeys = append(cfg.extraKeys, k)
+			}
+		}
+	})
+}
+
+// WithExtraKeyAsStr convert extraKey to a string type when retrieving value from context
+// Not recommended for use, only for compatibility with certain situations
+//
+// For more information, refer to the documentation at
+// `https://pkg.go.dev/context#WithValue`
+func WithExtraKeyAsStr() Option {
+	return option(func(cfg *config) {
+		cfg.extraKeyAsStr = true
+	})
+}

code/base-project/base-go-v3/log/utils.go

@@ -0,0 +1,50 @@
+// Copyright 2022 CloudWeGo Authors.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package log
+
+import (
+	"github.com/cloudwego/hertz/pkg/common/hlog"
+	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+)
+
+// InArray check if a string in a slice
+func InArray(key ExtraKey, arr []ExtraKey) bool {
+	for _, k := range arr {
+		if k == key {
+			return true
+		}
+	}
+	return false
+}
+
+func hLevelToZapLevel(level hlog.Level) zapcore.Level {
+	var lvl zapcore.Level
+	switch level {
+	case hlog.LevelTrace, hlog.LevelDebug:
+		lvl = zap.DebugLevel
+	case hlog.LevelInfo:
+		lvl = zap.InfoLevel
+	case hlog.LevelWarn, hlog.LevelNotice:
+		lvl = zap.WarnLevel
+	case hlog.LevelError:
+		lvl = zap.ErrorLevel
+	case hlog.LevelFatal:
+		lvl = zap.FatalLevel
+	default:
+		lvl = zap.WarnLevel
+	}
+	return lvl
+}

code/base-project/base-go-v3/main.go

@@ -5,8 +5,8 @@ import (
 	"cn/oneao/base-go/db"
 	"cn/oneao/base-go/log"
 	"cn/oneao/base-go/utils/idgen"
-
 	"github.com/cloudwego/hertz/pkg/app/server"
+	"github.com/cloudwego/hertz/pkg/common/hlog"
 	"github.com/hertz-contrib/requestid"
 )
 
@@ -29,6 +29,15 @@ func main() {
 
 	log.InitLog()
 
+	type User struct {
+		Name string
+		Age  int
+	}
+
+	u := User{"tom", 20}
+
+	hlog.Info(u)
+
 	register(h)
 	h.Spin()
 }