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/subconfigs"
	"github.com/hashicorp/consul/api"
)

// Register 注册服务到Consul
func Register(serviceName, configHostname string, port int, consulConfig *subconfigs.ConsulConfig) error {
	// 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 {
		return fmt.Errorf("创建Consul客户端失败: %v", err)
	}

	// 4. 测试连接
	_, err = client.Agent().Self()
	if err != nil {
		return fmt.Errorf("无法连接到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 {
		return fmt.Errorf("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 nil
}

// CheckRegistration 检查服务是否已注册
func CheckRegistration(serviceName, configHostname string, port int, consulConfig *subconfigs.ConsulConfig) error {
	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)
}

// Deregister 注销服务
func Deregister(serviceName, ip string, port int, consulConfig *subconfigs.ConsulConfig) error {
	serviceID := fmt.Sprintf("%s-%s-%d", serviceName, ip, port)

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

	// 注销服务
	err = client.Agent().ServiceDeregister(serviceID)
	if err != nil {
		return fmt.Errorf("从consul注销服务失败 (ID: %s): %v", serviceID, err)
	}

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