Skip to content

Commit

Permalink
fix: speed up cloud linter command
Browse files Browse the repository at this point in the history
  • Loading branch information
bruuuuuuuce committed Apr 11, 2024
1 parent a90a0ae commit 2f38ee0
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 6 deletions.
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -7,3 +7,5 @@ client-cli.iml
bin/
momento.exe
obj/
*.gz

16 changes: 10 additions & 6 deletions momento/src/commands/cloud_linter/linter_cli.rs
Original file line number Diff line number Diff line change
Expand Up @@ -28,17 +28,21 @@ pub async fn run_cloud_linter(region: String) -> Result<(), CliError> {
let output_file_path = "linter_results.json.gz";
check_output_is_writable(output_file_path).await?;

let quota =
Quota::per_second(core::num::NonZeroU32::new(1).expect("should create non-zero quota"));
let limiter = Arc::new(RateLimiter::direct(quota));
let fast_quota =
Quota::per_second(core::num::NonZeroU32::new(10).expect("should create non-zero fast_quota"));
let fast_limiter = Arc::new(RateLimiter::direct(fast_quota));

let mut resources = get_ddb_resources(&config, Arc::clone(&limiter)).await?;
let slow_quota =
Quota::per_second(core::num::NonZeroU32::new(1).expect("should create non-zero slow_quota"));
let slow_limiter = Arc::new(RateLimiter::direct(slow_quota));

let mut resources = get_ddb_resources(&config, Arc::clone(&slow_limiter)).await?;

let mut elasticache_resources =
get_elasticache_resources(&config, Arc::clone(&limiter)).await?;
get_elasticache_resources(&config, Arc::clone(&fast_limiter)).await?;
resources.append(&mut elasticache_resources);

let resources = append_metrics_to_resources(&config, Arc::clone(&limiter), resources).await?;
let resources = append_metrics_to_resources(&config, Arc::clone(&fast_limiter), resources).await?;

let data_format = DataFormat { resources };

Expand Down

0 comments on commit 2f38ee0

Please sign in to comment.