Merge pull request #3 from FishGoddess/develop

v0.2.1

Author: FishGoddess

HISTORY.md

@@ -1,5 +1,11 @@
 ## 📜 历史版本的特性介绍 (Features in old versions)
 
+### v0.2.1
+
+> 此版本发布于 2025-04-12
+
+* 优化代码
+
 ### v0.2.0
 
 > 此版本发布于 2025-04-12

_icons/coverage.svg

@@ -17,8 +17,8 @@ var (
 type config struct {
 	fastFailed bool
 
-	newPoolFullErr   func(ctx context.Context) error
-	newPoolClosedErr func(ctx context.Context) error
+	newPoolFullErrFunc   func(ctx context.Context) error
+	newPoolClosedErrFunc func(ctx context.Context) error
 }
 
 func newDefaultConfig() *config {
@@ -31,14 +31,30 @@ func newDefaultConfig() *config {
 	}
 
 	conf := &config{
-		fastFailed:       false,
-		newPoolFullErr:   newPoolFullErr,
-		newPoolClosedErr: newPoolClosedErr,
+		fastFailed:           false,
+		newPoolFullErrFunc:   newPoolFullErr,
+		newPoolClosedErrFunc: newPoolClosedErr,
 	}
 
 	return conf
 }
 
+func (c *config) newPoolFullErr(ctx context.Context) error {
+	if c.newPoolFullErrFunc == nil {
+		return nil
+	}
+
+	return c.newPoolFullErrFunc(ctx)
+}
+
+func (c *config) newPoolClosedErr(ctx context.Context) error {
+	if c.newPoolClosedErrFunc == nil {
+		return nil
+	}
+
+	return c.newPoolClosedErrFunc(ctx)
+}
+
 type Option func(conf *config)
 
 func (o Option) ApplyTo(conf *config) {
@@ -56,7 +72,7 @@ func WithFastFailed() Option {
 func WithPoolFullErr(newPoolFullErr func(ctx context.Context) error) Option {
 	return func(conf *config) {
 		if newPoolFullErr != nil {
-			conf.newPoolFullErr = newPoolFullErr
+			conf.newPoolFullErrFunc = newPoolFullErr
 		}
 	}
 }
@@ -65,7 +81,7 @@ func WithPoolFullErr(newPoolFullErr func(ctx context.Context) error) Option {
 func WithPoolClosedErr(newPoolClosedErr func(ctx context.Context) error) Option {
 	return func(conf *config) {
 		if newPoolClosedErr != nil {
-			conf.newPoolClosedErr = newPoolClosedErr
+			conf.newPoolClosedErrFunc = newPoolClosedErr
 		}
 	}
 }

option.go

@@ -26,17 +26,17 @@ func TestWithPoolFullErr(t *testing.T) {
 		return nil
 	}
 
-	conf := &config{newPoolFullErr: nil}
+	conf := &config{newPoolFullErrFunc: nil}
 	WithPoolFullErr(newPoolFullErr)(conf)
 
-	if fmt.Sprintf("%p", conf.newPoolFullErr) != fmt.Sprintf("%p", newPoolFullErr) {
-		t.Fatalf("conf.newPoolFullErr %p is wrong", conf.newPoolFullErr)
+	if fmt.Sprintf("%p", conf.newPoolFullErrFunc) != fmt.Sprintf("%p", newPoolFullErr) {
+		t.Fatalf("conf.newPoolFullErr %p is wrong", conf.newPoolFullErrFunc)
 	}
 
 	WithPoolFullErr(nil)(conf)
 
-	if fmt.Sprintf("%p", conf.newPoolFullErr) != fmt.Sprintf("%p", newPoolFullErr) {
-		t.Fatalf("conf.newPoolFullErr %p is wrong", conf.newPoolFullErr)
+	if fmt.Sprintf("%p", conf.newPoolFullErrFunc) != fmt.Sprintf("%p", newPoolFullErr) {
+		t.Fatalf("conf.newPoolFullErr %p is wrong", conf.newPoolFullErrFunc)
 	}
 }
 
@@ -46,16 +46,16 @@ func TestWithPoolClosedErr(t *testing.T) {
 		return nil
 	}
 
-	conf := &config{newPoolClosedErr: nil}
+	conf := &config{newPoolClosedErrFunc: nil}
 	WithPoolClosedErr(newPoolClosedErr)(conf)
 
-	if fmt.Sprintf("%p", conf.newPoolClosedErr) != fmt.Sprintf("%p", newPoolClosedErr) {
-		t.Fatalf("conf.newPoolClosedErr %p is wrong", conf.newPoolClosedErr)
+	if fmt.Sprintf("%p", conf.newPoolClosedErrFunc) != fmt.Sprintf("%p", newPoolClosedErr) {
+		t.Fatalf("conf.newPoolClosedErr %p is wrong", conf.newPoolClosedErrFunc)
 	}
 
 	WithPoolClosedErr(nil)(conf)
 
-	if fmt.Sprintf("%p", conf.newPoolClosedErr) != fmt.Sprintf("%p", newPoolClosedErr) {
-		t.Fatalf("conf.newPoolClosedErr %p is wrong", conf.newPoolClosedErr)
+	if fmt.Sprintf("%p", conf.newPoolClosedErrFunc) != fmt.Sprintf("%p", newPoolClosedErr) {
+		t.Fatalf("conf.newPoolClosedErr %p is wrong", conf.newPoolClosedErrFunc)
 	}
 }