PluralKit/PluralKit.Core/Database/Utils/ConnectionUtils.cs

17 lines
455 B
C#

using System.Data.Common;
using Dapper;
namespace PluralKit.Core;
public static class ConnectionUtils
{
public static async IAsyncEnumerable<T> QueryStreamAsync<T>(this IPKConnection conn, string sql, object param)
{
await using var reader = (DbDataReader)await conn.ExecuteReaderAsync(sql, param);
var parser = reader.GetRowParser<T>();
while (await reader.ReadAsync())
yield return parser(reader);
}
}