1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102
| package main
import ( "context" "log" "time"
"go.etcd.io/etcd/clientv3" )
type ServiceRegister struct { cli *clientv3.Client leaseID clientv3.LeaseID keepAliveChan <-chan *clientv3.LeaseKeepAliveResponse key string val string }
func NewServiceRegister(endpoints []string, key, val string, lease int64) (*ServiceRegister, error) { cli, err := clientv3.New(clientv3.Config{ Endpoints: endpoints, DialTimeout: 5 * time.Second, }) if err != nil { log.Fatal(err) }
ser := &ServiceRegister{ cli: cli, key: key, val: val, }
if err := ser.putKeyWithLease(lease); err != nil { return nil, err }
return ser, nil }
func (s *ServiceRegister) putKeyWithLease(lease int64) error { resp, err := s.cli.Grant(context.Background(), lease) if err != nil { return err } _, err = s.cli.Put(context.Background(), s.key, s.val, clientv3.WithLease(resp.ID)) if err != nil { return err } leaseRespChan, err := s.cli.KeepAlive(context.Background(), resp.ID)
if err != nil { return err } s.leaseID = resp.ID log.Println(s.leaseID) s.keepAliveChan = leaseRespChan log.Printf("Put key:%s val:%s success!", s.key, s.val) return nil }
func (s *ServiceRegister) ListenLeaseRespChan() { for leaseKeepResp := range s.keepAliveChan { log.Println("续约成功", leaseKeepResp) } log.Println("关闭续租") }
func (s *ServiceRegister) Close() error { if _, err := s.cli.Revoke(context.Background(), s.leaseID); err != nil { return err } log.Println("撤销租约") return s.cli.Close() }
func main() { var endpoints = []string{"localhost:2379"} ser, err := NewServiceRegister(endpoints, "/web/node1", "localhost:8000", 5) if err != nil { log.Fatalln(err) } go ser.ListenLeaseRespChan() select { } }
|