1. 程式人生 > 實用技巧 >如何剔掉 sql 語句中的尾巴,我用 C# 苦思了五種辦法

如何剔掉 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

相信很多朋友用這個方法的場景大多在於剔除尾部的空格,哈哈,其實它還有一個隱藏功能,不僅可以剔除空格,還可以剔除任意多個指定的字元,這就