Skip to content

Commit 407fbbb

Browse files
chore(style): align new_agent_control_fqn and new_agent_control_id (#982)
1 parent d7919f3 commit 407fbbb

File tree

6 files changed

+18
-29
lines changed

6 files changed

+18
-29
lines changed

agent-control/src/agent_control/agent_control.rs

Lines changed: 2 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,9 @@
1-
use super::config::{
2-
AgentControlDynamicConfig, AgentID, AgentTypeFQN, SubAgentConfig, SubAgentsMap,
3-
};
1+
use super::config::{AgentControlDynamicConfig, AgentID, SubAgentConfig, SubAgentsMap};
42
use super::config_storer::loader_storer::{
53
AgentControlDynamicConfigDeleter, AgentControlDynamicConfigLoader,
64
AgentControlDynamicConfigStorer,
75
};
8-
use crate::agent_control::{
9-
defaults::{AGENT_CONTROL_NAMESPACE, AGENT_CONTROL_TYPE, AGENT_CONTROL_VERSION},
10-
error::AgentError,
11-
};
6+
use crate::agent_control::error::AgentError;
127
use crate::event::{
138
channel::{EventConsumer, EventPublisher},
149
AgentControlEvent, ApplicationEvent, OpAMPEvent, SubAgentEvent,
@@ -409,17 +404,6 @@ where
409404
}
410405
}
411406

412-
pub fn agent_control_fqn() -> AgentTypeFQN {
413-
AgentTypeFQN::try_from(
414-
format!(
415-
"{}/{}:{}",
416-
AGENT_CONTROL_NAMESPACE, AGENT_CONTROL_TYPE, AGENT_CONTROL_VERSION
417-
)
418-
.as_str(),
419-
)
420-
.unwrap()
421-
}
422-
423407
////////////////////////////////////////////////////////////////////////////////////
424408
// Tests
425409
////////////////////////////////////////////////////////////////////////////////////

agent-control/src/agent_control/config.rs

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
use super::http_server::config::ServerConfig;
2-
use crate::agent_control::agent_control_fqn;
32
use crate::agent_control::defaults::{
43
default_capabilities, default_sub_agent_custom_capabilities, AGENT_CONTROL_ID,
4+
AGENT_CONTROL_NAMESPACE, AGENT_CONTROL_TYPE, AGENT_CONTROL_VERSION,
55
};
66
use crate::http::proxy::ProxyConfig;
77
use crate::logging::config::LoggingConfig;
@@ -221,6 +221,13 @@ impl AgentTypeFQN {
221221
pub fn version(&self) -> String {
222222
self.0.chars().skip_while(|&i| i != ':').skip(1).collect()
223223
}
224+
225+
pub fn new_agent_control_fqn() -> Self {
226+
AgentTypeFQN(format!(
227+
"{}/{}:{}",
228+
AGENT_CONTROL_NAMESPACE, AGENT_CONTROL_TYPE, AGENT_CONTROL_VERSION
229+
))
230+
}
224231
}
225232

226233
impl Display for AgentTypeFQN {
@@ -379,7 +386,7 @@ impl AgentTypeFQN {
379386

380387
pub(crate) fn get_custom_capabilities(&self) -> Option<CustomCapabilities> {
381388
//TODO: We should move this to EffectiveAgent
382-
if self.eq(&agent_control_fqn()) {
389+
if self.eq(&AgentTypeFQN::new_agent_control_fqn()) {
383390
// Agent_Control does not have custom capabilities for now
384391
return None;
385392
}

agent-control/src/agent_control/run/k8s.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
1-
use crate::agent_control::config::{AgentID, K8sConfig};
1+
use crate::agent_control::config::{AgentID, AgentTypeFQN, K8sConfig};
22
use crate::agent_control::config_storer::loader_storer::AgentControlConfigLoader;
33
use crate::agent_control::config_storer::store::AgentControlConfigStore;
44
use crate::agent_control::defaults::{
55
AGENT_CONTROL_VERSION, FLEET_ID_ATTRIBUTE_KEY, HOST_NAME_ATTRIBUTE_KEY,
66
OPAMP_AGENT_VERSION_ATTRIBUTE_KEY, OPAMP_CHART_VERSION_ATTRIBUTE_KEY,
77
};
88
use crate::agent_control::run::AgentControlRunner;
9-
use crate::agent_control::{agent_control_fqn, AgentControl};
9+
use crate::agent_control::AgentControl;
1010
use crate::agent_type::renderer::TemplateRenderer;
1111
#[cfg_attr(test, mockall_double::double)]
1212
use crate::k8s::client::SyncK8sClient;
@@ -119,7 +119,7 @@ impl AgentControlRunner {
119119
builder,
120120
&instance_id_getter,
121121
AgentID::new_agent_control_id(),
122-
&agent_control_fqn(),
122+
&AgentTypeFQN::new_agent_control_fqn(),
123123
additional_identifying_attributes,
124124
non_identifying_attributes,
125125
)

agent-control/src/agent_control/run/on_host.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
1-
use crate::agent_control::config::AgentID;
1+
use crate::agent_control::config::{AgentID, AgentTypeFQN};
22
use crate::agent_control::config_storer::loader_storer::AgentControlConfigLoader;
33
use crate::agent_control::config_storer::store::AgentControlConfigStore;
44
use crate::agent_control::defaults::{
55
AGENT_CONTROL_VERSION, FLEET_ID_ATTRIBUTE_KEY, HOST_ID_ATTRIBUTE_KEY, HOST_NAME_ATTRIBUTE_KEY,
66
OPAMP_AGENT_VERSION_ATTRIBUTE_KEY, SUB_AGENT_DIR,
77
};
88
use crate::agent_control::run::AgentControlRunner;
9-
use crate::agent_control::{agent_control_fqn, AgentControl};
9+
use crate::agent_control::AgentControl;
1010
use crate::agent_type::variable::definition::VariableDefinition;
1111
use crate::opamp::effective_config::loader::DefaultEffectiveConfigLoaderBuilder;
1212
use crate::opamp::instance_id::getter::InstanceIDWithIdentifiersGetter;
@@ -127,7 +127,7 @@ impl AgentControlRunner {
127127
builder,
128128
&instance_id_getter,
129129
AgentID::new_agent_control_id(),
130-
&agent_control_fqn(),
130+
&AgentTypeFQN::new_agent_control_fqn(),
131131
HashMap::from([(
132132
OPAMP_AGENT_VERSION_ATTRIBUTE_KEY.to_string(),
133133
DescriptionValueType::String(AGENT_CONTROL_VERSION.to_string()),

agent-control/src/opamp/remote_config/validators/certificate_fetcher.rs

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
use crate::opamp::remote_config::validators::signature::Certificate;
22
use std::sync::Mutex;
3-
use std::time::SystemTime;
43
use thiserror::Error;
54
use tracing::debug;
65
use tracing::log::error;

agent-control/src/opamp/remote_config/validators/signature.rs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,6 @@ impl SignatureValidator {
6161
#[cfg(test)]
6262
mod tests {
6363
use super::*;
64-
use crate::agent_control::agent_control_fqn;
6564
use crate::agent_control::config::AgentID;
6665
use crate::opamp::remote_config::hash::Hash;
6766
use crate::opamp::remote_config::signature::Signature;
@@ -105,7 +104,7 @@ mod tests {
105104
Hash::new("test".to_string()),
106105
None,
107106
);
108-
let agent_type = agent_control_fqn();
107+
let agent_type = AgentTypeFQN::new_agent_control_fqn();
109108

110109
assert!(signature_validator.validate(&agent_type, &rc).unwrap());
111110
}

0 commit comments

Comments
 (0)