import (
    "github.com/baowk/dilu-core/core"
    "github.com/baowk/dilu-rd/adapter"
)

// 创建注册中心客户端
cfg := adapter.DefaultConfig("consul", []string{"localhost:8500"})
registry, err := adapter.NewDiluRegistry(cfg)
if err != nil {
    log.Fatal(err)
}

// 注入 dilu-core，启动时自动注册，关闭时自动注销
core.GetApp().SetRegistry(registry)

import (
    "github.com/baowk/dilu-rd/config"
    "github.com/baowk/dilu-rd/rd"
)

cfg := &config.Config{
    Enable:    true,
    Driver:    "consul", // 或 "etcd"
    Endpoints: []string{"localhost:8500"},
    Scheme:    "http",
    Timeout:   10 * time.Second,
    Registers: []*config.RegisterNode{
        {
            Name:        "my-service",
            Addr:        "192.168.1.100",
            Port:        8080,
            Protocol:    "http",
            HealthCheck: "http://192.168.1.100:8080/health",
            Tags:        []string{"dev"},
        },
    },
    Discoveries: []*config.DiscoveryNode{
        {
            Enable:              true,
            Name:                "my-service",
            SchedulingAlgorithm: "robin", // robin | random
        },
    },
}

client, err := rd.NewRDClient(cfg)
if err != nil {
    log.Fatal(err)
}
defer client.Deregister()

// 获取可用服务节点
node, err := client.GetService("my-service", clientIP)
fmt.Println(node.GetUrl()) // http://192.168.1.100:8080