// Copyright (c) .NET Foundation. All rights reserved.

// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

using System.Collections.Generic;

using System.Diagnostics;

using JetBrains.Annotations;

namespace Microsoft.EntityFrameworkCore.Internal

{

/// <summary>

/// This API supports the Entity Framework Core infrastructure and is not intended to be used

/// directly from your code. This API may change or be removed in future releases.

/// </summary>

[DebuggerStepThrough]

internal static class DictionaryExtensions

{

/// <summary>

/// This API supports the Entity Framework Core infrastructure and is not intended to be used

/// directly from your code. This API may change or be removed in future releases.

/// </summary>

public static TValue GetOrAddNew<TKey, TValue>(

[NotNull] this IDictionary<TKey, TValue> source,

[NotNull] TKey key)

where TValue : new()

{

if (!source.TryGetValue(key, out var value))

{

value = new TValue();

source.Add(key, value);

}

return value;

}

/// <summary>

/// This API supports the Entity Framework Core infrastructure and is not intended to be used

/// directly from your code. This API may change or be removed in future releases.

/// </summary>

public static TValue Find<TKey, TValue>(

[NotNull] this IDictionary<TKey, TValue> source,

[NotNull] TKey key)

=> !source.TryGetValue(key, out var value) ? default : value;

}

}