一个基于 SourceGenerator 生成从 dbReader转换为 class 数据的性能测试实验

睑缘

于 2024-09-13 10:46:07 发布

阅读量37

点赞数

文章标签： java 开发语言

好奇

SourceGenerator 出现开始，好几年了，虽然一直好奇用SourceGenerator 生成代码与 emit 等动态生成的代码会有多少差距，

但是一直特别懒，不想搞

其实 dapper aot 项目做了类似事情，不过功能特别积极，还引用了实验特性，所以还是想更为简单客观对比

本次乘着自己暂时性不懒了，做了一个基于 SourceGenerator 生成从 dbReader转换为 class 数据的测试

no generate code when

Generic Type (如果不用 emit 动态生成，还真无法处理未知类型 T)
Anonymous Type (SourceGenerator 生成时机要早于匿名类生成，所以还没机会生成)

generate code

具体怎么做的就这里不写了，感兴趣参考 https://github.com/fs7744/SlowestEM

生成的代码带有一定 db结果动态类型处理，以此更接近实际使用


// <auto-generated/>
#pragma warning disable 8019 //disable 'unnecessary using directive' warning
using System;
using System.Data;
using System.Runtime.CompilerServices;
using System.Collections.Generic;

namespace SlowestEM.Generator
{
    public static partial class Dog_Accessors
    {
        public static IEnumerable<BenchmarkTest.Dog> Read(IDataReader reader)
        {
            var s = new List<Action<BenchmarkTest.Dog, IDataReader>>(reader.FieldCount);
            for (int i = 0; i < reader.FieldCount; i++)
            {
                var j = i;
                switch (reader.GetName(j).ToLower())
                {
                    
                    case "age": 
                    {
                        // int?
                        
                        var needConvert = typeof(int) != reader.GetFieldType(i);
                        s.Add((d,r) => d.Age = DBExtensions.ReadToInt32Nullable(r,j,needConvert));
                         
                    }
                    break;
                    case "name": 
                    {
                        // string
                        
                        var needConvert = typeof(string) != reader.GetFieldType(i);
                        s.Add((d,r) => d.Name = DBExtensions.ReadToString(r,j,needConvert));
                         
                    }
                    break;
                    case "weight": 
                    {
                        // float?
                        
                        var needConvert = typeof(float) != reader.GetFieldType(i);
                        s.Add((d,r) => d.Weight = DBExtensions.ReadToFloatNullable(r,j,needConvert));
                         
                    }
                    break;
                    default:
                        break;
                }
            }
            while (reader.Read())
            {
                var d = new BenchmarkTest.Dog();
                foreach (var item in s)
                {
                    item?.Invoke(d,reader);
                }
                yield return d;
            }
        }
    }
}

测试结果

mock db，避免 db层实现性能和没有正确处理数据类型装箱拆箱问题

[Benchmark(Baseline = true), BenchmarkCategory("1")]
public void SetClassFirst()
{
    Dog dog;
    try
    {
        connection.Open();
        var cmd = connection.CreateCommand();
        cmd.CommandText = "select ";
        using (var reader = cmd.ExecuteReader(CommandBehavior.Default))
        {
            if (reader.Read())
            {
                dog = new Dog();
                dog.Name = reader.GetString(0);
                dog.Age = reader.GetInt32(1);
                dog.Weight = reader.GetFloat(2);
            }
        }
    }
    finally
    {
        connection.Close();
    }
}

[Benchmark, BenchmarkCategory("1")]
public void DapperMappingFirst()
{
    var dogs = connection.QueryFirst<Dog>("select ");
}

[Benchmark, BenchmarkCategory("1")]
public void SourceGeneratorMappingFirst()
{
    Dog dog;
    try
    {
        connection.Open();
        var cmd = connection.CreateCommand();
        cmd.CommandText = "select ";
        using (var reader = cmd.ExecuteReader(CommandBehavior.Default))
        {
            dog = reader.ReadTo<Dog>().FirstOrDefault();
        }
    }
    finally
    {
        connection.Close();
    }
}


BenchmarkDotNet v0.13.12, Windows 10 (10.0.19045.4651/22H2/2022Update)
Intel Core i7-10700 CPU 2.90GHz, 1 CPU, 16 logical and 8 physical cores
.NET SDK 9.0.100-preview.5.24307.3
  [Host]     : .NET 8.0.6 (8.0.624.26715), X64 RyuJIT AVX2
  DefaultJob : .NET 8.0.6 (8.0.624.26715), X64 RyuJIT AVX2

Method	Categories	Mean	Error	StdDev	Ratio	RatioSD	Gen0	Gen1	Allocated	Alloc Ratio
SetClassFirst	1 实体	18.38 ns	0.378 ns	0.316 ns	1.00	0.00	0.0181	-	152 B	1.00
SourceGeneratorMappingFirst	1 实体	183.31 ns	3.525 ns	3.462 ns	9.98	0.14	0.0899	-	752 B	4.95
DapperMappingFirst	1 实体	1,336.69 ns	5.777 ns	5.121 ns	72.77	1.30	0.0343	-	288 B	1.89

SetClass	1000 实体	7,700.08 ns	87.311 ns	68.167 ns	1.00	0.00	6.7749	1.1139	56712 B	1.00
SourceGeneratorMapping	1000 实体	23,428.85 ns	262.698 ns	232.875 ns	3.04	0.03	6.8359	1.1292	57312 B	1.01
DapperMapping	1000 实体	48,880.92 ns	682.693 ns	533.002 ns	6.35	0.06	13.4888	2.1362	113048 B	1.99

原创作者: fs7744 转载于: https://www.cnblogs.com/fs7744/p/18332993