PluralKit/PluralKit.Core/Database/Utils/ConnectionUtils.cs
2021-08-27 11:03:47 -04:00

19 lines
531 B
C#

using System.Collections.Generic;
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);
}
}
}