feat: add 03 builder

Author: mohuishou

03_builder/builder.go

@@ -0,0 +1,96 @@
+package builder
+
+import "fmt"
+
+const (
+	defaultMaxTotal = 10
+	defaultMaxIdle  = 9
+	defaultMinIdle  = 1
+)
+
+// ResourcePoolConfig resource pool
+type ResourcePoolConfig struct {
+	name     string
+	maxTotal int
+	maxIdle  int
+	minIdle  int
+}
+
+// ResourcePoolConfigBuilder 用于构建 ResourcePoolConfig
+type ResourcePoolConfigBuilder struct {
+	name     string
+	maxTotal int
+	maxIdle  int
+	minIdle  int
+}
+
+// SetName SetName
+func (b *ResourcePoolConfigBuilder) SetName(name string) error {
+	if name == "" {
+		return fmt.Errorf("name can not be empty")
+	}
+	b.name = name
+	return nil
+}
+
+// SetMinIdle SetMinIdle
+func (b *ResourcePoolConfigBuilder) SetMinIdle(minIdle int) error {
+	if minIdle < 0 {
+		return fmt.Errorf("max tatal cannot < 0, input: %d", minIdle)
+	}
+	b.minIdle = minIdle
+	return nil
+}
+
+// SetMaxIdle SetMaxIdle
+func (b *ResourcePoolConfigBuilder) SetMaxIdle(maxIdle int) error {
+	if maxIdle < 0 {
+		return fmt.Errorf("max tatal cannot < 0, input: %d", maxIdle)
+	}
+	b.maxIdle = maxIdle
+	return nil
+}
+
+// SetMaxTotal SetMaxTotal
+func (b *ResourcePoolConfigBuilder) SetMaxTotal(maxTotal int) error {
+	if maxTotal <= 0 {
+		return fmt.Errorf("max tatal cannot <= 0, input: %d", maxTotal)
+	}
+	b.maxTotal = maxTotal
+	return nil
+}
+
+// Build Build
+func (b *ResourcePoolConfigBuilder) Build() (*ResourcePoolConfig, error) {
+	if b.name == "" {
+		return nil, fmt.Errorf("name can not be empty")
+	}
+
+	// 设置默认值
+	if b.minIdle == 0 {
+		b.minIdle = defaultMinIdle
+	}
+
+	if b.maxIdle == 0 {
+		b.maxIdle = defaultMaxIdle
+	}
+
+	if b.maxTotal == 0 {
+		b.maxTotal = defaultMaxTotal
+	}
+
+	if b.maxTotal < b.maxIdle {
+		return nil, fmt.Errorf("max total(%d) cannot < max idle(%d)", b.maxTotal, b.maxIdle)
+	}
+
+	if b.minIdle > b.maxIdle {
+		return nil, fmt.Errorf("max idle(%d) cannot < min idle(%d)", b.maxIdle, b.minIdle)
+	}
+
+	return &ResourcePoolConfig{
+		name:     b.name,
+		maxTotal: b.maxTotal,
+		maxIdle:  b.maxIdle,
+		minIdle:  b.minIdle,
+	}, nil
+}

03_builder/builder_test.go

@@ -0,0 +1,58 @@
+package builder
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestResourcePoolConfigBuilder_Build(t *testing.T) {
+	tests := []struct {
+		name    string
+		builder *ResourcePoolConfigBuilder
+		want    *ResourcePoolConfig
+		wantErr bool
+	}{
+		{
+			name: "name empty",
+			builder: &ResourcePoolConfigBuilder{
+				name:     "",
+				maxTotal: 0,
+			},
+			want:    nil,
+			wantErr: true,
+		},
+		{
+			name: "maxIdle < minIdle",
+			builder: &ResourcePoolConfigBuilder{
+				name:     "test",
+				maxTotal: 0,
+				maxIdle:  10,
+				minIdle:  20,
+			},
+			want:    nil,
+			wantErr: true,
+		},
+		{
+			name: "success",
+			builder: &ResourcePoolConfigBuilder{
+				name: "test",
+			},
+			want: &ResourcePoolConfig{
+				name:     "test",
+				maxTotal: defaultMaxTotal,
+				maxIdle:  defaultMaxIdle,
+				minIdle:  defaultMinIdle,
+			},
+			wantErr: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := tt.builder.Build()
+			require.Equalf(t, tt.wantErr, err != nil, "Build() error = %v, wantErr %v", err, tt.wantErr)
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}

03_builder/option.go

@@ -0,0 +1,45 @@
+package builder
+
+import "fmt"
+
+// ResourcePoolConfigOption option
+type ResourcePoolConfigOption struct {
+	maxTotal int
+	maxIdle  int
+	minIdle  int
+}
+
+// ResourcePoolConfigOptFunc to set option
+type ResourcePoolConfigOptFunc func(option *ResourcePoolConfigOption)
+
+// NewResourcePoolConfig NewResourcePoolConfig
+func NewResourcePoolConfig(name string, opts ...ResourcePoolConfigOptFunc) (*ResourcePoolConfig, error) {
+	if name == "" {
+		return nil, fmt.Errorf("name can not be empty")
+	}
+
+	option := &ResourcePoolConfigOption{
+		maxTotal: 10,
+		maxIdle:  9,
+		minIdle:  1,
+	}
+
+	for _, opt := range opts {
+		opt(option)
+	}
+
+	if option.maxTotal < 0 || option.maxIdle < 0 || option.minIdle < 0 {
+		return nil, fmt.Errorf("args err, option: %v", option)
+	}
+
+	if option.maxTotal < option.maxIdle || option.minIdle > option.maxIdle {
+		return nil, fmt.Errorf("args err, option: %v", option)
+	}
+
+	return &ResourcePoolConfig{
+		name:     name,
+		maxTotal: option.maxTotal,
+		maxIdle:  option.maxIdle,
+		minIdle:  option.minIdle,
+	}, nil
+}

03_builder/option_test.go

@@ -0,0 +1,55 @@
+package builder
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestNewResourcePoolConfig(t *testing.T) {
+	type args struct {
+		name string
+		opts []ResourcePoolConfigOptFunc
+	}
+	tests := []struct {
+		name    string
+		args    args
+		want    *ResourcePoolConfig
+		wantErr bool
+	}{
+		{
+			name: "name empty",
+			args: args{
+				name: "",
+			},
+			want:    nil,
+			wantErr: true,
+		},
+		{
+			name: "success",
+			args: args{
+				name: "test",
+				opts: []ResourcePoolConfigOptFunc{
+					func(option *ResourcePoolConfigOption) {
+						option.minIdle = 2
+					},
+				},
+			},
+			want: &ResourcePoolConfig{
+				name:     "test",
+				maxTotal: 10,
+				maxIdle:  9,
+				minIdle:  2,
+			},
+			wantErr: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := NewResourcePoolConfig(tt.args.name, tt.args.opts...)
+			require.Equalf(t, tt.wantErr, err != nil, "error = %v, wantErr %v", err, tt.wantErr)
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}

readme.md

@@ -20,7 +20,7 @@ go 设计模式实现，包含 23 种常见的设计模式实现
   - [工厂方法](./02_factory/022_factory_method)
   - [抽象工厂(不常用)](./02_factory/023_abstract_factory)
   - [DI 容器](./02_factory/024_di)
-- 建造者模式
+- [建造者模式](./03_builder) [![](https://img.shields.io/badge/BLOG-%E7%82%B9%E5%87%BB%E6%9F%A5%E7%9C%8B-success?style=flat&cacheSeconds=360000)](https://lailin.xyz/post/builder.html)
 
 ### 不常用