dotnet
diff --git a/‎src/AspNetCore/WebApi/src/Asp.Versioning.Mvc/ApplicationModels/ApiBehaviorSpecification.cs
Copy file name to clipboardExpand all lines: src/AspNetCore/WebApi/src/Asp.Versioning.Mvc/ApplicationModels/ApiBehaviorSpecification.cs
+15-9Lines changed: 15 additions & 9 deletions b/‎src/AspNetCore/WebApi/src/Asp.Versioning.Mvc/ApplicationModels/ApiBehaviorSpecification.cs
Copy file name to clipboardExpand all lines: src/AspNetCore/WebApi/src/Asp.Versioning.Mvc/ApplicationModels/ApiBehaviorSpecification.cs
+15-9Lines changed: 15 additions & 9 deletions
@@ -3,6 +3,7 @@
 namespace Asp.Versioning.ApplicationModels;
 
 using Microsoft.AspNetCore.Mvc.ApplicationModels;
+using Microsoft.AspNetCore.Mvc.Infrastructure;
 using System.Reflection;
 
 /// <summary>
@@ -11,17 +12,22 @@ namespace Asp.Versioning.ApplicationModels;
 [CLSCompliant( false )]
 public sealed class ApiBehaviorSpecification : IApiControllerSpecification
 {
-    static ApiBehaviorSpecification()
+    /// <inheritdoc />
+    public bool IsSatisfiedBy( ControllerModel controller )
     {
-        const string ApiBehaviorApplicationModelProviderTypeName = "Microsoft.AspNetCore.Mvc.ApplicationModels.ApiBehaviorApplicationModelProvider, Microsoft.AspNetCore.Mvc.Core";
-        var type = Type.GetType( ApiBehaviorApplicationModelProviderTypeName, throwOnError: true )!;
-        var method = type.GetRuntimeMethods().Single( m => m.Name == "IsApiController" );
+        if ( controller == null )
+        {
+            throw new ArgumentNullException( nameof( controller ) );
+        }
 
-        IsApiController = (Func<ControllerModel, bool>) method.CreateDelegate( typeof( Func<ControllerModel, bool> ) );
-    }
+        // REF: https://github.com/dotnet/aspnetcore/blob/main/src/Mvc/Mvc.Core/src/ApplicationModels/ApiBehaviorApplicationModelProvider.cs
+        if ( controller.Attributes.OfType<IApiBehaviorMetadata>().Any() )
+        {
+            return true;
+        }
 
-    private static Func<ControllerModel, bool> IsApiController { get; }
+        var assembly = controller.ControllerType.Assembly;
 
-    /// <inheritdoc />
-    public bool IsSatisfiedBy( ControllerModel controller ) => IsApiController( controller );
+        return assembly.GetCustomAttributes().OfType<IApiBehaviorMetadata>().Any();
+    }
 }