【ETCD】[源码阅读]深度解析 EtcdServer 的 processInternalRaftRequestOnce 方法

原创

已于 2024-12-12 09:09:34 修改 · 1.6k 阅读

16 ·

CC 4.0 BY-SA版权

文章标签：

#etcd

于 2024-12-12 07:22:40 首次发布

在分布式系统中，etcd 的一致性与高效性得益于其强大的 Raft 协议模块。而 processInternalRaftRequestOnce 是 etcd 服务器处理内部 Raft 请求的核心方法之一。本文将从源码角度解析这个方法的逻辑流程，帮助读者更好地理解 etcd 的内部实现。

方法源码

func (s *EtcdServer) processInternalRaftRequestOnce(ctx context.Context, r pb.InternalRaftRequest) (*applyResult, error) {
   
   
	ai := s.getAppliedIndex()
	ci := s.getCommittedIndex()
	if ci > ai+maxGapBetweenApplyAndCommitIndex {
   
   
		return nil, ErrTooManyRequests
	}

	r.Header = &pb.RequestHeader{
   
   
		ID: s.reqIDGen.Next(),
	}

	// check authinfo if it is not InternalAuthenticateRequest
	if r.Authenticate == nil {
   
   
		authInfo, err := s.AuthInfoFromCtx(ctx)
		if err != nil {
   
   
			return nil, err
		}
		if authInfo != nil {
   
   
			r.Header.Username = authInfo.Username
			r.Header.AuthRevision = authInfo.Revision
		}
	}

	data, err := r.Marshal()
	if err != nil {
   
   
		return nil, err
	}

	if len(data) > int(s.Cfg.MaxRequestBytes) {
   
   
		return nil, ErrRequestTooLarge
	}

	id := r.ID
	if id == 0 {
   
   
		id = r.Header.ID
	}
	ch := s.w.Register(id)

	cctx, cancel := context.WithTimeout(ctx, s.Cfg.ReqTimeout())
	defer cancel()

	start := time.Now()
	err = s.r.Propose(cctx, data)
	if err != nil {
   
   
		proposalsFailed.Inc()
		s.w.Trigger(id, nil) // GC wait
		return nil, err
	}
	proposalsPending.Inc()
	defer proposalsPending.Dec()

	select {
   
   
	case x := <-ch:
		return x.(*applyResult), nil
	case <-cctx.Done():
		proposalsFailed