一：背景
1. 講故事
這幾天都在修復bug真的太忙了，期間也遇到了一個挺有趣bug，和大家分享一下，這是一塊sql挺複雜的報表相關業務，不知道哪一位大佬在錯綜複雜的 巢狀 + 平行 if判斷中sql拼接在某些UI組合下出問題了，最終的 sql 架構類似這樣的。
var sql = "select 1 union all select 2 union all select 3 union all";
這種sql到資料庫去肯定是報錯的，有些朋友可能想說這還不簡單，在相關的 if 判斷中不要追加這個 union all 就好了，這確實是一個根治的辦法，但現實情況這一塊的業務太複雜了，也不太敢改裡面的程式碼，改的沒問題還好，改出問題你得兜著走，所以最保險的辦法就是怎麼去掉 union all 這個大尾巴，所以我乾脆思考了一會，想出瞭如下五種辦法。
二：剔除 union all 的五大方式
1. 最原始的 for 迴圈
最簡單的辦法就是通過 for 迴圈搞定，我可以倒序判斷最後幾個字元是不是關鍵詞 union all 就可以了，如下程式碼所示：
static void Main(string[] args)
{
var sql = "select 1 union all select 2 union all select 3 union all";
var keyword = "union all";
var isall = true;
int i = 0;
for (i = 1; i <= keyword.Length; i++)
{
if (keyword[keyword.Length - i] != sql[sql.Length - i])
{
isall = false;
break;
}
}
if (isall)
{
var query = sql.Substring(0, sql.Length - i + 1);
Console.WriteLine(query);
}
}
從程式碼中可以看出，只要在倒序的過程中，有一個字元和 keyword 中的不符，那就直接跳出，否則就是全匹配，拿到最後的 i 進行 Substring 擷取即可。
2. 使用 Substring 搞定
第一種方式確實可以實現，但實現的並不輕鬆，畢竟大家都是用 C# 寫程式碼而不是 C，為了這點小功能寫了這麼多程式碼，顯得太 low 了，所以儘量能用類庫的方法就用類庫的方法吧，改進措施很簡單，可以從 sql 尾部切出 keyword.length 個字元，也就是： start：sql.length - keyword.length，然後判斷一下是否和 keyword 相等即可，程式碼修改如下：
static void Main(string[] args)
{
var sql = "select 1 union all select 2 union all select 3 union all";
var keyword = "union all";
var isSucc = sql.Substring(sql.Length - keyword.Length) == keyword;
if (isSucc)
{
var query = sql.Substring(0, sql.Length - keyword.Length);
Console.WriteLine(query);
}
}
3. 使用 LastIndexOf
第二種方式寫出來的程式碼確實比較簡潔，但大家有沒有發現一個問題，我為了獲取最後的 string 做了兩次 substring 操作，也就是說在託管堆中生成了兩個 string 物件，那能不能免掉一個 substring 呢？給 gc 減輕一些負擔，這就可以用到 LastIndexOf 方法了，程式碼如下：
static void Main(string[] args)
{
var sql = "select 1 union all select 2 union all select 3 union all";
var keyword = "union all";
var index = sql.LastIndexOf(keyword);
if (sql.Length - index == keyword.Length)
{
var query = sql.Substring(0, index);
Console.WriteLine(query);
}
}
思想很簡單，就是判斷最後出現的 union all的位置到尾部的距離 是否恰好和 keyword.length 一致，如果是的話那 keyword 就是 sql 的大尾巴，這裡的 if 寫的有點難懂，其實還可以使用 EndsWith 再優化一下程式碼：
static void Main(string[] args)
{
var sql = "select 1 union all select 2 union all select 3 union all";
var keyword = "union all";
if (sql.EndsWith(keyword))
{
var query = sql.Substring(0, sql.Length - keyword.Length);
Console.WriteLine(query);
}
}
4. 使用 Split 切割
前面幾種方式都是在 string 上做文章，要麼 substring，要麼 LastIndexOf，要麼 EndsWith，其實也可以跳出這個定勢思維，轉換成陣列進行處理，用 union all 作為分隔符切割字串，如果陣列的最後一個元素為 string.Empty，那就表明 sql 尾巴就是 keyword, 對吧，程式碼修改如下：
static void Main(string[] args)
{
var sql = "select 1 union all select 2 union all select 3 union all";
var keyword = "union all";
var arr = sql.Split("union all");
if (string.IsNullOrEmpty(arr[arr.Length - 1]))
{
var query = string.Join(keyword, arr.Take(arr.Length - 1));
Console.WriteLine(query);
}
}
5. 使用 TrimEnd
相信很多朋友用這個方法的場景大多在於剔除尾部的空格，哈哈，其實它還有一個隱藏功能，不僅可以剔除空格，還可以剔除任意多個指定的字元，這就??了，不信的話可以看看 TrimEnd 方法簽名即可：
public sealed class String
{
public string TrimEnd(params char[] trimChars)
{
if (trimChars == null || trimChars.Length == 0)
{
return TrimHelper(1);
}
return TrimHelper(trimChars, 1);
}
}
可以看到 trimChars 是一個字元陣列，你可以灌入你想剔除的任意多個字元，有了這個思想，我可以將 keyword 轉成 char[] 再灌入到 TrimEnd 即可，程式碼如下：
static void Main(string[] args)
{
var sql = "select 1 union all select 2 union all select 3 union all";
var keyword = "union all";
var query = sql.TrimEnd(keyword.ToArray());
Console.WriteLine(query);
}
可以看出已成功剔除，此時我很好奇底層到底是怎麼實現的，原始碼如下：
private string TrimHelper(char[] trimChars, int trimType)
{
int num = Length - 1;
int i = 0;
if (trimType != 1)
{
for (i = 0; i < Length; i++)
{
int num2 = 0;
char c = this[i];
for (num2 = 0; num2 < trimChars.Length && trimChars[num2] != c; num2++)
{
}
if (num2 == trimChars.Length)
{
break;
}
}
}
if (trimType != 0)
{
for (num = Length - 1; num >= i; num--)
{
int num3 = 0;
char c2 = this[num];
for (num3 = 0; num3 < trimChars.Length && trimChars[num3] != c2; num3++)
{
}
if (num3 == trimChars.Length)
{
break;
}
}
}
return CreateTrimmedString(i, num);
}
private string CreateTrimmedString(int start, int end)
{
int num = end - start + 1;
if (num == Length)
{
return this;
}
if (num == 0)
{
return Empty;
}
return InternalSubString(start, num);
}
從原始碼中可以看出，它的程式碼邏輯是從後往前篩查，一直到遇見的字元不在 char[] 中為止，這裡就有一個潛在的坑，大家要注意了，比如下面sql的結果：
static void Main(string[] args)
{
var sql = "select * from english_l union all";
var keyword = "union all";
var query = sql.TrimEnd(keyword.ToArray());
Console.WriteLine(query);
}
可以看到： english_l 中的 l 也被意外剔除，這就不是我想看到的結果。
此坑需要感謝公號暱稱為：
神仙氏族的友情提醒 ?
四：總結
五種方式各有利弊，不管是簡單粗暴的，基於效能的，靈活巧妙的，都能達到最終的目的，暫時就想到這5種，腦仁已經疼了???， 更多好玩的寫法，歡迎大家留言討論哈！
更多高質量乾貨：參見我的 GitHub: dotnetfly
