src/YesSql.Core/Data/WorkerQueryKey.cs - main - aprogrammer/yessql

src/YesSql.Core/Data/WorkerQueryKey.cs
201 строка5 KB
Mike Alhayek
Add Format and apply rules (#544)
03 май 2024, 03:22
03 май 2024, 03:22c748f4d
using System;
using System.Collections.Generic;

namespace YesSql.Data
{
    /// <summary>
    /// An instance of <see cref="WorkerQueryKey"/> represents the state of <see cref="WorkerQueryKey"/>.
    /// </summary>
    public class WorkerQueryKey : IEquatable<WorkerQueryKey>
    {
        private readonly string _prefix;
        private readonly long _id;
        private readonly long[] _ids;
        private readonly Dictionary<string, object> _parameters;
        private readonly int _hashCode;

        public WorkerQueryKey(string prefix, long[] ids)
        {
            ArgumentNullException.ThrowIfNull(prefix);
            ArgumentNullException.ThrowIfNull(ids);

            _prefix = prefix;
            _parameters = null;
            _ids = ids;
            _hashCode = BuildHashCode();
        }

        public WorkerQueryKey(string prefix, long id)
        {
            ArgumentNullException.ThrowIfNull(prefix);

            _prefix = prefix;
            _parameters = null;
            _id = id;
            _hashCode = BuildHashCode();
        }

        public WorkerQueryKey(string prefix, Dictionary<string, object> parameters)
        {
            ArgumentNullException.ThrowIfNull(prefix);
            ArgumentNullException.ThrowIfNull(parameters);

            _prefix = prefix;
            _parameters = parameters;
            _ids = null;
            _hashCode = BuildHashCode();
        }

        /// <inheritdoc />
        public override bool Equals(object obj)
        {
            if (obj is WorkerQueryKey other)
            {
                return Equals(other);
            }

            return false;
        }

        /// <inheritdoc />
        public bool Equals(WorkerQueryKey other)
        {
            if (!string.Equals(other._prefix, _prefix, StringComparison.Ordinal))
            {
                return false;
            }

            if (_parameters != null || other._parameters != null)
            {
                return SameParameters(_parameters, other._parameters);
            }

            if (_ids != null || other._ids != null)
            {
                return SameIds(_ids, other._ids);
            }

            return true;
        }

        private int BuildHashCode()
        {
            var hashCode = new HashCode();

            hashCode.Add(_prefix);

            if (_parameters != null)
            {
                foreach (var parameter in _parameters)
                {
                    if (parameter.Key != null)
                    {
                        hashCode.Add(parameter.Key);
                    }

                    if (parameter.Value != null)
                    {
                        hashCode.Add(parameter.Value);
                    }
                }
            }

            if (_ids != null)
            {
                foreach (var id in _ids)
                {
                    hashCode.Add(id);
                }
            }

            if (_id != 0)
            {
                hashCode.Add(_id);
            }

            return hashCode.ToHashCode();
        }

        /// <inheritdoc />
        public override int GetHashCode()
        {
            return _hashCode;
        }

        private static bool SameParameters(Dictionary<string, object> values1, Dictionary<string, object> values2)
        {
            if (values1 == values2)
            {
                return true;
            }

            if ((values1 == null && values2 != null) || (values1 != null && values2 == null) || values1.Count != values2.Count)
            {
                return false;
            }

            foreach (var entry1 in values1)
            {
                var key1 = entry1.Key;

                if (!values2.TryGetValue(key1, out var value2))
                {
                    return false;
                }

                var value1 = entry1.Value;

                if (value1 == null)
                {
                    if (value2 != null)
                    {
                        return false;
                    }
                }
                else
                {
                    if (!value1.Equals(value2))
                    {
                        return false;
                    }
                }
            }

            return true;
        }

        private static bool SameIds(long[] values1, long[] values2)
        {
            // If one is not null both need to be non-null
            if (!(values1 != null && values2 != null))
            {
                return false;
            }

            if (values1.Length != values2.Length)
            {
                return false;
            }

            for (var i = 0; i < values1.Length; i++)
            {
                if (values1[i] != values2[i])
                {
                    return false;
                }
            }

            return true;
        }

        public static bool operator ==(WorkerQueryKey left, WorkerQueryKey right)
        {
            return left.Equals(right);
        }

        public static bool operator !=(WorkerQueryKey left, WorkerQueryKey right)
        {
            return !(left == right);
        }
    }
}