欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

c# 剔除sql语句'尾巴'的五种方法

程序员文章站 2022-06-26 12:19:55
一:背景1. 讲故事这几天都在修复bug真的太忙了,期间也遇到了一个挺有趣bug,和大家分享一下,这是一块sql挺复杂的报表相关业务,不知道哪一位大佬在错综复杂的 嵌套 + 平行 if判断中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);
      }
    }

c# 剔除sql语句'尾巴'的五种方法

从代码中可以看出,只要在倒序的过程中,有一个字符和 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);
      }
    }

c# 剔除sql语句'尾巴'的五种方法

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

c# 剔除sql语句'尾巴'的五种方法

思想很简单,就是判断最后出现的 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);
      }
    }

c# 剔除sql语句'尾巴'的五种方法

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

c# 剔除sql语句'尾巴'的五种方法

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

c# 剔除sql语句'尾巴'的五种方法

可以看出已成功剔除,此时我很好奇底层到底是怎么实现的,源码如下:

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

c# 剔除sql语句'尾巴'的五种方法

四:总结

五种方式各有利弊,不管是简单粗暴的,基于性能的,灵活巧妙的,都能达到最终的目的,暂时就想到这5种,脑仁已经疼了????????????, 更多好玩的写法,欢迎大家留言讨论哈!

以上就是c# 剔除sql语句'尾巴'的五种方法的详细内容,更多关于c# sql语句的资料请关注其它相关文章!

相关标签: c# sql语句