c# 剔除sql语句'尾巴'的五种方法-C/C++开发

这篇文章主要介绍了c# 剔除sql语句'尾巴'的五种方法，

一：背景

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


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


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


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


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


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