go-base/consul/register_consul.go

// consul/register.go
package consul

import (
	"fmt"
	"log"

	"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
	}
	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. 注册服务
	registration := &api.AgentServiceRegistration{
		ID:      fmt.Sprintf("%s-%s-%d", serviceName, ip, port),
		Name:    serviceName,
		Address: ip,
		Port:    port,
		Check: &api.AgentServiceCheck{
			HTTP:     fmt.Sprintf("http://%s:%d/health", ip, port),
			Interval: "10s",
		},
	}

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

	log.Printf("✅ 服务已注册到Consul服务器: %s", consulConfig.Address)
	return nil
}

// Deregister 注销服务
// Deregister 注销服务
func Deregister(serviceName, ip string, port int) {
	// 获取IP
	//ip := netutil.GetServiceIP(configHostname)
	serviceID := fmt.Sprintf("%s-%s-%d", serviceName, ip, port)

	// 创建客户端
	client, err := api.NewClient(api.DefaultConfig())
	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)
}