| | | 1 | | // Copyright (c) ZeroC, Inc. |
| | | 2 | | |
| | | 3 | | using System.Collections.Frozen; |
| | | 4 | | |
| | | 5 | | namespace IceRpc; |
| | | 6 | | |
| | | 7 | | /// <summary>Provides methods for routing incoming requests to dispatchers.</summary> |
| | | 8 | | /// <example> |
| | | 9 | | /// The following example shows how you would install a middleware, and map a service. |
| | | 10 | | /// <code source="../../docfx/examples/IceRpc.Examples/RouterExamples.cs" region="CreatingAndUsingTheRouterWithMiddlewar |
| | | 11 | | /// You can easily create your own middleware and add it to the router. The next example shows how you can create a |
| | | 12 | | /// middleware using an <see cref="InlineDispatcher"/> and add it to the router with |
| | | 13 | | /// <see cref="Use(Func{IDispatcher, IDispatcher})"/>. |
| | | 14 | | /// <code source="../../docfx/examples/IceRpc.Examples/RouterExamples.cs" region="CreatingAndUsingTheRouterWithAnInlineD |
| | | 15 | | /// </example> |
| | | 16 | | /// <remarks><para>The <see cref="Router"/> class allows you to define a dispatch pipeline for customizing how incoming |
| | | 17 | | /// requests are processed. You utilize the Router class for creating routes with various middleware, sub-routers, |
| | | 18 | | /// and dispatchers.</para> |
| | | 19 | | /// <para>Incoming requests flow through the dispatch pipeline. An incoming request is first processed by the router's |
| | | 20 | | /// middleware and then routed to a target dispatcher based on its path. The target dispatcher returns an outgoing |
| | | 21 | | /// response that goes through the dispatch pipeline in the opposite direction.</para> |
| | | 22 | | /// <para>The routing algorithm determines how incoming requests are routed to dispatchers. The router first checks if |
| | | 23 | | /// a dispatcher is registered with the request's path, which corresponds to dispatchers registered using |
| | | 24 | | /// <see cref="Map(string, IDispatcher)"/>. If there isn't a dispatcher registered for the request's path, the router |
| | | 25 | | /// looks for dispatchers registered with a matching prefix, which corresponds to dispatchers installed using |
| | | 26 | | /// <see cref="Mount(string, IDispatcher)"/>. When searching for a matching prefix, the router starts with the request |
| | | 27 | | /// path and successively tries chopping segments from the end of the path until either the path is exhausted or a |
| | | 28 | | /// dispatcher matching the prefix is found. Finally, if the router cannot find any dispatcher, it returns an |
| | | 29 | | /// <see cref="OutgoingResponse"/> with a <see cref="StatusCode.NotFound"/> status code.</para></remarks> |
| | | 30 | | public sealed class Router : IDispatcher |
| | | 31 | | { |
| | | 32 | | /// <summary>Gets the absolute path-prefix of this router. The absolute path of a service added to this |
| | | 33 | | /// Router is: <c>$"{AbsolutePrefix}{path}"</c> where <c>path</c> corresponds to the argument given to |
| | | 34 | | /// <see cref="Map(string, IDispatcher)" />.</summary> |
| | | 35 | | /// <value>The absolute prefix of this router. It is either an empty string or a string with two or more |
| | | 36 | | /// characters starting with a <c>/</c>.</value> |
| | 208 | 37 | | public string AbsolutePrefix { get; } = ""; |
| | | 38 | | |
| | | 39 | | // When searching in the prefixMatchRoutes, we search up to MaxSegments before giving up. This prevents a |
| | | 40 | | // a malicious client from sending a request with a huge number of segments (/a/a/a/a/a/a/a/a/a/a...) that |
| | | 41 | | // results in numerous unsuccessful lookups. |
| | | 42 | | private const int MaxSegments = 10; |
| | | 43 | | |
| | | 44 | | private readonly Lazy<IDispatcher> _dispatcher; |
| | 53 | 45 | | private readonly Dictionary<string, IDispatcher> _exactMatchRoutes = new(); |
| | | 46 | | |
| | 53 | 47 | | private readonly Stack<Func<IDispatcher, IDispatcher>> _middlewareStack = new(); |
| | | 48 | | |
| | 53 | 49 | | private readonly Dictionary<string, IDispatcher> _prefixMatchRoutes = new(); |
| | | 50 | | |
| | | 51 | | /// <summary>Constructs a top-level router.</summary> |
| | 106 | 52 | | public Router() => _dispatcher = new Lazy<IDispatcher>(CreateDispatchPipeline); |
| | | 53 | | |
| | | 54 | | /// <summary>Constructs a router with an absolute prefix.</summary> |
| | | 55 | | /// <param name="absolutePrefix">The absolute prefix of the new router. It must start with a <c>/</c>.</param> |
| | | 56 | | /// <exception cref="FormatException">Thrown if <paramref name="absolutePrefix" /> is not a valid path. |
| | | 57 | | /// </exception> |
| | | 58 | | public Router(string absolutePrefix) |
| | 16 | 59 | | : this() |
| | 16 | 60 | | { |
| | 16 | 61 | | ServiceAddress.CheckPath(absolutePrefix); |
| | 15 | 62 | | absolutePrefix = NormalizePrefix(absolutePrefix); |
| | 15 | 63 | | AbsolutePrefix = absolutePrefix.Length > 1 ? absolutePrefix : ""; |
| | 15 | 64 | | } |
| | | 65 | | |
| | | 66 | | /// <inheritdoc/> |
| | | 67 | | public ValueTask<OutgoingResponse> DispatchAsync( |
| | | 68 | | IncomingRequest request, |
| | | 69 | | CancellationToken cancellationToken = default) => |
| | 119 | 70 | | _dispatcher.Value.DispatchAsync(request, cancellationToken); |
| | | 71 | | |
| | | 72 | | /// <summary>Registers a route with a path. If there is an existing route at the same path, it is replaced. |
| | | 73 | | /// </summary> |
| | | 74 | | /// <param name="path">The path of this route. It must match exactly the path of the request. In particular, it |
| | | 75 | | /// must start with a <c>/</c>.</param> |
| | | 76 | | /// <param name="dispatcher">The target of this route. It is typically a service.</param> |
| | | 77 | | /// <returns>This router.</returns> |
| | | 78 | | /// <exception cref="FormatException">Thrown if <paramref name="path" /> is not a valid path.</exception> |
| | | 79 | | /// <exception cref="InvalidOperationException">Thrown if <see cref="IDispatcher.DispatchAsync" /> was already |
| | | 80 | | /// called on this router.</exception> |
| | | 81 | | /// <seealso cref="Mount" /> |
| | | 82 | | public Router Map(string path, IDispatcher dispatcher) |
| | 26 | 83 | | { |
| | 26 | 84 | | if (_dispatcher.IsValueCreated) |
| | 1 | 85 | | { |
| | 1 | 86 | | throw new InvalidOperationException( |
| | 1 | 87 | | $"Cannot call {nameof(Map)} after calling {nameof(IDispatcher.DispatchAsync)}."); |
| | | 88 | | } |
| | 25 | 89 | | ServiceAddress.CheckPath(path); |
| | 25 | 90 | | _exactMatchRoutes[path] = dispatcher; |
| | 25 | 91 | | return this; |
| | 25 | 92 | | } |
| | | 93 | | |
| | | 94 | | /// <summary>Registers a route with a prefix. If there is an existing route at the same prefix, it is replaced. |
| | | 95 | | /// </summary> |
| | | 96 | | /// <param name="prefix">The prefix of this route. This prefix will be compared with the start of the path of |
| | | 97 | | /// the request.</param> |
| | | 98 | | /// <param name="dispatcher">The target of this route.</param> |
| | | 99 | | /// <returns>This router.</returns> |
| | | 100 | | /// <exception cref="FormatException">Thrown if <paramref name="prefix" /> is not a valid path.</exception> |
| | | 101 | | /// <exception cref="InvalidOperationException">Thrown if <see cref="IDispatcher.DispatchAsync" /> was already |
| | | 102 | | /// called on this router.</exception> |
| | | 103 | | /// <seealso cref="Map(string, IDispatcher)" /> |
| | | 104 | | public Router Mount(string prefix, IDispatcher dispatcher) |
| | 24 | 105 | | { |
| | 24 | 106 | | if (_dispatcher.IsValueCreated) |
| | 1 | 107 | | { |
| | 1 | 108 | | throw new InvalidOperationException( |
| | 1 | 109 | | $"Cannot call {nameof(Mount)} after calling {nameof(IDispatcher.DispatchAsync)}."); |
| | | 110 | | } |
| | 23 | 111 | | ServiceAddress.CheckPath(prefix); |
| | 21 | 112 | | prefix = NormalizePrefix(prefix); |
| | 21 | 113 | | _prefixMatchRoutes[prefix] = dispatcher; |
| | 21 | 114 | | return this; |
| | 21 | 115 | | } |
| | | 116 | | |
| | | 117 | | /// <summary>Installs a middleware in this router. A middleware must be installed before calling |
| | | 118 | | /// <see cref="IDispatcher.DispatchAsync" />.</summary> |
| | | 119 | | /// <param name="middleware">The middleware to install.</param> |
| | | 120 | | /// <returns>This router.</returns> |
| | | 121 | | /// <exception cref="InvalidOperationException">Thrown if <see cref="IDispatcher.DispatchAsync" /> was already |
| | | 122 | | /// called on this router.</exception> |
| | | 123 | | public Router Use(Func<IDispatcher, IDispatcher> middleware) |
| | 38 | 124 | | { |
| | 38 | 125 | | if (_dispatcher.IsValueCreated) |
| | 1 | 126 | | { |
| | 1 | 127 | | throw new InvalidOperationException( |
| | 1 | 128 | | $"All middleware must be registered before calling {nameof(IDispatcher.DispatchAsync)}."); |
| | | 129 | | } |
| | 37 | 130 | | _middlewareStack.Push(middleware); |
| | 37 | 131 | | return this; |
| | 37 | 132 | | } |
| | | 133 | | |
| | | 134 | | /// <summary>Returns a string that represents this router.</summary> |
| | | 135 | | /// <returns>A string that represents this router.</returns> |
| | 0 | 136 | | public override string ToString() => AbsolutePrefix.Length > 0 ? $"router({AbsolutePrefix})" : "router"; |
| | | 137 | | |
| | | 138 | | // Trim trailing slashes but keep the leading slash. |
| | | 139 | | private static string NormalizePrefix(string prefix) |
| | 68 | 140 | | { |
| | 68 | 141 | | if (prefix.Length > 1) |
| | 58 | 142 | | { |
| | 58 | 143 | | prefix = prefix.TrimEnd('/'); |
| | 58 | 144 | | if (prefix.Length == 0) |
| | 2 | 145 | | { |
| | 2 | 146 | | prefix = "/"; |
| | 2 | 147 | | } |
| | 58 | 148 | | } |
| | 68 | 149 | | return prefix; |
| | 68 | 150 | | } |
| | | 151 | | |
| | | 152 | | private IDispatcher CreateDispatchPipeline() |
| | 43 | 153 | | { |
| | 43 | 154 | | var exactMatchRoutes = _exactMatchRoutes.ToFrozenDictionary(); |
| | 43 | 155 | | var prefixMatchRoutes = _prefixMatchRoutes.ToFrozenDictionary(); |
| | | 156 | | |
| | | 157 | | // The last dispatcher of the pipeline: |
| | 43 | 158 | | IDispatcher dispatchPipeline = new InlineDispatcher( |
| | 43 | 159 | | (request, cancellationToken) => |
| | 118 | 160 | | { |
| | 118 | 161 | | string path = request.Path; |
| | 43 | 162 | | |
| | 118 | 163 | | if (AbsolutePrefix.Length > 0) |
| | 8 | 164 | | { |
| | 43 | 165 | | // Remove AbsolutePrefix from path. AbsolutePrefix starts with a '/' but usually does not end with |
| | 43 | 166 | | // one. |
| | 8 | 167 | | path = path.StartsWith(AbsolutePrefix, StringComparison.Ordinal) ? |
| | 8 | 168 | | (path.Length == AbsolutePrefix.Length ? "/" : path[AbsolutePrefix.Length..]) : |
| | 8 | 169 | | throw new InvalidOperationException( |
| | 8 | 170 | | $"Received request for path '{path}' in router mounted at '{AbsolutePrefix}'."); |
| | 8 | 171 | | } |
| | 43 | 172 | | // else there is nothing to remove |
| | 43 | 173 | | |
| | 43 | 174 | | // First check for an exact match |
| | 118 | 175 | | if (exactMatchRoutes.TryGetValue(path, out IDispatcher? dispatcher)) |
| | 101 | 176 | | { |
| | 101 | 177 | | return dispatcher.DispatchAsync(request, cancellationToken); |
| | 43 | 178 | | } |
| | 43 | 179 | | else |
| | 17 | 180 | | { |
| | 43 | 181 | | // Then a prefix match |
| | 17 | 182 | | string prefix = NormalizePrefix(path); |
| | 43 | 183 | | |
| | 100 | 184 | | foreach (int _ in Enumerable.Range(0, MaxSegments)) |
| | 33 | 185 | | { |
| | 33 | 186 | | if (prefixMatchRoutes.TryGetValue(prefix, out dispatcher)) |
| | 16 | 187 | | { |
| | 16 | 188 | | return dispatcher.DispatchAsync(request, cancellationToken); |
| | 43 | 189 | | } |
| | 43 | 190 | | |
| | 17 | 191 | | if (prefix == "/") |
| | 1 | 192 | | { |
| | 1 | 193 | | return new(new OutgoingResponse(request, StatusCode.NotFound)); |
| | 43 | 194 | | } |
| | 43 | 195 | | |
| | 43 | 196 | | // Cut last segment |
| | 16 | 197 | | int lastSlashPos = prefix.LastIndexOf('/'); |
| | 16 | 198 | | prefix = lastSlashPos > 0 ? NormalizePrefix(prefix[..lastSlashPos]) : "/"; |
| | 43 | 199 | | // and try again with the new shorter prefix |
| | 16 | 200 | | } |
| | 0 | 201 | | return new(new OutgoingResponse(request, StatusCode.InvalidData, "Too many segments in path.")); |
| | 43 | 202 | | } |
| | 161 | 203 | | }); |
| | | 204 | | |
| | 203 | 205 | | foreach (Func<IDispatcher, IDispatcher> middleware in _middlewareStack) |
| | 37 | 206 | | { |
| | 37 | 207 | | dispatchPipeline = middleware(dispatchPipeline); |
| | 37 | 208 | | } |
| | 43 | 209 | | _middlewareStack.Clear(); // we no longer need these functions |
| | 43 | 210 | | return dispatchPipeline; |
| | 43 | 211 | | } |
| | | 212 | | } |