| | | 1 | | // Copyright (c) ZeroC, Inc. |
| | | 2 | | |
| | | 3 | | using System.Runtime.CompilerServices; |
| | | 4 | | |
| | | 5 | | namespace IceRpc.Transports.Internal; |
| | | 6 | | |
| | | 7 | | /// <summary>The flag enum extensions help with atomically set/get enumeration flags on an integer value. The |
| | | 8 | | /// enumeration underlying type must be an integer as well.</summary> |
| | | 9 | | internal static class FlagEnumExtensions |
| | | 10 | | { |
| | | 11 | | internal static bool TrySetFlag<T>(this ref int source, T flag) where T : Enum |
| | 93731 | 12 | | { |
| | 93731 | 13 | | int flagValue = Unsafe.As<T, int>(ref flag); |
| | 93731 | 14 | | int previousValue = Interlocked.Or(ref source, flagValue); |
| | 93731 | 15 | | return previousValue != (previousValue | flagValue); |
| | 93731 | 16 | | } |
| | | 17 | | |
| | | 18 | | internal static bool TrySetFlag<T>(this ref int source, T flag, out int newValue) where T : Enum |
| | 12990 | 19 | | { |
| | 12990 | 20 | | int flagValue = Unsafe.As<T, int>(ref flag); |
| | 12990 | 21 | | int previousValue = Interlocked.Or(ref source, flagValue); |
| | 12990 | 22 | | newValue = previousValue | flagValue; |
| | 12990 | 23 | | return previousValue != newValue; |
| | 12990 | 24 | | } |
| | | 25 | | |
| | | 26 | | internal static bool HasFlag<T>(this ref int source, T flag) where T : Enum |
| | 234551 | 27 | | { |
| | 234551 | 28 | | int flagValue = Unsafe.As<T, int>(ref flag); |
| | 234551 | 29 | | return (Volatile.Read(ref source) & flagValue) == flagValue; |
| | 234551 | 30 | | } |
| | | 31 | | |
| | | 32 | | internal static void ClearFlag<T>(this ref int source, T flag) where T : Enum |
| | 86697 | 33 | | { |
| | 86697 | 34 | | int flagValue = Unsafe.As<T, int>(ref flag); |
| | 86697 | 35 | | int previousValue = Interlocked.And(ref source, ~flagValue); |
| | 86697 | 36 | | if (previousValue == (previousValue & ~flagValue)) |
| | 0 | 37 | | { |
| | 0 | 38 | | throw new InvalidOperationException("The state was already cleared."); |
| | | 39 | | } |
| | 86697 | 40 | | } |
| | | 41 | | } |