asim: add AllocatorSync to asim node

This commit adds the AllocatorSync struct to the asim node. Each
MMAllocator will have a corresponding AllocatorSync responsible for coordinating
changes between the various queues and the MMA logic. This struct is currently
unused and will be integrated in future commits.

Epic: none
Release note: none

Author: wenyihu6

pkg/kv/kvserver/asim/state/BUILD.bazel

@@ -26,6 +26,7 @@ go_library(
         "//pkg/kv/kvserver/allocator",
         "//pkg/kv/kvserver/allocator/allocatorimpl",
         "//pkg/kv/kvserver/allocator/mmaprototype",
+        "//pkg/kv/kvserver/allocator/mmaprototypehelpers",
         "//pkg/kv/kvserver/allocator/storepool",
         "//pkg/kv/kvserver/asim/config",
         "//pkg/kv/kvserver/asim/workload",

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

@@ -23,6 +23,7 @@ import (
 	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/allocator"
 	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/allocator/allocatorimpl"
 	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/allocator/mmaprototype"
+	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/allocator/mmaprototypehelpers"
 	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/allocator/storepool"
 	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/asim/config"
 	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/asim/workload"
@@ -426,6 +427,7 @@ func (s *state) AddNode() Node {
 		stores:      []StoreID{},
 		storepool:   sp,
 		mmAllocator: mmAllocator,
+		as:          mmaprototypehelpers.NewAllocatorSync(sp, mmAllocator),
 	}
 	s.nodes[nodeID] = node
 	s.SetNodeLiveness(nodeID, livenesspb.NodeLivenessStatus_LIVE)
@@ -1400,6 +1402,7 @@ type node struct {
 	stores      []StoreID
 	storepool   *storepool.StorePool
 	mmAllocator mmaprototype.Allocator
+	as          *mmaprototypehelpers.AllocatorSync
 }
 
 // NodeID returns the ID of this node.
@@ -1421,6 +1424,10 @@ func (n *node) MMAllocator() mmaprototype.Allocator {
 	return n.mmAllocator
 }
 
+func (n *node) AllocatorSync() *mmaprototypehelpers.AllocatorSync {
+	return n.as
+}
+
 // store is an implementation of the Store interface.
 type store struct {
 	storeID StoreID

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

@@ -13,6 +13,7 @@ import (
 	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/allocator"
 	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/allocator/allocatorimpl"
 	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/allocator/mmaprototype"
+	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/allocator/mmaprototypehelpers"
 	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/allocator/storepool"
 	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/asim/workload"
 	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/liveness/livenesspb"
@@ -217,6 +218,8 @@ type Node interface {
 	Descriptor() roachpb.NodeDescriptor
 	// TODO(wenyihu6): use this in mma store rebalancer
 	MMAllocator() mmaprototype.Allocator
+	// AllocatorSync returns the AllocatorSync for this node.
+	AllocatorSync() *mmaprototypehelpers.AllocatorSync
 }
 
 // Store is a container for replicas.