consoleViewMap = new ConcurrentHashMap<>(16);
29 |
30 |
31 | private static final Gson gson = new GsonBuilder().setDateFormat("yyyy-MM-dd HH:mm:ss")
32 | //当Map的key为复杂对象时,需要开启该方法
33 | .enableComplexMapKeySerialization()
34 | //当字段值为空或null时，依然对该字段进行转换
35 | .serializeNulls()
36 | //防止特殊字符出现乱码
37 | .disableHtmlEscaping().create();
38 |
39 | /**
40 | * Sql语句还原，整个插件的核心就是该方法
41 | *
42 | * @param parametersLine 参数
43 | * @return
44 | */
45 | private static SqlVO restoreSql(final String parametersLine) {
46 | final String[] split = parametersLine.split(Config.Idea.getParameters());
47 | if (split.length == 2) {
48 | final String s = split[1];
49 | return gson.fromJson(s, SqlVO.class);
50 | }
51 | return null;
52 | }
53 |
54 |
55 | public static void printsInit(ConsoleView consoleView) {
56 | if (Config.Idea.getWelcomeMessage()) {
57 | consoleView.print(" ============================================================================ " + "\n", ConsoleViewContentType.NORMAL_OUTPUT);
58 | consoleView.print("\n", ConsoleViewContentType.NORMAL_OUTPUT);
59 | consoleView.print(" MyBatis Log EasyPlus" + "\n", ConsoleViewContentType.NORMAL_OUTPUT);
60 | consoleView.print("\n", ConsoleViewContentType.NORMAL_OUTPUT);
61 | consoleView.print(" A mybatis javaagent framework :)" + "\n", ConsoleViewContentType.NORMAL_OUTPUT);
62 | consoleView.print("\n", ConsoleViewContentType.NORMAL_OUTPUT);
63 | consoleView.print(" https://github.com/Link-Kou/intellij-mybaitslog" + "\n", ConsoleViewContentType.NORMAL_OUTPUT);
64 | consoleView.print("\n", ConsoleViewContentType.NORMAL_OUTPUT);
65 | consoleView.print(" ============================================================================ " + "\n", ConsoleViewContentType.NORMAL_OUTPUT);
66 | }
67 | }
68 |
69 | public static void prints(Project project, String currentLine) {
70 | final String parameters = Config.Idea.getParameters();
71 | if (currentLine.contains(parameters)) {
72 | //序号前缀字符串
73 | final SqlVO sqlVO = restoreSql(currentLine);
74 | if (null != sqlVO) {
75 | String completesql = sqlVO.getCompleteSql().replaceAll("\t|\r|\n", "");
76 | final String id = sqlVO.getId();
77 | final String parameter = sqlVO.getParameter();
78 | if (Config.Idea.getFormatSql()) {
79 | completesql = new BasicFormatter().format(completesql);
80 | }
81 | //序号
82 | PrintlnUtil.println(project, Config.SQL_START_LINE + id + "\n", ConsoleViewContentType.USER_INPUT);
83 | PrintlnUtil.printlnSqlType(project, Config.SQL_MIDDLE_LINE, completesql + "\n");
84 | PrintlnUtil.printlnSqlType(project, Config.SQL_MIDDLE_LINE, parameter + "\n");
85 | PrintlnUtil.println(project, Config.SQL_END_LINE + "\n", ConsoleViewContentType.USER_INPUT);
86 | }
87 | }
88 | }
89 |
90 | /**
91 | * 输出语句
92 | *
93 | * @param project 项目
94 | * @param rowLine 行数据
95 | * @param consoleViewContentType 输出颜色
96 | */
97 | public static void println(Project project, String rowLine, ConsoleViewContentType consoleViewContentType) {
98 | ConsoleView consoleView = consoleViewMap.get(project);
99 | if (consoleView != null) {
100 | consoleView.print(rowLine, consoleViewContentType);
101 | }
102 | }
103 |
104 | public static void setConsoleView(Project project, ConsoleView consoleView) {
105 | consoleViewMap.put(project, consoleView);
106 | }
107 |
108 | public static ConsoleView getConsoleView(Project project) {
109 | return consoleViewMap.get(project);
110 | }
111 |
112 | /**
113 | * 获取Sql语句类型
114 | *
115 | * @param sql 语句
116 | * @return String
117 | */
118 | private static String getSqlType(String sql) {
119 | if (StringUtils.isNotBlank(sql)) {
120 | String lowerLine = sql.toLowerCase().trim();
121 | if (lowerLine.startsWith("insert")) {
122 | return "insert";
123 | }
124 | if (lowerLine.startsWith("update")) {
125 | return "update";
126 | }
127 | if (lowerLine.startsWith("delete")) {
128 | return "delete";
129 | }
130 | if (lowerLine.startsWith("select")) {
131 | return "select";
132 | }
133 | }
134 | return "other";
135 | }
136 |
137 |
138 | /**
139 | * SQL 输出语句
140 | *
141 | * @param rowLine 行数据
142 | */
143 | public static void printlnSqlType(Project project, String title, String rowLine) {
144 | final String sqlType = getSqlType(rowLine);
145 | final ConsoleViewContentType systemOutput = ConsoleViewContentType.SYSTEM_OUTPUT;
146 | final TextAttributes attributes = systemOutput.getAttributes();
147 | final ConsoleViewContentType styleName = new ConsoleViewContentType("styleName", new TextAttributes(attributes.getForegroundColor(), attributes.getBackgroundColor(), attributes.getEffectColor(), attributes.getEffectType(), attributes.getFontType()));
148 | switch (sqlType) {
149 | case "select":
150 | case "insert":
151 | case "update":
152 | case "delete":
153 | final Color color = Config.Idea.getColor(sqlType);
154 | styleName.getAttributes().setForegroundColor(color);
155 | println(project, title + rowLine, styleName);
156 | break;
157 | default:
158 | final Color color1 = Config.Idea.getColor("other");
159 | styleName.getAttributes().setForegroundColor(color1);
160 | println(project, title + rowLine, styleName);
161 | }
162 | }
163 | }
164 |
--------------------------------------------------------------------------------
/mybatis-test/src/test/resources/mapper/EmployeesMapper.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 |
6 |
7 |
8 |
9 |
10 |
11 |
12 |
13 |
14 |
15 |
25 |
26 |
27 |
53 |
54 |
55 |
79 |
80 |
81 |
82 | insert into employees(emp_no, birth_date, first_name, last_name, gender, hire_date)
83 | values (#{empNo}, #{birthDate}, #{firstName}, #{lastName}, #{gender}, #{hireDate})
84 |
85 |
86 |
87 | insert into employees(emp_no, birth_date, first_name, last_name, gender, hire_date)
88 | values
89 |
90 | (#{entity.empNo}, #{entity.birthDate}, #{entity.firstName}, #{entity.lastName}, #{entity.gender},
91 | #{entity.hireDate})
92 |
93 |
94 |
95 |
96 | insert into employees(emp_no, birth_date, first_name, last_name, gender, hire_date)
97 | values
98 |
99 | (#{entity.empNo}, #{entity.birthDate}, #{entity.firstName}, #{entity.lastName}, #{entity.gender},
100 | #{entity.hireDate})
101 |
102 | on duplicate key update
103 | emp_no = values(emp_no),
104 | birth_date = values(birth_date),
105 | first_name = values(first_name),
106 | last_name = values(last_name),
107 | gender = values(gender),
108 | hire_date = values(hire_date)
109 |
110 |
111 |
112 |
113 | update employees
114 |
115 |
116 | emp_no = #{empNo},
117 |
118 |
119 | birth_date = #{birthDate},
120 |
121 |
122 | first_name = #{firstName},
123 |
124 |
125 | last_name = #{lastName},
126 |
127 |
128 | gender = #{gender},
129 |
130 |
131 | hire_date = #{hireDate},
132 |
133 |
134 | where emp_no = #{empNo}
135 |
136 |
137 |
150 |
151 |
152 |
162 |
163 |
164 |
174 |
175 |
176 |
186 |
187 |
196 |
197 |
198 |
199 |
--------------------------------------------------------------------------------
/mybatis-test/src/test/java/com/linkkou/mybatislog/mybatis/boot/plugins/QueryPaginatorInterceptor.java:
--------------------------------------------------------------------------------
1 | /*
2 | * The MIT License (MIT)
3 | *
4 | * Copyright (c) 2014-2017 abel533@gmail.com
5 | *
6 | * Permission is hereby granted, free of charge, to any person obtaining a copy
7 | * of this software and associated documentation files (the "Software"), to deal
8 | * in the Software without restriction, including without limitation the rights
9 | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 | * copies of the Software, and to permit persons to whom the Software is
11 | * furnished to do so, subject to the following conditions:
12 | *
13 | * The above copyright notice and this permission notice shall be included in
14 | * all copies or substantial portions of the Software.
15 | *
16 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 | * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22 | * THE SOFTWARE.
23 | */
24 |
25 | package com.linkkou.mybatislog.mybatis.boot.plugins;
26 |
27 | import net.sf.jsqlparser.JSQLParserException;
28 | import net.sf.jsqlparser.parser.CCJSqlParserUtil;
29 | import net.sf.jsqlparser.statement.select.PlainSelect;
30 | import net.sf.jsqlparser.statement.select.Select;
31 | import net.sf.jsqlparser.util.TablesNamesFinder;
32 | import org.apache.ibatis.cache.CacheKey;
33 | import org.apache.ibatis.executor.Executor;
34 | import org.apache.ibatis.mapping.*;
35 | import org.apache.ibatis.plugin.*;
36 | import org.apache.ibatis.session.ResultHandler;
37 | import org.apache.ibatis.session.RowBounds;
38 |
39 | import java.lang.reflect.Field;
40 | import java.util.ArrayList;
41 | import java.util.List;
42 | import java.util.Properties;
43 |
44 | /**
45 | * 目前只有支持MySQL分页支持
46 | * 需要开启allowMultiQueries
47 | *
48 | * @author lk
49 | * @version 1.0.0
50 | */
51 | @Intercepts(
52 | {
53 | @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}),
54 | @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class, CacheKey.class, BoundSql.class})
55 | }
56 | )
57 | //@SuppressWarnings()
58 | public class QueryPaginatorInterceptor implements Interceptor {
59 |
60 | private final static String QUERY = "query";
61 | /**
62 | * FOUND_ROWS == true
63 | * COUNT(*) == false
64 | */
65 | private boolean PAGINATORTYPE = false;
66 |
67 | /**
68 | * 是否支持多条语句执行
69 | */
70 | protected boolean ALLOWMULTIQUERIES = true;
71 |
72 | public void setPaginatortype(boolean paginator) {
73 | this.PAGINATORTYPE = paginator;
74 | }
75 |
76 | @Override
77 | public Object intercept(Invocation invocation) throws Throwable {
78 | final Object[] queryArgs = invocation.getArgs();
79 | if (invocation.getTarget() instanceof Executor && ALLOWMULTIQUERIES) {
80 | final MappedStatement mappedStatement = (MappedStatement) queryArgs[0];
81 | if (SqlCommandType.SELECT == mappedStatement.getSqlCommandType()) {
82 | if (QUERY.equals(invocation.getMethod().getName()) && queryArgs.length == 4) {
83 | final Object MappedStatement = queryArgs[0];
84 | final Object parameter = queryArgs[1];
85 | final BoundSql boundSql = mappedStatement.getBoundSql(parameter);
86 |
87 |
88 | MappedStatement newMs = ProxyResultSetHandler(mappedStatement, new BoundSqlSqlSource(getPagingSql(mappedStatement, boundSql)));
89 | queryArgs[0] = newMs;
90 | }
91 | }
92 | }
93 | return invocation.proceed();
94 | }
95 |
96 | /**
97 | * 返回分页sql语句
98 | *
99 | * @param mappedStatement
100 | * @param boundSql
101 | * @return
102 | */
103 | private BoundSql getPagingSql(MappedStatement mappedStatement, BoundSql boundSql) throws JSQLParserException {
104 | return getSqlCountTrue(mappedStatement, boundSql);
105 | }
106 |
107 | /**
108 | * ALLOWMULTIQUERIES = true count(*) 模式
109 | *
110 | * select rowid, fid
111 | * from t_sysconfig_carspec_config tscc
112 | * WHERE fcar_name = #{car}
113 | * LIMIT
114 | * #{offset},#{rows}
115 | *
116 | *
117 | * @param mappedStatement
118 | * @param boundSql
119 | * @return
120 | */
121 | private BoundSql getSqlCountTrue(MappedStatement mappedStatement, BoundSql boundSql) throws JSQLParserException {
122 | String sql = boundSql.getSql();
123 | String newSql = String.format("select count(*) from (%s);", sql);
124 | List newparameterMappings = new ArrayList<>();
125 | List parameterMappings = boundSql.getParameterMappings();
126 | newparameterMappings.addAll(parameterMappings);
127 | //Limit 去两个参数，所有参数以Mapper顺序为准
128 | for (int i = 0; i < parameterMappings.size() - 2; i++) {
129 | newparameterMappings.add(parameterMappings.get(i));
130 | }
131 | BoundSql countBoundSql = new BoundSql(mappedStatement.getConfiguration(), newSql, newparameterMappings, boundSql.getParameterObject());
132 | return countBoundSql;
133 | }
134 |
135 | /**
136 | * 新加Mapper结构
137 | *
138 | * @param ms
139 | * @param newSqlSource
140 | * @return
141 | */
142 | private MappedStatement ProxyResultSetHandler(MappedStatement ms, SqlSource newSqlSource) {
143 | MappedStatement.Builder builder = new MappedStatement.Builder(ms.getConfiguration(), ms.getId(), newSqlSource, ms.getSqlCommandType());
144 | builder.resource(ms.getResource());
145 | builder.fetchSize(ms.getFetchSize());
146 | builder.statementType(ms.getStatementType());
147 | builder.keyGenerator(ms.getKeyGenerator());
148 | if (ms.getKeyProperties() != null && ms.getKeyProperties().length > 0) {
149 | builder.keyProperty(ms.getKeyProperties()[0]);
150 | }
151 | builder.timeout(ms.getTimeout());
152 | builder.parameterMap(ms.getParameterMap());
153 | builder.resultMaps(newResultMap(ms.getResultMaps()));
154 | builder.resultSetType(ms.getResultSetType());
155 | builder.cache(ms.getCache());
156 | builder.flushCacheRequired(ms.isFlushCacheRequired());
157 | builder.useCache(ms.isUseCache());
158 | return builder.build();
159 | }
160 |
161 | /**
162 | * 构建统一的返回值
163 | * 支持 allowMultiQueries=true的时候ResultMap会返回多值
164 | */
165 | protected List newResultMap(List lrm) {
166 | ResultMap resultMap = new ResultMap.Builder(null, lrm.size() > 0 ? lrm.get(0).getId() : "", Integer.class, new ArrayList()).build();
167 | List list = new ArrayList<>();
168 | if (lrm.size() > 0) {
169 | list.add(lrm.get(0));
170 | }
171 | list.add(resultMap);
172 | return list;
173 | }
174 |
175 | @Override
176 | public Object plugin(Object target) {
177 | return Plugin.wrap(target, this);
178 | }
179 |
180 | @Override
181 | public void setProperties(Properties properties) {
182 | }
183 |
184 | public static class BoundSqlSqlSource implements SqlSource {
185 | private BoundSql boundSql;
186 |
187 | public BoundSqlSqlSource(BoundSql boundSql) {
188 | this.boundSql = boundSql;
189 | }
190 |
191 | @Override
192 | public BoundSql getBoundSql(Object parameterObject) {
193 | return boundSql;
194 | }
195 | }
196 |
197 | }
198 |
--------------------------------------------------------------------------------
/gradlew:
--------------------------------------------------------------------------------
1 | #!/bin/sh
2 |
3 | #
4 | # Copyright © 2015-2021 the original authors.
5 | #
6 | # Licensed under the Apache License, Version 2.0 (the "License");
7 | # you may not use this file except in compliance with the License.
8 | # You may obtain a copy of the License at
9 | #
10 | # https://www.apache.org/licenses/LICENSE-2.0
11 | #
12 | # Unless required by applicable law or agreed to in writing, software
13 | # distributed under the License is distributed on an "AS IS" BASIS,
14 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | # See the License for the specific language governing permissions and
16 | # limitations under the License.
17 | #
18 |
19 | ##############################################################################
20 | #
21 | # Gradle start up script for POSIX generated by Gradle.
22 | #
23 | # Important for running:
24 | #
25 | # (1) You need a POSIX-compliant shell to run this script. If your /bin/sh is
26 | # noncompliant, but you have some other compliant shell such as ksh or
27 | # bash, then to run this script, type that shell name before the whole
28 | # command line, like:
29 | #
30 | # ksh Gradle
31 | #
32 | # Busybox and similar reduced shells will NOT work, because this script
33 | # requires all of these POSIX shell features:
34 | # * functions;
35 | # * expansions «$var», «${var}», «${var:-default}», «${var+SET}»,
36 | # «${var#prefix}», «${var%suffix}», and «$( cmd )»;
37 | # * compound commands having a testable exit status, especially «case»;
38 | # * various built-in commands including «command», «set», and «ulimit».
39 | #
40 | # Important for patching:
41 | #
42 | # (2) This script targets any POSIX shell, so it avoids extensions provided
43 | # by Bash, Ksh, etc; in particular arrays are avoided.
44 | #
45 | # The "traditional" practice of packing multiple parameters into a
46 | # space-separated string is a well documented source of bugs and security
47 | # problems, so this is (mostly) avoided, by progressively accumulating
48 | # options in "$@", and eventually passing that to Java.
49 | #
50 | # Where the inherited environment variables (DEFAULT_JVM_OPTS, JAVA_OPTS,
51 | # and GRADLE_OPTS) rely on word-splitting, this is performed explicitly;
52 | # see the in-line comments for details.
53 | #
54 | # There are tweaks for specific operating systems such as AIX, CygWin,
55 | # Darwin, MinGW, and NonStop.
56 | #
57 | # (3) This script is generated from the Groovy template
58 | # https://github.com/gradle/gradle/blob/master/subprojects/plugins/src/main/resources/org/gradle/api/internal/plugins/unixStartScript.txt
59 | # within the Gradle project.
60 | #
61 | # You can find Gradle at https://github.com/gradle/gradle/.
62 | #
63 | ##############################################################################
64 |
65 | # Attempt to set APP_HOME
66 |
67 | # Resolve links: $0 may be a link
68 | app_path=$0
69 |
70 | # Need this for daisy-chained symlinks.
71 | while
72 | APP_HOME=${app_path%"${app_path##*/}"} # leaves a trailing /; empty if no leading path
73 | [ -h "$app_path" ]
74 | do
75 | ls=$( ls -ld "$app_path" )
76 | link=${ls#*' -> '}
77 | case $link in #(
78 | /*) app_path=$link ;; #(
79 | *) app_path=$APP_HOME$link ;;
80 | esac
81 | done
82 |
83 | APP_HOME=$( cd "${APP_HOME:-./}" && pwd -P ) || exit
84 |
85 | APP_NAME="Gradle"
86 | APP_BASE_NAME=${0##*/}
87 |
88 | # Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
89 | DEFAULT_JVM_OPTS='"-Xmx64m" "-Xms64m"'
90 |
91 | # Use the maximum available, or set MAX_FD != -1 to use that value.
92 | MAX_FD=maximum
93 |
94 | warn () {
95 | echo "$*"
96 | } >&2
97 |
98 | die () {
99 | echo
100 | echo "$*"
101 | echo
102 | exit 1
103 | } >&2
104 |
105 | # OS specific support (must be 'true' or 'false').
106 | cygwin=false
107 | msys=false
108 | darwin=false
109 | nonstop=false
110 | case "$( uname )" in #(
111 | CYGWIN* ) cygwin=true ;; #(
112 | Darwin* ) darwin=true ;; #(
113 | MSYS* | MINGW* ) msys=true ;; #(
114 | NONSTOP* ) nonstop=true ;;
115 | esac
116 |
117 | CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
118 |
119 |
120 | # Determine the Java command to use to start the JVM.
121 | if [ -n "$JAVA_HOME" ] ; then
122 | if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
123 | # IBM's JDK on AIX uses strange locations for the executables
124 | JAVACMD=$JAVA_HOME/jre/sh/java
125 | else
126 | JAVACMD=$JAVA_HOME/bin/java
127 | fi
128 | if [ ! -x "$JAVACMD" ] ; then
129 | die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME
130 |
131 | Please set the JAVA_HOME variable in your environment to match the
132 | location of your Java installation."
133 | fi
134 | else
135 | JAVACMD=java
136 | which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
137 |
138 | Please set the JAVA_HOME variable in your environment to match the
139 | location of your Java installation."
140 | fi
141 |
142 | # Increase the maximum file descriptors if we can.
143 | if ! "$cygwin" && ! "$darwin" && ! "$nonstop" ; then
144 | case $MAX_FD in #(
145 | max*)
146 | MAX_FD=$( ulimit -H -n ) ||
147 | warn "Could not query maximum file descriptor limit"
148 | esac
149 | case $MAX_FD in #(
150 | '' | soft) :;; #(
151 | *)
152 | ulimit -n "$MAX_FD" ||
153 | warn "Could not set maximum file descriptor limit to $MAX_FD"
154 | esac
155 | fi
156 |
157 | # Collect all arguments for the java command, stacking in reverse order:
158 | # * args from the command line
159 | # * the main class name
160 | # * -classpath
161 | # * -D...appname settings
162 | # * --module-path (only if needed)
163 | # * DEFAULT_JVM_OPTS, JAVA_OPTS, and GRADLE_OPTS environment variables.
164 |
165 | # For Cygwin or MSYS, switch paths to Windows format before running java
166 | if "$cygwin" || "$msys" ; then
167 | APP_HOME=$( cygpath --path --mixed "$APP_HOME" )
168 | CLASSPATH=$( cygpath --path --mixed "$CLASSPATH" )
169 |
170 | JAVACMD=$( cygpath --unix "$JAVACMD" )
171 |
172 | # Now convert the arguments - kludge to limit ourselves to /bin/sh
173 | for arg do
174 | if
175 | case $arg in #(
176 | -*) false ;; # don't mess with options #(
177 | /?*) t=${arg#/} t=/${t%%/*} # looks like a POSIX filepath
178 | [ -e "$t" ] ;; #(
179 | *) false ;;
180 | esac
181 | then
182 | arg=$( cygpath --path --ignore --mixed "$arg" )
183 | fi
184 | # Roll the args list around exactly as many times as the number of
185 | # args, so each arg winds up back in the position where it started, but
186 | # possibly modified.
187 | #
188 | # NB: a `for` loop captures its iteration list before it begins, so
189 | # changing the positional parameters here affects neither the number of
190 | # iterations, nor the values presented in `arg`.
191 | shift # remove old arg
192 | set -- "$@" "$arg" # push replacement arg
193 | done
194 | fi
195 |
196 | # Collect all arguments for the java command;
197 | # * $DEFAULT_JVM_OPTS, $JAVA_OPTS, and $GRADLE_OPTS can contain fragments of
198 | # shell script including quotes and variable substitutions, so put them in
199 | # double quotes to make sure that they get re-expanded; and
200 | # * put everything else in single quotes, so that it's not re-expanded.
201 |
202 | set -- \
203 | "-Dorg.gradle.appname=$APP_BASE_NAME" \
204 | -classpath "$CLASSPATH" \
205 | org.gradle.wrapper.GradleWrapperMain \
206 | "$@"
207 |
208 | # Stop when "xargs" is not available.
209 | if ! command -v xargs >/dev/null 2>&1
210 | then
211 | die "xargs is not available"
212 | fi
213 |
214 | # Use "xargs" to parse quoted args.
215 | #
216 | # With -n1 it outputs one arg per line, with the quotes and backslashes removed.
217 | #
218 | # In Bash we could simply go:
219 | #
220 | # readarray ARGS < <( xargs -n1 <<<"$var" ) &&
221 | # set -- "${ARGS[@]}" "$@"
222 | #
223 | # but POSIX shell has neither arrays nor command substitution, so instead we
224 | # post-process each arg (as a line of input to sed) to backslash-escape any
225 | # character that might be a shell metacharacter, then use eval to reverse
226 | # that process (while maintaining the separation between arguments), and wrap
227 | # the whole thing up as a single "set" statement.
228 | #
229 | # This will of course break if any of these variables contains a newline or
230 | # an unmatched quote.
231 | #
232 |
233 | eval "set -- $(
234 | printf '%s\n' "$DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS" |
235 | xargs -n1 |
236 | sed ' s~[^-[:alnum:]+,./:=@_]~\\&~g; ' |
237 | tr '\n' ' '
238 | )" '"$@"'
239 |
240 | exec "$JAVACMD" "$@"
241 |
--------------------------------------------------------------------------------
/mybatis-idea-plugin/src/main/java/com/plugins/mybaitslog/Config.java:
--------------------------------------------------------------------------------
1 | package com.plugins.mybaitslog;
2 |
3 | import com.intellij.ide.util.PropertiesComponent;
4 | import com.intellij.openapi.project.Project;
5 |
6 | import java.awt.*;
7 | import java.util.*;
8 | import java.util.List;
9 |
10 | /**
11 | * A Config Class
12 | *
13 | * @author lk
14 | * @version 1.0
15 | * date: 2023/3/30 18:08
16 | */
17 | public class Config {
18 |
19 | private static final String PROJECT_ID = "MyBatisLogPlugin";
20 |
21 | public final static String KEY_NAME = "==> SQLStructure: ";
22 | public final static String KEY_ERROR_NAME = "==> SQLStructureError: ";
23 |
24 | public static final String SQL_START_LINE = "- --> ";
25 | public static final String SQL_MIDDLE_LINE = " > ";
26 | public static final String SQL_END_LINE = "- --< ";
27 |
28 | public static class Idea {
29 |
30 | /**
31 | * 存储的key名称
32 | */
33 | private static final String DB_PARAMETERS_KEY = PROJECT_ID + "SQLStructure";
34 | /**
35 | * 存储的key名称 是否允许运行
36 | */
37 | public static final String DB_STARTUP_KEY = PROJECT_ID + "startup";
38 |
39 |
40 | public static final String FORMAT_SQL_1 = PROJECT_ID + "SQL_FORMAT";
41 |
42 | /**
43 | * 运行提示
44 | */
45 | public static final String NOTIFICATION = PROJECT_ID + "NOTIFICATION";
46 |
47 | /**
48 | * 欢迎信息
49 | */
50 | public static final String WELCOMEMESSAGE = PROJECT_ID + "WELCOMEMESSAGE";
51 |
52 | /**
53 | * 是否折叠
54 | */
55 | public static final String WHETHERFOLD = PROJECT_ID + "WHETHERFOLD";
56 |
57 | public static final String RMISERVER = PROJECT_ID + "RMISERVER";
58 |
59 | public static final String PARAMETERS = "SQLStructure:";
60 |
61 | public static final Map ColorMap = new HashMap() {{
62 | put("select", "42,199,180");
63 | put("update", "158,190,92");
64 | put("delete", "231,29,54");
65 | put("insert", "97,163,172");
66 | put("other", "252,255,253");
67 | }};
68 |
69 | public static final Map PerRunMap = new HashMap() {{
70 | put("org.jetbrains.idea.maven.execution", false);
71 | put("com.intellij.execution.junit", true);
72 | put("com.intellij.spring.boot.run", true);
73 | put("org.jetbrains.plugins.gradle", false);
74 | }};
75 |
76 | public static final ArrayList AddOpens = new ArrayList() {{
77 | add("--add-opens java.base/java.lang=ALL-UNNAMED");
78 | }};
79 |
80 | /**
81 | * 获取关键字常量配置
82 | *
83 | * @return string
84 | */
85 | public static String getParameters() {
86 | return PropertiesComponent.getInstance().getValue(DB_PARAMETERS_KEY, PARAMETERS);
87 | }
88 |
89 |
90 | /**
91 | * 设置配置
92 | *
93 | * @param value 值
94 | * @param defaultValue 默认
95 | */
96 | public static void setParameters(String value, String defaultValue) {
97 | PropertiesComponent.getInstance().setValue(DB_PARAMETERS_KEY, null == value ? defaultValue : value);
98 | }
99 |
100 | /**
101 | * 设置启动过滤
102 | *
103 | * @param value 值
104 | */
105 | public static void setStartup(int value) {
106 | PropertiesComponent.getInstance().setValue(DB_STARTUP_KEY, Integer.toString(value), "1");
107 | }
108 |
109 | /**
110 | * 获取启动过滤
111 | */
112 | public static boolean getStartup() {
113 | final String value = PropertiesComponent.getInstance().getValue(DB_STARTUP_KEY, "1");
114 | return Integer.parseInt(value) == 1;
115 | }
116 |
117 | /**
118 | * 运行通知
119 | *
120 | * @param value 值
121 | */
122 | public static void setRunNotification(int value) {
123 | PropertiesComponent.getInstance().setValue(NOTIFICATION, Integer.toString(value), "1");
124 | }
125 |
126 | /**
127 | * 运行通知
128 | */
129 | public static boolean getRunNotification() {
130 | final String value = PropertiesComponent.getInstance().getValue(NOTIFICATION, "1");
131 | return Integer.parseInt(value) == 1;
132 | }
133 |
134 | /**
135 | * 欢迎信息
136 | *
137 | * @param value 值
138 | */
139 | public static void setWelcomeMessage(int value) {
140 | PropertiesComponent.getInstance().setValue(WELCOMEMESSAGE, Integer.toString(value), "1");
141 | }
142 |
143 | /**
144 | * 欢迎信息
145 | */
146 | public static boolean getWelcomeMessage() {
147 | final String value = PropertiesComponent.getInstance().getValue(WELCOMEMESSAGE, "1");
148 | return Integer.parseInt(value) == 1;
149 | }
150 |
151 | /**
152 | * 是否折叠
153 | *
154 | * @param value 值
155 | */
156 | public static void setWhetherfold(int value) {
157 | PropertiesComponent.getInstance().setValue(WHETHERFOLD, Integer.toString(value), "1");
158 | }
159 |
160 | /**
161 | * 是否折叠
162 | */
163 | public static boolean getWhetherfold() {
164 | final String value = PropertiesComponent.getInstance().getValue(WHETHERFOLD, "1");
165 | return Integer.parseInt(value) == 1;
166 | }
167 |
168 | /**
169 | * 是否RMI输出
170 | *
171 | * @param value 值
172 | */
173 | public static void setRunRmi(int value) {
174 | PropertiesComponent.getInstance().setValue(RMISERVER, Integer.toString(value), "0");
175 | }
176 |
177 | /**
178 | * 是否RMI输出
179 | */
180 | public static boolean getRunRmi() {
181 | final String value = PropertiesComponent.getInstance().getValue(RMISERVER, "0");
182 | return Integer.parseInt(value) == 1;
183 | }
184 |
185 |
186 | /**
187 | * 设置格式化
188 | *
189 | * @param value 值
190 | */
191 | public static void setFormatSql(int value) {
192 | PropertiesComponent.getInstance().setValue(FORMAT_SQL_1, Integer.toString(value), "0");
193 | }
194 |
195 | /**
196 | * 设置格式化
197 | */
198 | public static boolean getFormatSql() {
199 | final String value = PropertiesComponent.getInstance().getValue(FORMAT_SQL_1, "0");
200 | return Integer.parseInt(value) == 1;
201 | }
202 |
203 | /**
204 | * 设置配置
205 | *
206 | * @param type 值
207 | */
208 | public static Color getColor(String type) {
209 | final String s = ColorMap.get(type);
210 | final String value = PropertiesComponent.getInstance().getValue("color:" + type, s);
211 | final String[] split = value.split(",");
212 | return new Color(Integer.parseInt(split[0]), Integer.parseInt(split[1]), Integer.parseInt(split[2]));
213 | }
214 |
215 | /**
216 | * 设置启动过滤
217 | *
218 | * @param value 值
219 | */
220 | public static void setColor(String type, Color value) {
221 | final String color = String.format("%s,%s,%s", value.getRed(), value.getGreen(), value.getBlue());
222 | PropertiesComponent.getInstance().setValue("color:" + type, color);
223 | }
224 |
225 | public static ArrayList getAddOpens() {
226 | final String value = PropertiesComponent.getInstance().getValue("addOpens:", String.join(";", AddOpens));
227 | return new ArrayList(Arrays.asList(value.split(";")));
228 | }
229 |
230 | public static void setAddOpens(String opens) {
231 | final List openlist = Arrays.asList(opens.split("\n"));
232 | PropertiesComponent.getInstance().setValue("addOpens:", String.join(";", openlist));
233 | }
234 |
235 | public static Map getPerRunMap() {
236 | final String value = PropertiesComponent.getInstance().getValue("perRunMap:");
237 | if (null == value) {
238 | return PerRunMap;
239 | }
240 | final String[] split1 = value.split(";");
241 | final List strings = Arrays.asList(split1);
242 | strings.forEach(e -> {
243 | final String[] split2 = e.split("=");
244 | if (!PerRunMap.containsKey(split2[0])) {
245 | PerRunMap.put(split2[0], Boolean.valueOf(split2[1]));
246 | }
247 | });
248 | return PerRunMap;
249 | }
250 |
251 | /**
252 | * 添加执行器
253 | *
254 | * @param name 执行器名称
255 | * @param run 是否可以运行
256 | * @param cover 是否替换 false=不替换 true=强制替换
257 | */
258 | public static void setPerRunMap(String name, boolean run, boolean cover) {
259 | final Boolean aBoolean = PerRunMap.containsKey(name);
260 | if (cover) {
261 | PerRunMap.put(name, run);
262 | }
263 | if (!cover && !aBoolean) {
264 | PluginUtil.Notificat_AddConfiguration();
265 | PerRunMap.put(name, run);
266 | }
267 | ArrayList value = new ArrayList();
268 | for (Map.Entry next : PerRunMap.entrySet()) {
269 | final String format = String.format("%s=%s", next.getKey(), next.getValue());
270 | value.add(format);
271 | }
272 | PropertiesComponent.getInstance().setValue("perRunMap:", String.join(";", value));
273 | }
274 |
275 |
276 | }
277 |
278 | public static String id;
279 | }
280 |
--------------------------------------------------------------------------------
/mybatis-plugin/src/main/java/com/linkkou/mybatis/log/LogInterceptor.java:
--------------------------------------------------------------------------------
1 | package com.linkkou.mybatis.log;
2 |
3 | import com.google.gson.Gson;
4 | import com.google.gson.GsonBuilder;
5 | import org.apache.ibatis.cache.CacheKey;
6 | import org.apache.ibatis.executor.Executor;
7 | import org.apache.ibatis.mapping.BoundSql;
8 | import org.apache.ibatis.mapping.MappedStatement;
9 | import org.apache.ibatis.mapping.ParameterMapping;
10 | import org.apache.ibatis.mapping.SqlSource;
11 | import org.apache.ibatis.plugin.*;
12 | import org.apache.ibatis.reflection.MetaObject;
13 | import org.apache.ibatis.scripting.defaults.RawSqlSource;
14 | import org.apache.ibatis.scripting.xmltags.DynamicContext;
15 | import org.apache.ibatis.scripting.xmltags.DynamicSqlSource;
16 | import org.apache.ibatis.scripting.xmltags.SqlNode;
17 | import org.apache.ibatis.session.Configuration;
18 | import org.apache.ibatis.session.ResultHandler;
19 | import org.apache.ibatis.session.RowBounds;
20 | import org.apache.ibatis.type.JdbcType;
21 | import org.apache.ibatis.type.TypeHandler;
22 | import org.javatuples.Pair;
23 |
24 | import java.lang.reflect.Field;
25 | import java.util.*;
26 | import java.util.regex.Matcher;
27 |
28 |
29 | /**
30 | * 用于构建完整的SQL，彻底解决通过？替换的问题存在的不正确的情况
31 | *
32 | * @author lk
33 | */
34 | @Intercepts({
35 | @Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class}),
36 | @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class, CacheKey.class, BoundSql.class}),
37 | @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}),
38 | @Signature(type = Executor.class, method = "queryCursor", args = {MappedStatement.class, Object.class, RowBounds.class})
39 | })
40 | public class LogInterceptor implements Interceptor {
41 |
42 | /**
43 | * 对象名称
44 | */
45 | private static final String ROOTSQLNODE = "rootSqlNode";
46 | private static final String ROOTSQLNODE2 = "_rootSqlNode_";
47 | /**
48 | * 参数数量
49 | */
50 | private static final Integer ARGSNUMBER = 2;
51 |
52 | private final String id;
53 |
54 | public static Gson gson = new GsonBuilder().setDateFormat("yyyy-MM-dd HH:mm:ss")
55 | //当Map的key为复杂对象时,需要开启该方法
56 | .enableComplexMapKeySerialization()
57 | //当字段值为空或null时，依然对该字段进行转换
58 | .serializeNulls()
59 | //防止特殊字符出现乱码
60 | .disableHtmlEscaping().create();
61 |
62 | public LogInterceptor(String id) {
63 | this.id = id;
64 | }
65 |
66 | @Override
67 | public Object intercept(Invocation invocation) throws Throwable {
68 | if (invocation.getArgs().length >= ARGSNUMBER) {
69 | final Pair args = getArgs(invocation);
70 | MappedStatement mappedStatement = args.getValue0();
71 | Object parameter = args.getValue1();
72 | try {
73 | final String originalSql = this.getOriginalSql(mappedStatement, parameter);
74 | if (originalSql != null) {
75 | BoundSql boundSql = mappedStatement.getBoundSql(parameter);
76 | Configuration configuration = mappedStatement.getConfiguration();
77 | // 通过配置信息和BoundSql对象来生成带值得sql语句
78 | final Pair>> completeSql = getCompleteSql(configuration, boundSql, originalSql);
79 | final SqlVO sqlVO = new SqlVO().setId(mappedStatement.getId())
80 | .setCompleteSql(completeSql.getValue0())
81 | .setParameter(gson.toJson(completeSql.getValue1()))
82 | .setOriginalSql(originalSql);
83 | final String json = gson.toJson(sqlVO);
84 | PrintlnLog.log("==> SQLStructure: " + json, this.id);
85 | }
86 | } catch (Exception e) {
87 | e.printStackTrace();
88 | }
89 | }
90 | return invocation.proceed();
91 | }
92 |
93 | private Pair getArgs(Invocation invocation) {
94 | final Object[] args = invocation.getArgs();
95 | return Pair.with((MappedStatement) args[0], args[1]);
96 | }
97 |
98 | /**
99 | * 生成对应的带有值得sql语句
100 | *
101 | * @param configuration 配置
102 | * @param boundSql 对象
103 | * @param originalSql sql
104 | * @return String
105 | */
106 | private Pair>> getCompleteSql(Configuration configuration, BoundSql boundSql, String originalSql) {
107 | Object parameterObject = boundSql.getParameterObject();
108 | List parameterMappings = boundSql.getParameterMappings();
109 | List> keyvalue = new ArrayList>();
110 | final int size = parameterMappings.size();
111 | if (size > 0 && parameterObject != null) {
112 | MetaObject metaObject = configuration.newMetaObject(parameterObject);
113 | for (ParameterMapping parameterMapping : parameterMappings) {
114 | String propertyName = parameterMapping.getProperty();
115 | final JdbcType jdbcType = parameterMapping.getJdbcType();
116 | final TypeHandler typeHandler = parameterMapping.getTypeHandler();
117 | final HashMap stringObjectHashMap = new HashMap<>();
118 | if (metaObject.hasGetter(propertyName)) {
119 | Object obj = metaObject.getValue(propertyName);
120 | final String parameterValue = getParameterTypeHandler(configuration, typeHandler, obj, jdbcType);
121 | stringObjectHashMap.put(propertyName, parameterValue);
122 | keyvalue.add(stringObjectHashMap);
123 | originalSql = replaceFirst(originalSql, propertyName, parameterValue);
124 | } else if (boundSql.hasAdditionalParameter(propertyName)) {
125 | Object obj = boundSql.getAdditionalParameter(propertyName);
126 | final String parameterValue = getParameterTypeHandler(configuration, typeHandler, obj, jdbcType);
127 | stringObjectHashMap.put(propertyName, parameterValue);
128 | keyvalue.add(stringObjectHashMap);
129 | originalSql = replaceFirst(originalSql, propertyName, parameterValue);
130 | } else if (!(parameterObject instanceof Map)) {
131 | //单个参数默认组合
132 | final String parameterValue = getParameterTypeHandler(configuration, typeHandler, metaObject.getOriginalObject(), jdbcType);
133 | stringObjectHashMap.put(propertyName, parameterValue);
134 | keyvalue.add(stringObjectHashMap);
135 | originalSql = replaceFirst(originalSql, propertyName, parameterValue);
136 | }
137 | }
138 | }
139 | return Pair.with(originalSql.replaceAll("[\\s]+", " "), keyvalue);
140 | }
141 |
142 | /**
143 | * 参数替换
144 | *
145 | * @param originalSql SQL
146 | * @param propertyName 名称
147 | * @param parameterValue 值
148 | * @return String
149 | */
150 | private String replaceFirst(String originalSql, String propertyName, String parameterValue) {
151 | //(\$|#)\{\s*epNo2((?!\{).)*}
152 | return originalSql.replaceFirst("(\\$|#)\\{\\s*" + Matcher.quoteReplacement(propertyName) + "((?!\\{).)*}", Matcher.quoteReplacement(parameterValue));
153 | }
154 |
155 | /**
156 | * 获取原始的SQL
157 | *
158 | * @param mappedStatement 对象
159 | * @param parameter 对象
160 | * @return String
161 | */
162 | private String getOriginalSql(MappedStatement mappedStatement, Object parameter) {
163 | final SqlSource sqlSource = mappedStatement.getSqlSource();
164 | if (sqlSource instanceof DynamicSqlSource) {
165 | return getDynamicSqlSource(mappedStatement, parameter);
166 | }
167 | if (sqlSource instanceof RawSqlSource) {
168 | return getRawSqlSource(sqlSource);
169 | }
170 | //(\'|\")(.*?)\?(.*?)+(\'|\")
171 | //(?!(\'|\"))\?(?!(\'|\"))
172 | //自定义SQL的拦截
173 | return null;
174 | }
175 |
176 |
177 | private String getDynamicSqlSource(MappedStatement mappedStatement, Object parameter) {
178 | final DynamicContext dynamicContext = new DynamicContext(mappedStatement.getConfiguration(), parameter);
179 | final SqlNode sqlNode = getSqlNode(mappedStatement.getSqlSource());
180 | if (sqlNode != null && sqlNode.apply(dynamicContext)) {
181 | return dynamicContext.getSql();
182 | }
183 | return null;
184 | }
185 |
186 | private String getRawSqlSource(SqlSource sqlSource) {
187 | final RawSqlSource dynamicSqlSource = (RawSqlSource) sqlSource;
188 | Field[] declaredFields = dynamicSqlSource.getClass().getDeclaredFields();
189 | for (Field declaredField : declaredFields) {
190 | if (ROOTSQLNODE2.equals(declaredField.getName())) {
191 | try {
192 | declaredField.setAccessible(true);
193 | return (String) declaredField.get(dynamicSqlSource);
194 | } catch (IllegalAccessException e) {
195 | throw new RuntimeException(e);
196 | }
197 | }
198 | }
199 | return null;
200 | }
201 |
202 | /**
203 | * 构建 SqlNode
204 | *
205 | * @param sqlSource 对象
206 | * @return SqlNode
207 | */
208 | private SqlNode getSqlNode(SqlSource sqlSource) {
209 | if (sqlSource instanceof DynamicSqlSource) {
210 | final DynamicSqlSource dynamicSqlSource = (DynamicSqlSource) sqlSource;
211 | Field[] declaredFields = dynamicSqlSource.getClass().getDeclaredFields();
212 | for (Field declaredField : declaredFields) {
213 | if (ROOTSQLNODE.equals(declaredField.getName())) {
214 | try {
215 | declaredField.setAccessible(true);
216 | return (SqlNode) declaredField.get(dynamicSqlSource);
217 | } catch (IllegalAccessException e) {
218 | throw new RuntimeException(e);
219 | }
220 | }
221 | }
222 | }
223 | return null;
224 | }
225 |
226 | /**
227 | * 如果是字符串对象则加上单引号返回，如果是日期则也需要转换成字符串形式，如果是其他则直接转换成字符串返回。
228 | *
229 | * @param configuration 对象
230 | * @param jdbcType 对象
231 | * @return String
232 | */
233 | private static String getParameterTypeHandler(Configuration configuration, TypeHandler typeHandler, Object value, JdbcType jdbcType) {
234 | try {
235 | TypeHandler
46 |
47 | 
23 |