一文了解 etcd watch 實現

羣裏討論服務發現,最簡單的模型使用 etcd watch 來實現消息訂閱,我司也是這方案。但從可運維及穩定性考慮,etcd 並不是最佳選擇,好多小夥伴所在的公司仍然採用 mysql 的推拉結合方式。這讓我想起來,很多人面試問如何實現延遲隊列,花裏胡哨一大堆解決方案,可能公司最後還是使用 mysql 輪詢 ...

扯遠了,來看一下 etcd 如何實現的 watch, 原理不算複雜,但要處理很多場景,比如不同 client 消費速度不同的問題,處理如何高效匹配 range 範圍等等。下面是整體架構圖

經典使用

先看一下經典的使用 case, 代碼做了簡化,網上有很多人說 watch 丟消息,基本就是姿勢錯誤導致的。

func watch(ctx context.Context, revision int64) {
 ......
 for {
  rch := watcher.Watch(ctx, "/somepath", clientv3.WithRev(revision), clientv3.WithPrefix())
  for wresp := range rch {
   // meet compacted error, use the compact revision.
   if wresp.CompactRevision != 0 {
    logging.Warn("required revision has been compacted, use the compact revision:%d, required-revision:%d", wresp.CompactRevision, revision)
    revision = wresp.CompactRevision
    break
   }
   if wresp.Canceled {
    logging.Warn("watcher is canceled with revision: %d error: %v", revision, wresp.Err())
    return
   }

   for _, ev := range wresp.Events {
    process(ev)
   }
   revision = wresp.Header.Revision
  }

  select {
  case <-ctx.Done():
   // server closed, return
   return
  default:
  }
 }
 ......
}

上面代碼是經典的使用方式,算是標準實踐,要注意處理 compact revision,  處理 ctx 超時,以及上層的 rewatch. 同時 watch 時可以指定 option, 接受歷史版本數據,訂閱一個範圍前輟等等

服務初始化

etcd 啓動時會註冊 WatchServer[1], pb.WatchServer 用於處理 watch 請求

// NewWatchServer returns a new watch server.
func NewWatchServer(s *etcdserver.EtcdServer) pb.WatchServer {
 return &watchServer{
  lg: s.Cfg.Logger,
  clusterID: int64(s.Cluster().ID()),
  memberID:  int64(s.ID()),
  maxRequestBytes: int(s.Cfg.MaxRequestBytes + grpcOverheadBytes),
  sg:        s,
  watchable: s.Watchable(),
  ag:        s,
 }
}

這裏面關注 watchable 就行,是一個接口,實際上實現是 mvcc.watchableStore

接收 watch 請求

func (ws *watchServer) Watch(stream pb.Watch_WatchServer) (err error) {
 sws := serverWatchStream{
  lg: ws.lg,

  clusterID: ws.clusterID,
  memberID:  ws.memberID,

  maxRequestBytes: ws.maxRequestBytes,

  sg:        ws.sg,
  watchable: ws.watchable,
  ag:        ws.ag,

  gRPCStream:  stream,
  watchStream: ws.watchable.NewWatchStream(),
  // chan for sending control response like watcher created and canceled.
  ctrlStream: make(chan *pb.WatchResponse, ctrlStreamBufLen),

  progress: make(map[mvcc.WatchID]bool),
  prevKV:   make(map[mvcc.WatchID]bool),
  fragment: make(map[mvcc.WatchID]bool),

  closec: make(chan struct{}),
 }

 sws.wg.Add(1)
 go func() {
  sws.sendLoop()
  sws.wg.Done()
 }()

 errc := make(chan error, 1)
 // Ideally recvLoop would also use sws.wg to signal its completion
 // but when stream.Context().Done() is closed, the stream's recv
 // may continue to block since it uses a different context, leading to
 // deadlock when calling sws.close().
 go func() {
  if rerr := sws.recvLoop(); rerr != nil {
   if isClientCtxErr(stream.Context().Err(), rerr) {
    if sws.lg != nil {
     sws.lg.Debug("failed to receive watch request from gRPC stream", zap.Error(rerr))
    } else {
     plog.Debugf("failed to receive watch request from gRPC stream (%q)", rerr.Error())
    }
   } else {
    if sws.lg != nil {
     sws.lg.Warn("failed to receive watch request from gRPC stream", zap.Error(rerr))
    } else {
     plog.Warningf("failed to receive watch request from gRPC stream (%q)", rerr.Error())
    }
    streamFailures.WithLabelValues("receive""watch").Inc()
   }
   errc <- rerr
  }
 }()

 select {
 case err = <-errc:
  close(sws.ctrlStream)

 case <-stream.Context().Done():
  err = stream.Context().Err()
  // the only server-side cancellation is noleader for now.
  if err == context.Canceled {
   err = rpctypes.ErrGRPCNoLeader
  }
 }

 sws.close()
 return err
}
  1. 每一個 watch 流都創建一個 serverWatchStream 結構體

  2. 開啓兩個 goroutine, sendLoop 用於發送 watch 消息到流中,recvLoop 接受請求

  3. select 阻塞直到流關閉,或是超時退出。

1. 接收 watch 請求 recvLoop

func (sws *serverWatchStream) recvLoop() error {
 for {
  req, err := sws.gRPCStream.Recv()
    ......
  switch uv := req.RequestUnion.(type) {
  case *pb.WatchRequest_CreateRequest:
   creq := uv.CreateRequest
   if len(creq.Key) == 0 {
    // \x00 is the smallest key
    creq.Key = []byte{0}
   }
   if len(creq.RangeEnd) == 0 {
    // force nil since watchstream.Watch distinguishes
    // between nil and []byte{} for single key / >=
    creq.RangeEnd = nil
   }
   if len(creq.RangeEnd) == 1 && creq.RangeEnd[0] == 0 {
    // support  >= key queries
    creq.RangeEnd = []byte{}
   }

   if !sws.isWatchPermitted(creq) {
    wr := &pb.WatchResponse{
     Header:       sws.newResponseHeader(sws.watchStream.Rev()),
     WatchId:      creq.WatchId,
     Canceled:     true,
     Created:      true,
     CancelReason: rpctypes.ErrGRPCPermissionDenied.Error(),
    }

    select {
    case sws.ctrlStream <- wr:
    case <-sws.closec:
    }
    return nil
   }

   filters := FiltersFromRequest(creq)

   wsrev := sws.watchStream.Rev()
   rev := creq.StartRevision
   if rev == 0 {
    rev = wsrev + 1
   }
   id, err := sws.watchStream.Watch(mvcc.WatchID(creq.WatchId), creq.Key, creq.RangeEnd, rev, filters...)
   if err == nil {
    sws.mu.Lock()
    if creq.ProgressNotify {
     sws.progress[id] = true
    }
    if creq.PrevKv {
     sws.prevKV[id] = true
    }
    if creq.Fragment {
     sws.fragment[id] = true
    }
    sws.mu.Unlock()
   }
   wr := &pb.WatchResponse{
    Header:   sws.newResponseHeader(wsrev),
    WatchId:  int64(id),
    Created:  true,
    Canceled: err != nil,
   }
   if err != nil {
    wr.CancelReason = err.Error()
   }
   select {
   case sws.ctrlStream <- wr:
   case <-sws.closec:
    return nil
   }

  case *pb.WatchRequest_CancelRequest:
   if uv.CancelRequest != nil {
    id := uv.CancelRequest.WatchId
    err := sws.watchStream.Cancel(mvcc.WatchID(id))
    if err == nil {
     sws.ctrlStream <- &pb.WatchResponse{
      Header:   sws.newResponseHeader(sws.watchStream.Rev()),
      WatchId:  id,
      Canceled: true,
     }
     sws.mu.Lock()
     delete(sws.progress, mvcc.WatchID(id))
     delete(sws.prevKV, mvcc.WatchID(id))
     delete(sws.fragment, mvcc.WatchID(id))
     sws.mu.Unlock()
    }
   }
  case *pb.WatchRequest_ProgressRequest:
   if uv.ProgressRequest != nil {
    sws.ctrlStream <- &pb.WatchResponse{
     Header:  sws.newResponseHeader(sws.watchStream.Rev()),
     WatchId: -1, // response is not associated with any WatchId and will be broadcast to all watch channels
    }
   }
  default:
   // we probably should not shutdown the entire stream when
   // receive an valid command.
   // so just do nothing instead.
   continue
  }
 }
}

recvLoopgRPCStream 讀出 req, 然後分別處理類型爲 CreateRequest, CancelRequest, ProgressRequest 的情況

  1. CreateRequest: 監聽的可能是一個範圍,所以構建 key 和 RangeEnd. 處理 StartRevision, 如果爲 0, 那麼使用當前系統最新的 Rev+1. 調用 mvcc 層的 watchStream.Watch, 返回一個 watchid, 將這個 id 寫到 ctrlStream 返回給 client

  2. CancelRequest: 還是調用 mvcc 層的 watchableStore.Cancel 取消訂閱,然後清除狀態信息

  3. ProgressRequest: broadcast 廣播當前系統的 Rev 版本

2. 接收 watch 請求 sendLoop

func (sws *serverWatchStream) sendLoop() {
 // watch ids that are currently active
 ids := make(map[mvcc.WatchID]struct{})
 // watch responses pending on a watch id creation message
 pending := make(map[mvcc.WatchID][]*pb.WatchResponse)

 interval := GetProgressReportInterval()
 progressTicker := time.NewTicker(interval)

 defer func() {
  progressTicker.Stop()
  // drain the chan to clean up pending events
  for ws := range sws.watchStream.Chan() {
   mvcc.ReportEventReceived(len(ws.Events))
  }
  for _, wrs := range pending {
   for _, ws := range wrs {
    mvcc.ReportEventReceived(len(ws.Events))
   }
  }
 }()

 for {
  select {
  case wresp, ok := <-sws.watchStream.Chan():
   if !ok {
    return
   }

   // TODO: evs is []mvccpb.Event type
   // either return []*mvccpb.Event from the mvcc package
   // or define protocol buffer with []mvccpb.Event.
   evs := wresp.Events
   events := make([]*mvccpb.Event, len(evs))
   sws.mu.RLock()
   needPrevKV := sws.prevKV[wresp.WatchID]
   sws.mu.RUnlock()
   for i := range evs {
    events[i] = &evs[i]
    if needPrevKV {
     opt := mvcc.RangeOptions{Rev: evs[i].Kv.ModRevision - 1}
     r, err := sws.watchable.Range(evs[i].Kv.Key, nil, opt)
     if err == nil && len(r.KVs) != 0 {
      events[i].PrevKv = &(r.KVs[0])
     }
    }
   }

   canceled := wresp.CompactRevision != 0
   wr := &pb.WatchResponse{
    Header:          sws.newResponseHeader(wresp.Revision),
    WatchId:         int64(wresp.WatchID),
    Events:          events,
    CompactRevision: wresp.CompactRevision,
    Canceled:        canceled,
   }

   if _, okID := ids[wresp.WatchID]; !okID {
    // buffer if id not yet announced
    wrs := append(pending[wresp.WatchID], wr)
    pending[wresp.WatchID] = wrs
    continue
   }

   mvcc.ReportEventReceived(len(evs))

   sws.mu.RLock()
   fragmented, ok := sws.fragment[wresp.WatchID]
   sws.mu.RUnlock()

   var serr error
   if !fragmented && !ok {
    serr = sws.gRPCStream.Send(wr)
   } else {
    serr = sendFragments(wr, sws.maxRequestBytes, sws.gRPCStream.Send)
   }

   ......
   sws.mu.Lock()
   if len(evs) > 0 && sws.progress[wresp.WatchID] {
    // elide next progress update if sent a key update
    sws.progress[wresp.WatchID] = false
   }
   sws.mu.Unlock()

  case c, ok := <-sws.ctrlStream:
   if !ok {
    return
   }

   if err := sws.gRPCStream.Send(c); err != nil {
    ......
   }

   // track id creation
   wid := mvcc.WatchID(c.WatchId)
   if c.Canceled {
    delete(ids, wid)
    continue
   }
   if c.Created {
    // flush buffered events
    ids[wid] = struct{}{}
    for _, v := range pending[wid] {
     mvcc.ReportEventReceived(len(v.Events))
     if err := sws.gRPCStream.Send(v); err != nil {
      ......
     }
    }
    delete(pending, wid)
   }

  case <-progressTicker.C:
   sws.mu.Lock()
   for id, ok := range sws.progress {
    if ok {
     sws.watchStream.RequestProgress(id)
    }
    sws.progress[id] = true
   }
   sws.mu.Unlock()

  case <-sws.closec:
   return
  }
 }
}

在 watchid 生成前,可能就有消息觸發了,此時還沒有 id, 所以消息會堆積到 pending 中。整個函數主要從 mvcc.watchStream.Chan() 中處理讀取訂閱的消息,處理 ctrlStream 控制消息和處理 progressTicker

  1. Chan(): 如果 needPrevKV, 需要填充。watchid 不存在的話,暫時移到 pending 隊列中。Fragment 查看是否需要分包,這裏閾值是 1.5M, 不需要的話直接調用 sws.gRPCStream.Send 發送即可。如果有數據發送的情況,sws.progress[wresp.WatchID] 置爲 false, 不用發進度消息

  2. ctrlStream: 讀取控制消息,這裏只要是獲取 watchid, 然後發送堆積的 pending 消息

  3. progressTicker: 定期調用 RequestProgress 生成進度消息,把當前 Rev 發給 client

MVCC watch

這一塊主要是看 mvcc.watchStream, 看下 Watch 如何實現

// Watch creates a new watcher in the stream and returns its WatchID.
func (ws *watchStream) Watch(id WatchID, key, end []byte, startRev int64, fcs ...FilterFunc) (WatchID, error) {
 // prevent wrong range where key >= end lexicographically
 // watch request with 'WithFromKey' has empty-byte range end
 if len(end) != 0 && bytes.Compare(key, end) != -1 {
  return -1, ErrEmptyWatcherRange
 }

 ws.mu.Lock()
 defer ws.mu.Unlock()
 if ws.closed {
  return -1, ErrEmptyWatcherRange
 }

 if id == AutoWatchID {
  for ws.watchers[ws.nextID] != nil {
   ws.nextID++
  }
  id = ws.nextID
  ws.nextID++
 } else if _, ok := ws.watchers[id]; ok {
  return -1, ErrWatcherDuplicateID
 }

 w, c := ws.watchable.watch(key, end, startRev, id, ws.ch, fcs...)

 ws.cancels[id] = c
 ws.watchers[id] = w
 return id, nil
}

主要是用來生成 watchid, 自增就可以了。

func (s *watchableStore) watch(key, end []byte, startRev int64, id WatchID, ch chan<- WatchResponse, fcs ...FilterFunc) (*watcher, cancelFunc) {
 wa := &watcher{
  key:    key,
  end:    end,
  minRev: startRev,
  id:     id,
  ch:     ch,
  fcs:    fcs,
 }

 s.mu.Lock()
 s.revMu.RLock()
 synced := startRev > s.store.currentRev || startRev == 0
 if synced {
  wa.minRev = s.store.currentRev + 1
  if startRev > wa.minRev {
   wa.minRev = startRev
  }
 }
 if synced {
  s.synced.add(wa)
 } else {
  slowWatcherGauge.Inc()
  s.unsynced.add(wa)
 }
 s.revMu.RUnlock()
 s.mu.Unlock()

 watcherGauge.Inc()

 return wa, func() { s.cancelWatcher(wa) }
}

watchableStore 一共有三個 group: synced, unsynced 與 victims, 當 client watch 時是從歷史記錄開始的,也就是說此時有一堆消息待發送給 client, 那麼將 watcher 結構體扔到 unsynced 組中,否則扔到 synced 組中。爲什麼這麼做呢?因爲消息處理有快慢,後面具體代碼再講,只要記住 watcher 會在這三個組中流轉即可,當然理想情況一直待在 synced 組中

func newWatchableStore(lg *zap.Logger, b backend.Backend, le lease.Lessor, ig ConsistentIndexGetter, cfg StoreConfig) *watchableStore {
 s := &watchableStore{
  store:    NewStore(lg, b, le, ig, cfg),
  victimc:  make(chan struct{}, 1),
  unsynced: newWatcherGroup(),
  synced:   newWatcherGroup(),
  stopc:    make(chan struct{}),
 }
 s.store.ReadView = &readView{s}
 s.store.WriteView = &writeView{s}
 if s.le != nil {
  // use this store as the deleter so revokes trigger watch events
  s.le.SetRangeDeleter(func() lease.TxnDelete { return s.Write(traceutil.TODO()) })
 }
 s.wg.Add(2)
 go s.syncWatchersLoop()
 go s.syncVictimsLoop()
 return s
}

newWatchableStore 時,會生成兩個異步 goroutine, syncWatchersLoop 用於將 unsynced 的 watcher 變成 synced watcher, syncVictimsLoop 用於將 victims 的消息儘可能的發送出。

MVCC 消息生成

底層 Txn 用 watchableStoreTxnWrite 封裝了一下,在調用 End 提交事務前,調用 notify 將變更的消息發送出去。

func (tw *watchableStoreTxnWrite) End() {
 changes := tw.Changes()
 if len(changes) == 0 {
  tw.TxnWrite.End()
  return
 }

 rev := tw.Rev() + 1
 evs := make([]mvccpb.Event, len(changes))
 for i, change := range changes {
  evs[i].Kv = &changes[i]
  if change.CreateRevision == 0 {
   evs[i].Type = mvccpb.DELETE
   evs[i].Kv.ModRevision = rev
  } else {
   evs[i].Type = mvccpb.PUT
  }
 }

 // end write txn under watchable store lock so the updates are visible
 // when asynchronous event posting checks the current store revision
 tw.s.mu.Lock()
 tw.s.notify(rev, evs)
 tw.TxnWrite.End()
 tw.s.mu.Unlock()
}

遍歷 changes, 判斷類型 mvccpb.DELETE 或是 mvccpb.PUT, 然後封裝成 envs 事件,調用 tw.s.notify 發送出去後提交。

// notify notifies the fact that given event at the given rev just happened to
// watchers that watch on the key of the event.
func (s *watchableStore) notify(rev int64, evs []mvccpb.Event) {
 var victim watcherBatch
 for w, eb := range newWatcherBatch(&s.synced, evs) {
  if eb.revs != 1 {
   if s.store != nil && s.store.lg != nil {
    s.store.lg.Panic(
     "unexpected multiple revisions in watch notification",
     zap.Int("number-of-revisions", eb.revs),
    )
   } else {
    plog.Panicf("unexpected multiple revisions in notification")
   }
  }
  if w.send(WatchResponse{WatchID: w.id, Events: eb.evs, Revision: rev}) {
   pendingEventsGauge.Add(float64(len(eb.evs)))
  } else {
   // move slow watcher to victims
   w.minRev = rev + 1
   if victim == nil {
    victim = make(watcherBatch)
   }
   w.victim = true
   victim[w] = eb
   s.synced.delete(w)
   slowWatcherGauge.Inc()
  }
 }
 s.addVictim(victim)
}

newWatcherBatch 用於從 synced 組中獲取待發送的 watcher, 然後調用 w.send 發送到 channel 裏面,如果 channel 滿了,那麼說明發送不出去,將 watcher 從 synced 組中刪除,並添加到 victim 組中,等後後續異步 goroutine syncVictimsLoop 處理。我們看一下,newWatcherBatch 實現

func newWatcherBatch(wg *watcherGroup, evs []mvccpb.Event) watcherBatch {
 if len(wg.watchers) == 0 {
  return nil
 }

 wb := make(watcherBatch)
 for _, ev := range evs {
  for w := range wg.watcherSetByKey(string(ev.Kv.Key)) {
   if ev.Kv.ModRevision >= w.minRev {
    // don't double notify
    wb.add(w, ev)
   }
  }
 }
 return wb
}

watcherSetByKey 用於返回滿足 ev.Kv.Key 的 watcher, 這裏內部實現使用 adt 紅黑樹,可以做到快速的範圍匹配。感興趣的可以看源代碼。

func (w *watcher) send(wr WatchResponse) bool {
 progressEvent := len(wr.Events) == 0

 if len(w.fcs) != 0 {
  ne := make([]mvccpb.Event, 0, len(wr.Events))
  for i := range wr.Events {
   filtered := false
   for _, filter := range w.fcs {
    if filter(wr.Events[i]) {
     filtered = true
     break
    }
   }
   if !filtered {
    ne = append(ne, wr.Events[i])
   }
  }
  wr.Events = ne
 }

 // if all events are filtered out, we should send nothing.
 if !progressEvent && len(wr.Events) == 0 {
  return true
 }
 select {
 case w.ch <- wr:
  return true
 default:
  return false
 }
}

send 函數先 apply filter 過濾一遍,然後發送到 w.ch 中,如果滿了則返回 false. 這個 w.ch 就是 v3rpc 使用的 channel, 有數據後就發送 http2 stream ...

慢速處理

1. 慢速處理 victim

func (s *watchableStore) syncVictimsLoop() {
 defer s.wg.Done()

 for {
  for s.moveVictims() != 0 {
   // try to update all victim watchers
  }
  s.mu.RLock()
  isEmpty := len(s.victims) == 0
  s.mu.RUnlock()

  var tickc <-chan time.Time
  if !isEmpty {
   tickc = time.After(10 * time.Millisecond)
  }

  select {
  case <-tickc:
  case <-s.victimc:
  case <-s.stopc:
   return
  }
 }
}

調用 moveVictims 嘗試去發送堆積的消息

// moveVictims tries to update watches with already pending event data
func (s *watchableStore) moveVictims() (moved int) {
 s.mu.Lock()
 victims := s.victims
 s.victims = nil
 s.mu.Unlock()

 var newVictim watcherBatch
 for _, wb := range victims {
  // try to send responses again
  for w, eb := range wb {
   // watcher has observed the store up to, but not including, w.minRev
   rev := w.minRev - 1
   if w.send(WatchResponse{WatchID: w.id, Events: eb.evs, Revision: rev}) {
    pendingEventsGauge.Add(float64(len(eb.evs)))
   } else {
    if newVictim == nil {
     newVictim = make(watcherBatch)
    }
    newVictim[w] = eb
    continue
   }
   moved++
  }

  // assign completed victim watchers to unsync/sync
  s.mu.Lock()
  s.store.revMu.RLock()
  curRev := s.store.currentRev
  for w, eb := range wb {
   if newVictim != nil && newVictim[w] != nil {
    // couldn't send watch response; stays victim
    continue
   }
   w.victim = false
   if eb.moreRev != 0 {
    w.minRev = eb.moreRev
   }
   if w.minRev <= curRev {
    s.unsynced.add(w)
   } else {
    slowWatcherGauge.Dec()
    s.synced.add(w)
   }
  }
  s.store.revMu.RUnlock()
  s.mu.Unlock()
 }

 if len(newVictim) > 0 {
  s.mu.Lock()
  s.victims = append(s.victims, newVictim)
  s.mu.Unlock()
 }

 return moved
}

代碼很簡單,先嚐試發送 victims 這些消息,如果失敗了,再放到 victims 中。成功了的話,還要看當前系統中的 Rev 是否與該 watcher.minRev 相等,再考濾放到 synced 組還是 unsynced 組中。

2. 慢速處理 unsynced

syncWatchersLoop 函數循環調用 syncWatchers 處理 unsynced 組數據

// syncWatchers syncs unsynced watchers by:
// 1. choose a set of watchers from the unsynced watcher group
// 2. iterate over the set to get the minimum revision and remove compacted watchers
// 3. use minimum revision to get all key-value pairs and send those events to watchers
// 4. remove synced watchers in set from unsynced group and move to synced group
func (s *watchableStore) syncWatchers() int {
 s.mu.Lock()
 defer s.mu.Unlock()

 if s.unsynced.size() == 0 {
  return 0
 }

 s.store.revMu.RLock()
 defer s.store.revMu.RUnlock()

 // in order to find key-value pairs from unsynced watchers, we need to
 // find min revision index, and these revisions can be used to
 // query the backend store of key-value pairs
 curRev := s.store.currentRev
 compactionRev := s.store.compactMainRev

 wg, minRev := s.unsynced.choose(maxWatchersPerSync, curRev, compactionRev)
 minBytes, maxBytes := newRevBytes(), newRevBytes()
 revToBytes(revision{main: minRev}, minBytes)
 revToBytes(revision{main: curRev + 1}, maxBytes)

 // UnsafeRange returns keys and values. And in boltdb, keys are revisions.
 // values are actual key-value pairs in backend.
 tx := s.store.b.ReadTx()
 tx.RLock()
 revs, vs := tx.UnsafeRange(keyBucketName, minBytes, maxBytes, 0)
 var evs []mvccpb.Event
 if s.store != nil && s.store.lg != nil {
  evs = kvsToEvents(s.store.lg, wg, revs, vs)
 } else {
  // TODO: remove this in v3.5
  evs = kvsToEvents(nil, wg, revs, vs)
 }
 tx.RUnlock()

 var victims watcherBatch
 wb := newWatcherBatch(wg, evs)
 for w := range wg.watchers {
  w.minRev = curRev + 1

  eb, ok := wb[w]
  if !ok {
   // bring un-notified watcher to synced
   s.synced.add(w)
   s.unsynced.delete(w)
   continue
  }

  if eb.moreRev != 0 {
   w.minRev = eb.moreRev
  }

  if w.send(WatchResponse{WatchID: w.id, Events: eb.evs, Revision: curRev}) {
   pendingEventsGauge.Add(float64(len(eb.evs)))
  } else {
   if victims == nil {
    victims = make(watcherBatch)
   }
   w.victim = true
  }

  if w.victim {
   victims[w] = eb
  } else {
   if eb.moreRev != 0 {
    // stay unsynced; more to read
    continue
   }
   s.synced.add(w)
  }
  s.unsynced.delete(w)
 }
 s.addVictim(victims)

 vsz := 0
 for _, v := range s.victims {
  vsz += len(v)
 }
 slowWatcherGauge.Set(float64(s.unsynced.size() + vsz))

 return s.unsynced.size()
}
  1. choose 從 unsynced 中選擇待發送數據的 watcher groups, 只看版本是否可用,即處於 [compactRev, curRev]

  2. UnsafeRange 從 boltdb 中獲取所有滿足條件的 keys/values

  3. 遍歷 watchers,開始發送符合條件的 keys/values, 成功了那麼從 unsynced 中刪除,再加到 synced 中,否則加到 victims 隊列中

小結

這次分享就這些,以後面還會分享更多 etcd 與 raft 的內容,如果感興趣,可以關注並點擊左下角的分享轉發哦 (:

參考資料

[1]

WatchServer: https://github.com/etcd-io/etcd/blob/master/etcdserver/api/v3rpc/grpc.go#L60,

本文由 Readfog 進行 AMP 轉碼,版權歸原作者所有。
來源https://mp.weixin.qq.com/s/zrv-rI45Ix06PJyMWOu0hA