asim: add node id for simulator replica

Previously, the simulator replica interface did not include a NodeID() method.
Future changes to replica_rankings.go will require this as part of the candidate
replica interface. This commit updates the simulator replica to implement
NodeID() method.

Epic: none
Release note: none

Author: wenyihu6

pkg/kv/kvserver/asim/state/impl.go

@@ -628,6 +628,7 @@ func (s *state) addReplica(
 
 	desc := rng.desc.AddReplica(roachpb.NodeID(nodeID), roachpb.StoreID(storeID), rtype)
 	replica := &replica{
+		nodeID:    nodeID,
 		replicaID: ReplicaID(desc.ReplicaID),
 		storeID:   storeID,
 		rangeID:   rangeID,
@@ -1607,6 +1608,7 @@ func (r *rng) Size() int64 {
 // replica is an implementation of the Replica interface.
 type replica struct {
 	replicaID  ReplicaID
+	nodeID     NodeID
 	storeID    StoreID
 	rangeID    RangeID
 	desc       roachpb.ReplicaDescriptor
@@ -1618,6 +1620,11 @@ func (r *replica) ReplicaID() ReplicaID {
 	return r.replicaID
 }
 
+// NodeID returns the ID of the node this replica is on.
+func (r *replica) NodeID() NodeID {
+	return r.nodeID
+}
+
 // StoreID returns the ID of the store this replica is on.
 func (r *replica) StoreID() StoreID {
 	return r.storeID

pkg/kv/kvserver/asim/state/state.go

@@ -270,6 +270,8 @@ type Range interface {
 type Replica interface {
 	// ReplicaID returns the ID of this replica.
 	ReplicaID() ReplicaID
+	// NodeID returns the ID of the node this replica is on.
+	NodeID() NodeID
 	// StoreID returns the ID of the store this replica is on.
 	StoreID() StoreID
 	// Descriptor returns the descriptor for this replica.