Refactored log merging

This commit is contained in:
Thorsten Sommer 2026-01-13 10:00:36 +01:00
parent 4f8266f255
commit 00237b1784
Signed by: tsommer
GPG Key ID: 371BBA77A02C0108
9 changed files with 111 additions and 45 deletions

View File

@ -185,6 +185,7 @@ internal sealed class Program
var rustLogger = app.Services.GetRequiredService<ILogger<RustService>>(); var rustLogger = app.Services.GetRequiredService<ILogger<RustService>>();
rust.SetLogger(rustLogger); rust.SetLogger(rustLogger);
rust.SetEncryptor(encryption); rust.SetEncryptor(encryption);
TerminalLogger.SetRustService(rust);
RUST_SERVICE = rust; RUST_SERVICE = rust;
ENCRYPTION = encryption; ENCRYPTION = encryption;

View File

@ -0,0 +1,9 @@
namespace AIStudio.Tools.Rust;
public readonly record struct LogEventRequest(
string Timestamp,
string Level,
string Category,
string Message,
string? Exception
);

View File

@ -0,0 +1,3 @@
namespace AIStudio.Tools.Rust;
public readonly record struct LogEventResponse(bool Success, string Issue);

View File

@ -12,4 +12,27 @@ public sealed partial class RustService
{ {
return await this.http.GetFromJsonAsync<GetLogPathsResponse>("/log/paths", this.jsonRustSerializerOptions); return await this.http.GetFromJsonAsync<GetLogPathsResponse>("/log/paths", this.jsonRustSerializerOptions);
} }
/// <summary>
/// Sends a log event to the Rust runtime.
/// </summary>
/// <param name="timestamp">The timestamp of the log event.</param>
/// <param name="level">The log level.</param>
/// <param name="category">The category of the log event.</param>
/// <param name="message">The log message.</param>
/// <param name="exception">Optional exception details.</param>
public void LogEvent(string timestamp, string level, string category, string message, string? exception = null)
{
try
{
// Fire-and-forget the log event to avoid blocking:
var request = new LogEventRequest(timestamp, level, category, message, exception);
_ = this.http.PostAsJsonAsync("/log/event", request, this.jsonRustSerializerOptions);
}
catch
{
Console.WriteLine("Failed to send log event to Rust service.");
// Ignore errors to avoid log loops
}
}
} }

View File

@ -1,3 +1,5 @@
using AIStudio.Tools.Services;
using Microsoft.Extensions.Logging.Abstractions; using Microsoft.Extensions.Logging.Abstractions;
using Microsoft.Extensions.Logging.Console; using Microsoft.Extensions.Logging.Console;
@ -6,6 +8,17 @@ namespace AIStudio.Tools;
public sealed class TerminalLogger() : ConsoleFormatter(FORMATTER_NAME) public sealed class TerminalLogger() : ConsoleFormatter(FORMATTER_NAME)
{ {
public const string FORMATTER_NAME = "AI Studio Terminal Logger"; public const string FORMATTER_NAME = "AI Studio Terminal Logger";
private static RustService? RUST_SERVICE;
/// <summary>
/// Sets the Rust service for logging events.
/// </summary>
/// <param name="service">The Rust service instance.</param>
public static void SetRustService(RustService service)
{
RUST_SERVICE = service;
}
public override void Write<TState>(in LogEntry<TState> logEntry, IExternalScopeProvider? scopeProvider, TextWriter textWriter) public override void Write<TState>(in LogEntry<TState> logEntry, IExternalScopeProvider? scopeProvider, TextWriter textWriter)
{ {
@ -13,11 +26,15 @@ public sealed class TerminalLogger() : ConsoleFormatter(FORMATTER_NAME)
var timestamp = DateTimeOffset.UtcNow.ToString("yyyy-MM-dd HH:mm:ss.fff"); var timestamp = DateTimeOffset.UtcNow.ToString("yyyy-MM-dd HH:mm:ss.fff");
var logLevel = logEntry.LogLevel.ToString(); var logLevel = logEntry.LogLevel.ToString();
var category = logEntry.Category; var category = logEntry.Category;
var exception = logEntry.Exception?.ToString();
textWriter.Write($"=> {timestamp} [{logLevel}] {category}: {message}");
if (logEntry.Exception is not null) textWriter.Write($"[{timestamp}] {logLevel} [{category}] {message}");
textWriter.Write($" Exception was = {logEntry.Exception}"); if (exception is not null)
textWriter.Write($" Exception: {exception}");
textWriter.WriteLine(); textWriter.WriteLine();
// Send log event to Rust via API (fire-and-forget):
RUST_SERVICE?.LogEvent(timestamp, logLevel, category, message, exception);
} }
} }

View File

@ -1,2 +1,3 @@
# v26.1.2, build 232 (2026-01-xx xx:xx UTC) # v26.1.2, build 232 (2026-01-xx xx:xx UTC)
- Added the option to hide specific assistants by configuration plugins. This is useful for enterprise environments in organizations. - Added the option to hide specific assistants by configuration plugins. This is useful for enterprise environments in organizations.
- Fixed a logging bug that prevented log events from being recorded in some cases.

View File

@ -2,7 +2,7 @@ use std::collections::HashMap;
use std::sync::{Arc, Mutex}; use std::sync::{Arc, Mutex};
use base64::Engine; use base64::Engine;
use base64::prelude::BASE64_STANDARD; use base64::prelude::BASE64_STANDARD;
use log::{debug, error, info, warn}; use log::{error, info, warn};
use once_cell::sync::Lazy; use once_cell::sync::Lazy;
use rocket::get; use rocket::get;
use tauri::api::process::{Command, CommandChild, CommandEvent}; use tauri::api::process::{Command, CommandChild, CommandEvent};
@ -101,43 +101,11 @@ pub fn start_dotnet_server() {
*server_spawn_clone.lock().unwrap() = Some(child); *server_spawn_clone.lock().unwrap() = Some(child);
// Log the output of the .NET server: // Log the output of the .NET server:
// NOTE: Log events are sent via structured HTTP API calls.
// This loop serves for fundamental output (e.g., startup errors).
while let Some(CommandEvent::Stdout(line)) = rx.recv().await { while let Some(CommandEvent::Stdout(line)) = rx.recv().await {
// Remove newline characters from the end:
let line = line.trim_end(); let line = line.trim_end();
info!(Source = ".NET Server (stdout)"; "{line}");
// Starts the line with '=>'?
if line.starts_with("=>") {
// Yes. This means that the line is a log message from the .NET server.
// The format is: '<YYYY-MM-dd HH:mm:ss.fff> [<log level>] <source>: <message>'.
// We try to parse this line and log it with the correct log level:
let line = line.trim_start_matches("=>").trim();
let parts = line.split_once(": ").unwrap();
let left_part = parts.0.trim();
let message = parts.1.trim();
let parts = left_part.split_once("] ").unwrap();
let level = parts.0.split_once("[").unwrap().1.trim();
let source = parts.1.trim();
match level {
"Trace" => debug!(Source = ".NET Server", Comp = source; "{message}"),
"Debug" => debug!(Source = ".NET Server", Comp = source; "{message}"),
"Information" => info!(Source = ".NET Server", Comp = source; "{message}"),
"Warning" => warn!(Source = ".NET Server", Comp = source; "{message}"),
"Error" => error!(Source = ".NET Server", Comp = source; "{message}"),
"Critical" => error!(Source = ".NET Server", Comp = source; "{message}"),
_ => error!(Source = ".NET Server", Comp = source; "{message} (unknown log level '{level}')"),
}
} else {
let lower_line = line.to_lowercase();
if lower_line.contains("error") {
error!(Source = ".NET Server"; "{line}");
} else if lower_line.contains("warning") {
warn!(Source = ".NET Server"; "{line}");
} else {
info!(Source = ".NET Server"; "{line}");
}
}
} }
}); });
} }

View File

@ -6,11 +6,11 @@ use std::path::{absolute, PathBuf};
use std::sync::OnceLock; use std::sync::OnceLock;
use flexi_logger::{DeferredNow, Duplicate, FileSpec, Logger, LoggerHandle}; use flexi_logger::{DeferredNow, Duplicate, FileSpec, Logger, LoggerHandle};
use flexi_logger::writers::FileLogWriter; use flexi_logger::writers::FileLogWriter;
use log::kv; use log::{debug, error, info, kv, warn};
use log::kv::{Key, Value, VisitSource}; use log::kv::{Key, Value, VisitSource};
use rocket::get; use rocket::{get, post};
use rocket::serde::json::Json; use rocket::serde::json::Json;
use rocket::serde::Serialize; use rocket::serde::{Deserialize, Serialize};
use crate::api_token::APIToken; use crate::api_token::APIToken;
use crate::environment::is_dev; use crate::environment::is_dev;
@ -231,9 +231,52 @@ pub async fn get_log_paths(_token: APIToken) -> Json<LogPathsResponse> {
}) })
} }
/// Logs an event from the .NET server.
#[post("/log/event", data = "<event>")]
pub fn log_event(_token: APIToken, event: Json<LogEvent>) -> Json<LogEventResponse> {
let event = event.into_inner();
let message = &event.message.as_str();
let category = &event.category.as_str();
// Log with the appropriate level
match event.level.as_str() {
"Trace" | "Debug" => debug!(Source = ".NET Server", Comp = category; "{message}"),
"Information" => info!(Source = ".NET Server", Comp = category; "{message}"),
"Warning" => warn!(Source = ".NET Server", Comp = category; "{message}"),
"Error" | "Critical" => {
if let Some(ref ex) = event.exception {
error!(Source = ".NET Server", Comp = category; "{message} Exception: {ex}")
} else {
error!(Source = ".NET Server", Comp = category; "{message}")
}
},
_ => error!(Source = ".NET Server", Comp = category; "{message} (unknown log level '{}')", event.level),
}
Json(LogEventResponse { success: true, issue: String::new() })
}
/// The response the get log paths request. /// The response the get log paths request.
#[derive(Serialize)] #[derive(Serialize)]
pub struct LogPathsResponse { pub struct LogPathsResponse {
log_startup_path: String, log_startup_path: String,
log_app_path: String, log_app_path: String,
}
/// A log event from the .NET server.
#[derive(Deserialize)]
#[allow(unused)]
pub struct LogEvent {
timestamp: String,
level: String,
category: String,
message: String,
exception: Option<String>,
}
/// The response to a log event request.
#[derive(Serialize)]
pub struct LogEventResponse {
success: bool,
issue: String,
} }

View File

@ -86,6 +86,7 @@ pub fn start_runtime_api() {
crate::environment::read_enterprise_env_config_server_url, crate::environment::read_enterprise_env_config_server_url,
crate::file_data::extract_data, crate::file_data::extract_data,
crate::log::get_log_paths, crate::log::get_log_paths,
crate::log::log_event,
]) ])
.ignite().await.unwrap() .ignite().await.unwrap()
.launch().await.unwrap(); .launch().await.unwrap();