Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Telemetry, Logging] Emit {OriginalFormat} as the last property of the log record #6111

Merged
merged 6 commits into from
Mar 18, 2025
Merged
Changes from 4 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 10 additions & 7 deletions src/Generators/Microsoft.Gen.Logging/Emission/Emitter.Method.cs
Original file line number Diff line number Diff line change
@@ -102,7 +102,8 @@ private void GenLogMethod(LoggingMethod lm)
OutLn($"[{GeneratorUtilities.GeneratedCodeAttribute}] static string ({lambdaStateName}, {exceptionLambdaName}) =>");
OutOpenBrace();

if (GenVariableAssignments(lm, lambdaStateName, numReservedUnclassifiedTags, numReservedClassifiedTags))
if (!string.IsNullOrEmpty(lm.Message) &&
GenVariableAssignments(lm, lambdaStateName, numReservedUnclassifiedTags, numReservedClassifiedTags))
{
var mapped = Parsing.TemplateProcessor.MapTemplates(lm.Message, t =>
{
@@ -318,7 +319,14 @@ void GenTagWrites(LoggingMethod lm, string stateName, out int numReservedUnclass
{
OutLn();
OutLn($"_ = {stateName}.ReserveTagSpace({numReservedUnclassifiedTags});");

int count = numReservedUnclassifiedTags;

if (!string.IsNullOrEmpty(lm.Message))
{
OutLn($"{stateName}.TagArray[{--count}] = new(\"{{OriginalFormat}}\", {EscapeMessageString(lm.Message)});");
}

foreach (var p in lm.Parameters)
{
if (NeedsASlot(p) && !p.HasDataClassification)
@@ -367,11 +375,6 @@ void GenTagWrites(LoggingMethod lm, string stateName, out int numReservedUnclass
});
}
}

if (!string.IsNullOrEmpty(lm.Message))
{
OutLn($"{stateName}.TagArray[{--count}] = new(\"{{OriginalFormat}}\", {EscapeMessageString(lm.Message)});");
}
}

if (numReservedClassifiedTags > 0)
@@ -516,7 +519,7 @@ bool GenVariableAssignments(LoggingMethod lm, string lambdaStateName, int numRes
{
bool generatedAssignments = false;

int index = numReservedUnclassifiedTags - 1;
int index = numReservedUnclassifiedTags - 2;
foreach (var p in lm.Parameters)
{
if (NeedsASlot(p) && !p.HasDataClassification)
Original file line number Diff line number Diff line change
@@ -50,17 +50,19 @@ public void SetIncomingTags(IReadOnlyList<KeyValuePair<string, object?>> value)
{
get
{
if (index < _incomingTagCount)
// Keep the order of the if-else statements. It ensures that the {OriginalFormat}
// property is always the last tag in the list.
if (index < StaticTags!.Length)
{
return _incomingTags![index];
return StaticTags[index];
}
else if (index < _incomingTagCount + _extraTags.Count)
else if (index < StaticTags.Length + _extraTags.Count)
{
return _extraTags[index - _incomingTagCount];
return _extraTags[index - StaticTags.Length];
}
else
{
return StaticTags![index - _incomingTagCount - _extraTags.Count];
return _incomingTags![index - StaticTags.Length - _extraTags.Count];
}
}
}
Original file line number Diff line number Diff line change
@@ -56,21 +56,23 @@ public void SetIncomingTags(LoggerMessageState value)
{
get
{
if (index < _incomingTagsCount)
// Keep the order of the if-else statements. It ensures that the {OriginalFormat}
// property is always the last tag in the list.
if (index < _redactedTagsCount)
{
return _incomingTags![index];
return _redactedTags![index];
}
else if (index < _incomingTagsCount + _redactedTagsCount)
else if (index < _redactedTagsCount + _extraTags.Count)
{
return _redactedTags![index - _incomingTagsCount];
return _extraTags[index - _redactedTagsCount];
}
else if (index < _incomingTagsCount + _redactedTagsCount + _extraTags.Count)
else if (index < _redactedTagsCount + _extraTags.Count + StaticTags!.Length)
{
return _extraTags[index - _incomingTagsCount - _redactedTagsCount];
return StaticTags[index - _redactedTagsCount - _extraTags.Count];
}
else
{
return StaticTags![index - _incomingTagsCount - _redactedTagsCount - _extraTags.Count];
return _incomingTags![index - _redactedTagsCount - _extraTags.Count - StaticTags.Length];
}
}
}
Original file line number Diff line number Diff line change
@@ -322,24 +322,6 @@ private void LegacyPath<TState>(LogLevel logLevel, EventId eventId, TState state
joiner.Formatter = formatter;
joiner.State = state;

switch (state)
{
case IReadOnlyList<KeyValuePair<string, object?>> stateList:
joiner.SetIncomingTags(stateList);
break;

case IEnumerable<KeyValuePair<string, object?>> stateList:
joiner.EnrichmentTagCollector.AddRange(stateList);
break;

case null:
break;

default:
joiner.EnrichmentTagCollector.Add("{OriginalFormat}", state);
break;
}

List<Exception>? exceptions = null;

// enrich
@@ -356,12 +338,30 @@ private void LegacyPath<TState>(LogLevel logLevel, EventId eventId, TState state
}
}

// one last dedicated bit of enrichment
// enrich log data with exception information
if (exception != null)
{
RecordException(exception, joiner.EnrichmentTagCollector, config);
}

switch (state)
{
case IReadOnlyList<KeyValuePair<string, object?>> stateList:
joiner.SetIncomingTags(stateList);
break;

case IEnumerable<KeyValuePair<string, object?>> stateList:
joiner.EnrichmentTagCollector.AddRange(stateList);
break;

case null:
break;

default:
joiner.EnrichmentTagCollector.Add("{OriginalFormat}", state);
break;
}

bool? samplingDecision = null;
for (int i = 0; i < loggers.Length; i++)
{
Original file line number Diff line number Diff line change
@@ -5,6 +5,7 @@
using System.Collections.Generic;
using System.Diagnostics.CodeAnalysis;
using System.Globalization;
using System.Linq;
using System.Threading;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Logging.Testing;
@@ -923,6 +924,34 @@ public void FormattableTest()
Assert.Equal("Converted!", collector.LatestRecord.StructuredState!.GetValue("p1"));
}

[Fact]
public void OriginalFormatTest()
{
using var logger = Utils.GetLogger();
var collector = logger.FakeLogCollector;

OriginalFormatTestExtensions.M0(logger);
Assert.Equal(1, collector.Count);
EnsureOriginalFormatIsLastProperty(collector.LatestRecord, "M0", "M0");

collector.Clear();
OriginalFormatTestExtensions.M1(logger, "0");
Assert.Equal(1, collector.Count);
EnsureOriginalFormatIsLastProperty(collector.LatestRecord, "M1 {p0}", "M1 0");

collector.Clear();
OriginalFormatTestExtensions.M2(logger, "0", "1");
Assert.Equal(1, collector.Count);
EnsureOriginalFormatIsLastProperty(collector.LatestRecord, "M2 {p0}{p1}", "M2 01");

collector.Clear();
OriginalFormatTestExtensions.M3(logger, "0", "1", "2");
Assert.Equal(1, collector.Count);
Assert.Empty(collector.LatestRecord.Message);
Assert.Equal(3, collector.LatestRecord.StructuredState!.Count);
AssertLastState(collector, new("p0", "0"), new("p1", "1"), new("p2", "2"));
}

private static void AssertLastState(FakeLogCollector collector, params KeyValuePair<string, string?>[] expected)
{
var rol = (IReadOnlyList<KeyValuePair<string, string>>)collector.LatestRecord.State!;
@@ -949,4 +978,15 @@ private static void TestCollection(int expected, FakeLogCollector collector)
}
}
}

private static void EnsureOriginalFormatIsLastProperty(FakeLogRecord record, string format, string message)
{
var state = (IReadOnlyList<KeyValuePair<string, string>>)record.State!;
Assert.NotNull(state);

var pair = state.Last();
Assert.Equal("{OriginalFormat}", pair.Key);
Assert.Equal(format, pair.Value);
Assert.Equal(message, record.Message);
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using Microsoft.Extensions.Logging;

namespace TestClasses
{
internal static partial class OriginalFormatTestExtensions
{
[LoggerMessage(0, LogLevel.Information, "M0")]
public static partial void M0(ILogger logger);

[LoggerMessage(1, LogLevel.Information, "M1 {p0}")]
public static partial void M1(ILogger logger, string p0);

[LoggerMessage(2, LogLevel.Information, "M2 {p0}{p1}")]
public static partial void M2(ILogger logger, string p0, string p1);

[LoggerMessage(LogLevel.Information)]
public static partial void M3(ILogger logger, string p0, string p1, string p2);
}
}
Original file line number Diff line number Diff line change
@@ -23,9 +23,9 @@ namespace Test
var state = global::Microsoft.Extensions.Logging.LoggerMessageHelper.ThreadLocalState;

_ = state.ReserveTagSpace(3);
state.TagArray[2] = new("classWithNullablePropertyParam.NullableDateTime", classWithNullablePropertyParam?.NullableDateTime);
state.TagArray[1] = new("classWithNullablePropertyParam.NonNullableDateTime", classWithNullablePropertyParam?.NonNullableDateTime);
state.TagArray[0] = new("{OriginalFormat}", "Testing nullable property within class here...");
state.TagArray[2] = new("{OriginalFormat}", "Testing nullable property within class here...");
state.TagArray[1] = new("classWithNullablePropertyParam.NullableDateTime", classWithNullablePropertyParam?.NullableDateTime);
state.TagArray[0] = new("classWithNullablePropertyParam.NonNullableDateTime", classWithNullablePropertyParam?.NonNullableDateTime);

logger.Log(
global::Microsoft.Extensions.Logging.LogLevel.Information,
Original file line number Diff line number Diff line change
@@ -6,9 +6,6 @@
Name: SystemResourcesInfo
},
State: [
{
{OriginalFormat}: System resources information: CpuLimit = {cpuLimit}, CpuRequest = {cpuRequest}, MemoryLimit = {memoryLimit}, MemoryRequest = {memoryRequest}.
},
{
memoryRequest: 1048576
},
@@ -20,12 +17,12 @@
},
{
cpuLimit: 20
},
{
{OriginalFormat}: System resources information: CpuLimit = {cpuLimit}, CpuRequest = {cpuRequest}, MemoryLimit = {memoryLimit}, MemoryRequest = {memoryRequest}.
}
],
StructuredState: [
{
{OriginalFormat}: System resources information: CpuLimit = {cpuLimit}, CpuRequest = {cpuRequest}, MemoryLimit = {memoryLimit}, MemoryRequest = {memoryRequest}.
},
{
memoryRequest: 1048576
},
@@ -37,6 +34,9 @@
},
{
cpuLimit: 20
},
{
{OriginalFormat}: System resources information: CpuLimit = {cpuLimit}, CpuRequest = {cpuRequest}, MemoryLimit = {memoryLimit}, MemoryRequest = {memoryRequest}.
}
],
Message: System resources information: CpuLimit = 20, CpuRequest = 0.078125, MemoryLimit = 1048576, MemoryRequest = 1048576.,
Original file line number Diff line number Diff line change
@@ -5,9 +5,6 @@
Name: SnapshotReceived
},
State: [
{
{OriginalFormat}: Snapshot received: TotalTimeSinceStart={totalTimeSinceStart}, KernelTimeSinceStart={kernelTimeSinceStart}, UserTimeSinceStart={userTimeSinceStart}, MemoryUsageInBytes={memoryUsageInBytes}.
},
{
memoryUsageInBytes: 500
},
@@ -19,12 +16,12 @@
},
{
totalTimeSinceStart: 730119.00:00:00
},
{
{OriginalFormat}: Snapshot received: TotalTimeSinceStart={totalTimeSinceStart}, KernelTimeSinceStart={kernelTimeSinceStart}, UserTimeSinceStart={userTimeSinceStart}, MemoryUsageInBytes={memoryUsageInBytes}.
}
],
StructuredState: [
{
{OriginalFormat}: Snapshot received: TotalTimeSinceStart={totalTimeSinceStart}, KernelTimeSinceStart={kernelTimeSinceStart}, UserTimeSinceStart={userTimeSinceStart}, MemoryUsageInBytes={memoryUsageInBytes}.
},
{
memoryUsageInBytes: 500
},
@@ -36,6 +33,9 @@
},
{
totalTimeSinceStart: 730119.00:00:00
},
{
{OriginalFormat}: Snapshot received: TotalTimeSinceStart={totalTimeSinceStart}, KernelTimeSinceStart={kernelTimeSinceStart}, UserTimeSinceStart={userTimeSinceStart}, MemoryUsageInBytes={memoryUsageInBytes}.
}
],
Message: Snapshot received: TotalTimeSinceStart=730119.00:00:00, KernelTimeSinceStart=00:00:01, UserTimeSinceStart=00:00:01, MemoryUsageInBytes=500.,
Original file line number Diff line number Diff line change
@@ -27,9 +27,6 @@
Name: SystemResourcesInfo
},
State: [
{
{OriginalFormat}: System resources information: CpuLimit = {cpuLimit}, CpuRequest = {cpuRequest}, MemoryLimit = {memoryLimit}, MemoryRequest = {memoryRequest}.
},
{
memoryRequest: 2000
},
@@ -41,12 +38,12 @@
},
{
cpuLimit: 1
},
{
{OriginalFormat}: System resources information: CpuLimit = {cpuLimit}, CpuRequest = {cpuRequest}, MemoryLimit = {memoryLimit}, MemoryRequest = {memoryRequest}.
}
],
StructuredState: [
{
{OriginalFormat}: System resources information: CpuLimit = {cpuLimit}, CpuRequest = {cpuRequest}, MemoryLimit = {memoryLimit}, MemoryRequest = {memoryRequest}.
},
{
memoryRequest: 2000
},
@@ -58,6 +55,9 @@
},
{
cpuLimit: 1
},
{
{OriginalFormat}: System resources information: CpuLimit = {cpuLimit}, CpuRequest = {cpuRequest}, MemoryLimit = {memoryLimit}, MemoryRequest = {memoryRequest}.
}
],
Message: System resources information: CpuLimit = 1, CpuRequest = 1, MemoryLimit = 2000, MemoryRequest = 2000.,
Original file line number Diff line number Diff line change
@@ -27,9 +27,6 @@
Name: SystemResourcesInfo
},
State: [
{
{OriginalFormat}: System resources information: CpuLimit = {cpuLimit}, CpuRequest = {cpuRequest}, MemoryLimit = {memoryLimit}, MemoryRequest = {memoryRequest}.
},
{
memoryRequest: 2000
},
@@ -41,12 +38,12 @@
},
{
cpuLimit: 1
},
{
{OriginalFormat}: System resources information: CpuLimit = {cpuLimit}, CpuRequest = {cpuRequest}, MemoryLimit = {memoryLimit}, MemoryRequest = {memoryRequest}.
}
],
StructuredState: [
{
{OriginalFormat}: System resources information: CpuLimit = {cpuLimit}, CpuRequest = {cpuRequest}, MemoryLimit = {memoryLimit}, MemoryRequest = {memoryRequest}.
},
{
memoryRequest: 2000
},
@@ -58,6 +55,9 @@
},
{
cpuLimit: 1
},
{
{OriginalFormat}: System resources information: CpuLimit = {cpuLimit}, CpuRequest = {cpuRequest}, MemoryLimit = {memoryLimit}, MemoryRequest = {memoryRequest}.
}
],
Message: System resources information: CpuLimit = 1, CpuRequest = 1, MemoryLimit = 2000, MemoryRequest = 2000.,
Loading