Refactor async queries + add IAsyncEnumerable support

Author: Joy-less

SQLiteSharp/AsyncTableQuery.cs

@@ -939,10 +939,6 @@ public Task<int> ExecuteAsync(string query, params IEnumerable<object?> paramete
     public Task RunInTransactionAsync(Action action) {
         return Task.Run(() => RunInTransaction(action));
     }
-    /// <inheritdoc cref="Table{T}"/>
-    public AsyncTableQuery<T> TableAsync<T>() where T : new() {
-        return new AsyncTableQuery<T>(Table<T>());
-    }
     /// <inheritdoc cref="ExecuteScalar{T}(string, IEnumerable{object?})"/>
     public Task<T> ExecuteScalarAsync<T>(string query, params IEnumerable<object?> parameters) {
         return Task.Run(() => ExecuteScalar<T>(query, parameters));

SQLiteSharp/SQLiteConnection.cs

@@ -7,5 +7,6 @@
   </PropertyGroup>
   <ItemGroup>
     <PackageReference Include="SQLitePCLRaw.bundle_e_sqlcipher" Version="2.1.10" />
+    <PackageReference Include="System.Linq.Async" Version="6.0.1" />
   </ItemGroup>
 </Project>

SQLiteSharp/SQLiteSharp.csproj

@@ -1,11 +1,11 @@
 using System.Collections;
-using System.Linq.Expressions;
-using System.Reflection;
 using System.Text;
+using System.Reflection;
+using System.Linq.Expressions;
 
 namespace SQLiteSharp;
 
-public class TableQuery<T>(SQLiteConnection connection, TableMap table) : IEnumerable<T>, IEnumerable {
+public class TableQuery<T>(SQLiteConnection connection, TableMap table) : IEnumerable<T>, IEnumerable, IAsyncEnumerable<T> {
     public SQLiteConnection Connection { get; } = connection;
     public TableMap Table { get; } = table;
 
@@ -423,6 +423,10 @@ public IEnumerator<T> GetEnumerator() {
     IEnumerator IEnumerable.GetEnumerator() {
         return GetEnumerator();
     }
+    /// <inheritdoc cref="GetEnumerator()"/>
+    public IAsyncEnumerator<T> GetAsyncEnumerator(CancellationToken cancelToken = default) {
+        return this.ToAsyncEnumerable().GetAsyncEnumerator(cancelToken);
+    }
 
     /// <summary>
     /// Returns a list of all the elements matching the query.
@@ -460,4 +464,45 @@ public T First(Expression<Func<T, bool>> predicate) {
     public T? FirstOrDefault(Expression<Func<T, bool>> predicate) {
         return Where(predicate).FirstOrDefault();
     }
+
+    /// <inheritdoc cref="ToList()"/>
+    public Task<List<T>> ToListAsync() {
+        return Task.Run(ToList);
+    }
+    /// <inheritdoc cref="ToArray()"/>
+    public Task<T[]> ToArrayAsync() {
+        return Task.Run(ToArray);
+    }
+    /// <inheritdoc cref="Count()"/>
+    public Task<int> CountAsync() {
+        return Task.Run(Count);
+    }
+    /// <inheritdoc cref="Count(Expression{Func{T, bool}})"/>
+    public Task<int> CountAsync(Expression<Func<T, bool>> predicate) {
+        return Task.Run(() => Count(predicate));
+    }
+    /// <inheritdoc cref="ElementAt(int)"/>
+    public Task<T> ElementAtAsync(int index) {
+        return Task.Run(() => ElementAt(index));
+    }
+    /// <inheritdoc cref="First()"/>
+    public Task<T> FirstAsync() {
+        return Task.Run(First);
+    }
+    /// <inheritdoc cref="FirstOrDefault()"/>
+    public Task<T?> FirstOrDefaultAsync() {
+        return Task.Run(FirstOrDefault);
+    }
+    /// <inheritdoc cref="First(Expression{Func{T, bool}})"/>
+    public Task<T> FirstAsync(Expression<Func<T, bool>> predicate) {
+        return Task.Run(() => First(predicate));
+    }
+    /// <inheritdoc cref="FirstOrDefault(Expression{Func{T, bool}})"/>
+    public Task<T?> FirstOrDefaultAsync(Expression<Func<T, bool>> predicate) {
+        return Task.Run(() => FirstOrDefault(predicate));
+    }
+    /// <inheritdoc cref="Delete(Expression{Func{T, bool}}?)"/>
+    public Task<int> DeleteAsync(Expression<Func<T, bool>> predicate) {
+        return Task.Run(() => Delete(predicate));
+    }
 }