refactor: update file handling in write_validated_key_to_tier_files to use AsyncWrite trait

main
Xerxes-2 2025-08-19 16:56:07 +10:00
parent 5144b988cd
commit e42b952649
No known key found for this signature in database
GPG Key ID: A6C508165D76B601
1 changed files with 3 additions and 3 deletions

View File

@ -1,14 +1,14 @@
use crate::error::ValidatorError; use crate::error::ValidatorError;
use crate::types::{GeminiKey, KeyTier, ValidatedKey}; use crate::types::{GeminiKey, KeyTier, ValidatedKey};
use std::{fs, io::Write}; use std::{fs, io::Write};
use tokio::io::{AsyncWriteExt, BufWriter}; use tokio::io::{AsyncWrite, AsyncWriteExt};
use toml::Value; use toml::Value;
use tracing::info; use tracing::info;
// Write valid key to appropriate tier file // Write valid key to appropriate tier file
pub async fn write_validated_key_to_tier_files( pub async fn write_validated_key_to_tier_files(
free_file: &mut BufWriter<tokio::fs::File>, free_file: &mut (impl AsyncWrite + Unpin),
paid_file: &mut BufWriter<tokio::fs::File>, paid_file: &mut (impl AsyncWrite + Unpin),
validated_key: &ValidatedKey, validated_key: &ValidatedKey,
) -> Result<(), ValidatorError> { ) -> Result<(), ValidatorError> {
match validated_key.tier { match validated_key.tier {