nadekobot/src/Nadeko.Common/Collections/ConcurrentHashSet.cs

using System.Diagnostics;

namespace System.Collections.Generic;

[DebuggerDisplay("{_backingStore.Count}")]
public sealed class ConcurrentHashSet<T> : IReadOnlyCollection<T>, ICollection<T> where T : notnull
{
    private readonly ConcurrentDictionary<T, bool> _backingStore;
    
    public ConcurrentHashSet()
        => _backingStore = new();

    public ConcurrentHashSet(IEnumerable<T> values, IEqualityComparer<T>? comparer = null)
        => _backingStore = new(values.Select(x => new KeyValuePair<T, bool>(x, true)), comparer);

    public IEnumerator<T> GetEnumerator()
        => _backingStore.Keys.GetEnumerator();

    IEnumerator IEnumerable.GetEnumerator()
        => GetEnumerator();

    /// <summary>
    ///     Adds the specified item to the <see cref="ConcurrentHashSet{T}" />.
    /// </summary>
    /// <param name="item">The item to add.</param>
    /// <returns>
    ///     true if the items was added to the <see cref="ConcurrentHashSet{T}" />
    ///     successfully; false if it already exists.
    /// </returns>
    /// <exception cref="T:System.OverflowException">
    ///     The <see cref="ConcurrentHashSet{T}" />
    ///     contains too many items.
    /// </exception>
    public bool Add(T item)
        => _backingStore.TryAdd(item, true);

    void ICollection<T>.Add(T item)
        => Add(item);

    public void Clear()
        => _backingStore.Clear();

    public bool Contains(T item)
        => _backingStore.ContainsKey(item);

    public void CopyTo(T[] array, int arrayIndex)
    {
        ArgumentNullException.ThrowIfNull(array);
        
        if (arrayIndex < 0)
            throw new ArgumentOutOfRangeException(nameof(arrayIndex));
        
        if (arrayIndex >= array.Length)
            throw new ArgumentOutOfRangeException(nameof(arrayIndex));

        CopyToInternal(array, arrayIndex);
    }

    private void CopyToInternal(T[] array, int arrayIndex)
    {
        var len = array.Length;
        foreach (var (k, _) in _backingStore)
        {
            if (arrayIndex >= len)
                throw new IndexOutOfRangeException(nameof(arrayIndex));
            
            array[arrayIndex++] = k;
        }
    }

    bool ICollection<T>.Remove(T item)
        => TryRemove(item);

    public bool TryRemove(T item)
        => _backingStore.TryRemove(item, out _);

    public void RemoveWhere(Func<T, bool> predicate)
    {
        foreach (var elem in this.Where(predicate))
            TryRemove(elem);
    }

    public int Count
        => _backingStore.Count;

    public bool IsReadOnly
        => false;
}