feat: add WithHook to customize the hook function for zerolog

hertz-contrib · Jan 27, 2024 · 4179a32 · 4179a32
1 parent 8af1031
commit 4179a32
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/logging/zerolog/option.go b/logging/zerolog/option.go
@@ -76,6 +76,13 @@ func WithRecordStackTraceInSpan(recordStackTraceInSpan bool) Option {
 	})
 }
 
+// WithHook configure a custom hook function option
+func WithHook(fn zerolog.HookFunc) Option {
+	return option(func(cfg *config) {
+		cfg.hookFunc = fn
+	})
+}
+
 func (cfg config) getZerologHookFn() zerolog.HookFunc {
 	if cfg.hookFunc != nil {
 		return cfg.hookFunc