mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2026-02-13 06:21:37 +00:00
Refactored log merging
This commit is contained in:
parent
4f8266f255
commit
00237b1784
@ -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;
|
||||||
|
|||||||
9
app/MindWork AI Studio/Tools/Rust/LogEventRequest.cs
Normal file
9
app/MindWork AI Studio/Tools/Rust/LogEventRequest.cs
Normal file
@ -0,0 +1,9 @@
|
|||||||
|
namespace AIStudio.Tools.Rust;
|
||||||
|
|
||||||
|
public readonly record struct LogEventRequest(
|
||||||
|
string Timestamp,
|
||||||
|
string Level,
|
||||||
|
string Category,
|
||||||
|
string Message,
|
||||||
|
string? Exception
|
||||||
|
);
|
||||||
3
app/MindWork AI Studio/Tools/Rust/LogEventResponse.cs
Normal file
3
app/MindWork AI Studio/Tools/Rust/LogEventResponse.cs
Normal file
@ -0,0 +1,3 @@
|
|||||||
|
namespace AIStudio.Tools.Rust;
|
||||||
|
|
||||||
|
public readonly record struct LogEventResponse(bool Success, string Issue);
|
||||||
@ -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
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@ -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);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -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.
|
||||||
@ -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}");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|||||||
@ -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,
|
||||||
}
|
}
|
||||||
@ -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();
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user