package fgin

import (
	"fmt"

	"github.com/hashicorp/consul/api"
)

func initConsul() {
	if fginConfig.Consul == (Consul{}) {
		// 没有服务注册中心
		return
	}
	consulConfig := api.DefaultConfig()
	consulConfig.Address = fginConfig.Consul.Address
	consulConfig.Token = fginConfig.Consul.Token
	consulClient, err := api.NewClient(consulConfig)
	if err != nil {
		panic(err)
	}
	// 创建服务实例
	service := &api.AgentServiceRegistration{
		ID:      fginConfig.ServiceName,
		Name:    fginConfig.ServiceName,
		Tags:    []string{fginConfig.Tag},
		Port:    fginConfig.Port,
		Address: fginConfig.Ip,
		Check: &api.AgentServiceCheck{
			Interval:                       "5s",
			Timeout:                        "2s",
			GRPC:                           fmt.Sprintf("%v:%v", fginConfig.Ip, fginConfig.Port),
			DeregisterCriticalServiceAfter: "30s", // 如果服务失败，30s后自动删除
		},
	}

	// 注册服务到 Consul
	err = consulClient.Agent().ServiceRegister(service)
	if err != nil {
		panic(err)
	}
}