源码生成器应用
This commit is contained in:
parent
cab0e1ed08
commit
021153a319
@ -62,7 +62,7 @@ Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "0.Docs", "0.Docs", "{D8346C
|
||||
readme.md = readme.md
|
||||
EndProjectSection
|
||||
EndProject
|
||||
Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "JiShe.CollectBus.SourceExtend", "shared\JiShe.CollectBus.SourceExtend\JiShe.CollectBus.SourceExtend.csproj", "{A34A4EA6-AEB0-4199-8E3A-6F042DBB51D9}"
|
||||
Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "JiShe.CollectBus.IncrementalGenerator", "shared\JiShe.CollectBus.IncrementalGenerator\JiShe.CollectBus.IncrementalGenerator.csproj", "{828D20BC-454E-0467-B29D-429B25F1E0CA}"
|
||||
EndProject
|
||||
Global
|
||||
GlobalSection(SolutionConfigurationPlatforms) = preSolution
|
||||
@ -150,10 +150,10 @@ Global
|
||||
{75B7D419-C261-577D-58D6-AA3ACED9129F}.Debug|Any CPU.Build.0 = Debug|Any CPU
|
||||
{75B7D419-C261-577D-58D6-AA3ACED9129F}.Release|Any CPU.ActiveCfg = Release|Any CPU
|
||||
{75B7D419-C261-577D-58D6-AA3ACED9129F}.Release|Any CPU.Build.0 = Release|Any CPU
|
||||
{A34A4EA6-AEB0-4199-8E3A-6F042DBB51D9}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
|
||||
{A34A4EA6-AEB0-4199-8E3A-6F042DBB51D9}.Debug|Any CPU.Build.0 = Debug|Any CPU
|
||||
{A34A4EA6-AEB0-4199-8E3A-6F042DBB51D9}.Release|Any CPU.ActiveCfg = Release|Any CPU
|
||||
{A34A4EA6-AEB0-4199-8E3A-6F042DBB51D9}.Release|Any CPU.Build.0 = Release|Any CPU
|
||||
{828D20BC-454E-0467-B29D-429B25F1E0CA}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
|
||||
{828D20BC-454E-0467-B29D-429B25F1E0CA}.Debug|Any CPU.Build.0 = Debug|Any CPU
|
||||
{828D20BC-454E-0467-B29D-429B25F1E0CA}.Release|Any CPU.ActiveCfg = Release|Any CPU
|
||||
{828D20BC-454E-0467-B29D-429B25F1E0CA}.Release|Any CPU.Build.0 = Release|Any CPU
|
||||
EndGlobalSection
|
||||
GlobalSection(SolutionProperties) = preSolution
|
||||
HideSolutionNode = FALSE
|
||||
@ -179,7 +179,7 @@ Global
|
||||
{8A61DF78-069B-40B5-8811-614E2960443E} = {3C3F9DB2-EC97-4464-B49F-BF1A0C2B46DC}
|
||||
{E27377CC-E2D3-4237-060F-96EA214D3129} = {3C3F9DB2-EC97-4464-B49F-BF1A0C2B46DC}
|
||||
{75B7D419-C261-577D-58D6-AA3ACED9129F} = {3C3F9DB2-EC97-4464-B49F-BF1A0C2B46DC}
|
||||
{A34A4EA6-AEB0-4199-8E3A-6F042DBB51D9} = {EBF7C01F-9B4F-48E6-8418-2CBFDA51EB0B}
|
||||
{828D20BC-454E-0467-B29D-429B25F1E0CA} = {EBF7C01F-9B4F-48E6-8418-2CBFDA51EB0B}
|
||||
EndGlobalSection
|
||||
GlobalSection(ExtensibilityGlobals) = postSolution
|
||||
SolutionGuid = {4324B3B4-B60B-4E3C-91D8-59576B4E26DD}
|
||||
|
||||
@ -11,6 +11,5 @@
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ProjectReference Include="..\..\shared\JiShe.CollectBus.Common\JiShe.CollectBus.Common.csproj" />
|
||||
<ProjectReference Include="..\..\shared\JiShe.CollectBus.SourceExtend\JiShe.CollectBus.SourceExtend.csproj" OutputItemType="Analyzer" ReferenceOutputAssembly="false"/>
|
||||
</ItemGroup>
|
||||
</Project>
|
||||
|
||||
@ -1,12 +1,6 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Text;
|
||||
using System.Threading.Tasks;
|
||||
using JiShe.CollectBus.Common.Attributes;
|
||||
using JiShe.CollectBus.Common;
|
||||
using JiShe.CollectBus.IoTDB.Attribute;
|
||||
using JiShe.CollectBus.IoTDB.Enums;
|
||||
using JiShe.CollectBus.IoTDB.Provider;
|
||||
|
||||
namespace JiShe.CollectBus.IoTDB.Model
|
||||
{
|
||||
|
||||
@ -1,11 +1,6 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Text;
|
||||
using System.Threading.Tasks;
|
||||
using JiShe.CollectBus.Common;
|
||||
using JiShe.CollectBus.IoTDB.Attribute;
|
||||
using JiShe.CollectBus.IoTDB.Enums;
|
||||
using JiShe.CollectBus.IoTDB.Provider;
|
||||
|
||||
namespace JiShe.CollectBus.IoTDB.Model
|
||||
{
|
||||
@ -13,6 +8,7 @@ namespace JiShe.CollectBus.IoTDB.Model
|
||||
/// Tree模型单项数据实体
|
||||
/// </summary>
|
||||
[EntityType(EntityTypeEnum.TreeModel)]
|
||||
[GenerateAccessors]
|
||||
public class TreeModelSingleMeasuringEntity<T> : IoTEntity
|
||||
{
|
||||
/// <summary>
|
||||
|
||||
@ -30,6 +30,8 @@
|
||||
<ProjectReference Include="..\..\modules\JiShe.CollectBus.FreeRedis\JiShe.CollectBus.FreeRedis.csproj" />
|
||||
<ProjectReference Include="..\..\modules\JiShe.CollectBus.Kafka\JiShe.CollectBus.Kafka.csproj" />
|
||||
<ProjectReference Include="..\..\modules\JiShe.CollectBus.MongoDB\JiShe.CollectBus.MongoDB.csproj" />
|
||||
|
||||
<ProjectReference Include="..\..\shared\JiShe.CollectBus.IncrementalGenerator\JiShe.CollectBus.IncrementalGenerator.csproj" OutputItemType="Analyzer" ReferenceOutputAssembly="false" />
|
||||
</ItemGroup>
|
||||
|
||||
</Project>
|
||||
|
||||
@ -66,6 +66,7 @@ public class SampleAppService : CollectBusAppService, ISampleAppService, IKafkaS
|
||||
IssuedMessageHexString = "messageHexString",
|
||||
Timestamps = testTime// DateTimeOffset.UtcNow.ToUnixTimeNanoseconds()//testTime.GetDateTimeOffset().ToUnixTimeNanoseconds(),
|
||||
};
|
||||
|
||||
await _iotDBProvider.InsertAsync(meter);
|
||||
}
|
||||
|
||||
|
||||
@ -1,10 +1,12 @@
|
||||
using JiShe.CollectBus.IoTDB.Attribute;
|
||||
using JiShe.CollectBus.Common;
|
||||
using JiShe.CollectBus.IoTDB.Attribute;
|
||||
using JiShe.CollectBus.IoTDB.Enums;
|
||||
using JiShe.CollectBus.IoTDB.Model;
|
||||
|
||||
namespace JiShe.CollectBus.Ammeters
|
||||
{
|
||||
[EntityType(EntityTypeEnum.TreeModel)]
|
||||
[GenerateAccessors]
|
||||
public class ElectricityMeterTreeModel : IoTEntity
|
||||
{
|
||||
[ATTRIBUTEColumn]
|
||||
|
||||
@ -1,16 +1,8 @@
|
||||
using JiShe.CollectBus.Common.Encrypt;
|
||||
using JiShe.CollectBus.Common.Enums;
|
||||
using JiShe.CollectBus.Common.Models;
|
||||
using JiShe.CollectBus.Common;
|
||||
using JiShe.CollectBus.IoTDB.Attribute;
|
||||
using JiShe.CollectBus.IoTDB.Enums;
|
||||
using JiShe.CollectBus.IoTDB.Model;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Text;
|
||||
using System.Threading.Tasks;
|
||||
using Volo.Abp.Domain.Entities;
|
||||
using Volo.Abp.Domain.Entities.Auditing;
|
||||
|
||||
namespace JiShe.CollectBus.IotSystems.MeterReadingRecords
|
||||
{
|
||||
@ -18,6 +10,7 @@ namespace JiShe.CollectBus.IotSystems.MeterReadingRecords
|
||||
/// 抄读任务数据
|
||||
/// </summary>
|
||||
[EntityType(EntityTypeEnum.TableModel)]
|
||||
[GenerateAccessors]
|
||||
public class MeterReadingTelemetryPacketInfo : IoTEntity
|
||||
{
|
||||
/// <summary>
|
||||
|
||||
@ -25,7 +25,7 @@
|
||||
<ProjectReference Include="..\..\modules\JiShe.CollectBus.IoTDB\JiShe.CollectBus.IoTDB.csproj" />
|
||||
<ProjectReference Include="..\..\shared\JiShe.CollectBus.Common\JiShe.CollectBus.Common.csproj" OutputItemType="Analyzer" ReferenceOutputAssembly="false" />
|
||||
<ProjectReference Include="..\..\shared\JiShe.CollectBus.Domain.Shared\JiShe.CollectBus.Domain.Shared.csproj" />
|
||||
</ItemGroup>
|
||||
</ItemGroup>
|
||||
|
||||
<ItemGroup>
|
||||
<PackageReference Include="Volo.Abp.Emailing" Version="8.3.3" />
|
||||
|
||||
@ -1,10 +1,6 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Text;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace JiShe.CollectBus.SourceExtend
|
||||
namespace JiShe.CollectBus.Common
|
||||
{
|
||||
/// <summary>
|
||||
/// 标记需要生成源码的类
|
||||
@ -26,7 +26,6 @@
|
||||
<PackageReference Include="Serilog" Version="4.1.0" />
|
||||
<PackageReference Include="Volo.Abp.Core" Version="8.3.3" />
|
||||
<PackageReference Include="Volo.Abp.Ddd.Domain" Version="8.3.3" />
|
||||
|
||||
</ItemGroup>
|
||||
|
||||
</Project>
|
||||
|
||||
@ -8,12 +8,12 @@
|
||||
/// <summary>
|
||||
/// 服务器标识
|
||||
/// </summary>
|
||||
public required string ServerTagName { get; set; }
|
||||
public string ServerTagName { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// 系统类型
|
||||
/// </summary>
|
||||
public required string SystemType { get; set; }
|
||||
public string SystemType { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// 首次采集时间
|
||||
@ -23,31 +23,31 @@
|
||||
/// <summary>
|
||||
/// 自动验证时间
|
||||
/// </summary>
|
||||
public required string AutomaticVerificationTime { get; set; }
|
||||
public string AutomaticVerificationTime { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// 自动获取终端版时间
|
||||
/// </summary>
|
||||
public required string AutomaticTerminalVersionTime { get; set; }
|
||||
public string AutomaticTerminalVersionTime { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// 自动获取远程通信模块(SIM)版本时间
|
||||
/// </summary>
|
||||
public required string AutomaticTelematicsModuleTime { get; set; }
|
||||
public string AutomaticTelematicsModuleTime { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// 日冻结抄读时间
|
||||
/// </summary>
|
||||
public required string AutomaticDayFreezeTime { get; set; }
|
||||
public string AutomaticDayFreezeTime { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// 月冻结抄读时间
|
||||
/// </summary>
|
||||
public required string AutomaticMonthFreezeTime { get; set; }
|
||||
public string AutomaticMonthFreezeTime { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// 默认协议插件
|
||||
/// </summary>
|
||||
public required string DefaultProtocolPlugin { get; set; }
|
||||
public string DefaultProtocolPlugin { get; set; }
|
||||
}
|
||||
}
|
||||
|
||||
@ -0,0 +1,223 @@
|
||||
using Microsoft.CodeAnalysis;
|
||||
using Microsoft.CodeAnalysis.CSharp.Syntax;
|
||||
using Microsoft.CodeAnalysis.Text;
|
||||
using System.Collections.Generic;
|
||||
using System.Diagnostics;
|
||||
using System.Linq;
|
||||
using System.Text;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace JiShe.CollectBus.IncrementalGenerator
|
||||
{
|
||||
/// <summary>
|
||||
/// 复杂类型源生成器
|
||||
/// </summary>
|
||||
[Generator(LanguageNames.CSharp)]
|
||||
public class ComplexTypeSourceGenerator : IIncrementalGenerator
|
||||
{
|
||||
private const string AttributeFullName = "JiShe.CollectBus.Common.GenerateAccessorsAttribute";
|
||||
|
||||
public void Initialize(IncrementalGeneratorInitializationContext context)
|
||||
{
|
||||
|
||||
Debugger.Launch();
|
||||
|
||||
context.RegisterPostInitializationOutput(ctx =>
|
||||
{
|
||||
ctx.AddSource("DebugInit.g.cs", "// Generator initialized");
|
||||
});
|
||||
|
||||
// 步骤1:筛选带有 [GenerateAccessors] 的类
|
||||
var classDeclarations = context.SyntaxProvider
|
||||
.CreateSyntaxProvider(
|
||||
predicate: static (s, _) => IsClassWithAttribute(s),
|
||||
transform: static (ctx, _) => GetClassDeclaration(ctx))
|
||||
.Where(static c => c is not null);
|
||||
|
||||
// 步骤2:合并编译信息
|
||||
var compilationAndClasses = context.CompilationProvider.Combine(classDeclarations.Collect());
|
||||
|
||||
context.RegisterSourceOutput(compilationAndClasses, (spc, source) =>
|
||||
GenerateCode(source.Left, source.Right!, spc));
|
||||
}
|
||||
|
||||
private static bool IsClassWithAttribute(SyntaxNode node) => node is ClassDeclarationSyntax cds && cds.AttributeLists.Count > 0;
|
||||
|
||||
//private static ClassDeclarationSyntax? GetClassDeclaration(GeneratorSyntaxContext context)
|
||||
//{
|
||||
// var classDecl = (ClassDeclarationSyntax)context.Node;
|
||||
// foreach (var attributeList in classDecl.AttributeLists)
|
||||
// {
|
||||
// foreach (var attribute in attributeList.Attributes)
|
||||
// {
|
||||
// var symbol = context.SemanticModel.GetSymbolInfo(attribute).Symbol;
|
||||
// if (symbol is IMethodSymbol ctor)
|
||||
// {
|
||||
// string sdd = ctor.ContainingType.Name;
|
||||
// var attributeType = context.SemanticModel.Compilation.GetTypeByMetadataName(AttributeFullName);
|
||||
// if (ctor.ContainingType?.OriginalDefinition?.Equals(attributeType, SymbolEqualityComparer.Default) == true)
|
||||
// {
|
||||
// return classDecl;
|
||||
// }
|
||||
// }
|
||||
|
||||
// }
|
||||
// }
|
||||
// return null;
|
||||
//}
|
||||
|
||||
private static ClassDeclarationSyntax? GetClassDeclaration(GeneratorSyntaxContext context)
|
||||
{
|
||||
var classDecl = (ClassDeclarationSyntax)context.Node;
|
||||
var attributeType = context.SemanticModel.Compilation.GetTypeByMetadataName(AttributeFullName);
|
||||
|
||||
foreach (var attribute in classDecl.AttributeLists.SelectMany(al => al.Attributes))
|
||||
{
|
||||
var symbol = context.SemanticModel.GetSymbolInfo(attribute).Symbol;
|
||||
if (symbol is IMethodSymbol ctor &&
|
||||
ctor.ContainingType?.OriginalDefinition?.Equals(attributeType, SymbolEqualityComparer.Default) == true)
|
||||
{
|
||||
return classDecl;
|
||||
}
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void GenerateCode(
|
||||
Compilation compilation,
|
||||
IEnumerable<ClassDeclarationSyntax> classes,
|
||||
SourceProductionContext context)
|
||||
{
|
||||
var processedTypes = new HashSet<ITypeSymbol>(SymbolEqualityComparer.Default);
|
||||
|
||||
|
||||
foreach (var classDecl in classes.Distinct())
|
||||
{
|
||||
var model = compilation.GetSemanticModel(classDecl.SyntaxTree);
|
||||
var classSymbol = model.GetDeclaredSymbol(classDecl) as INamedTypeSymbol;
|
||||
|
||||
if (classSymbol == null || !processedTypes.Add(classSymbol)) continue;
|
||||
|
||||
context.ReportDiagnostic(Diagnostic.Create(
|
||||
new DiagnosticDescriptor(
|
||||
"PA001",
|
||||
"Generated Accessors",
|
||||
$"Generating accessors for {classSymbol.Name}",
|
||||
"Performance",
|
||||
DiagnosticSeverity.Info,
|
||||
true),
|
||||
Location.None));
|
||||
|
||||
context.ReportDiagnostic(Diagnostic.Create(
|
||||
new DiagnosticDescriptor("PA002", "Class Found",
|
||||
$"Processing class: {classSymbol.Name}", "Debug", DiagnosticSeverity.Warning, true),
|
||||
Location.None));
|
||||
|
||||
var code = BuildAccessorsForType(classSymbol, compilation, processedTypes);
|
||||
context.AddSource($"{classSymbol.Name}Extension.g.cs", SourceText.From(code, Encoding.UTF8));
|
||||
}
|
||||
}
|
||||
|
||||
private static string BuildAccessorsForType(
|
||||
INamedTypeSymbol classSymbol,
|
||||
Compilation compilation,
|
||||
HashSet<ITypeSymbol> processedTypes)
|
||||
{
|
||||
var code = new StringBuilder();
|
||||
code.AppendLine("// <auto-generated/>");
|
||||
code.AppendLine("#nullable enable");
|
||||
code.AppendLine($"namespace {classSymbol.ContainingNamespace.ToDisplayString()};");
|
||||
code.AppendLine();
|
||||
|
||||
code.AppendLine($"public static class {classSymbol.Name}Extension{GetGenericParams(classSymbol)}");
|
||||
code.AppendLine("{");
|
||||
|
||||
foreach (var prop in classSymbol.GetMembers().OfType<IPropertySymbol>())
|
||||
{
|
||||
if (prop.IsIndexer) continue;
|
||||
|
||||
GeneratePropertyAccessors(prop, code, compilation, processedTypes);
|
||||
}
|
||||
|
||||
code.AppendLine("}");
|
||||
return code.ToString();
|
||||
}
|
||||
|
||||
private static string GetGenericParams(INamedTypeSymbol symbol)
|
||||
=> symbol.IsGenericType ? $"<{string.Join(", ", symbol.TypeParameters.Select(t => t.Name))}>" : "";
|
||||
|
||||
private static void GeneratePropertyAccessors(
|
||||
IPropertySymbol prop,
|
||||
StringBuilder code,
|
||||
Compilation compilation,
|
||||
HashSet<ITypeSymbol> processedTypes)
|
||||
{
|
||||
// 关键修复点1:安全类型转换
|
||||
if (prop.Type is not ITypeSymbol propType) return;
|
||||
|
||||
code.AppendLine($" // Processing property: {prop.Name}");
|
||||
|
||||
// 处理元组类型
|
||||
if (propType is INamedTypeSymbol { IsTupleType: true } tupleType)
|
||||
{
|
||||
GenerateTupleAccessors(prop, tupleType, code);
|
||||
}
|
||||
else if (propType is INamedTypeSymbol namedType)
|
||||
{
|
||||
GenerateStandardAccessors(prop, namedType, code);
|
||||
ProcessNestedType(namedType, compilation, processedTypes);
|
||||
}
|
||||
}
|
||||
|
||||
private static void GenerateTupleAccessors(IPropertySymbol prop, INamedTypeSymbol tupleType, StringBuilder code)
|
||||
{
|
||||
var elements = tupleType.TupleElements;
|
||||
var parentType = prop.ContainingType.ToDisplayString();
|
||||
|
||||
for (int i = 0; i < elements.Length; i++)
|
||||
{
|
||||
var element = elements[i];
|
||||
if (element.Type is not ITypeSymbol elementType) continue;
|
||||
|
||||
var elementName = element.CorrespondingTupleField?.Name ?? $"Item{i + 1}";
|
||||
code.AppendLine($" public static {elementType.ToDisplayString()} Get{prop.Name}_{elementName}({parentType} obj) => obj.{prop.Name}.{elementName};");
|
||||
|
||||
if (prop.SetMethod != null)
|
||||
{
|
||||
var assignments = elements.Select((e, idx) =>
|
||||
idx == i ? "value" : $"obj.{prop.Name}.{e.Name}");
|
||||
code.AppendLine($" public static void Set{prop.Name}_{elementName}({parentType} obj, {elementType.ToDisplayString()} value) => obj.{prop.Name} = ({string.Join(", ", assignments)});");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private static void GenerateStandardAccessors(IPropertySymbol prop, INamedTypeSymbol propType, StringBuilder code)
|
||||
{
|
||||
var parentType = prop.ContainingType.ToDisplayString();
|
||||
code.AppendLine($" public static {propType.ToDisplayString()} Get{prop.Name}({parentType} obj) => obj.{prop.Name};");
|
||||
|
||||
if (prop.SetMethod != null)
|
||||
{
|
||||
code.AppendLine($" public static void Set{prop.Name}({parentType} obj, {propType.ToDisplayString()} value) => obj.{prop.Name} = value;");
|
||||
}
|
||||
}
|
||||
|
||||
private static void ProcessNestedType(ITypeSymbol typeSymbol, Compilation compilation, HashSet<ITypeSymbol> processedTypes)
|
||||
{
|
||||
if (typeSymbol is not INamedTypeSymbol namedType) return;
|
||||
if (!ShouldProcessNestedType(namedType)) return;
|
||||
if (!processedTypes.Add(namedType)) return;
|
||||
|
||||
var code = BuildAccessorsForType(namedType, compilation, processedTypes);
|
||||
}
|
||||
|
||||
private static bool ShouldProcessNestedType(INamedTypeSymbol symbol)
|
||||
{
|
||||
return symbol.DeclaredAccessibility == Accessibility.Public &&
|
||||
!symbol.IsTupleType &&
|
||||
!symbol.IsAnonymousType &&
|
||||
!symbol.IsImplicitlyDeclared &&
|
||||
!symbol.Name.StartsWith("<");
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,14 @@
|
||||
<Project Sdk="Microsoft.NET.Sdk">
|
||||
|
||||
<PropertyGroup>
|
||||
<Nullable>enable</Nullable>
|
||||
<TargetFramework>netstandard2.0</TargetFramework>
|
||||
<!--<TargetFramework>net8.0</TargetFramework>-->
|
||||
<EnforceExtendedAnalyzerRules>true</EnforceExtendedAnalyzerRules>
|
||||
<LangVersion>latest</LangVersion>
|
||||
</PropertyGroup>
|
||||
<ItemGroup>
|
||||
<PackageReference Include="Microsoft.CodeAnalysis.Analyzers" Version="3.11.0" PrivateAssets="all" />
|
||||
<PackageReference Include="Microsoft.CodeAnalysis.CSharp" Version="4.13.0" PrivateAssets="all" />
|
||||
</ItemGroup>
|
||||
</Project>
|
||||
@ -1,187 +0,0 @@
|
||||
using Microsoft.CodeAnalysis;
|
||||
using Microsoft.CodeAnalysis.CSharp.Syntax;
|
||||
using Microsoft.CodeAnalysis.Text;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Text;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace JiShe.CollectBus.SourceExtend
|
||||
{
|
||||
/// <summary>
|
||||
/// 复杂类型源生成器
|
||||
/// </summary>
|
||||
[Generator]
|
||||
public sealed class PropertyAccessorGenerator : IIncrementalGenerator
|
||||
{
|
||||
private const string AttributeName = "GenerateAccessors";
|
||||
|
||||
public void Initialize(IncrementalGeneratorInitializationContext context)
|
||||
{
|
||||
// 步骤1:筛选带有 [GenerateAccessors] 的类
|
||||
var classDeclarations = context.SyntaxProvider
|
||||
.CreateSyntaxProvider(
|
||||
predicate: static (s, _) => IsClassWithAttribute(s),
|
||||
transform: static (ctx, _) => GetClassDeclaration(ctx))
|
||||
.Where(static c => c is not null);
|
||||
|
||||
// 步骤2:编译符号并生成代码
|
||||
var compilationAndClasses = context.CompilationProvider.Combine(classDeclarations.Collect());
|
||||
|
||||
context.RegisterSourceOutput(compilationAndClasses, (spc, source) =>
|
||||
GenerateCode(source.Left, source.Right!, spc));
|
||||
}
|
||||
|
||||
private static bool IsClassWithAttribute(SyntaxNode node)
|
||||
{
|
||||
return node is ClassDeclarationSyntax { AttributeLists.Count: > 0 };
|
||||
}
|
||||
|
||||
private static ClassDeclarationSyntax? GetClassDeclaration(GeneratorSyntaxContext context)
|
||||
{
|
||||
var classDecl = (ClassDeclarationSyntax)context.Node;
|
||||
foreach (var attributeList in classDecl.AttributeLists)
|
||||
{
|
||||
foreach (var attribute in attributeList.Attributes)
|
||||
{
|
||||
var symbol = context.SemanticModel.GetSymbolInfo(attribute).Symbol;
|
||||
if (symbol?.ContainingType.ToDisplayString() == $"{nameof(PropertyAccessorGenerator)}.{AttributeName}")
|
||||
{
|
||||
return classDecl;
|
||||
}
|
||||
}
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
private static void GenerateCode(
|
||||
Compilation compilation,
|
||||
IEnumerable<ClassDeclarationSyntax> classes,
|
||||
SourceProductionContext context)
|
||||
{
|
||||
var processedTypes = new HashSet<ITypeSymbol>(SymbolEqualityComparer.Default);
|
||||
|
||||
foreach (var classDecl in classes.Distinct())
|
||||
{
|
||||
var model = compilation.GetSemanticModel(classDecl.SyntaxTree);
|
||||
var classSymbol = model.GetDeclaredSymbol(classDecl)!;
|
||||
|
||||
if (!processedTypes.Add(classSymbol)) continue;
|
||||
|
||||
var code = BuildAccessorsForType(classSymbol, compilation, processedTypes);
|
||||
context.AddSource($"{classSymbol.Name}Accessors.g.cs", code);
|
||||
}
|
||||
}
|
||||
|
||||
private static string BuildAccessorsForType(
|
||||
INamedTypeSymbol classSymbol,
|
||||
Compilation compilation,
|
||||
HashSet<ITypeSymbol> processedTypes)
|
||||
{
|
||||
var code = new StringBuilder();
|
||||
code.AppendLine("// <auto-generated/>");
|
||||
code.AppendLine("#nullable enable");
|
||||
code.AppendLine();
|
||||
code.AppendLine($"namespace {classSymbol.ContainingNamespace.ToDisplayString()};");
|
||||
code.AppendLine();
|
||||
|
||||
// 处理泛型参数
|
||||
var genericParams = classSymbol.IsGenericType
|
||||
? $"<{string.Join(", ", classSymbol.TypeParameters.Select(t => t.Name))}>"
|
||||
: "";
|
||||
|
||||
code.AppendLine($"public static class {classSymbol.Name}Accessors{genericParams}");
|
||||
code.AppendLine("{");
|
||||
|
||||
foreach (var prop in classSymbol.GetMembers().OfType<IPropertySymbol>())
|
||||
{
|
||||
if (prop.IsIndexer) continue;
|
||||
|
||||
GeneratePropertyAccessors(prop, code, compilation, processedTypes);
|
||||
}
|
||||
|
||||
code.AppendLine("}");
|
||||
return code.ToString();
|
||||
}
|
||||
|
||||
private static void GeneratePropertyAccessors(
|
||||
IPropertySymbol prop,
|
||||
StringBuilder code,
|
||||
Compilation compilation,
|
||||
HashSet<ITypeSymbol> processedTypes)
|
||||
{
|
||||
var propType = prop.Type;
|
||||
var parentType = prop.ContainingType.ToDisplayString();
|
||||
|
||||
// 处理元组类型
|
||||
if (propType.IsTupleType && propType is INamedTypeSymbol tupleType)
|
||||
{
|
||||
GenerateTupleAccessors(prop, tupleType, code);
|
||||
}
|
||||
else if (propType is INamedTypeSymbol namedType)
|
||||
{
|
||||
GenerateStandardAccessors(prop, namedType, code);
|
||||
ProcessNestedType(namedType, compilation, processedTypes);
|
||||
}
|
||||
}
|
||||
|
||||
private static void GenerateTupleAccessors(IPropertySymbol prop, INamedTypeSymbol tupleType, StringBuilder code)
|
||||
{
|
||||
var elements = tupleType.TupleElements;
|
||||
var parentType = prop.ContainingType.ToDisplayString();
|
||||
|
||||
for (int i = 0; i < elements.Length; i++)
|
||||
{
|
||||
var element = elements[i];
|
||||
var elementType = element.Type.ToDisplayString();
|
||||
var elementName = element.CorrespondingTupleField?.Name ?? $"Item{i + 1}";
|
||||
|
||||
code.AppendLine($" public static {elementType} Get{prop.Name}_{elementName}({parentType} obj)");
|
||||
code.AppendLine($" => obj.{prop.Name}.{elementName};");
|
||||
|
||||
if (prop.SetMethod != null)
|
||||
{
|
||||
var assignments = elements.Select((e, idx) =>
|
||||
idx == i ? "value" : $"obj.{prop.Name}.{e.Name}");
|
||||
|
||||
code.AppendLine($" public static void Set{prop.Name}_{elementName}({parentType} obj, {elementType} value)");
|
||||
code.AppendLine($" => obj.{prop.Name} = ({string.Join(", ", assignments)});");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private static void GenerateStandardAccessors(IPropertySymbol prop, INamedTypeSymbol propType, StringBuilder code)
|
||||
{
|
||||
var typeName = propType.ToDisplayString();
|
||||
var parentType = prop.ContainingType.ToDisplayString();
|
||||
|
||||
code.AppendLine($" public static {typeName} Get{prop.Name}({parentType} obj)");
|
||||
code.AppendLine($" => obj.{prop.Name};");
|
||||
|
||||
if (prop.SetMethod != null)
|
||||
{
|
||||
code.AppendLine($" public static void Set{prop.Name}({parentType} obj, {typeName} value)");
|
||||
code.AppendLine($" => obj.{prop.Name} = value;");
|
||||
}
|
||||
}
|
||||
|
||||
private static void ProcessNestedType(INamedTypeSymbol typeSymbol, Compilation compilation, HashSet<ITypeSymbol> processedTypes)
|
||||
{
|
||||
if (ShouldGenerateForType(typeSymbol) && processedTypes.Add(typeSymbol))
|
||||
{
|
||||
var code = BuildAccessorsForType(typeSymbol, compilation, processedTypes);
|
||||
// 注意:实际项目中需要通过 SourceProductionContext 添加源代码
|
||||
}
|
||||
}
|
||||
|
||||
private static bool ShouldGenerateForType(INamedTypeSymbol typeSymbol)
|
||||
{
|
||||
return !typeSymbol.IsTupleType &&
|
||||
!typeSymbol.IsAnonymousType &&
|
||||
!typeSymbol.IsNativeIntegerType &&
|
||||
typeSymbol.DeclaredAccessibility == Accessibility.Public &&
|
||||
!typeSymbol.Name.StartsWith("<");
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -1,12 +0,0 @@
|
||||
<Project Sdk="Microsoft.NET.Sdk">
|
||||
|
||||
<PropertyGroup>
|
||||
<TargetFramework>netstandard2.1</TargetFramework>
|
||||
<LangVersion>12.0</LangVersion>
|
||||
<Nullable>enable</Nullable>
|
||||
<EnforceExtendedAnalyzerRules>true</EnforceExtendedAnalyzerRules>
|
||||
</PropertyGroup>
|
||||
<ItemGroup>
|
||||
<PackageReference Include="Microsoft.CodeAnalysis.CSharp" Version="4.13.0" />
|
||||
</ItemGroup>
|
||||
</Project>
|
||||
@ -5,7 +5,7 @@
|
||||
<Nullable>enable</Nullable>
|
||||
<ImplicitUsings>enable</ImplicitUsings>
|
||||
<GenerateDocumentationFile>True</GenerateDocumentationFile>
|
||||
</PropertyGroup>
|
||||
</PropertyGroup>
|
||||
|
||||
<ItemGroup>
|
||||
<Compile Remove="logs\**" />
|
||||
@ -57,7 +57,7 @@
|
||||
<ProjectReference Include="..\..\services\JiShe.CollectBus.Application.Contracts\JiShe.CollectBus.Application.Contracts.csproj" />
|
||||
<ProjectReference Include="..\..\services\JiShe.CollectBus.Application\JiShe.CollectBus.Application.csproj" />
|
||||
<ProjectReference Include="..\JiShe.CollectBus.HttpApi\JiShe.CollectBus.HttpApi.csproj" />
|
||||
</ItemGroup>
|
||||
</ItemGroup>
|
||||
|
||||
<ItemGroup>
|
||||
<Content Update="appsettings.json">
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user