如何剔掉 sql 語句中的尾巴,我用 C# 苦思了五種辦法
一:背景
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
相信很多朋友用這個方法的場景大多在於剔除尾部的空格,哈哈,其實它還有一個隱藏功能,不僅可以剔除空格,還可以剔除任意多個指定的字元,這就