OC格式化SQL语句函数，提取表名

最新推荐文章于 2024-10-12 11:42:17 发布
最新推荐文章于 2024-10-12 11:42:17 发布
阅读量635 收藏
点赞数
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/mlibai/article/details/47123685
版权

返回值是一个数组：

第一个元素：是查询的table名，如果查询的表不是基本表，返回的是自动创建的临时表名。

第二个元素：是格式化后的sql语句。

NSArray* analyseSQLString(NSString *sqlString, NSInteger page, NSInteger pageSize){
    __block NSString *selftable, *_sql, *sqlString2;
    sqlString2 = sqlString;

    // 格式化SQL语句为，符号前后又空格，逗号前没空格，复合符号间无空格。将字段值存入数组，替换为?
    NSCharacterSet *spaceCharSet = [NSCharacterSet whitespaceAndNewlineCharacterSet];
    NSCharacterSet *symbolCharSet = [NSCharacterSet characterSetWithCharactersInString:@"+-x/><=,"];
    sqlString2 = [[sqlString2 stringByTrimmingCharactersInSet:spaceCharSet] lowercaseString];
    BOOL meetSpaceChar = NO; // 遇到空白字符
    BOOL meetQuoteString = NO; // 上一个是单引号
    BOOL meetSymbolChar = NO;    // 遇到 + - x ÷ > < =
    BOOL meetValueString = NO;
    NSMutableArray *allSqlValues = [NSMutableArray array];
    NSMutableString *tmpSqlString = [NSMutableString string];
    NSMutableString *sqlValueString = [NSMutableString string];

    for (NSInteger i = 0; i < sqlString2.length; i ++) {
        NSString *oneCharString = [sqlString2 substringWithRange:NSMakeRange(i, 1)];
        unichar aChar = [sqlString2 characterAtIndex:i];
        if ([oneCharString isEqualToString:@"'"]) { // 遇到值
            meetValueString = YES;
            i ++; // 跳过当前的 单引号
            while (meetValueString && i < sqlString2.length) {
                oneCharString = [sqlString2 substringWithRange:NSMakeRange(i, 1)];
                aChar = [sqlString2 characterAtIndex:i];
                if ([oneCharString isEqualToString:@"'"]) {
                    if (meetQuoteString) {                // 如果当前是个单引号，且上一个也是单引号，加入单引号
                        [sqlValueString appendString:@"'"];
                        meetQuoteString = NO;
                    }else{
                        if (i == sqlString2.length - 1) { // 是最后一个字符时
                            meetValueString = NO;
                        }else{
                            meetQuoteString = YES;
                        }
                    }
                    i ++;
                }else{
                    if (meetQuoteString) {  // 当前不是单引号，而上一个是单引号，说明值结束了
                        meetValueString = NO;
                        meetQuoteString = NO;
                    }else{
                        [sqlValueString appendString:oneCharString];
                        i ++;
                    }
                }
            } // 以单引号后面第一个字符结束循环 或 sql语句遍历完毕
            if (meetValueString && i == sqlString2.length) { // 循环结束了，没有找到值的结束分号
                NSLog(@"SQL语句不正确。");
                return nil;
            }else{
                [allSqlValues addObject:[sqlValueString copy]];
                [sqlValueString setString:@""];
                if (meetSymbolChar) {
                    [tmpSqlString appendFormat:@" %@", @"?"];
                }else{
                    [tmpSqlString appendString:@"?"];
                }
            }
            meetSpaceChar = NO;
            meetSymbolChar = NO;
        }
        if (i < sqlString2.length) {
            if ([spaceCharSet characterIsMember:aChar]) { // 是空白字符
                if (!meetSpaceChar) { // 上一个不是空白字符
                    if (!meetSymbolChar) { // 上一个不是符号
                        [tmpSqlString appendString:@" "];
                        meetSpaceChar = YES;
                        meetSymbolChar = NO;
                    }
                }
            }else{// 不是空白字符
                if ([symbolCharSet characterIsMember:aChar]) { // 是符号
                    if (!meetSymbolChar) {
                        meetSymbolChar = YES;
                        if (meetSpaceChar) { // 上一个是空白字符
                            [tmpSqlString appendString:oneCharString];
                        }else{ // 上一个不是空白字符
                            if ([oneCharString isEqualToString:@","]) {
                                [tmpSqlString appendString:oneCharString];
                            }else{
                                [tmpSqlString appendFormat:@" %@", oneCharString];
                            }
                        }
                    }else{ // 上一个也是符号
                        [tmpSqlString appendString:oneCharString];
                    }
                }else{ // 不是符号，也不是空格
                    if (meetSymbolChar) {
                        [tmpSqlString appendFormat:@" %@", oneCharString]; // 前一个是符号，添加空格
                    }else{
                        [tmpSqlString appendString:oneCharString]; // 前一个不是符号，直接添加
                    }
                    meetSymbolChar = NO;
                }
                meetSpaceChar = NO;
            }
        }
    }
    sqlString2 = [NSString stringWithString:tmpSqlString];
    
    // 找出表名
    NSRange selectRange = [sqlString2 rangeOfString:@"select "];
    NSRange fromRange = [sqlString2 rangeOfString:@" from "];
    NSRange whereRange = [sqlString2 rangeOfString:@" where "];
    NSRange groupRange = [sqlString2 rangeOfString:@" group " options:NSBackwardsSearch];
    NSRange orderRange = [sqlString2 rangeOfString:@" order by " options:NSBackwardsSearch];
    NSRange limitRange = [sqlString2 rangeOfString:@" limit " options:NSBackwardsSearch];
    NSRange fieldsRange = NSMakeRange(selectRange.location + selectRange.length, fromRange.location - selectRange.location - selectRange.length);
    
    if (selectRange.length == 0 || fromRange.length == 0) {
        NSLog(@"SQL错误！");
        return nil;
    }
    
    // 格式化字段
    NSString *fields = [sqlString2 substringWithRange:fieldsRange];
    __block NSCharacterSet *charSet = [NSCharacterSet characterSetWithCharactersInString:@"`()"];
    BOOL meetRowID = NO;
    if ([fields containsString:@"rowid"]) {
        NSArray *array = [fields componentsSeparatedByString:@","];
        [tmpSqlString setString:@""];
        for (NSInteger i = 0; i < array.count; i ++) {
            NSString *str = [[array[i] stringByTrimmingCharactersInSet:spaceCharSet] stringByTrimmingCharactersInSet:charSet];
            if ([str isEqualToString:@"rowid"]) {
                meetRowID = YES;
            }
            if (!([str containsString:@" "] || [str isEqualToString:@"*"])) {
                str = [NSString stringWithFormat:@"`%@`", str];
            }
            if (i == 0) {
                [tmpSqlString appendString:str];
            }else{
                [tmpSqlString appendFormat:@", %@", str];
            }
        }
        fields = [NSString stringWithString:tmpSqlString]; // fields的替换要放到最后，sqlString下面还要使用
    }
    
    NSInteger minRangeLocation = 0;
    if (groupRange.length > 0) {
        minRangeLocation = groupRange.location;
    }
    if (orderRange.length > 0) {
        if (minRangeLocation > orderRange.location) {
            minRangeLocation = orderRange.location;
        }
    }
    if (limitRange.length > 0) {
        if (minRangeLocation > limitRange.location) {
            minRangeLocation = limitRange.location;
        }
    }
    
    __block BOOL writable;
    NSString *(^getTmpTableName)() = ^{
        return [NSString stringWithFormat:@"tmpTable%05u", arc4random()%1000000];
    };
    void (^TestAsInTableString)(NSRange, NSString *) = ^(NSRange tableRange, NSString *tmp){
        NSRange asRange = [tmp rangeOfString:@" as " options:NSBackwardsSearch];
        if (asRange.length > 0) {
            selftable = [tmp substringFromIndex:asRange.location + asRange.length];
            writable = NO;
        }else{
            if ([tmp containsString:@" "]) {
                NSString *tmpTableName = getTmpTableName();
                while ([sqlString2 containsString:tmpTableName]) {
                    tmpTableName = getTmpTableName();
                }
                selftable = tmpTableName;
                if ([tmp hasPrefix:@"("]) {
                    sqlString2 = [sqlString2 stringByReplacingCharactersInRange:tableRange withString:[NSString stringWithFormat:@"%@ as `%@`", tmp, tmpTableName]];
                }else{
                    sqlString2 = [sqlString2 stringByReplacingCharactersInRange:tableRange withString:[NSString stringWithFormat:@"(%@) as `%@`", tmp, tmpTableName]];
                }
                
                writable = NO;
            }else{
                selftable = [tmp stringByTrimmingCharactersInSet:charSet];
                sqlString2 = [sqlString2 stringByReplacingCharactersInRange:tableRange withString:[NSString stringWithFormat:@"`%@`", tmp]];
            }
        }
    };
    
    if (whereRange.length == 0) {
        if (minRangeLocation == 0) {
            NSString *tmp = [sqlString2 substringFromIndex:(fromRange.length + fromRange.location)];
            NSRange tableRange = [sqlString2 rangeOfString:tmp];
            TestAsInTableString(tableRange, tmp);
        }else{
            NSRange tableRange = NSMakeRange(fromRange.location + fromRange.length, 0);
            tableRange.length = minRangeLocation - tableRange.location;
            NSString *tmp = [sqlString2 substringWithRange:tableRange];
            TestAsInTableString(tableRange, tmp);
        }
    }else{
        NSRange tableRange = NSMakeRange(fromRange.location + fromRange.length, 0);
        tableRange.length = whereRange.location - tableRange.location;
        NSString *tmp = [sqlString2 substringWithRange:tableRange];
        if ([tmp containsString:@" "]) {
            NSInteger meetStart = 0;
            for (NSInteger i = tableRange.location; i < sqlString2.length; i ++) {
                unichar aChar = [sqlString2 characterAtIndex:i];
                if (meetStart == 0) {
                    if (aChar == '(') {
                        meetStart ++;
                    }
                }else{
                    if (aChar == ')') {
                        meetStart --;
                    }
                    if (meetStart == 0) {
                        tableRange.length = i - tableRange.location + 1;
                        break;
                    }
                }
            }
            if (tableRange.location + tableRange.length <= whereRange.location) {
                tableRange.length = whereRange.location - tableRange.location;
                NSString *tmp = [sqlString2 substringWithRange:tableRange];
                TestAsInTableString(tableRange, tmp);
            }else{
                NSString *tmp = [sqlString2 substringFromIndex:tableRange.location + tableRange.length];
                whereRange = [tmp rangeOfString:@" where "];
                if (whereRange.length == 0) {
                    tmp = [sqlString2 substringFromIndex:fromRange.location + fromRange.length];
                    tableRange = [sqlString2 rangeOfString:tmp];
                    TestAsInTableString(tableRange, tmp);
                }else{
                    if (minRangeLocation == 0) {
                        tableRange.length += whereRange.location;
                        tmp = [sqlString2 substringWithRange:tableRange];
                        TestAsInTableString(tableRange, tmp);
                    }else{
                        tableRange.length = minRangeLocation - tableRange.location;
                        tmp = [sqlString2 substringWithRange:tableRange];
                        TestAsInTableString(tableRange, tmp);
                    }
                }
            }
        }else{
            selftable = tmp;
            TestAsInTableString(tableRange, tmp);
        }
    }
    
    if (!meetRowID) {
        sqlString2 = [sqlString2 stringByReplacingCharactersInRange:fieldsRange withString:[NSString stringWithFormat:@"`rowid`, %@", fields]];
    }else{
        sqlString2 = [sqlString2 stringByReplacingCharactersInRange:fieldsRange withString:fields];
    }
    
    NSRange ranges[allSqlValues.count];
    
    for (NSInteger i = 0; i < allSqlValues.count; i ++) {
        NSRange range = NSMakeRange(0, sqlString2.length);
        if (i > 0) {
            range.location = ranges[i - 1].location + 1;
            range.length -= range.location;
        }
        ranges[i] = [sqlString2 rangeOfString:@"?" options:NSCaseInsensitiveSearch range:range];
    }
    for (NSInteger i = allSqlValues.count; i > 0; i --) {
        sqlString2 = [sqlString2 stringByReplacingCharactersInRange:ranges[i-1] withString:[NSString stringWithFormat:@"'%@'", [allSqlValues[i-1] stringByReplacingOccurrencesOfString:@"'" withString:@"''"]]];
    }
    
    if (pageSize > 0) {
        _sql = [[NSString alloc] initWithFormat:@"select * from (%@) as %@ limit %lu offset %lu", sqlString2, selftable, pageSize, pageSize * page];
    }else{
        _sql = [[NSString alloc] initWithString:sqlString2];;
    }

    return @[selftable, _sql];
}


mlibai
关注
从多条sql语句中提取所有表名，并排序
qq_39153893的博客
01-15 755
不多说，直接上代码，都是从网上找资料学习，然后整合写出来的。 package com.neo; import java.util.ArrayList; import java.util.Collections; import java.util.HashSet; import java.util.List; import java.util.Set; import java.util.regex.Matcher; import java.util.regex.Pattern; public clas.
SQL Compare使用教程:在Argfiles中使用SQL Compare命令行
RoffeyYang的博客
03-23 619
SQL Compare是一款比较和同步SQL Server数据库结构的工具。现有超过150,000的数据库管理员、开发人员和测试人员在使用它。当测试本地数据库，暂存或激活远程服务器的数据库时，SQL Compare将分配数据库的过程自动化。 点击下载SQL Compare试用版 Phil Factor演示了如何使用XML argfiles将参数传递给SQL Compare，从而消除了每个目标数据库所需的许多修改数据库模式比较和部署过程所涉及的繁琐脚本。 有很多方法可以使用PowerShell自动执行需要执行
提取sql语句中的表名 (python)
onwingsofsong的博客
10-30 7977
import ply.lex as lex, re def extract_table_name_from_sql(sql_str): # remove the /* */ comments q = re.sub(r"/\*[^*]*\*+(?:[^*/][^*]*\*+)*/", "", sql_str) # remove whole line -- and #...
SelectParser：免费从sql语句中提取表名的最强工具
如意机反光镜裸的博客
05-04 2967
从sql语句中提取表名的免费强力工具
提取sql查询语句中的表名 (from 和where 中间的table_name)
时刻学习
06-16 2858
select substr(execute_sql,locate('from',execute_sql)+4,locate('where',execute_sql)-5) from table_name 其中 execute_sql是列名,存储的是sql语句。 另外需要注意的是 sql语句中from和表名表名和where只隔了一个字符串的距离.此sql才有效.如果不是，需要读者自己调试sql. ...
获取sql中的表名
qq_23114283的博客
04-01 1973
项目上遇到一个需求，在很多sql种提取出所有表名。代码如下： # 这个脚本有一个前提，就是from接表名之后必定有一个where。通常情况下也都是如此 strSql = "select * from mm a, bb b where a > 0; select * from cc a, dd, qq, aa b where a > 0;" def check_ch(ch): if ch >= '0' and ch <= '9': return F
Documentos:estudio ocódigo文件
03-29
Markdown语言可能是用来编写这些文档的工具，它简单易学，能快速生成格式化的文本。 6. **问题解决与调试**: 在编程过程中，遇到并解决问题是常态。学习如何阅读和理解错误信息，使用调试工具，以及编写测试用例来...
GLP(Grafna +Loki +Promtail)日志可视化企业级实战
MyySophia的博客
07-17 2250
GLP全称: grafana loki promtail, 一个轻量级的云原生日志检索系统。对于数据，Loki 不会构建全文索引，而是通过 Label 的方式去构建索引，并通过 Grep 查询匹配对应的 COS。对于存储，Index 和 Chunks，是可以刷到廉价的第三方 COS 对象存储中。这样跟 ELK 一对比，成本的优化是巨大的。
【SQL标准战】
!... # 摘要 本文详细探讨了SQL标准的发展历程和其理论基础，深入分析了SQL语言中数据定义（DDL）、数据操纵（DML）、数据查询...通过对规范化数据库设计、数据分析技巧和性能优化策略的实践应用，文章揭示了SQL在现代数据
【递归调用与动态SQL】：深入理解存储过程的高级操作
!... # 摘要 存储过程和递归调用是数据库管理和程序设计中的核心概念，它们对于处理复杂数据结构和逻辑至关重要。...随后，文章转向动态SQL的原理与实现，探讨如何构建和管理动态SQL语句及其在存储过程中的高级
SQL提取数据库表名及字段名等信息代码示例
12-16
本文向大家介绍了使用SQL语句提取数据库所有表的表名、字段名的实例代码，在SQLserver 中进行了测试，具体内容如下： --查询所有用户表所有字段的特征 SELECT D.Name as TableName, A.colorder AS ColOrder, A.name AS Name, COLUMNPROPERTY(A.ID,A.Name, 'IsIdentity') AS IsIdentity, CASE WHEN EXISTS (SELECT 1 FROM dbo.sysobjects WHERE Xtype = 'PK' AND Name IN (SELECT Nam
解析SQL提取表名并输出文本文件
04-26
本小工具实现了解析sql文件，并将文件中存在的所有表名解析出来，保存成文本文件，在存储过程及数据处理上对工作有帮助，方便实用。 如果觉得好，作者会增加一些参数，将此小工具做得更完善。
如何获取一条SQL语句中涉及的表名
gjc592的博客
12-19 2454
点击上方蓝字关注我在数据库操作和SQL查询的开发过程中，有时候我们为了动态生成查询、进行权限控制、进行查询优化或者其他一些与数据库交互相关、数据库监控等的需求，需要从SQL语句中提取表名。本文分别使用正则表达式和使用SQL解析库的方式来获取。当然实际使用中需要进行优化，本次只是做初步的获取操作。1. 使用正则表达式正则表达式是一种强大的文本匹配工具，通过定义模式，可以从文本中提取...
纯文本分析方法，从复杂的sql查询语句中提取所有来源表名
Jacky_kplin的博客
06-08 5189
如何从复杂的sql查询语句中提取所有来源表名 本文使用Python实现 一、背景 前段时间开发了一个小工具用来检测一些sql脚本，其中有一个步骤需要将查询语句的各个来源表都提取出来。本来借助数据库是可以实现的，但该方法有局限性。所以我一直在想，能不能通过纯文本分析的方法来解决这个问题。 网上虽然能找到类似的解决方法，但基本都不是通用的脚本，只能解决某些句式，考虑的场景太少了，想要找到通用的脚本，只能靠自己写了。 最后经过一番费力的分析拆解，终于写出来一个相对来说比较通用的脚本。 二、题外话 先说一下借助数据
从sql中提取表名称，删除sql中的某个查询条件
weixin_44716895的博客
11-21 3237
注：内容均以java实现 这件事情，起始我们可以通过jsqlparser这款良心工具来完成。但是拿人家的东西来用就没机会锻炼自己了，正好最近工作上碰到了这个问题，就顺便来写个日志记录下吧。 这里做一个前提假设：sql没有语法错误 提取表名 虽然sql没有语法错误，但无法保证sql中带有注释符号，有的sql还带着很大的空行以及很长的空格。【我们公司的客服和老板都热爱能写sql，他们就是慢sql和不规范sql的缔造者，甚至一度引起公司内部服务器瘫痪，嘘！】 言归正传，先把sql按一个规则垂直分割，请围观java
如何利用python来提取SQL语句中的表名称
theskylife的博客
09-08 4264
在一些场景下，我们需要从一个SQL语句中提取出对应的表名，那么如何利用python来提取对应的表名称？本文带你了解如何使用正确的姿势打开
SQL —— 获取数据库中的所有表名
看了就会暴富的博客！
11-18 442
问题 如何获取数据库中的所有表名 解决 SELECT table_name FROM information_schema.TABLES
mysql缓存设置
wasd986523的博客
07-22 426
缓存设置my.cnf https://www.cnblogs.com/caicairui/archive/2017/08/12/7350698.html query_cache_type = ON query_cache_type= 1 query_cache_size= 512M show variables like 'have_query_cache'; show variables...
如何批量从sql语句中提取表名
最新发布
如意机反光镜裸的博客
10-12 1067
如何批量从sql语句中提取表名
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值