快捷導(dǎo)航

c# 剔除sql語句'尾巴'的五種方法

更新時間：2020年11月21日 17:02:45 作者：一線碼農(nóng)

這篇文章主要介紹了c# 剔除sql語句'尾巴'的五種方法，

一：背景

1. 講故事

這幾天都在修復(fù)bug真的太忙了，期間也遇到了一個挺有趣bug，和大家分享一下，這是一塊sql挺復(fù)雜的報表相關(guān)業(yè)務(wù)，不知道哪一位大佬在錯綜復(fù)雜的嵌套 + 平行 if判斷中sql拼接在某些UI組合下出問題了，最終的 sql 架構(gòu)類似這樣的。

var sql = "select 1 union all select 2 union all select 3 union all";

這種sql到數(shù)據(jù)庫去肯定是報錯的，有些朋友可能想說這還不簡單，在相關(guān)的 if 判斷中不要追加這個 union all 就好了，這確實是一個根治的辦法，但現(xiàn)實情況這一塊的業(yè)務(wù)太復(fù)雜了，也不太敢改里面的代碼，改的沒問題還好，改出問題你得兜著走，所以最保險的辦法就是怎么去掉 union all 這個大尾巴，所以我干脆思考了一會，想出了如下五種辦法。

二：剔除 union all 的五大方式

1. 最原始的 for 循環(huán)

最簡單的辦法就是通過 for 循環(huán)搞定，我可以倒序判斷最后幾個字符是不是關(guān)鍵詞 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 搞定

第一種方式確實可以實現(xiàn)，但實現(xiàn)的并不輕松，畢竟大家都是用 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

第二種方式寫出來的代碼確實比較簡潔，但大家有沒有發(fā)現(xiàn)一個問題，我為了獲取最后的 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);
      }
    }

思想很簡單，就是判斷最后出現(xiàn)的 union all的位置到尾部的距離是否恰好和 keyword.length 一致，如果是的話那 keyword 就是 sql 的大尾巴，這里的 if 寫的有點難懂，其實還可以使用 EndsWith 再優(yōu)化一下代碼：

    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，其實也可以跳出這個定勢思維，轉(zhuǎn)換成數(shù)組進行處理，用 union all 作為分隔符切割字符串，如果數(shù)組的最后一個元素為 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 是一個字符數(shù)組，你可以灌入你想剔除的任意多個字符，有了這個思想，我可以將 keyword 轉(zhuǎn)成 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);
    }

可以看出已成功剔除，此時我很好奇底層到底是怎么實現(xiàn)的，源碼如下：

    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的結(jié)果：

    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);
    }

四：總結(jié)

五種方式各有利弊，不管是簡單粗暴的，基于性能的，靈活巧妙的，都能達到最終的目的，暫時就想到這5種，腦仁已經(jīng)疼了😂😂😂，更多好玩的寫法，歡迎大家留言討論哈！

以上就是c# 剔除sql語句'尾巴'的五種方法的詳細內(nèi)容，更多關(guān)于c# sql語句的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: