Added option to use the controller action as the throttle key

WebApiThrottle/Models/EndpointThrottlingMethod.cs

@@ -0,0 +1,8 @@
+namespace WebApiThrottle.Models
+{
+    public enum EndpointThrottlingMethod
+    {
+        Url,
+        Action
+    }
+}

WebApiThrottle/Models/RequestIdentity.cs

@@ -18,6 +18,10 @@ public class RequestIdentity
 
         public string Endpoint { get; set; }
 
+        public string ActionName { get; set; }
+
+        public string ControllerName { get; set; }
+
         public bool ForceWhiteList { get; set; }
     }
 }

WebApiThrottle/ThrottlePolicy.cs

@@ -6,6 +6,8 @@
 
 namespace WebApiThrottle
 {
+    using Models;
+
     /// <summary>
     /// Rate limits policy
     /// </summary>
@@ -80,6 +82,8 @@ public ThrottlePolicy(long? perSecond = null, long? perMinute = null, long? perH
         /// </summary>
         public bool EndpointThrottling { get; set; }
 
+        public EndpointThrottlingMethod EndpointThrottleMethod { get; set; }
+
         public List<string> EndpointWhitelist { get; set; }
 
         public IDictionary<string, RateLimits> EndpointRules { get; set; }

WebApiThrottle/ThrottlingCore.cs

@@ -8,6 +8,8 @@
 
 namespace WebApiThrottle
 {
+    using Models;
+
     /// <summary>
     /// Common code shared between ThrottlingHandler and ThrottlingFilter
     /// </summary>
@@ -135,7 +137,17 @@ internal string ComputeThrottleKey(RequestIdentity requestIdentity, RateLimitPer
 
             if (Policy.EndpointThrottling)
             {
-                keyValues.Add(requestIdentity.Endpoint);
+                if (Policy.EndpointThrottleMethod == EndpointThrottlingMethod.Action &&
+                    !string.IsNullOrWhiteSpace(requestIdentity.ActionName) &&
+                    !string.IsNullOrWhiteSpace(requestIdentity.ControllerName))
+                {
+                    //TODO: think of a better way to do this
+                    keyValues.Add(requestIdentity.ControllerName + "." + requestIdentity.ActionName);
+                }
+                else if (Policy.EndpointThrottleMethod == EndpointThrottlingMethod.Url)
+                {
+                    keyValues.Add(requestIdentity.Endpoint);
+                }
             }
 
             keyValues.Add(period.ToString());

WebApiThrottle/ThrottlingFilter.cs

@@ -228,6 +228,14 @@ protected virtual RequestIdentity SetIdentity(HttpRequestMessage request)
             entry.ClientIp = core.GetClientIp(request).ToString();
             entry.Endpoint = request.RequestUri.AbsolutePath.ToLowerInvariant();
             entry.ClientKey = request.Headers.Contains("Authorization-Token") ? request.Headers.GetValues("Authorization-Token").First() : "anon";
+
+            var actionDescriptor = request.GetActionDescriptor();
+            if (actionDescriptor == null)
+                return entry;
+
+            entry.ActionName = actionDescriptor.ActionName;
+            if(actionDescriptor.ControllerDescriptor != null)
+                entry.ControllerName = actionDescriptor.ControllerDescriptor.ControllerName;
 
             return entry;
         }

WebApiThrottle/WebApiThrottle.csproj

@@ -69,6 +69,7 @@
   </ItemGroup>
   <ItemGroup>
     <Compile Include="Attributes\DisableThrottingAttribute.cs" />
+    <Compile Include="Models\EndpointThrottlingMethod.cs" />
     <Compile Include="Net\DefaultIpAddressParser.cs" />
     <Compile Include="Net\HttpRequestExtensions.cs" />
     <Compile Include="Net\IIpAddressParser.cs" />