| | | 1 | | namespace TeleFlow.Telegram.Internal.Handlers; |
| | | 2 | | |
| | | 3 | | /// <summary> |
| | | 4 | | /// Groups callback handler candidates by state and payload shape so callback dispatch can preserve |
| | | 5 | | /// typed-before-raw routing without scanning unrelated callback handlers on every callback query. |
| | | 6 | | /// </summary> |
| | | 7 | | internal sealed class TelegramCallbackHandlerCandidateSet |
| | | 8 | | { |
| | | 9 | | private readonly IReadOnlyDictionary<string, IReadOnlyList<TelegramHandlerCandidate>> _statefulTypedByState; |
| | | 10 | | private readonly IReadOnlyDictionary<string, IReadOnlyList<TelegramHandlerCandidate>> _statefulRawByState; |
| | | 11 | | |
| | | 12 | | private TelegramCallbackHandlerCandidateSet( |
| | | 13 | | IReadOnlyList<TelegramHandlerCandidate> statelessTyped, |
| | | 14 | | IReadOnlyList<TelegramHandlerCandidate> statelessRaw, |
| | | 15 | | IReadOnlyDictionary<string, IReadOnlyList<TelegramHandlerCandidate>> statefulTypedByState, |
| | | 16 | | IReadOnlyDictionary<string, IReadOnlyList<TelegramHandlerCandidate>> statefulRawByState) |
| | | 17 | | { |
| | | 18 | | StatelessTyped = statelessTyped; |
| | | 19 | | StatelessRaw = statelessRaw; |
| | 202 | 20 | | _statefulTypedByState = statefulTypedByState; |
| | 202 | 21 | | _statefulRawByState = statefulRawByState; |
| | 202 | 22 | | } |
| | | 23 | | |
| | | 24 | | public IReadOnlyList<TelegramHandlerCandidate> StatelessTyped { get; } |
| | | 25 | | |
| | | 26 | | public IReadOnlyList<TelegramHandlerCandidate> StatelessRaw { get; } |
| | | 27 | | |
| | 169 | 28 | | public bool HasStatefulCandidates => _statefulTypedByState.Count > 0 || _statefulRawByState.Count > 0; |
| | | 29 | | |
| | | 30 | | public IReadOnlyList<TelegramHandlerCandidate> GetStatefulTypedCandidates(string currentState) |
| | | 31 | | { |
| | 1 | 32 | | return _statefulTypedByState.TryGetValue(currentState, out var candidates) |
| | 1 | 33 | | ? candidates |
| | 1 | 34 | | : TelegramHandlerCandidateStateIndex.EmptyCandidates; |
| | | 35 | | } |
| | | 36 | | |
| | | 37 | | public IReadOnlyList<TelegramHandlerCandidate> GetStatefulRawCandidates(string currentState) |
| | | 38 | | { |
| | 1 | 39 | | return _statefulRawByState.TryGetValue(currentState, out var candidates) |
| | 1 | 40 | | ? candidates |
| | 1 | 41 | | : TelegramHandlerCandidateStateIndex.EmptyCandidates; |
| | | 42 | | } |
| | | 43 | | |
| | | 44 | | public static TelegramCallbackHandlerCandidateSet Create(IReadOnlyList<TelegramHandlerDescriptor> handlers) |
| | | 45 | | { |
| | 202 | 46 | | ArgumentNullException.ThrowIfNull(handlers); |
| | | 47 | | |
| | 202 | 48 | | var statelessTyped = new List<TelegramHandlerCandidate>(); |
| | 202 | 49 | | var statelessRaw = new List<TelegramHandlerCandidate>(); |
| | 202 | 50 | | var statefulTyped = new Dictionary<string, List<TelegramHandlerCandidate>>(StringComparer.Ordinal); |
| | 202 | 51 | | var statefulRaw = new Dictionary<string, List<TelegramHandlerCandidate>>(StringComparer.Ordinal); |
| | | 52 | | |
| | 922 | 53 | | for (var index = 0; index < handlers.Count; index++) |
| | | 54 | | { |
| | 259 | 55 | | var handler = handlers[index]; |
| | 259 | 56 | | var candidate = TelegramHandlerCandidate.Create(handler); |
| | 259 | 57 | | var isTyped = handler.CallbackPayloadType is not null; |
| | | 58 | | |
| | 259 | 59 | | if (handler.States.Count == 0) |
| | | 60 | | { |
| | 258 | 61 | | if (isTyped) |
| | | 62 | | { |
| | 43 | 63 | | statelessTyped.Add(candidate); |
| | | 64 | | } |
| | | 65 | | else |
| | | 66 | | { |
| | 215 | 67 | | statelessRaw.Add(candidate); |
| | | 68 | | } |
| | | 69 | | |
| | 215 | 70 | | continue; |
| | | 71 | | } |
| | | 72 | | |
| | 1 | 73 | | TelegramHandlerCandidateStateIndex.Add( |
| | 1 | 74 | | isTyped ? statefulTyped : statefulRaw, |
| | 1 | 75 | | handler.States, |
| | 1 | 76 | | candidate); |
| | | 77 | | } |
| | | 78 | | |
| | 202 | 79 | | return new TelegramCallbackHandlerCandidateSet( |
| | 202 | 80 | | statelessTyped.ToArray(), |
| | 202 | 81 | | statelessRaw.ToArray(), |
| | 202 | 82 | | TelegramHandlerCandidateStateIndex.Freeze(statefulTyped), |
| | 202 | 83 | | TelegramHandlerCandidateStateIndex.Freeze(statefulRaw)); |
| | | 84 | | } |
| | | 85 | | } |