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

Improve gateway scheduling and runtime admission

Browse Source
This commit is contained in:
elky
2026-06-24 01:53:45 +08:00
parent cf0af8fa1e
commit d336d1a7fa
87 changed files with 9671 additions and 804 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/aether-runtime-state/src/redis/client.rs
+26 -9
View File
@@ -106,24 +106,28 @@ impl RedisConnectionRouter {
&client,
connection_manager_config(command_timeout_ms),
RedisConnectionLane::Fast,
command_timeout_ms,
)
.await?;
let stream = connect_lane(
&client,
connection_manager_config(command_timeout_ms),
RedisConnectionLane::Stream,
command_timeout_ms,
)
.await?;
let blocking_stream = connect_lane(
&client,
connection_manager_config(command_timeout_ms),
RedisConnectionLane::BlockingStream,
command_timeout_ms,
)
.await?;
let admin = connect_lane(
&client,
connection_manager_config(command_timeout_ms),
RedisConnectionLane::Admin,
command_timeout_ms,
)
.await?;
info!(
@@ -228,16 +232,29 @@ async fn connect_lane(
client: &RedisClient,
config: redis::aio::ConnectionManagerConfig,
lane: RedisConnectionLane,
command_timeout_ms: Option<u64>,
) -> Result<RedisManagedConnection, DataLayerError> {
client
.get_connection_manager_with_config(config)
.await
.map_err(|err| {
DataLayerError::Redis(format!(
"failed to initialize runtime redis {} lane: {err}",
lane.as_str()
))
})
let connect = client.get_connection_manager_with_config(config);
let result = if let Some(timeout_ms) = command_timeout_ms {
match tokio::time::timeout(Duration::from_millis(timeout_ms), connect).await {
Ok(result) => result,
Err(_) => {
return Err(DataLayerError::TimedOut(format!(
"runtime redis {} lane connection exceeded {}ms timeout",
lane.as_str(),
timeout_ms
)));
}
}
} else {
connect.await
};
result.map_err(|err| {
DataLayerError::Redis(format!(
"failed to initialize runtime redis {} lane: {err}",
lane.as_str()
))
})
}
#[cfg(test)]