Merge pull request #9 from L2ncE/feat/add_option_for_register_with_co…

…nfig

feat: add option for `NewConsulRegisterWithConfig`

consul_registry.go

@@ -69,13 +69,21 @@ func NewConsulRegister(address string, opts ...Option) (registry.Registry, error
 }
 
 // NewConsulRegisterWithConfig create a new registry using consul, with a custom config.
-func NewConsulRegisterWithConfig(config *api.Config) (*consulRegistry, error) {
+func NewConsulRegisterWithConfig(config *api.Config, opts ...Option) (*consulRegistry, error) {
 	client, err := api.NewClient(config)
 	if err != nil {
 		return nil, err
 	}
 
-	return &consulRegistry{consulClient: client}, nil
+	op := options{
+		check: defaultCheck(),
+	}
+
+	for _, option := range opts {
+		option(&op)
+	}
+
+	return &consulRegistry{consulClient: client, opts: op}, nil
 }
 
 // Register register a service to consul.

consul_test.go

@@ -81,7 +81,11 @@ func TestNewConsulRegisterWithConfig(t *testing.T) {
 		Address:   consulAddr,
 		WaitTime:  5 * time.Second,
 		Namespace: "TEST-NS",
-	})
+	}, WithCheck(&consulapi.AgentServiceCheck{
+		Interval:                       "7s",
+		Timeout:                        "5s",
+		DeregisterCriticalServiceAfter: "15s",
+	}))
 	assert.NoError(t, err)
 }
 

example/custom-config/server/main.go

@@ -43,7 +43,11 @@ func main() {
 		Address: "127.0.0.1:8500",
 		Token:   "TEST-MY-TOKEN",
 	}
-	r, err := consul.NewConsulRegisterWithConfig(&consulConfig)
+	r, err := consul.NewConsulRegisterWithConfig(&consulConfig, consul.WithCheck(&consulapi.AgentServiceCheck{
+		Interval:                       "7s",
+		Timeout:                        "5s",
+		DeregisterCriticalServiceAfter: "15s",
+	}))
 	if err != nil {
 		log.Fatal(err)
 	}