.NET Core 对象池的使用
昨天在『.NET 大牛之路』技术群和大家聊到了对象池的话题,今天展开详细讲讲这个知识点。
池这个概念大家都很熟悉,比如我们经常听到数据库连接池和线程池。它是一种基于使用预先分配资源集合的性能优化思想。
简单说,对象池就是对象的容器,旨在优化资源的使用,通过在一个容器中池化对象,并根据需要重复使用这些池化对象来满足性能上的需求。当一个对象被激活时,便被从池中取出。当对象被停用时,它又被放回池中,等待下一个请求。对象池一般用于对象的初始化过程代价较大或使用频率较高的场景。
那在 .NET 中如何实现或使用对象池呢?
在 ASP.NET Core 框架里已经内置了一个对象池功能的实现:Microsoft.Extensions.ObjectPool
。如果是控制台应用程序,可以单独安装这个扩展库。
池化策略
首先,要使用 ObjectPool
,需要创建一个池化策略,告诉对象池你将如何创建对象,以及如何归还对象。
该策略通过实现接口 IPooledObjectPolicy
来定义,下面是一个最简单的策略实现:
public class FooPooledObjectPolicy : IPooledObjectPolicy<Foo>
{
public Foo Create()
{
return new Foo();
}
public bool Return(Foo obj)
{
return true;
}
}
如果每次编码都要定义这样的策略,会比较麻烦,可以自己定义一个通用的泛型实现。Microsoft.Extensions.ObjectPool
中也提供了一个默认的泛型实现:DefaultPooledObjectPolicy<T>
。如果不需要定义复杂的构造逻辑,使用默认的就行。下面我们来看看怎么使用。
对象池的使用
对象池使用的原则是:有借有还,再借不难。
当对象池中没有实例时,则创建实例并返回给调用组件;当对象池中已有实例时,则直接取一个现有实例返回给调用组件。而且这个过程是线程安全的。
Microsoft.Extensions.ObjectPool
提供了默认的对象池实现:DefaultObjectPool<T>
,它提供了借 Get
和还 Return
操作接口。创建对象池时需要提供池化策略 IPooledObjectPolicy<T>
作为其构造参数。
var policy = new DefaultPooledObjectPolicy<Foo>();
var pool = new DefaultObjectPool<Foo>(policy);
我们来看一个常规示例(C# 9.0 单文件完整代码):
using Microsoft.Extensions.ObjectPool;
using System;
var policy = new DefaultPooledObjectPolicy<Foo>();
var pool = new DefaultObjectPool<Foo>(policy);
// 借
var item1 = pool.Get();
// 还
pool.Return(item1);
Console.WriteLine("item 1: {0}", item1.Id);
// 借
var item2 = pool.Get();
// 还
pool.Return(item2);
Console.WriteLine("item 2: {0}", item2.Id);
Console.ReadKey();
public class Foo
{
public string Id { get; set; } = Guid.NewGuid().ToString("N");
}
打印结果:
通过打印的 Id 知道,item1
和 item2
是同一样对象。
我们再来看看只借不还会是什么样子:
// ...
// 借
var item1 = pool.Get();
Console.WriteLine("item 1: {0}", item1.Id);
// 再借
var item2 = pool.Get();
Console.WriteLine("item 2: {0}", item2.Id);
// ...
打印结果:
可以看到,两个对象是不同的实例。所以,当调用组件从对象池中借走一个对象实例,使用完后应立即归还给对象池,以便重复使用,避免因构造新对象消耗过多资源。
指定对象池容量
在创建 DefaultObjectPool<T>
时,还可以指定第二个参数:对象池的容量。它表示最大可从该对象池取出的对象数量,指定数量以外的被取走的对象将不会被池化。我来演示一下,大家就知道什么意思了,请看示例:
using Microsoft.Extensions.ObjectPool;
using System;
var policy = new DefaultPooledObjectPolicy<Foo>();
// 指定容量为 2。
var pool = new DefaultObjectPool<Foo>(policy, 2);
// 借走 3 个
var item1 = pool.Get();
Console.WriteLine("item 1: {0}", item1.Id);
var item2 = pool.Get();
Console.WriteLine("item 2: {0}", item2.Id);
var item3 = pool.Get();
Console.WriteLine("item 3: {0}", item3.Id);
// 再还会 3 个
pool.Return(item1);
pool.Return(item2);
pool.Return(item3);
// 再借走 3 个
var item4 = pool.Get();
Console.WriteLine("item 4: {0}", item4.Id);
var item5 = pool.Get();
Console.WriteLine("item 5: {0}", item5.Id);
var item6 = pool.Get();
Console.WriteLine("item 6: {0}", item6.Id);
Console.ReadKey();
注意示例代码中我给对象池指定了容量为 2,然后借走 3 个再归还 3 个,后面再借走 3 个。来看看打印结果:
我们看到,item1
与 item4
是同一个对象,item2
与 item5
是同一个对象。item3
与 item6
却不是同一个对象。
也就是说,当对象从池中取出超过指定容量的对象数量,虽然归还了相同数量的对象,但对象池只允许容纳 2 个对象,第三个对象不会被池化。
在 ASP.NET Core 中使用
ASP.NET Core 框架内置好了 Microsoft.Extensions.ObjectPool
,不需要单独安装。官方文档有个基于 ASP.NET Core 的使用示例:
https://docs.microsoft.com/en-us/aspnet/core/performance/objectpool
这个例子把 StringBuilder
做了池化。我这里就直接贴官方的例子了,为了更直观些,我把无关的代码简化掉了。
先定义一个中间件:
public class BirthdayMiddleware
{
private readonly RequestDelegate _next;
public BirthdayMiddleware(RequestDelegate next)
{
_next = next;
}
public async Task InvokeAsync(HttpContext context, ObjectPool<StringBuilder> builderPool)
{
var stringBuilder = builderPool.Get();
try
{
stringBuilder.Append("Hi");
// 其它处理
await context.Response.WriteAsync(stringBuilder.ToString());
}
finally // 即使出错也要保证归还对象
{
builderPool.Return(stringBuilder);
}
}
}
在 Startup
中注册相应的服务和中间件:
public class Startup
{
public void ConfigureServices(IServiceCollection services)
{
services.TryAddSingleton<ObjectPoolProvider, DefaultObjectPoolProvider>();
services.TryAddSingleton<ObjectPool<StringBuilder>>(serviceProvider =>
{
var provider = serviceProvider.GetRequiredService<ObjectPoolProvider>();
var policy = new StringBuilderPooledObjectPolicy();
return provider.Create(policy);
});
}
public void Configure(IApplicationBuilder app, IHostingEnvironment env)
{
app.UseMiddleware<BirthdayMiddleware>();
}
}
这个示例用了 DefaultObjectPoolProvider
,它是默认的对象池 Provider,所以你也可以自定义自己的对象池 Provider。
总结
Microsoft.Extensions.ObjectPool
提供的对象池功能还是挺灵活的。普通场景使用使用默认的池化策略、默认的对象池和默认的对象池提供者就可以满足需求,也可以自定义其中任意某部件来实现比较特殊或复杂的需求。
对象池的使用原则是:有借有还,再借不难。当调用组件从对象池中借走一个对象实例,使用完后应立即归还给对象池,以便重复利用,避免因过多的对象初始化影响系统性能。