From f126073ce786b510b670db13af24dad26fb9f74c Mon Sep 17 00:00:00 2001 From: Xerxes-2 Date: Tue, 19 Aug 2025 17:31:10 +1000 Subject: [PATCH] refactor: rename write_validated_key_to_tier_files to write_validated_key_to_tier_writers for consistency --- src/adapters/mod.rs | 2 +- src/adapters/output/local.rs | 10 +++++----- src/validation/validation_service.rs | 4 ++-- 3 files changed, 8 insertions(+), 8 deletions(-) diff --git a/src/adapters/mod.rs b/src/adapters/mod.rs index 2a50a7c..de2d31e 100644 --- a/src/adapters/mod.rs +++ b/src/adapters/mod.rs @@ -2,4 +2,4 @@ pub mod input; pub mod output; pub use input::load_keys_from_txt; -pub use output::write_validated_key_to_tier_files; +pub use output::write_validated_key_to_tier_writers; diff --git a/src/adapters/output/local.rs b/src/adapters/output/local.rs index 2086e88..78a6a57 100644 --- a/src/adapters/output/local.rs +++ b/src/adapters/output/local.rs @@ -6,9 +6,9 @@ use toml::Value; use tracing::info; // Write valid key to appropriate tier file -pub async fn write_validated_key_to_tier_files( - free_file: &mut W, - paid_file: &mut W, +pub async fn write_validated_key_to_tier_writers( + free_writer: &mut W, + paid_writer: &mut W, validated_key: &ValidatedKey, ) -> Result<(), ValidatorError> where @@ -16,12 +16,12 @@ where { match validated_key.tier { KeyTier::Free => { - free_file + free_writer .write_all(format!("{}\n", validated_key.key.as_ref()).as_bytes()) .await?; } KeyTier::Paid => { - paid_file + paid_writer .write_all(format!("{}\n", validated_key.key.as_ref()).as_bytes()) .await?; } diff --git a/src/validation/validation_service.rs b/src/validation/validation_service.rs index fe692ec..6ad6736 100644 --- a/src/validation/validation_service.rs +++ b/src/validation/validation_service.rs @@ -1,5 +1,5 @@ use super::key_validator::{test_cache_content_api, test_generate_content_api}; -use crate::adapters::{load_keys_from_txt, write_validated_key_to_tier_files}; +use crate::adapters::{load_keys_from_txt, write_validated_key_to_tier_writers}; use crate::config::KeyCheckerConfig; use crate::error::ValidatorError; use crate::types::GeminiKey; @@ -93,7 +93,7 @@ impl ValidationService { // Process validated keys and write to appropriate tier files while let Some(valid_key) = valid_keys_stream.next().await { - if let Err(e) = write_validated_key_to_tier_files( + if let Err(e) = write_validated_key_to_tier_writers( &mut free_buffer_writer, &mut paid_buffer_writer, &valid_key,