源生成器替代反射

2025-10-06 17:46:59 +08:00 · 2024-10-10 22:17:28 +08:00
parent f5aa996516
commit 5c83a2f97b
63 changed files with 375 additions and 239 deletions
--- a/linker.gen/InterfaceSourceGenerator.cs
+++ b/linker.gen/InterfaceSourceGenerator.cs
@@ -0,0 +1,114 @@
+using Microsoft.CodeAnalysis.CSharp.Syntax;
+using Microsoft.CodeAnalysis;
+using System.Linq;
+using System.Collections.Generic;
+using Microsoft.CodeAnalysis.Text;
+using System.Text;
+using System.Diagnostics;
+
+namespace linker.gen
+{
+
+    [Generator(LanguageNames.CSharp)]
+    public class InterfaceSourceGenerator : IIncrementalGenerator
+    {
+        private List<GeneratorInfo> generators = new List<GeneratorInfo> {
+             new GeneratorInfo{ ClassName="FlowTransfer", ClassNameSpace="linker.plugins.flow", InterfaceName="linker.plugins.flow.IFlow" },
+             new GeneratorInfo{ ClassName="RelayValidatorTransfer", ClassNameSpace="linker.plugins.relay.validator", InterfaceName="linker.plugins.relay.validator.IRelayValidator" },
+             new GeneratorInfo{ ClassName="SignInArgsTransfer", ClassNameSpace="linker.plugins.signIn.args", InterfaceName="linker.plugins.signIn.args.ISignInArgs" },
+             new GeneratorInfo{ ClassName="RelayTransfer", ClassNameSpace="linker.plugins.relay", InterfaceName="linker.plugins.relay.transport.ITransport" },
+             new GeneratorInfo{ ClassName="ResolverTransfer", ClassNameSpace="linker.plugins.resolver", InterfaceName="linker.plugins.resolver.IResolver" },
+             new GeneratorInfo{ ClassName="TunnelExcludeIPTransfer", ClassNameSpace="linker.plugins.tunnel.excludeip", InterfaceName="linker.plugins.tunnel.excludeip.ITunnelExcludeIP" },
+             new GeneratorInfo{ ClassName="StartupTransfer", ClassNameSpace="linker.startup", InterfaceName="linker.startup.IStartup", Instance=true },
+             new GeneratorInfo{ ClassName="MessengerResolverTypes", ClassNameSpace="linker.plugins.messenger", InterfaceName="linker.plugins.messenger.IMessenger"},
+             new GeneratorInfo{ ClassName="ApiClientServer", ClassNameSpace="linker.plugins.capi", InterfaceName="linker.plugins.capi.IApiClientController"},
+        };
+
+        public void Initialize(IncrementalGeneratorInitializationContext context)
+        {
+            IncrementalValueProvider<Compilation> compilations = context.CompilationProvider.Select((compilation, cancellationToken) => compilation);
+
+            context.RegisterSourceOutput(compilations, (sourceProductionContext, compilation) =>
+            {
+                foreach (GeneratorInfo info in generators)
+                {
+                    var iFlowSymbol = compilation.GetTypeByMetadataName(info.InterfaceName);
+                    List<string> types = new List<string> { };
+                    List<string> classs = new List<string> { };
+                    List<string> namespaces = new List<string> { };
+
+                    foreach (var syntaxTree in compilation.SyntaxTrees)
+                    {
+                        if (syntaxTree == null)
+                        {
+                            continue;
+                        }
+                        var root = syntaxTree.GetRoot(sourceProductionContext.CancellationToken);
+                        var classDeclarationSyntaxs = root
+                            .DescendantNodes(descendIntoTrivia: false)
+                            .OfType<ClassDeclarationSyntax>();
+
+                        foreach (var classDeclarationSyntax in classDeclarationSyntaxs)
+                        {
+                            var model = compilation.GetSemanticModel(classDeclarationSyntax.SyntaxTree);
+                            var classSymbol = model.GetDeclaredSymbol(classDeclarationSyntax) as INamedTypeSymbol;
+                            if (classSymbol.AllInterfaces.Contains(iFlowSymbol))
+                            {
+                                types.Add($"typeof({classDeclarationSyntax.Identifier.Text})");
+
+                                if (info.Instance)
+                                    classs.Add($"new {classDeclarationSyntax.Identifier.Text}()");
+
+                                var namespaceDecl = classDeclarationSyntax.FirstAncestorOrSelf<NamespaceDeclarationSyntax>();
+                                if (namespaceDecl != null)
+                                {
+                                    namespaces.Add($"using {namespaceDecl.Name.ToString()};");
+                                }
+
+                            }
+
+                        }
+                    }
+
+                    var source = $@"
+                    using System;
+                    using System.Collections.Generic;
+                    {string.Join("\r\n", namespaces)}
+
+                    namespace {info.ClassNameSpace}
+                    {{
+                        public partial class {info.ClassName}
+                        {{
+                             public static List<Type> GetSourceGeneratorTypes()
+                             {{
+                                return new List<Type> {{
+                                    {string.Join(",", types)}
+                                }};
+                             }}
+                             public static List<{info.InterfaceName}> GetSourceGeneratorInstances()
+                             {{
+                                return new List<{info.InterfaceName}> {{
+                                    {string.Join(",", classs)}
+                                }};
+                             }}
+                        }}
+                    }}";
+
+                    var sourceText = SourceText.From(source, Encoding.UTF8);
+                    sourceProductionContext.AddSource($"{info.ClassName}Instances.g.cs", sourceText);
+                }
+
+
+            });
+        }
+
+
+        public sealed class GeneratorInfo
+        {
+            public string ClassName { get; set; }
+            public string ClassNameSpace { get; set; }
+            public string InterfaceName { get; set; }
+            public bool Instance { get; set; }
+        }
+    }
+}