go-base/consul/register_consul.go

// consul/register.go
package consul

import (
	"fmt"
	"log"
	"time"

	"git.x2erp.com/qdy/go-base/client"
	"git.x2erp.com/qdy/go-base/config"
	"git.x2erp.com/qdy/go-base/config/subconfigs"

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

type ConsulFactory struct {
	config config.IConfig
	ip     string
}

// Register 注册服务到Consul
func Register(cfg config.IConfig) *ConsulFactory {

	serviceName := cfg.GetServiceConfig().ServiceName
	port := cfg.GetServiceConfig().Port
	consulConfig := cfg.GetConsulConfig()
	configHostname := client.GetServiceIP("")

	if consulConfig == nil || !consulConfig.IsConfigured() {
		log.Println("未配置consul注册发现中心，微服务按单体服务运行.")
		return &ConsulFactory{
			config: cfg,
			ip:     configHostname,
		}
	}

	// 1. 获取IP
	ip := client.GetServiceIP(configHostname)
	log.Printf("注册服务 %s -> %s:%d 到 %s", serviceName, ip, port, consulConfig.Address)

	// 2. 创建Consul客户端配置
	apiConfig := api.DefaultConfig()
	if consulConfig.Address != "" {
		apiConfig.Address = consulConfig.Address
		log.Printf("使用Consul地址: %s", consulConfig.Address)
	} else {
		log.Printf("⚠️ 使用默认Consul地址: %s", apiConfig.Address)
	}

	if consulConfig.Token != "" {
		apiConfig.Token = consulConfig.Token
	}
	if consulConfig.Scheme != "" {
		apiConfig.Scheme = consulConfig.Scheme
	}

	// 3. 创建客户端
	client, err := api.NewClient(apiConfig)
	if err != nil {
		log.Fatalf("创建Consul客户端失败: %v", err)
	}

	// 4. 测试连接
	_, err = client.Agent().Self()
	if err != nil {
		log.Fatalf("无法连接到Consul服务器 %s: %v", apiConfig.Address, err)
	}
	log.Printf("✅ 成功连接到Consul服务器: %s", apiConfig.Address)

	// 5. 注册服务（完整健康检查配置）
	registration := &api.AgentServiceRegistration{
		ID:      fmt.Sprintf("%s-%s-%d", serviceName, ip, port),
		Name:    serviceName,
		Address: ip,
		Port:    port,
		Tags:    []string{"http", "microservice"},
		Check: &api.AgentServiceCheck{
			HTTP:                           fmt.Sprintf("http://%s:%d/health", ip, port),
			Interval:                       "10s",
			Timeout:                        "5s",      // 关键：设置超时
			DeregisterCriticalServiceAfter: "1m",      // 关键：失败后注销时间
			TLSSkipVerify:                  true,      // 跳过TLS验证
			Status:                         "passing", // 初始状态
		},
		Meta: map[string]string{
			"registered_at": time.Now().Format(time.RFC3339),
			"port":          fmt.Sprintf("%d", port),
		},
	}

	err = client.Agent().ServiceRegister(registration)
	if err != nil {
		log.Fatalf("consul注册失败: %v", err)
	}

	// 打印详细注册信息
	//log.Printf("✅ 服务已注册到Consul服务器: %s", consulConfig.Address)
	log.Printf("📋注册详情:")
	log.Printf("-服务名称: %s", serviceName)
	log.Printf("-服务地址: %s:%d", ip, port)
	log.Printf("-服务ID: %s", registration.ID)
	log.Printf("-健康检查: %s", registration.Check.HTTP)
	log.Printf("-检查间隔: %s", registration.Check.Interval)
	log.Printf("-检查超时: %s", registration.Check.Timeout)
	log.Printf("-失败注销: %s", registration.Check.DeregisterCriticalServiceAfter)

	return &ConsulFactory{
		config: cfg,
		ip:     configHostname,
	}
}

// CheckRegistration 检查服务是否已注册
func CheckRegistration(serviceName, configHostname string, port int, consulConfig *subconfigs.ConsulConfig) error {
	if consulConfig == nil || !consulConfig.IsConfigured() {
		log.Println("未配置consul注册发现中心，微服务按单体服务运行.")
		return nil
	}
	ip := client.GetServiceIP(configHostname)
	serviceID := fmt.Sprintf("%s-%s-%d", serviceName, ip, port)

	// 创建客户端
	apiConfig := api.DefaultConfig()
	if consulConfig.Address != "" {
		apiConfig.Address = consulConfig.Address
	}
	client, err := api.NewClient(apiConfig)
	if err != nil {
		return fmt.Errorf("创建Consul客户端失败: %v", err)
	}

	// 检查服务
	services, err := client.Agent().Services()
	if err != nil {
		return fmt.Errorf("获取服务列表失败: %v", err)
	}

	if svc, exists := services[serviceID]; exists {
		log.Printf("✅ 服务已注册:")
		log.Printf("   服务ID: %s", svc.ID)
		log.Printf("   服务名: %s", svc.Service)
		log.Printf("   地址: %s:%d", svc.Address, svc.Port)

		// 检查健康状态
		checks, err := client.Agent().Checks()
		if err == nil {
			for checkID, check := range checks {
				if check.ServiceID == serviceID {
					log.Printf("   健康检查[%s]: %s", checkID, check.Status)
				}
			}
		}
		return nil
	}

	return fmt.Errorf("服务未注册: %s", serviceID)
}

func (c *ConsulFactory) GetName() string {
	return "ConsulFactory"
}

// Close 注销服务
func (c *ConsulFactory) Close() {

	serviceName := c.config.GetServiceConfig().ServiceName
	port := c.config.GetServiceConfig().Port
	consulConfig := c.config.GetConsulConfig()

	if consulConfig == nil || !consulConfig.IsConfigured() {
		log.Println("未配置consul注册发现中心,服务不用到注册中心进行注销.")

	}

	serviceID := fmt.Sprintf("%s-%s-%d", serviceName, c.ip, port)

	// 创建客户端
	apiConfig := api.DefaultConfig()
	if consulConfig != nil && consulConfig.Address != "" {
		apiConfig.Address = consulConfig.Address
	}
	client, err := api.NewClient(apiConfig)
	if err != nil {
		log.Printf("创建consul客户端失败: %v", err)
		return
	}

	// 注销服务
	err = client.Agent().ServiceDeregister(serviceID)
	if err != nil {

		log.Printf("从consul注销服务失败 (ID: %s): %v", serviceID, err)
		return
	}

	log.Printf("✅ 已从consul注销服务: %s", serviceID)
	return
}