C#去除SQL语句末尾'union all'的五种方法

"这篇文章主要介绍了如何在C#中去除SQL语句末尾的`UNION ALL`关键字，针对这个问题，作者分享了五种不同的解决方法。这些方法适用于处理复杂的SQL拼接逻辑，尤其是在报表业务中，避免因拼接错误导致的SQL语法错误。" 文章详细内容： 在开发过程中，可能会遇到SQL语句拼接错误的情况，特别是在报表业务中，由于条件分支复杂，容易在`UNION ALL`操作中留下多余的尾巴，导致SQL语句无法执行。本文将介绍五种方法来解决这个问题，帮助开发者更安全地处理这种情况。 ### 一、背景 在修复bug时，作者遇到了一个有趣的案例。一个复杂的SQL报表业务中，由于条件判断和拼接操作的复杂性，导致生成的SQL语句在某些特定的UI组合下出现`UNION ALL`关键字未被正确处理的问题。例如： ```sql var sql = "select 1 union all select 2 union all select 3 union all"; ``` 这样的SQL语句在执行时会报错。理想的解决方案是避免在不应添加`UNION ALL`的地方进行拼接，但由于业务代码复杂，直接修改可能引入新的风险。因此，本文将探讨如何在不改变原有逻辑的情况下，剔除SQL语句末尾的`UNION ALL`。 ### 二、剔除`UNION ALL`的五大方式 #### 1. 最原始的`for`循环 这种方法通过倒序遍历SQL字符串和关键词`UNION ALL`，检查它们是否完全匹配。如果匹配，则截取SQL语句，去除`UNION ALL`。代码如下： ```csharp static void Main(string[] args) { var sql = "select 1 union all select 2 union all select 3 union all"; var keyword = "unionall"; var isAllMatch = true; int i = 0; for (i = 1; i <= keyword.Length; i++) { if (keyword[keyword.Length - i] != sql[sql.Length - i]) { isAllMatch = false; break; } } if (isAllMatch) { var query = sql.Substring(0, sql.Length - i + 1); Console.WriteLine(query); } } ``` #### 2. 使用`Substring`简化 虽然`for`循环可以解决问题，但代码显得过于繁琐。可以利用C#的`Substring`方法简化代码，直接从SQL字符串的尾部切出`UNION ALL`的长度，如下： ```csharp var sql = "select 1 union all select 2 union all select 3 union all"; var keyword = "unionall"; var query = sql.EndsWith(keyword) ? sql.Substring(0, sql.Length - keyword.Length) : sql; Console.WriteLine(query); ``` #### 3. 使用`LastIndexOf`和`Substring` 还可以通过`LastIndexOf`找到`UNION ALL`的位置，然后使用`Substring`截取。这样可以处理`UNION ALL`出现在中间的情况： ```csharp var sql = "select 1 union all select 2 union all select 3 union all"; var index = sql.LastIndexOf("union all"); var query = index > 0 ? sql.Substring(0, index) : sql; Console.WriteLine(query); ``` #### 4. 使用正则表达式 利用正则表达式可以更灵活地处理复杂的匹配和替换： ```csharp using System.Text.RegularExpressions; var sql = "select 1 union all select 2 union all select 3 union all"; var regex = new Regex("(\\s*union\\s+all\\s*)+$"); var query = regex.Replace(sql, ""); Console.WriteLine(query); ``` #### 5. 自定义扩展方法 创建一个自定义的扩展方法，用于处理这类问题，使代码更具可读性： ```csharp public static class StringExtensions { public static string RemoveTrailingUnionAll(this string sql) { var index = sql.LastIndexOf("union all", StringComparison.OrdinalIgnoreCase); return index > 0 ? sql.Substring(0, index) : sql; } } var sql = "select 1 union all select 2 union all select 3 union all"; var query = sql.RemoveTrailingUnionAll(); Console.WriteLine(query); ``` 以上五种方法均能在不同程度上解决SQL语句尾部`UNION ALL`剔除的问题。选择哪种方法取决于具体需求和代码风格，但尽量保持代码简洁、可读性强是良好的编程习惯。