exegroup/HTTPListenAndServe移动到eghttp并增加HTTPServer；

exegroup/eghttp/actor.go

@@ -0,0 +1,73 @@
+package eghttp
+
+import (
+	"context"
+	"net"
+	"net/http"
+	"strconv"
+	"sync/atomic"
+	"time"
+)
+
+const (
+	DefaultAddr              = ":8080"
+	DefaultReadHeaderTimeout = time.Second
+)
+
+type Option interface {
+	apply(server *http.Server)
+}
+
+func WithPort(port int) Option {
+	return optionFunc(func(server *http.Server) {
+		server.Addr = net.JoinHostPort("", strconv.Itoa(port))
+	})
+}
+
+func WithHandler(handler http.Handler) Option {
+	return optionFunc(func(server *http.Server) {
+		server.Handler = handler
+	})
+}
+
+// WithServerOption 使用 fn 配置 [http.Server];
+func WithServerOption(fn func(server *http.Server)) Option {
+	return optionFunc(fn)
+}
+
+type optionFunc func(server *http.Server)
+
+func (fn optionFunc) apply(server *http.Server) {
+	fn(server)
+}
+
+// HTTPListenAndServe 创建 [http.Server] 并提供启动和停止函数;
+// opts 按照顺序应用到 [http.Server] 上.
+func HTTPListenAndServe(opts ...Option) (func(ctx context.Context) error, func(ctx context.Context)) {
+	server := &http.Server{
+		Addr:              DefaultAddr,
+		ReadHeaderTimeout: DefaultReadHeaderTimeout,
+	}
+	for _, opt := range opts {
+		opt.apply(server)
+	}
+	return HTTPServer(server)
+}
+
+// HTTPServer 提供 [http.Server] 的启动和停止函数;
+func HTTPServer(server *http.Server) (func(ctx context.Context) error, func(ctx context.Context)) {
+	inShutdown := &atomic.Bool{}
+	c := make(chan error, 1)
+	goFunc := func(_ context.Context) error {
+		err := server.ListenAndServe()
+		if inShutdown.Load() {
+			err = <-c
+		}
+		return err
+	}
+	stopFunc := func(ctx context.Context) {
+		inShutdown.Store(true)
+		c <- server.Shutdown(ctx)
+	}
+	return goFunc, stopFunc
+}

exegroup/httpserver.go

@@ -1,33 +0,0 @@
-package exegroup
-
-import (
-	"context"
-	"net"
-	"net/http"
-	"strconv"
-	"sync/atomic"
-	"time"
-)
-
-// HTTPListenAndServe 提供 [http.Server] 的启动和停止函数;
-func HTTPListenAndServe(port int, handler http.Handler) (func(ctx context.Context) error, func(ctx context.Context)) {
-	server := &http.Server{
-		Addr:              net.JoinHostPort("", strconv.Itoa(port)),
-		Handler:           handler,
-		ReadHeaderTimeout: time.Second,
-	}
-	inShutdown := &atomic.Bool{}
-	c := make(chan error, 1)
-	goFunc := func(_ context.Context) error {
-		err := server.ListenAndServe()
-		if inShutdown.Load() {
-			err = <-c
-		}
-		return err
-	}
-	stopFunc := func(ctx context.Context) {
-		inShutdown.Store(true)
-		c <- server.Shutdown(ctx)
-	}
-	return goFunc, stopFunc
-}

log/README.md

@@ -54,8 +54,7 @@ log.Logger()
 - `SetReportStack` 设置生成调用栈;
 - `SetReportStackLevel` 当日志等级小于设定值时强制生成调用栈;
 - `Reset` 把 Logger 恢复到初始状态;
-- `AddBeforeExit` 新增调用 `Logger.Exit` 时优先执行的函数, 先增加的后执行;
-- `SetExit` 设置 `Logger` 的退出函数(`Logger.Exit`), 当日志等级为 `LevelFatal` 时调用这个函数;
+- `AddBeforeExit` 增加 Exit 在调用 [os.Exit] 前执行的函数, 先增加的后执行;
 
 ### 日志生成
 
@@ -85,11 +84,13 @@ log.Logger()
 
 ### 其他用法
 
-关闭日志生成 `log.Logger().SetLevel(logsdk.LevelDisabled)`
+关闭日志生成, `log.Logger().SetLevel(logsdk.LevelDisabled)`;
 
-关闭强制生成调用栈 `log.Logger().SetReportStackLevel(logsdk.LevelDisabled)`
+关闭强制生成调用栈, `log.Logger().SetReportStackLevel(logsdk.LevelDisabled)`;
 
-mock 实现 mock 日志处理器对生成的日志进行处理
+记录 panic, 在 `defer v = recover()` 后执行 `log.Z(context.Background(), v)`, 其中 `Z` 需要是附加调用栈的等级;
+
+mock, 实现 mock 日志处理器对生成的日志进行处理;
 
 ---
 

log/log.go

@@ -145,7 +145,8 @@ func Fatalf(ctx context.Context, format string, args ...any) {
 	globalLogger.AddCallerSkip(1).Fatalf(ctx, format, args...)
 }
 
-// Panicf 格式化输出 LevelPanic 等级的日志
+// Panicf 格式化输出 LevelPanic 等级的日志,
+// 即使 Logger 日志等级高于 LevelPanic 也会 panic.
 func Panicf(ctx context.Context, format string, args ...any) {
 	globalLogger.AddCallerSkip(1).Panicf(ctx, format, args...)
 }

log/logsdk/entry.go

@@ -169,19 +169,19 @@ func (entry Entry) Fatalf(ctx context.Context, format string, args ...any) {
 	entry.logger.Exit(1)
 }
 
-// Panicf 格式化输出 LevelPanic 等级的日志
+// Panicf 格式化输出 LevelPanic 等级的日志,
 // 即使 Logger 日志等级高于 LevelPanic 也会 panic.
 func (entry Entry) Panicf(ctx context.Context, format string, args ...any) {
 	entry.log(ctx, LevelPanic, fmt.Sprintf(format, args...))
 }
 
 func (entry Entry) log(ctx context.Context, level Level, message string) {
-	newEntry := entry.copy()
 	defer func() {
 		if level == LevelPanic {
 			panic(message)
 		}
 	}()
+	newEntry := entry.copy()
 	if newEntry.logger.GetLevel() < level {
 		return
 	}

log/logsdk/level.go

@@ -64,14 +64,14 @@ const (
 	// LevelCount 日志等级的个数
 	LevelCount = 7
 
-	// LevelOffset 是把 LevelInfo 等级的日志值偏移到零值的偏移量
-	LevelOffset = 4
+	// levelOffset 是把 LevelInfo 等级的日志值偏移到零值的偏移量
+	levelOffset = 4
 )
 
 const (
 	// LevelDisabled 表示不处理任何等级的日志,
 	// 其他日志等级的值越小表示日志严重程度越高.
-	LevelDisabled Level = iota - LevelOffset - 1
+	LevelDisabled Level = iota - levelOffset - 1
 	LevelPanic
 	LevelFatal
 	LevelError

log/logsdk/logger.go

@@ -20,11 +20,11 @@ type entry = Entry
 // Logger 中保存了日志所需的全局配置,
 // 使用 Logger 处理日志.
 type Logger struct {
-	exit            func(code int)   // protected by lock
-	beforeExit      []func(code int) // protected by lock
-	levelProcessors levelProcessors  // protected by lock
+	beforeExitFns   []func()        // protected by lock
+	levelProcessors levelProcessors // protected by lock
 	entry
 	lock             sync.RWMutex
+	exitOnce         sync.Once
 	level            atomic.Int32
 	reportStackLevel atomic.Int32
 	callerSkip       atomic.Int32
@@ -36,7 +36,7 @@ type Logger struct {
 func (logger *Logger) AddProcessor(levels []Level, processor EntryProcessor) {
 	logger.lock.Lock()
 	for _, level := range levels {
-		logger.levelProcessors[level+LevelOffset] = append(logger.levelProcessors[level+LevelOffset], processor)
+		logger.levelProcessors[level+levelOffset] = append(logger.levelProcessors[level+levelOffset], processor)
 	}
 	logger.lock.Unlock()
 }
@@ -98,8 +98,7 @@ func (logger *Logger) SetReportStackLevel(level Level) {
 // Reset 把 Logger 重置到初始状态
 func (logger *Logger) Reset() {
 	logger.lock.Lock()
-	logger.exit = nil
-	logger.beforeExit = nil
+	logger.beforeExitFns = nil
 	logger.levelProcessors = levelProcessors{}
 	logger.lock.Unlock()
 
@@ -110,41 +109,38 @@ func (logger *Logger) Reset() {
 	logger.SetReportStack(false)
 }
 
-// Exit 退出程序, 执行的具体过程可以通过 SetExit 指定
+// BeforeExit 按照先添加后执行的顺序执行 AddBeforeExit 添加的函数,
+// AddBeforeExit 只会执行1次.
+func (logger *Logger) BeforeExit() {
+	logger.exitOnce.Do(func() {
+		logger.lock.RLock()
+		beforeExitFns := make([]func(), len(logger.beforeExitFns))
+		copy(beforeExitFns, logger.beforeExitFns)
+		logger.lock.RUnlock()
+		for i := len(beforeExitFns) - 1; i >= 0; i-- {
+			beforeExitFns[i]()
+		}
+	})
+}
+
+// Exit 执行 BeforeExit 后调用 [os.Exit]退出程序.
 func (logger *Logger) Exit(code int) {
-	logger.lock.RLock()
-	exit := logger.exit
-	beforeExit := make([]func(int), len(logger.beforeExit))
-	copy(beforeExit, logger.beforeExit)
-	logger.lock.RUnlock()
-	for i := len(beforeExit) - 1; i >= 0; i-- {
-		beforeExit[i](code)
-	}
-	if exit == nil {
-		exit = os.Exit
-	}
-	exit(code)
+	logger.BeforeExit()
+	os.Exit(code)
 }
 
-// SetExit 指定退出程序时执行的函数
-func (logger *Logger) SetExit(fn func(code int)) {
-	logger.lock.Lock()
-	logger.exit = fn
-	logger.lock.Unlock()
-}
-
-// AddBeforeExit 增加 Exit 在调用 SetExit 指定的函数前执行的函数,
+// AddBeforeExit 增加 Exit 在调用 [os.Exit] 前执行的函数,
 // 先增加的后执行.
-func (logger *Logger) AddBeforeExit(fn ...func(code int)) {
+func (logger *Logger) AddBeforeExit(fn ...func()) {
 	logger.lock.Lock()
-	logger.beforeExit = append(logger.beforeExit, fn...)
+	logger.beforeExitFns = append(logger.beforeExitFns, fn...)
 	logger.lock.Unlock()
 }
 
 func (logger *Logger) getLevelProcessors(level Level) []EntryProcessor {
 	logger.lock.RLock()
 	defer logger.lock.RUnlock()
-	return logger.levelProcessors[level+LevelOffset]
+	return logger.levelProcessors[level+levelOffset]
 }
 
 func (logger *Logger) newEntry() Entry {

log/logsdk/runtime.go

@@ -8,6 +8,8 @@ const (
 	maximumFrames       = 32
 	getCallerSkipOffset = 2
 	entrySkipOffset     = 2
+
+	runtimeMain = "runtime.main"
 )
 
 // Frame 调用相关信息
@@ -44,6 +46,9 @@ func getStack(skip, maximumFrames int) []Frame {
 	for {
 		frame, more := frames.Next()
 		if frame.PC != 0 {
+			if frame.Function == runtimeMain {
+				break
+			}
 			stack = append(stack, Frame{
 				Function: frame.Function,
 				File:     frame.File,