Masuit.Tools/Masuit.Tools.Abstractions/Systems/LookupX.cs

using System.Collections;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Linq;

namespace Masuit.Tools.Systems;

public class LookupX<TKey, TElement> : IEnumerable<KeyValuePair<TKey, List<TElement>>>
{
    private readonly IDictionary<TKey, List<TElement>> _dictionary;

    public LookupX(Dictionary<TKey, List<TElement>> dic)
    {
        _dictionary = dic;
    }

    public LookupX(ConcurrentDictionary<TKey, List<TElement>> dic)
    {
        _dictionary = dic;
    }

    public IEnumerator<KeyValuePair<TKey, List<TElement>>> GetEnumerator()
    {
        return _dictionary.Select(pair => pair).GetEnumerator();
    }

    IEnumerator IEnumerable.GetEnumerator()
    {
        return GetEnumerator();
    }

    public bool Contains(TKey key)
    {
        return _dictionary.ContainsKey(key);
    }

    public int Count => _dictionary.Count;

    public List<TElement> this[TKey key] => _dictionary.TryGetValue(key, out var value) ? value : new List<TElement>();
}