GeneralBots/generalbots
1
0
Fork 0
Code Activity Actions 5

fix: start.bas redundant execution with atomic SET NX guard
All checks were successful
BotServer CI / build (push) Successful in 42s
Details

Browse source
This commit is contained in:
Rodrigo Rodriguez 2026-04-30 16:45:37 -03:00
parent 5c54f7bc72
commit debe29a6dc
1 changed files with 23 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
botserver/src/core/bot/mod.rs
View file

@ -772,23 +772,37 @@ let system_prompt = if !message.active_switchers.is_empty() {
{
// Execute start.bas on first message only (use Redis flag to prevent re-execution)
let actual_session_id = session.id.to_string();
let start_bas_flag_key = format!("start_bas_executed:{}", actual_session_id);
// Check Redis flag - only execute start.bas once per session
// We use SET NX to atomically check and set the flag
let should_execute_start_bas = if let Some(cache) = &self.state.cache {
if let Ok(mut conn) = cache.get_multiplexed_async_connection().await {
let key = format!("start_bas_executed:{}", actual_session_id);
let already_executed: Option<String> = redis::cmd("GET")
.arg(&key)
let was_set: Option<String> = redis::cmd("SET")
.arg(&start_bas_flag_key)
.arg("1")
.arg("NX")
.arg("EX")
.arg(86400) // 24h TTL
.query_async(&mut conn)
.await
.ok()
.flatten();
already_executed.is_none()
.ok();
// If SET NX returned "OK" (Some("OK")), it means it was the first time
// If it returned nil (None), it means it was already set
let is_first_time = was_set.is_some();
if !is_first_time {
trace!("start.bas already executed for session {}, skipping in stream_response", actual_session_id);
}
is_first_time
} else {
true // No cache connection, execute as fallback
warn!("Failed to get Redis connection for start.bas guard, skipping to avoid redundant execution");
false
}
} else {
true // No cache configured, execute as fallback
trace!("No cache configured, skipping start.bas guard in stream_response");
false
};
if should_execute_start_bas {
@ -845,20 +859,7 @@ let system_prompt = if !message.active_switchers.is_empty() {
match result {
Ok(Ok(())) => {
trace!("start.bas completed successfully for session {}", actual_session_id);
// Set Redis flag so start.bas does not run again for this session
if let Some(cache) = &self.state.cache {
if let Ok(mut conn) = cache.get_multiplexed_async_connection().await {
let key = format!("start_bas_executed:{}", actual_session_id);
let _: Result<(), redis::RedisError> = redis::cmd("SET")
.arg(&key)
.arg("1")
.arg("EX")
.arg(86400) // 24h TTL
.query_async(&mut conn)
.await;
}
}
info!("start.bas completed successfully for session {}", actual_session_id);
}
Ok(Err(e)) => {
error!("start.bas error for session {}: {}", actual_session_id, e);