Skip to content

Commit 4fcb4f1

Browse files
committed
feat: refactor prom metric actions into modules
to reduce the amount of `cfg` macros needed
1 parent 5ee8464 commit 4fcb4f1

File tree

5 files changed

+164
-144
lines changed

5 files changed

+164
-144
lines changed

stacks-signer/src/client/stacks_client.rs

Lines changed: 17 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -323,8 +323,10 @@ impl StacksClient {
323323
block,
324324
chain_id: self.chain_id,
325325
};
326-
let timer =
327-
crate::monitoring::new_rpc_call_timer(&self.block_proposal_path(), &self.http_origin);
326+
let timer = crate::monitoring::actions::new_rpc_call_timer(
327+
&self.block_proposal_path(),
328+
&self.http_origin,
329+
);
328330
let send_request = || {
329331
self.stacks_node_client
330332
.post(self.block_proposal_path())
@@ -399,7 +401,8 @@ impl StacksClient {
399401
"{}{RPC_TENURE_FORKING_INFO_PATH}/:start/:stop",
400402
self.http_origin
401403
);
402-
let timer = crate::monitoring::new_rpc_call_timer(&metrics_path, &self.http_origin);
404+
let timer =
405+
crate::monitoring::actions::new_rpc_call_timer(&metrics_path, &self.http_origin);
403406
let send_request = || {
404407
self.stacks_node_client
405408
.get(&path)
@@ -420,7 +423,7 @@ impl StacksClient {
420423
pub fn get_current_and_last_sortition(&self) -> Result<CurrentAndLastSortition, ClientError> {
421424
debug!("StacksClient: Getting current and prior sortition");
422425
let path = format!("{}/latest_and_last", self.sortition_info_path());
423-
let timer = crate::monitoring::new_rpc_call_timer(&path, &self.http_origin);
426+
let timer = crate::monitoring::actions::new_rpc_call_timer(&path, &self.http_origin);
424427
let send_request = || {
425428
self.stacks_node_client.get(&path).send().map_err(|e| {
426429
warn!("Signer failed to request latest sortition"; "err" => ?e);
@@ -460,8 +463,10 @@ impl StacksClient {
460463
/// Get the current peer info data from the stacks node
461464
pub fn get_peer_info(&self) -> Result<PeerInfo, ClientError> {
462465
debug!("StacksClient: Getting peer info");
463-
let timer =
464-
crate::monitoring::new_rpc_call_timer(&self.core_info_path(), &self.http_origin);
466+
let timer = crate::monitoring::actions::new_rpc_call_timer(
467+
&self.core_info_path(),
468+
&self.http_origin,
469+
);
465470
let send_request = || {
466471
self.stacks_node_client
467472
.get(self.core_info_path())
@@ -485,7 +490,7 @@ impl StacksClient {
485490
debug!("StacksClient: Getting reward set signers";
486491
"reward_cycle" => reward_cycle,
487492
);
488-
let timer = crate::monitoring::new_rpc_call_timer(
493+
let timer = crate::monitoring::actions::new_rpc_call_timer(
489494
&format!("{}/v3/stacker_set/:reward_cycle", self.http_origin),
490495
&self.http_origin,
491496
);
@@ -521,7 +526,8 @@ impl StacksClient {
521526
/// Retrieve the current pox data from the stacks node
522527
pub fn get_pox_data(&self) -> Result<RPCPoxInfoData, ClientError> {
523528
debug!("StacksClient: Getting pox data");
524-
let timer = crate::monitoring::new_rpc_call_timer(&self.pox_path(), &self.http_origin);
529+
let timer =
530+
crate::monitoring::actions::new_rpc_call_timer(&self.pox_path(), &self.http_origin);
525531
let send_request = || {
526532
self.stacks_node_client
527533
.get(self.pox_path())
@@ -572,7 +578,7 @@ impl StacksClient {
572578
"address" => %address,
573579
);
574580
let timer_label = format!("{}/v2/accounts/:principal", self.http_origin);
575-
let timer = crate::monitoring::new_rpc_call_timer(&timer_label, &self.http_origin);
581+
let timer = crate::monitoring::actions::new_rpc_call_timer(&timer_label, &self.http_origin);
576582
let send_request = || {
577583
self.stacks_node_client
578584
.get(self.accounts_path(address))
@@ -628,7 +634,7 @@ impl StacksClient {
628634
"block_height" => %block.header.chain_length,
629635
);
630636
let path = format!("{}{}?broadcast=1", self.http_origin, postblock_v3::PATH);
631-
let timer = crate::monitoring::new_rpc_call_timer(&path, &self.http_origin);
637+
let timer = crate::monitoring::actions::new_rpc_call_timer(&path, &self.http_origin);
632638
let send_request = || {
633639
self.stacks_node_client
634640
.post(&path)
@@ -678,7 +684,7 @@ impl StacksClient {
678684
"{}/v2/contracts/call-read/:principal/{contract_name}/{function_name}",
679685
self.http_origin
680686
);
681-
let timer = crate::monitoring::new_rpc_call_timer(&timer_label, &self.http_origin);
687+
let timer = crate::monitoring::actions::new_rpc_call_timer(&timer_label, &self.http_origin);
682688
let response = self
683689
.stacks_node_client
684690
.post(path)

stacks-signer/src/lib.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ impl<S: Signer<T> + Send + 'static, T: SignerEventTrait + 'static> SpawnedSigner
125125
);
126126
let (res_send, res_recv) = channel();
127127
let ev = SignerEventReceiver::new(config.network.is_mainnet());
128-
crate::monitoring::start_serving_monitoring_metrics(config.clone()).ok();
128+
crate::monitoring::actions::start_serving_monitoring_metrics(config.clone()).ok();
129129
let runloop = RunLoop::new(config.clone());
130130
let mut signer: RunLoopSigner<S, T> = libsigner::Signer::new(runloop, ev, res_send);
131131
let running_signer = signer.spawn(endpoint).expect("Failed to spawn signer");

0 commit comments

Comments
 (0)