优化代码

Author: FishGoddess

option.go

@@ -71,13 +71,17 @@ func WithFastFailed() Option {
 // WithPoolFullErr sets newPoolFullErr to config.
 func WithPoolFullErr(newPoolFullErr func(ctx context.Context) error) Option {
 	return func(conf *config) {
-		conf.newPoolFullErrFunc = newPoolFullErr
+		if newPoolFullErr != nil {
+			conf.newPoolFullErrFunc = newPoolFullErr
+		}
 	}
 }
 
 // WithPoolClosedErr sets newPoolClosedErr to config.
 func WithPoolClosedErr(newPoolClosedErr func(ctx context.Context) error) Option {
 	return func(conf *config) {
-		conf.newPoolClosedErrFunc = newPoolClosedErr
+		if newPoolClosedErr != nil {
+			conf.newPoolClosedErrFunc = newPoolClosedErr
+		}
 	}
 }

option_test.go

@@ -32,6 +32,12 @@ func TestWithPoolFullErr(t *testing.T) {
 	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.newPoolFullErrFunc) != fmt.Sprintf("%p", newPoolFullErr) {
+		t.Fatalf("conf.newPoolFullErr %p is wrong", conf.newPoolFullErrFunc)
+	}
 }
 
 // go test -v -cover -run=^TestWithPoolClosedErr$
@@ -46,4 +52,10 @@ func TestWithPoolClosedErr(t *testing.T) {
 	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.newPoolClosedErrFunc) != fmt.Sprintf("%p", newPoolClosedErr) {
+		t.Fatalf("conf.newPoolClosedErr %p is wrong", conf.newPoolClosedErrFunc)
+	}
 }