src/Serilog/Rendering/MessageTemplateRenderer.cs - dev - aprogrammer/serilog

src/Serilog/Rendering/MessageTemplateRenderer.cs
107 строк3 KB
Ivan Maximov
Add space settings (#1885)
28 мар 2023, 03:50
28 мар 2023, 03:50f72a7a3
// Copyright 2013-2017 Serilog Contributors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

namespace Serilog.Rendering;

static class MessageTemplateRenderer
{
    static readonly JsonValueFormatter JsonValueFormatter = new("$type");

    [MethodImpl(MethodImplOptions.AggressiveInlining)]
    public static void Render(MessageTemplate messageTemplate, IReadOnlyDictionary<string, LogEventPropertyValue> properties, TextWriter output, string? format = null, IFormatProvider? formatProvider = null)
    {
        bool isLiteral = false, isJson = false;

        if (format != null)
        {
            for (var i = 0; i < format.Length; ++i)
            {
                if (format[i] == 'l')
                    isLiteral = true;
                else if (format[i] == 'j')
                    isJson = true;
            }
        }

        for (var ti = 0; ti < messageTemplate.TokenArray.Length; ++ti)
        {
            var token = messageTemplate.TokenArray[ti];
            if (token is TextToken tt)
            {
                RenderTextToken(tt, output);
            }
            else
            {
                var pt = (PropertyToken)token;
                RenderPropertyToken(pt, properties, output, formatProvider, isLiteral, isJson);
            }
        }
    }

    public static void RenderTextToken(TextToken tt, TextWriter output)
    {
        output.Write(tt.Text);
    }

    public static void RenderPropertyToken(PropertyToken pt, IReadOnlyDictionary<string, LogEventPropertyValue> properties, TextWriter output, IFormatProvider? formatProvider, bool isLiteral, bool isJson)
    {
        if (!properties.TryGetValue(pt.PropertyName, out var propertyValue))
        {
            output.Write(pt.RawText);
            return;
        }

        if (!pt.Alignment.HasValue)
        {
            RenderValue(propertyValue, isLiteral, isJson, output, pt.Format, formatProvider);
            return;
        }

        using var valueOutput = ReusableStringWriter.GetOrCreate();
        RenderValue(propertyValue, isLiteral, isJson, valueOutput, pt.Format, formatProvider);
        var sb = valueOutput.GetStringBuilder();

        if (sb.Length >= pt.Alignment.Value.Width)
        {
#if FEATURE_WRITE_STRINGBUILDER
            output.Write(sb);
#else
            output.Write(sb.ToString());
#endif
            return;
        }

#if FEATURE_WRITE_STRINGBUILDER
        Padding.Apply(output, sb, pt.Alignment.Value);
#else
        Padding.Apply(output, sb.ToString(), pt.Alignment.Value);
#endif
    }

    static void RenderValue(LogEventPropertyValue propertyValue, bool literal, bool json, TextWriter output, string? format, IFormatProvider? formatProvider)
    {
        if (literal && propertyValue is ScalarValue { Value: string str })
        {
            output.Write(str);
        }
        else if (json && format == null)
        {
            JsonValueFormatter.Format(propertyValue, output);
        }
        else
        {
            propertyValue.Render(output, format, formatProvider);
        }
    }
}