百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 热门文章 > 正文

如何剔掉 sql 语句中的尾巴,我用 C# 苦思了五种办法

bigegpt 2024-08-14 14:44 2 浏览

一:背景

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

相信很多朋友用这个方法的场景大多在于剔除尾部的空格,哈哈,其实它还有一个隐藏功能,不仅可以剔除空格,还可以剔除任意多个指定的字符,这就了,不信的话可以看看 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);
        }

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


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

可以看到: english_l 中的 l 也被意外剔除,这就不是我想看到的结果。

此坑需要感谢公号昵称为:神仙氏族 的友情提醒

四:总结

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

更多高质量干货:参见我的 GitHub: dotnetfly**

相关推荐

为3D手游打造, Visual Studio Unity扩展下载

IT之家(www.ithome.com):为3D手游打造,VisualStudioUnity扩展下载7月30日消息,微软正式发布升级版VisualStudioToolsforUnity扩...

由ArcMap属性字段自增引出字段计算器使用Python的技巧

1.前言前些日子有人问我ArcMap中要让某个字段的值实现自增有什么方法?我首先想到像SQLServer中对于数值型字段可以设置自增。所以我打开ArcCatalog查看发现只提供默认值,没办法只能看...

微软首次回答 HoloLens 相关问题,终于爆料了

fengo2015/04/2115:11注:本文作者张静是NVIDIAGPU架构师,微信公众号“黑客与画家”(HackerAndPainter),知乎专栏地址。欢迎各位童鞋与他交流探讨。...

C#指针的应用(c#指针类型)

C#在有限的范围内支持指针。C#的指针只不过是一个持有另一类型内存地址的变量。但是在C#中,指针只能被声明为持有值类型和数组的内存地址。与引用类型不同,指针类型不被默认的垃圾收集机制所跟踪。出于同...

C# 堆栈(Stack)(c# 堆栈中定位调用messagebox 的地方)

C#集合在C#中,堆栈(Stack)是一种后进先出(LIFO,LastInFirstOut)的数据结构。堆栈(Stack)适用于存储和按顺序处理数据,其中最新添加的元素会最先被移除。堆...

欢迎回来:Fortran意外重回流行编程语言20强榜单

TIOBE指数是用来确定一种编程语言受欢迎程度的指标之一。它并不表明哪种编程语言是最好的,也不表明哪种编程语言写的代码行数最多,而是利用在谷歌、维基百科、必应、亚马逊、YouTube等各种引擎和网站上...

C#+NET MAUI实现跨平台/终端(linux,win,ios等)解决方案

简介.NETMulti-platformAppUI(.NETMAUI)是一个跨平台的框架,用于使用C#和XAML创建移动和桌面应用程序。使用.NETMAUI,您可以用一套代码库开发可以在A...

C#代码安全红线:SQL注入防护终极方案,让你的系统固若金汤

在数字化时代,应用系统的安全性至关重要。而SQL注入攻击,长期盘踞在OWASP(OpenWebApplicationSecurityProject)漏洞榜单的前列,成为众多基于数据库的应用系统...

C# (一)状态机模式(状态机代码实现)

最近空闲,炒炒隔夜饭,以前这些模式在自己项目种应用过不少,但一直没有像别人那样写一个系列,最近年纪大了,很多东西都忘记了,特别AI的兴起,更少写代码了,反正没什么事情,自己在重写一遍吧。创建型模式(5...

C# 中 Predicate 详解(c#中的replace)

Predicate泛型委托:表示定义一组条件并确定指定对象是否符合这些条件的方法。此委托由Array和List类的几种方法使用,用于在集合中搜索元素。Predicate<T>...

C#中$的用法?(c#中&&什么意思)

文章来自AI问答。在C#中,$符号用于字符串插值(StringInterpolation)。字符串插值是C#6.0引入的一种特性,它允许你在字符串中直接嵌入表达式,而不需要使用string.For...

C#并行编程:Parallel类(c# 并行处理)

在Parallel类中提供了三个静态方法作为结构化并行的基本形式:Parallel.Invoke方法:并行执行一组委托。Parallel.For方法:执行与C#for循环等价的并行方法。Parall...

颠覆认知!用Span重构foreach循环竟让数据处理快如闪电

在C#编程的世界里,数据处理效率始终是开发者们关注的焦点。随着项目规模的扩大和数据量的激增,哪怕是细微的性能提升,都可能对整个应用的响应速度和用户体验产生深远影响。近年来,C#引入的Span<T...

Unity3D手游开发实践《腾讯桌球》客户端开发经验总结

本次分享总结,起源于腾讯桌球项目,但是不仅仅限于项目本身。虽然基于Unity3D,很多东西同样适用于Cocos。本文从以下10大点进行阐述:1.架构设计2.原生插件/平台交互3.版本与补丁4.用脚本,...

.NET 7 AOT 的使用以及 .NET 与 Go 互相调用

目录背景C#部分环境要求创建一个控制台项目体验AOT编译C#调用库函数减少体积C#导出函数C#调用C#生成的AOTGolang部分安装GCCGolang导出函数.NETC#...