修复Mysql

2026-07-26 10:19:46 +08:00
parent 634ef69a27
commit 72297aa23f
31 changed files with 1642 additions and 1192 deletions
@@ -0,0 +1,47 @@
+using System.Linq.Expressions;
+
+namespace Jiaowu.Api.Infrastructure.Persistence;
+
+public static class QueryableCollectionExtensions
+{
+    public static IQueryable<TEntity> WhereIn<TEntity, TValue>(
+        this IQueryable<TEntity> source,
+        IEnumerable<TValue> values,
+        Expression<Func<TEntity, TValue>> valueSelector)
+    {
+        var predicate = BuildPredicate(values, valueSelector, Expression.OrElse, false);
+        return source.Where(predicate);
+    }
+
+    public static IQueryable<TEntity> WhereNotIn<TEntity, TValue>(
+        this IQueryable<TEntity> source,
+        IEnumerable<TValue> values,
+        Expression<Func<TEntity, TValue>> valueSelector)
+    {
+        var predicate = BuildPredicate(values, valueSelector, Expression.AndAlso, true);
+        return source.Where(predicate);
+    }
+
+    private static Expression<Func<TEntity, bool>> BuildPredicate<TEntity, TValue>(
+        IEnumerable<TValue> values,
+        Expression<Func<TEntity, TValue>> valueSelector,
+        Func<Expression, Expression, BinaryExpression> combine,
+        bool negate)
+    {
+        Expression? body = null;
+        foreach (var value in values.Distinct())
+        {
+            Expression comparison = Expression.Equal(
+                valueSelector.Body,
+                Expression.Constant(value, typeof(TValue)));
+            if (negate)
+                comparison = Expression.Not(comparison);
+            body = body is null ? comparison : combine(body, comparison);
+        }
+
+        body ??= Expression.Constant(negate);
+        return Expression.Lambda<Func<TEntity, bool>>(
+            body,
+            valueSelector.Parameters);
+    }
+}
@@ -0,0 +1,44 @@
+using System.Data;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.Storage;
+
+namespace Jiaowu.Api.Infrastructure.Persistence;
+
+public static class RetriableTransactionExtensions
+{
+    public static Task<TResult> ExecuteInRetriableTransactionAsync<TResult>(
+        this DbContext db,
+        Func<IDbContextTransaction, Task<TResult>> operation,
+        CancellationToken cancellationToken,
+        IsolationLevel? isolationLevel = null)
+    {
+        var strategy = db.Database.CreateExecutionStrategy();
+        return strategy.ExecuteAsync(async () =>
+        {
+            await using var transaction = isolationLevel.HasValue
+                ? await db.Database.BeginTransactionAsync(
+                    isolationLevel.Value,
+                    cancellationToken)
+                : await db.Database.BeginTransactionAsync(cancellationToken);
+            return await operation(transaction);
+        });
+    }
+
+    public static Task ExecuteInRetriableTransactionAsync(
+        this DbContext db,
+        Func<IDbContextTransaction, Task> operation,
+        CancellationToken cancellationToken,
+        IsolationLevel? isolationLevel = null)
+    {
+        var strategy = db.Database.CreateExecutionStrategy();
+        return strategy.ExecuteAsync(async () =>
+        {
+            await using var transaction = isolationLevel.HasValue
+                ? await db.Database.BeginTransactionAsync(
+                    isolationLevel.Value,
+                    cancellationToken)
+                : await db.Database.BeginTransactionAsync(cancellationToken);
+            await operation(transaction);
+        });
+    }
+}