Fix fail.

byme8 · Jun 17, 2024 · 582d692 · 582d692
1 parent 1b01f75
commit 582d692
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/src/Apparatus.AOT.Reflection.SourceGenerator/KeyOf/KeyOfAnalyzer.cs b/src/Apparatus.AOT.Reflection.SourceGenerator/KeyOf/KeyOfAnalyzer.cs
@@ -16,7 +16,7 @@ public static void AnalyzeKeyOfUsages(
         {
             var keyOfs = parameters
                 .Where(o => o.RefKind != RefKind.Out)
-                .Select((o, i) => (Index: i, Type: (INamedTypeSymbol)o.Type))
+                .Select((o, i) => (Index: i, Type: o.Type as INamedTypeSymbol))
                 .Where(o => IsKeyOf(o.Type))
                 .ToArray();
 
@@ -87,19 +87,19 @@ expressionInvocation.Expression is IdentifierNameSyntax identifier &&
                     {
                         continue;
                     }
-                    
+
                     if (propertySymbol.Symbol is IParameterSymbol parameterSymbol &&
                         IsKeyOf(parameterSymbol.Type as INamedTypeSymbol))
                     {
                         continue;
                     }
-                    
+
                     if (propertySymbol.Symbol is IFieldSymbol fieldSymbol &&
                         IsKeyOf(fieldSymbol.Type as INamedTypeSymbol))
                     {
                         continue;
                     }
-                    
+
                     if (propertySymbol.Symbol is IPropertySymbol property &&
                         IsKeyOf(property.Type as INamedTypeSymbol))
                     {
@@ -132,10 +132,11 @@ expressionInvocation.Expression is IdentifierNameSyntax identifier &&
             }
         }
 
-        public static bool IsKeyOf(INamedTypeSymbol type)
-            => type.ConstructedFrom.ToString().StartsWith("Apparatus.AOT.Reflection.KeyOf<");
+        public static bool IsKeyOf(INamedTypeSymbol? type)
+            => type?.ConstructedFrom.ToString().StartsWith("Apparatus.AOT.Reflection.KeyOf<") ?? false;
 
         public static string Version { get; } = typeof(KeyOfAnalyzer).Assembly.GetName().Version.ToString();
+
         public static string CodeGenerationAttribute { get; } = $@"[System.CodeDom.Compiler.GeneratedCode(""AOT.Reflection"", ""{Version}"")]";
     }
 }