chore(qrm): support cpu/memory write-only state

pkg/agent/qrm-plugins/cpu/dynamicpolicy/policy.go

@@ -134,6 +134,7 @@ func NewDynamicPolicy(agentCtx *agent.GenericContext, conf *config.Configuration
 	}
 
 	state.SetReadonlyState(stateImpl)
+	state.SetWriteOnlyState(stateImpl)
 
 	wrappedEmitter := agentCtx.EmitterPool.GetDefaultMetricsEmitter().WithTags(agentName, metrics.MetricTag{
 		Key: util.QRMPluginPolicyTagName,

pkg/agent/qrm-plugins/cpu/dynamicpolicy/state/state.go

@@ -562,6 +562,11 @@ type ReadonlyState interface {
 	reader
 }
 
+// WriteOnlyState interface only provides methods for update pod assignments
+type WriteOnlyState interface {
+	writer
+}
+
 type GenerateMachineStateFromPodEntriesFunc func(topology *machine.CPUTopology, podEntries PodEntries) (NUMANodeMap, error)
 
 var (
@@ -589,3 +594,29 @@ func SetReadonlyState(state ReadonlyState) {
 
 	readonlyState = state
 }
+
+var (
+	writeOnlyStateLock sync.RWMutex
+	writeOnlyState     WriteOnlyState
+)
+
+// GetWriteOnlyState returns state.WriteOnlyState to provides a way
+// to update the states of the plugin
+func GetWriteOnlyState() (WriteOnlyState, error) {
+	writeOnlyStateLock.RLock()
+	defer writeOnlyStateLock.RUnlock()
+
+	if writeOnlyState == nil {
+		return nil, fmt.Errorf("readonlyState isn't set")
+	}
+	return writeOnlyState, nil
+}
+
+// SetWriteOnlyState sets state.WriteOnlyState to be used by the plugin
+// this function should only be called once during initialization
+func SetWriteOnlyState(state WriteOnlyState) {
+	readonlyStateLock.Lock()
+	defer readonlyStateLock.Unlock()
+
+	writeOnlyState = state
+}

pkg/agent/qrm-plugins/cpu/nativepolicy/policy.go

@@ -99,6 +99,7 @@ func NewNativePolicy(agentCtx *agent.GenericContext, conf *config.Configuration,
 	}
 
 	state.SetReadonlyState(stateImpl)
+	state.SetWriteOnlyState(stateImpl)
 
 	wrappedEmitter := agentCtx.EmitterPool.GetDefaultMetricsEmitter().WithTags(agentName, metrics.MetricTag{
 		Key: util.QRMPluginPolicyTagName,

pkg/agent/qrm-plugins/memory/dynamicpolicy/policy.go

@@ -177,6 +177,7 @@ func NewDynamicPolicy(agentCtx *agent.GenericContext, conf *config.Configuration
 	}
 
 	state.SetReadonlyState(stateImpl)
+	state.SetWriteOnlyState(stateImpl)
 
 	wrappedEmitter := agentCtx.EmitterPool.GetDefaultMetricsEmitter().WithTags(agentName, metrics.MetricTag{
 		Key: util.QRMPluginPolicyTagName,

pkg/agent/qrm-plugins/memory/dynamicpolicy/state/state.go

@@ -380,6 +380,11 @@ type ReadonlyState interface {
 	GetReservedMemory() map[v1.ResourceName]map[int]uint64
 }
 
+// WriteOnlyState interface only provides methods for update pod assignments
+type WriteOnlyState interface {
+	writer
+}
+
 // State interface provides methods for tracking and setting pod assignments
 type State interface {
 	writer
@@ -411,3 +416,29 @@ func SetReadonlyState(state ReadonlyState) {
 
 	readonlyState = state
 }
+
+var (
+	writeOnlyStateLock sync.RWMutex
+	writeOnlyState     WriteOnlyState
+)
+
+// GetWriteOnlyState returns state.WriteOnlyState to provides a way
+// to update the states of the plugin
+func GetWriteOnlyState() (WriteOnlyState, error) {
+	writeOnlyStateLock.RLock()
+	defer writeOnlyStateLock.RUnlock()
+
+	if writeOnlyState == nil {
+		return nil, fmt.Errorf("readonlyState isn't set")
+	}
+	return writeOnlyState, nil
+}
+
+// SetWriteOnlyState sets state.WriteOnlyState to be used by the plugin
+// this function should only be called once during initialization
+func SetWriteOnlyState(state WriteOnlyState) {
+	readonlyStateLock.Lock()
+	defer readonlyStateLock.Unlock()
+
+	writeOnlyState = state
+}