feat: Unify AgentControlDynamicConfig traits into one AgentControlDynamicConfigRepository

Author: alvarocabanas

agent-control/src/agent_control/agent_control.rs

@@ -1,11 +1,7 @@
 use super::config::{
     AgentControlConfig, AgentControlDynamicConfig, SubAgentsMap, sub_agents_difference,
 };
-use super::config_storer::loader_storer::{
-    AgentControlDynamicConfigLoader, AgentControlRemoteConfigDeleter,
-    AgentControlRemoteConfigHashGetter, AgentControlRemoteConfigHashStateUpdater,
-    AgentControlRemoteConfigStorer,
-};
+use super::config_repository::repository::AgentControlDynamicConfigRepository;
 use super::resource_cleaner::ResourceCleaner;
 use super::version_updater::VersionUpdater;
 use crate::agent_control::config_validator::DynamicConfigValidator;
@@ -35,11 +31,7 @@ use tracing::{debug, error, info, instrument, trace, warn};
 pub struct AgentControl<S, O, SL, DV, RC, VU>
 where
     O: StartedClient,
-    SL: AgentControlRemoteConfigStorer
-        + AgentControlDynamicConfigLoader
-        + AgentControlRemoteConfigDeleter
-        + AgentControlRemoteConfigHashStateUpdater
-        + AgentControlRemoteConfigHashGetter,
+    SL: AgentControlDynamicConfigRepository,
     S: SubAgentBuilder,
     DV: DynamicConfigValidator,
     RC: ResourceCleaner,
@@ -63,11 +55,7 @@ impl<S, O, SL, DV, RC, VU> AgentControl<S, O, SL, DV, RC, VU>
 where
     O: StartedClient,
     S: SubAgentBuilder,
-    SL: AgentControlRemoteConfigStorer
-        + AgentControlDynamicConfigLoader
-        + AgentControlRemoteConfigDeleter
-        + AgentControlRemoteConfigHashStateUpdater
-        + AgentControlRemoteConfigHashGetter,
+    SL: AgentControlDynamicConfigRepository,
     DV: DynamicConfigValidator,
     RC: ResourceCleaner,
     VU: VersionUpdater,
@@ -448,7 +436,7 @@ mod tests {
     use crate::agent_control::config::{
         AgentControlConfig, AgentControlDynamicConfig, SubAgentConfig,
     };
-    use crate::agent_control::config_storer::loader_storer::tests::MockAgentControlDynamicConfigStore;
+    use crate::agent_control::config_repository::repository::tests::MockAgentControlDynamicConfigStore;
     use crate::agent_control::config_validator::DynamicConfigValidatorError;
     use crate::agent_control::config_validator::tests::MockDynamicConfigValidator;
     use crate::agent_control::resource_cleaner::ResourceCleanerError;

…src/agent_control/config_storer/store.rs …agent_control/config_repository/store.rsagent-control/src/agent_control/config_storer/store.rs renamed to agent-control/src/agent_control/config_repository/store.rs agent-control/src/agent_control/config_storer/store.rs renamed to agent-control/src/agent_control/config_repository/store.rs

@@ -2,10 +2,8 @@ use crate::agent_control::agent_id::AgentID;
 use crate::agent_control::config::{
     AgentControlConfig, AgentControlConfigError, AgentControlDynamicConfig,
 };
-use crate::agent_control::config_storer::loader_storer::{
-    AgentControlConfigLoader, AgentControlDynamicConfigLoader, AgentControlRemoteConfigDeleter,
-    AgentControlRemoteConfigHashGetter, AgentControlRemoteConfigHashStateUpdater,
-    AgentControlRemoteConfigStorer,
+use crate::agent_control::config_repository::repository::{
+    AgentControlConfigLoader, AgentControlDynamicConfigRepository,
 };
 use crate::agent_control::defaults::{AGENT_CONTROL_CONFIG_ENV_VAR_PREFIX, default_capabilities};
 use crate::opamp::remote_config::hash::{ConfigState, Hash};
@@ -36,55 +34,35 @@ where
     }
 }
 
-impl<Y> AgentControlDynamicConfigLoader for AgentControlConfigStore<Y>
+impl<Y> AgentControlDynamicConfigRepository for AgentControlConfigStore<Y>
 where
     Y: ConfigRepository,
 {
     fn load(&self) -> Result<AgentControlDynamicConfig, AgentControlConfigError> {
         Ok(self._load_config()?.dynamic)
     }
-}
-
-impl<Y> AgentControlRemoteConfigDeleter for AgentControlConfigStore<Y>
-where
-    Y: ConfigRepository,
-{
-    fn delete(&self) -> Result<(), AgentControlConfigError> {
-        self.values_repository
-            .delete_remote(&self.agent_control_id)?;
-        Ok(())
-    }
-}
 
-impl<Y> AgentControlRemoteConfigStorer for AgentControlConfigStore<Y>
-where
-    Y: ConfigRepository,
-{
     fn store(&self, config: &RemoteConfig) -> Result<(), AgentControlConfigError> {
         self.values_repository
             .store_remote(&self.agent_control_id, config)?;
         Ok(())
     }
-}
 
-impl<Y> AgentControlRemoteConfigHashStateUpdater for AgentControlConfigStore<Y>
-where
-    Y: ConfigRepository,
-{
     fn update_hash_state(&self, state: &ConfigState) -> Result<(), AgentControlConfigError> {
         self.values_repository
             .update_hash_state(&self.agent_control_id, state)?;
         Ok(())
     }
-}
 
-impl<Y> AgentControlRemoteConfigHashGetter for AgentControlConfigStore<Y>
-where
-    Y: ConfigRepository,
-{
     fn get_hash(&self) -> Result<Option<Hash>, AgentControlConfigError> {
         Ok(self.values_repository.get_hash(&self.agent_control_id)?)
     }
+
+    fn delete(&self) -> Result<(), AgentControlConfigError> {
+        self.values_repository
+            .delete_remote(&self.agent_control_id)?;
+        Ok(())
+    }
 }
 
 impl<V> AgentControlConfigStore<V>

agent-control/src/config_migrate/migration/agent_config_getter.rs

@@ -1,5 +1,5 @@
 use crate::agent_control::config::{AgentControlConfigError, AgentControlDynamicConfig};
-use crate::agent_control::config_storer::loader_storer::AgentControlDynamicConfigLoader;
+use crate::agent_control::config_repository::repository::AgentControlDynamicConfigRepository;
 use crate::agent_type::agent_type_id::AgentTypeID;
 use semver::VersionReq;
 use thiserror::Error;
@@ -16,15 +16,15 @@ pub enum ConversionError {
 
 pub struct AgentConfigGetter<SL>
 where
-    SL: AgentControlDynamicConfigLoader,
+    SL: AgentControlDynamicConfigRepository,
 {
     pub(super) sub_agents_config_loader: SL,
 }
 
 #[cfg_attr(test, mockall::automock)]
 impl<SL> AgentConfigGetter<SL>
 where
-    SL: AgentControlDynamicConfigLoader + 'static,
+    SL: AgentControlDynamicConfigRepository + 'static,
 {
     pub fn new(sub_agents_config_loader: SL) -> Self {
         Self {
@@ -69,13 +69,23 @@ pub(crate) mod tests {
     use super::*;
     use crate::agent_control::agent_id::AgentID;
     use crate::agent_control::config::{AgentControlDynamicConfig, SubAgentConfig};
+    use crate::opamp::remote_config::hash::{ConfigState, Hash};
+    use crate::values::config::RemoteConfig;
     use mockall::mock;
     use std::collections::HashMap;
 
     mock! {
         pub AgentControlDynamicConfigLoader {}
-        impl AgentControlDynamicConfigLoader for AgentControlDynamicConfigLoader {
+        impl AgentControlDynamicConfigRepository for AgentControlDynamicConfigLoader {
             fn load(&self) -> Result<AgentControlDynamicConfig, AgentControlConfigError>;
+
+            fn store(&self, config: &RemoteConfig) -> Result<(), AgentControlConfigError>;
+
+            fn update_hash_state(&self, state: &ConfigState) -> Result<(), AgentControlConfigError>;
+
+            fn get_hash(&self) -> Result<Option<Hash>, AgentControlConfigError>;
+
+            fn delete(&self) -> Result<(), AgentControlConfigError>;
         }
     }
 

agent-control/src/config_migrate/migration/migrator.rs

@@ -1,6 +1,6 @@
 use crate::agent_control::config::AgentControlConfigError;
-use crate::agent_control::config_storer::loader_storer::AgentControlDynamicConfigLoader;
-use crate::agent_control::config_storer::store::AgentControlConfigStore;
+use crate::agent_control::config_repository::repository::AgentControlDynamicConfigRepository;
+use crate::agent_control::config_repository::store::AgentControlConfigStore;
 use crate::agent_type::agent_type_registry::AgentRegistry;
 use crate::agent_type::embedded_registry::EmbeddedRegistry;
 #[cfg_attr(test, mockall_double::double)]
@@ -39,7 +39,7 @@ pub enum MigratorError {
 
 pub struct ConfigMigrator<
     R: AgentRegistry,
-    SL: AgentControlDynamicConfigLoader + 'static,
+    SL: AgentControlDynamicConfigRepository + 'static,
     C: DirectoryManager,
     F: FileReader,
 > {

agent-control/tests/k8s/garbage_collector.rs

@@ -10,7 +10,10 @@ use super::tools::{
 use k8s_openapi::api::core::v1::Secret;
 use kube::{api::Api, core::TypeMeta};
 use mockall::mock;
+use newrelic_agent_control::agent_control::config_repository::repository::AgentControlDynamicConfigRepository;
+use newrelic_agent_control::opamp::remote_config::hash::{ConfigState, Hash};
 use newrelic_agent_control::sub_agent::k8s::supervisor::NotStartedSupervisorK8s;
+use newrelic_agent_control::values::config::RemoteConfig;
 use newrelic_agent_control::{
     agent_control::config::default_group_version_kinds, agent_type::agent_type_id::AgentTypeID,
 };
@@ -26,7 +29,7 @@ use newrelic_agent_control::{
 use newrelic_agent_control::{
     agent_control::{
         config::{AgentControlConfig, AgentControlConfigError, AgentControlDynamicConfig},
-        config_storer::loader_storer::{AgentControlConfigLoader, AgentControlDynamicConfigLoader},
+        config_repository::repository::AgentControlConfigLoader,
     },
     k8s::labels::Labels,
 };
@@ -49,10 +52,18 @@ mock! {
 
 // Setup AgentControlDynamicConfigLoader mock
 mock! {
-    pub AgentControlDynamicConfigLoader{}
+    pub AgentControlDynamicConfigRepository{}
 
-    impl AgentControlDynamicConfigLoader for AgentControlDynamicConfigLoader {
+    impl AgentControlDynamicConfigRepository for AgentControlDynamicConfigRepository {
         fn load(&self) -> Result<AgentControlDynamicConfig, AgentControlConfigError>;
+
+        fn store(&self, config: &RemoteConfig) -> Result<(), AgentControlConfigError>;
+
+        fn update_hash_state(&self, state: &ConfigState) -> Result<(), AgentControlConfigError>;
+
+        fn get_hash(&self) -> Result<Option<Hash>, AgentControlConfigError>;
+
+        fn delete(&self) -> Result<(), AgentControlConfigError>;
     }
 }
 

agent-control/tests/k8s/store.rs

@@ -4,11 +4,8 @@ use crate::k8s::tools::k8s_env::K8sEnv;
 use k8s_openapi::api::core::v1::ConfigMap;
 use kube::Api;
 use newrelic_agent_control::agent_control::agent_id::AgentID;
-use newrelic_agent_control::agent_control::config_storer::loader_storer::{
-    AgentControlDynamicConfigLoader, AgentControlRemoteConfigDeleter,
-    AgentControlRemoteConfigStorer,
-};
-use newrelic_agent_control::agent_control::config_storer::store::AgentControlConfigStore;
+use newrelic_agent_control::agent_control::config_repository::repository::AgentControlDynamicConfigRepository;
+use newrelic_agent_control::agent_control::config_repository::store::AgentControlConfigStore;
 use newrelic_agent_control::agent_control::defaults::default_capabilities;
 use newrelic_agent_control::k8s::client::SyncK8sClient;
 use newrelic_agent_control::k8s::labels::Labels;