使用AI 编写一个Mybatis SQL日志解析工具_mybatis日志解析工具

作者：繁依Fanyi0 | 2024-08-11 02:34:46

踩

mybatis日志解析工具

使用AI 编写一个Mybatis SQL日志解析工具

背景

尝试使用智谱清言 AI 编写一个工具，将Mybatis SQL日志解析为可以执行的SQL语句。

操作

准备初始代码

首先网站上找了一个有BUG的简单的实现 https://blog.csdn.net/x541211190/article/details/107410409

用提示词让AI分析代码

将以上代码贴给AI，并增加提示词分析上面代码的的功能，让AI分析代码。
分析结果如下：

这段HTML代码定义了一个简单的网页界面，目的是为了解析Mybatis框架的SQL日志，并将其转换为可执行的SQL语句。下面是代码的详细分析：
1. **HTML结构**：
   - 定义了一个标题“输入Mybatis SQL日志：”，和一个多行文本输入框`<textarea>`用于用户粘贴Mybatis的SQL日志。
   - 提供了两个按钮，“清空”用于清除文本框内容，“解析SQL”用于触发SQL解析功能。
   - 另有一个标题“解析为可执行SQL：”，和一个用于显示解析后SQL语句的多行文本框。
   - 提供了一个“复制SQL”按钮，用于将解析后的SQL语句复制到剪贴板。
   - 最后有一个`<div>`元素用于显示复制成功的提示信息。
2. **JavaScript功能**：
   - `f(obj)`函数：此函数接收textarea对象作为参数，提取其中的文本值，然后解析出SQL语句和参数。
     - 定位SQL语句的开始和结束位置，并提取出来。
     - 定位参数的开始和结束位置，并提取参数列表。
     - 遍历参数列表，根据参数的类型（字符串或时间戳为字符串类型，其他为数值类型），替换SQL语句中的占位符`?`。
     - 将解析后的SQL语句显示在页面上。
   - `copySQL()`函数：此函数用于将解析后的SQL语句复制到剪贴板。
     - 通过`select()`方法选择文本。
     - 使用`document.execCommand("Copy")`执行复制命令。
     - 显示复制成功的提示信息，并设置3秒后消失。
   - `clearLog(obj)`函数：此函数用于清空textarea中的内容。
3. **样式**：
   - 使用内联样式定义了文本框和按钮的大小、颜色等视觉样式。
   - 使用了`<font>`标签来定义标题的颜色，虽然这不是HTML5推荐的写法，但它仍然有效。
4. **用户交互**：
   - 用户可以将Mybatis的SQL日志粘贴到第一个文本框。
   - 点击“解析SQL”按钮后，JavaScript函数会解析日志，并将可执行的SQL语句显示在第二个文本框中。
   - 用户可以点击“复制SQL”按钮将解析后的SQL语句复制到剪贴板。
整体而言，这个网页界面是一个实用的工具，可以帮助开发者快速从Mybatis的日志中提取并转换出可执行的SQL语句，方便数据库的调试和优化。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

用提示词让AI 找出代码的问题

分析解析日志的算法存在哪些问题？ ，分析结果如下：

解析日志的算法存在以下问题：
1. **假设日志格式固定**：
   算法假设日志的格式是固定的，例如，`Preparing:`和`Parameters:`后面的内容格式和位置都是固定的。如果日志格式发生变化，例如空格数不同、换行位置变化或者增加了新的日志信息，算法可能无法正确解析。
2. **缺乏错误处理**：
   算法中没有错误处理机制。如果日志中不存在`Preparing:`或`Parameters:`，或者格式不正确，算法将无法正常工作，且不会提供任何错误信息。
3. **对参数类型的处理有限**：
   算法只处理了`String`和`Timestamp`类型的参数，将其用单引号括起来，其他类型直接替换。这种处理方式忽略了其他可能需要特殊处理的类型，例如`Date`或`Boolean`，也没有考虑到数值类型可能需要防止SQL注入的问题。
4. **参数解析不准确**：
   算法中参数解析的部分使用了`substring`和`indexOf`来提取参数值和类型，但如果参数值本身包含括号或者逗号，这将导致错误的解析。
5. **性能问题**：
   算法中使用了多次循环和字符串操作来解析日志，这可能在处理大量数据时导致性能问题。
6. **SQL注入风险**：
   算法直接将参数值替换到SQL语句中，对于数值类型没有进行任何转义或处理，这可能导致SQL注入攻击。
7. **缺乏对复杂SQL语句的支持**：
   如果SQL语句中本身就包含问号（?），算法会将这些问号错误地认为是参数占位符。
为了改进这个算法，可以考虑以下措施：
- 使用正则表达式来解析日志，这样可以更准确地处理不同格式的日志。
- 添加错误处理逻辑，对于无法解析的日志提供明确的错误信息。
- 对于不同类型的参数，使用适当的处理方式，例如对于数值类型使用参数化查询来防止SQL注入。
- 对于可能包含特殊字符的参数值，使用更可靠的解析方法，例如先提取所有参数，然后根据位置来替换SQL语句中的占位符。
- 提高算法性能，例如通过减少不必要的字符串操作和循环。
- 确保算法能够处理复杂的SQL语句，包括那些本身就包含问号的语句。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

用提示词让AI修复代码的指定问题

修复其中的2,3,4,5,7的问题 修复后的结果

<!DOCTYPE html>
<html>
<head>
    <meta charset="utf-8">
    <title>Mybatis SQL日志解析</title>
    <script type="text/javascript">
		function f(obj) {
			try {
				var textVa = obj.value;
				var logs = textVa.split('Preparing: '); // Split by "Preparing: "
				var results = [];
				logs.forEach(function(log) {
					// Check if this line contains "Preparing:" or "Parameters:"
					if (log.indexOf('Preparing:') === -1 && log.indexOf('Parameters:') === -1) {
						return; // Skip this line if it doesn't contain "Preparing:" or "Parameters:"
					}
					// 获取带问号的SQL语句
					var statementStartIndex = log.indexOf('Preparing: ');
					if (statementStartIndex !== -1) {
						var statementStr = log.substring(statementStartIndex + "Preparing: ".length).trim();
					} else {
						throw new Error("未找到SQL语句");
					}
					// 获取参数
					var parametersStartIndex = log.indexOf('Parameters: ');
					if (parametersStartIndex !== -1) {
						var parametersStr = log.substring(parametersStartIndex + "Parameters: ".length).trim();
					} else {
						// If "Parameters:" is not found, we assume there are no parameters
						parametersStr = "null";
					}
					// 解析参数
					var parameters = parametersStr.split(/,(?![^()]*\))/g); // 使用正则表达式正确分割参数
					// 替换SQL语句中的占位符
					for (var i = 0; i < parameters.length; i++) {
						var paramValue = parameters[i].trim();
						if (paramValue === "null") {
							// 如果参数是null，直接替换为NULL
							statementStr = statementStr.replace("?", "NULL");
						} else {
							// 否则，根据类型替换参数
							var typeStr = paramValue.substring(paramValue.indexOf("(") + 1, paramValue.indexOf(")"));
							paramValue = paramValue.substring(0, paramValue.indexOf("(")).trim();
							if (typeStr === "String" || typeStr === "Timestamp") {
								paramValue = "'" + paramValue.replace("'", "''") + "'";
							}
							statementStr = statementStr.replace("?", paramValue);
						}
					}
					results.push(statementStr);
				});
				document.getElementById("d1").innerHTML = results.join("\n\n");
			} catch (e) {
				console.error(e);
				alert("解析SQL时发生错误：" + e.message);
			}
		}
        function copySQL() {
            var SQL = document.getElementById("d1");
            navigator.clipboard.writeText(SQL.value).then(function() {
                var msg = document.getElementById("msg");
                msg.innerHTML = "已复制到剪切板";
                setTimeout(function () {
                    msg.innerHTML = "";
                }, 3000);
            }).catch(function(err) {
                console.error("复制失败：", err);
                alert("复制SQL时发生错误：" + err.message);
            });
        }
        function clearLog(obj) {
            obj.value = "";
        }
    </script>
</head>
<body>
<h2><font color="#00bfff"> 输入Mybatis SQL日志：</font></h2>
<textarea id="sqlLog" rows="13" cols="140" style="font-size:20px"></textarea>
<div style="border:0px deepskyblue solid;width:1425px;height:50px;text-align:right">
    <button style="color:mediumblue;width:100px;height:60px" type="button" onclick="clearLog(document.getElementById('sqlLog'))">清空</button>
    <button style="color:mediumblue;width:100px;height:60px" type="submit" onclick="f(document.getElementById('sqlLog'))">解析SQL</button>
</div>
<h2><font color="#32cd32">解析为可执行SQL：</font></h2>
<textarea id="d1" rows="13" cols="140" style="font-size:20px"></textarea>
<div style="border:0px deepskyblue solid;width:1425px;height:50px;text-align:right">
    <button style="color:mediumblue;width:100px;height:60px" type="button" onclick="copySQL()">复制SQL</button>
</div>
<div id="msg" style="color:cornflowerblue;border:0px black solid;width:800px;height:20px;text-align:right;font-style: initial;font-size: large"></div>
</body>
</html>

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

用提示词让AI 继续分析代码

分析的结果如下：

上面的代码是一个HTML页面，包含JavaScript脚本，用于解析Mybatis SQL日志并生成可执行的SQL语句。具体功能如下：
1. 页面中包含一个文本域(sqlLog)，用于输入Mybatis SQL日志。
2. 点击"解析SQL"按钮，会触发f()函数，对输入的SQL日志进行解析。
3. f()函数首先使用split()方法按"Preparing: "分割输入的日志字符串，然后遍历分割后的日志数组。
4. 对于每个日志条目，判断是否包含"Preparing:"或"Parameters:"，如果不包含则跳过。
5. 获取带问号的SQL语句和参数，然后解析参数。
6. 对于SQL语句中的每个问号占位符，根据参数的类型替换为实际的参数值。
7. 将解析后的SQL语句显示在另一个文本域(d1)中。
8. 点击"复制SQL"按钮，会触发copySQL()函数，将解析后的SQL语句复制到剪切板。
9. 点击"清空"按钮，会清空sqlLog文本域中的内容。
10. 解析过程中如果发生错误，会弹出错误提示框。

1
2
3
4
5
6
7
8
9
10
11
12

用提示词让AI修改为支持多组日志的版本

需要修改为支持多组Mybatis SQL日志的解析，并且提供的日志中可能有非Mybatis SQL日志的行。

AI 输出的版本如下：

<!DOCTYPE html>
<html>
<head>
    <meta charset="utf-8">
    <title>Mybatis SQL日志解析</title>
    <script type="text/javascript">
        function f(obj) {
            try {
                var textVa = obj.value;
                var logs = textVa.split('\n'); // Split by newline
                var results = [];
                var currentStatement = null;

                logs.forEach(function(log) {
                    // Check if this line contains "Preparing:" or "Parameters:"
                    if (log.indexOf('Preparing:') !== -1) {
                        if (currentStatement !== null) {
                            // If we have a current statement, push it to the results
                            results.push(currentStatement);
                        }
                        // Start a new statement
                        currentStatement = log.substring(log.indexOf('Preparing:') + "Preparing:".length).trim();
                    } else if (log.indexOf('Parameters:') !== -1 && currentStatement !== null) {
                        // If we find parameters and have a current statement, parse the parameters
                        var parametersStr = log.substring(log.indexOf('Parameters:') + "Parameters:".length).trim();
                        var parameters = parametersStr.split(/,(?![^()]*\))/g); // Use regex to split correctly
                        for (var i = 0; i < parameters.length; i++) {
                            var paramValue = parameters[i].trim();
                            if (paramValue === "null") {
                                currentStatement = currentStatement.replace("?", "NULL");
                            } else {
                                var typeStr = paramValue.substring(paramValue.indexOf("(") + 1, paramValue.indexOf(")"));
                                paramValue = paramValue.substring(0, paramValue.indexOf("(")).trim();
                                if (typeStr === "String" || typeStr === "Timestamp") {
                                    paramValue = "'" + paramValue.replace("'", "''") + "'";
                                }
                                currentStatement = currentStatement.replace("?", paramValue);
                            }
                        }
                        // Add the final statement to the results and reset currentStatement
                        results.push(currentStatement);
                        currentStatement = null;
                    }
                });

                // If there's a remaining statement, push it to the results
                if (currentStatement !== null) {
                    results.push(currentStatement);
                }

                document.getElementById("d1").value = results.join("\n\n");
            } catch (e) {
                console.error(e);
                alert("解析SQL时发生错误：" + e.message);
            }
        }
        function copySQL() {
            var SQL = document.getElementById("d1");
            navigator.clipboard.writeText(SQL.value).then(function() {
                var msg = document.getElementById("msg");
                msg.innerHTML = "已复制到剪切板";
                setTimeout(function () {
                    msg.innerHTML = "";
                }, 3000);
            }).catch(function(err) {
                console.error("复制失败：", err);
                alert("复制SQL时发生错误：" + err.message);
            });
        }
        function clearLog(obj) {
            obj.value = "";
        }
    </script>
</head>
<body>

<h2><font color="#00bfff"> 输入Mybatis SQL日志：</font></h2>
<textarea id="sqlLog" rows="13" cols="140" style="font-size:20px"></textarea>
<div style="border:0px deepskyblue solid;width:1425px;height:50px;text-align:right">
    <button style="color:mediumblue;width:100px;height:60px" type="button" onclick="clearLog(document.getElementById('sqlLog'))">清空</button>
    <button style="color:mediumblue;width:100px;height:60px" type="submit" onclick="f(document.getElementById('sqlLog'))">解析SQL</button>
</div>

<h2><font color="#32cd32">解析为可执行SQL：</font></h2>
<textarea id="d1" rows="13" cols="140" style="font-size:20px"></textarea>
<div style="border:0px deepskyblue solid;width:1425px;height:50px;text-align:right">
    <button style="color:mediumblue;width:100px;height:60px" type="button" onclick="copySQL()">复制SQL</button>
</div>

<div id="msg" style="color:cornflowerblue;border:0px black solid;width:800px;height:20px;text-align:right;font-style: initial;font-size: large"></div>

</body>
</html>

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94

测试验证

将代码保存为一个后缀为html的文件，用浏览器打开，随便复制一段Mybatis SQL日志，测试效果基本没问题。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/961769