41 lines
1.1 KiB
Go
41 lines
1.1 KiB
Go
package gap
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/hashicorp/consul/api"
|
|
"github.com/spf13/viper"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func Register() error {
|
|
cfg := api.DefaultConfig()
|
|
cfg.Address = viper.GetString("consul.addr")
|
|
|
|
client, err := api.NewClient(cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
httpBind := strings.SplitN(viper.GetString("bind"), ":", 2)
|
|
grpcBind := strings.SplitN(viper.GetString("grpc_bind"), ":", 2)
|
|
|
|
outboundIp, _ := GetOutboundIP()
|
|
port, _ := strconv.Atoi(httpBind[1])
|
|
|
|
registration := new(api.AgentServiceRegistration)
|
|
registration.ID = viper.GetString("id")
|
|
registration.Name = "Hydrogen.Passport"
|
|
registration.Address = outboundIp.String()
|
|
registration.Port = port
|
|
registration.Check = &api.AgentServiceCheck{
|
|
GRPC: fmt.Sprintf("%s:%s", outboundIp, grpcBind[1]),
|
|
HTTP: fmt.Sprintf("%s:%s/.well-known", outboundIp, httpBind[1]),
|
|
Timeout: "5s",
|
|
Interval: "1m",
|
|
DeregisterCriticalServiceAfter: "3m",
|
|
}
|
|
|
|
return client.Agent().ServiceRegister(registration)
|
|
}
|