src/Serilog/Core/LevelOverrideMap.cs - dev - aprogrammer/serilog

src/Serilog/Core/LevelOverrideMap.cs
87 строк3 KB
Travis Illig
Level override check is case-insensitive. (#2045)
24 апр 2024, 02:58
24 апр 2024, 02:5856a26bd
// Copyright 2016-2020 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.Core;

class LevelOverrideMap
{
    readonly LogEventLevel _defaultMinimumLevel;
    readonly LoggingLevelSwitch? _defaultLevelSwitch;

    readonly struct LevelOverride
    {
        public LevelOverride(string context, LoggingLevelSwitch levelSwitch)
        {
            Context = context;
            LevelSwitch = levelSwitch;
        }

        public string Context { get; }

        public LoggingLevelSwitch LevelSwitch { get; }
    }

    // There are two possible strategies to apply:
    //   1. Keep some bookkeeping data to consult when a new context is encountered, and a concurrent dictionary
    //        for exact matching ~ O(1), but slow and requires fences/locks; or,
    //   2. O(n) search over the raw configuration data every time (fast for small sets of overrides).
    // This implementation assumes there will only be a few overrides in each application, so chooses (2). This
    // is an assumption that's up for debate.
    readonly LevelOverride[] _overrides;

    public LevelOverrideMap(
        IDictionary<string, LoggingLevelSwitch> overrides,
        LogEventLevel defaultMinimumLevel,
        LoggingLevelSwitch? defaultLevelSwitch)
    {
        Guard.AgainstNull(overrides);

        _defaultLevelSwitch = defaultLevelSwitch;
        _defaultMinimumLevel = defaultLevelSwitch != null ? LevelAlias.Minimum : defaultMinimumLevel;

        // Descending order means that if we have a match, we're sure about it being the most specific.
        _overrides = overrides
            .OrderByDescending(o => o.Key)
            .Select(o => new LevelOverride(o.Key, o.Value))
            .ToArray();
    }

    public void GetEffectiveLevel(
#if FEATURE_SPAN
        ReadOnlySpan<char> context,
#else
        string context,
#endif
        out LogEventLevel minimumLevel,
        out LoggingLevelSwitch? levelSwitch)
    {
        foreach (var levelOverride in _overrides)
        {
            if (
                (
                    context.Length == levelOverride.Context.Length ||
                    (context.Length > levelOverride.Context.Length && context[levelOverride.Context.Length] == '.')
                ) &&
                context.StartsWith(levelOverride.Context, StringComparison.OrdinalIgnoreCase))
            {
                minimumLevel = LevelAlias.Minimum;
                levelSwitch = levelOverride.LevelSwitch;
                return;
            }
        }

        minimumLevel = _defaultMinimumLevel;
        levelSwitch = _defaultLevelSwitch;
    }
}