服务发现

package util

import (
    "context"
    "go.etcd.io/etcd/clientv3"
    "regexp"
    "time"
)

type Client struct {
    client   *clientv3.Client
    Services []*ServiceInfo
}

type ServiceInfo struct {
    ServiceID   string
    ServiceName string
    ServiceAddr string
}

func NewClient() *Client {
    config := clientv3.Config{
        Endpoints:   []string{"106.12.72.181:23791", "106.12.72.181:23792"},
        DialTimeout: 10 * time.Second,
    }
    client, _ := clientv3.New(config)
    return &Client{client: client}
}

func (this *Client) GetService() {
    kv := clientv3.NewKV(this.client)
    res, _ := kv.Get(context.TODO(), "/services", clientv3.WithPrefix())
    for _, item := range res.Kvs {
        this.parseService(item.Key, item.Value)
    }
}

func (this *Client) LoadService() error {
    kv := clientv3.NewKV(this.client)
    res, err := kv.Get(context.TODO(), "/services", clientv3.WithPrefix())
    if err != nil {
        return err
    }
    for _, item := range res.Kvs {
        this.parseService(item.Key, item.Value)
    }
    return nil
}
func (this *Client) parseService(key []byte, value []byte) {
    reg := regexp.MustCompile("/services/(\w+)/(\w+)")
    if reg.Match(key) {
        idandname := reg.FindSubmatch(key) //idandname是一个二维切片
         sid := idandname[1] //sid是一维切片，用string(sid)可以转成string
        sname := idandname[2]
        this.Services = append(this.Services, &ServiceInfo{ServiceID: string(sid),
            ServiceName: string(sname), ServiceAddr: string(value)})
    }
}

来自为知笔记(Wiz)