Method StreamExtSnapshot

worker/import.go:476–490

StreamExtSnapshot handles the stream of key-value pairs sent from proxy alpha. It receives a Forward flag from the stream to determine if the current node is the leader. If the node is the leader (Forward is true), it streams the data to its followers. Otherwise, it simply writes the data to BadgerD

(stream pb.Worker_StreamExtSnapshotServer)

// If the node is the leader (Forward is true), it streams the data to its followers.
// Otherwise, it simply writes the data to BadgerDB and flushes it.
func (w *grpcWorker) StreamExtSnapshot(stream pb.Worker_StreamExtSnapshotServer) error {
	glog.Info("[import] trying to update the import mode to false")
	defer x.ExtSnapshotStreamingState(false)

	// Receive the first message to check the Forward flag.
	// If Forward is true, this node is the leader and should forward the stream to its followers.
	// If Forward is false, the node just writes and flushes the data.
	forwardReq, err := stream.Recv()
	if err != nil {
		return err
	}

	glog.Infof("[import] received forward flag: %v", forwardReq.Forward)
	return streamInGroup(stream, forwardReq.Forward)
}

// postStreamProcessing handles the post-stream processing of data received from the buffer into the local BadgerDB.
// It loads the schema, updates the membership state, informs zero about tablets, resets caches, applies initial schema,