Merge pull request #27 from askuy/feature/askuyoptimize

governor
gotomicro · Dec 18, 2020 · 241ce81 · 241ce81
2 parents 64b6a8e + 005ce1c
commit 241ce81
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/server/egovernor/container.go b/server/egovernor/container.go
@@ -6,6 +6,8 @@ import (
 	"github.com/gotomicro/ego/core/elog"
 )
 
+type Option func(c *Container)
+
 type Container struct {
 	config *Config
 	name   string
@@ -35,6 +37,21 @@ func Load(key string) *Container {
 	return c
 }
 
-func (c *Container) Build() *Component {
+func WithHost(host string) Option {
+	return func(c *Container) {
+		c.config.Host = host
+	}
+}
+
+func WithPort(port int) Option {
+	return func(c *Container) {
+		c.config.Port = port
+	}
+}
+
+func (c *Container) Build(options ...Option) *Component {
+	for _, option := range options {
+		option(c)
+	}
 	return newComponent(c.name, c.config, c.logger)
 }