i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-09-02 01:10:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: correct API key expiry and dashboard savings (#361)

Browse Source
This commit is contained in:
Entropy.Xu
2026-04-30 00:35:32 +08:00
committed by GitHub
parent 9e2faa7e5b
commit 4d59d518d1
11 changed files with 228 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
crates/aether-data/src/repository/auth/memory.rs
View File

@@ -1,5 +1,6 @@
use std::collections::BTreeMap;
use std::sync::RwLock;
use std::time::{SystemTime, UNIX_EPOCH};
use async_trait::async_trait;
@@ -12,6 +13,13 @@ use super::types::{
use crate::repository::usage::{ApiKeyUsageContribution, ApiKeyUsageDelta};
use crate::DataLayerError;
fn current_unix_secs() -> u64 {
SystemTime::now()
.duration_since(UNIX_EPOCH)
.map(|duration| duration.as_secs())
.unwrap_or(0)
}
#[derive(Debug, Default)]
struct MemoryAuthApiKeyIndex {
by_api_key_id: BTreeMap<String, StoredAuthApiKeySnapshot>,
@@ -528,6 +536,7 @@ impl AuthApiKeyWriteRepository for InMemoryAuthApiKeySnapshotRepository {
)?
};
let now_unix_secs = current_unix_secs() as i64;
let export = StoredAuthApiKeyExportRecord::new(
record.user_id.clone(),
record.api_key_id.clone(),
@@ -556,7 +565,8 @@ impl AuthApiKeyWriteRepository for InMemoryAuthApiKeySnapshotRepository {
record.total_tokens as i64,
record.total_cost_usd,
false,
)?;
)?
.with_activity_timestamps(None, Some(now_unix_secs), Some(now_unix_secs))?;
index
.by_key_hash
@@ -649,6 +659,7 @@ impl AuthApiKeyWriteRepository for InMemoryAuthApiKeySnapshotRepository {
)?
};
let now_unix_secs = current_unix_secs() as i64;
let export = StoredAuthApiKeyExportRecord::new(
record.user_id.clone(),
record.api_key_id.clone(),
@@ -677,7 +688,8 @@ impl AuthApiKeyWriteRepository for InMemoryAuthApiKeySnapshotRepository {
record.total_tokens as i64,
record.total_cost_usd,
true,
)?;
)?
.with_activity_timestamps(None, Some(now_unix_secs), Some(now_unix_secs))?;
index
.by_key_hash