FreeSql 中解决 AsTable 分表查询的数据重复问题

在使用 FreeSql 进行分表查询时，开发人员可能会遇到一个特殊的问题：多次调用 AsTable 方法会导致查询结果出现重复数据。本文将深入分析这个问题产生的原因，并提供有效的解决方案。

问题现象

当开发人员使用 FreeSql 的分表功能时，可能会观察到以下异常现象：

1. 第一次查询返回 0 条记录（正常）
2. 第二次和第三次查询相同的条件却返回了 2 条记录（异常）
3. 第四次查询又恢复正常，返回 1 条记录

通过 SQL 日志可以发现，异常查询实际上执行了 UNION ALL 操作，将同一张表的数据合并了两次，导致结果集出现重复。

问题根源

这个问题源于 FreeSql 的 AsTable 方法的工作机制：

1. 多次调用积累：每次调用 AsTable 方法都会添加一个新的分表规则，而不是替换之前的规则
2. UNION ALL 机制：FreeSql 会将所有通过 AsTable 指定的表规则合并查询，使用 UNION ALL 连接
3. 工作单元影响：在事务或工作单元中，这些规则会被保留，影响后续查询

解决方案

方案一：扩展方法清除表规则

我们可以创建扩展方法来清除已积累的表规则：

public static class MyFreeSqlExtensions
{
    public static ISelect<T1> DisableGlobalAsTable<T1>(this ISelect<T1> that)
    {
        var s0p = that as Select0Provider;
        if (s0p != null) s0p._tableRules.Clear();
        return that;
    }
    
    public static IInsert<T1> DisableGlobalAsTable<T1>(this IInsert<T1> that) where T1 : class
    {
        var s0p = that as InsertProvider<T1>;
        if (s0p != null) s0p._tableRule = null;
        return that;
    }
    
    // 其他操作类型的类似扩展...
}

使用方法：

var data = await _repository.Select
    .DisableGlobalAsTable()
    .AsTable(date)
    .Where(...)
    .ToListAsync();

方案二：确保单次 AsTable 调用

另一种方法是确保每个查询只调用一次 AsTable 方法：

public static ISelect<T> SafeAsTable<T>(this ISelect<T> select, DateTime date)
{
    date = new DateTime(date.Year, date.Month, 1);
    
    // 数据有效性检查
    if (date > DateTime.Now.AddDays(1).Date || date < DateTime.Parse(DbBeginDate.BegDate))
    {
        return select;
    }
    
    // 清除已有规则
    var s0p = select as Select0Provider;
    if (s0p != null) s0p._tableRules.Clear();
    
    return select.AsTable((t, s) => 
        t == typeof(T) ? $"{s}_{date:yyyyMM}" : s);
}

最佳实践

1. 避免多次调用：确保每个查询链中只调用一次 AsTable 方法
2. 使用扩展方法：封装安全的 AsTable 方法，自动处理规则清除
3. 监控SQL：利用 FreeSql 的 AOP 功能监控生成的 SQL，及时发现异常
4. 明确表规则：在复杂场景中，明确指定表名而不是依赖动态规则

总结

FreeSql 的 AsTable 分表功能非常强大，但需要理解其 UNION ALL 的工作机制。通过适当的封装和规则管理，可以避免数据重复问题，确保查询结果的准确性。建议开发团队根据项目需求选择最适合的解决方案，并在代码审查时特别注意 AsTable 的使用方式。