Skip to content

Commit 0ad3972

Browse files
committed
Add function to inject_logs
1 parent 9692a22 commit 0ad3972

1 file changed

Lines changed: 10 additions & 5 deletions

File tree

crates/core/src/host/module_host.rs

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1021,7 +1021,7 @@ impl ModuleHost {
10211021
_ => None,
10221022
};
10231023
if let Some(log_message) = log_message {
1024-
self.inject_logs(LogLevel::Error, &log_message)
1024+
self.inject_logs(LogLevel::Error, reducer_name, &log_message)
10251025
}
10261026

10271027
res
@@ -1110,10 +1110,15 @@ impl ModuleHost {
11101110
tokio::join!(self.module.scheduler().closed(), self.job_tx.closed());
11111111
}
11121112

1113-
pub fn inject_logs(&self, log_level: LogLevel, message: &str) {
1114-
self.replica_ctx()
1115-
.logger
1116-
.write(log_level, &Record::injected(message), &())
1113+
pub fn inject_logs(&self, log_level: LogLevel, reducer_name: &str, message: &str) {
1114+
self.replica_ctx().logger.write(
1115+
log_level,
1116+
&Record {
1117+
function: Some(reducer_name),
1118+
..Record::injected(message)
1119+
},
1120+
&(),
1121+
)
11171122
}
11181123

11191124
/// Execute a one-off query and send the results to the given client.

0 commit comments

Comments
 (0)