com.magictools.mybatis.toolkit.PluginUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of magic-tools-all Show documentation
magic-tools 是一个小而全的Java工具类库，通过静态方法封装，降低相关API的学习成本，提高工作效率，使Java拥有函数式语言般的优雅，让Java语言也可以“甜甜的”。
The newest version!
/*
 * Copyright (c) 2011-2022, baomidou ([email protected]).
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.magictools.mybatis.toolkit;

import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.executor.parameter.ParameterHandler;
import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.ParameterMapping;
import org.apache.ibatis.reflection.MetaObject;
import org.apache.ibatis.reflection.SystemMetaObject;
import org.apache.ibatis.session.Configuration;

import java.lang.reflect.Proxy;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;

/**
 * 插件工具类
 *
 * @author TaoYu , hubin
 * @since 2017-06-20
 */
public abstract class PluginUtils {
	public static final String DELEGATE_BOUNDSQL_SQL = "delegate.boundSql.sql";

	/**
	 * 获得真正的处理对象,可能多层代理.
	 */
	@SuppressWarnings("unchecked")
	public static  T realTarget(Object target) {
		if (Proxy.isProxyClass(target.getClass())) {
			MetaObject metaObject = SystemMetaObject.forObject(target);
			return realTarget(metaObject.getValue("h.target"));
		}
		return (T) target;
	}

	/**
	 * 给 BoundSql 设置 additionalParameters
	 *
	 * @param boundSql             BoundSql
	 * @param additionalParameters additionalParameters
	 */
	public static void setAdditionalParameter(BoundSql boundSql, Map additionalParameters) {
		additionalParameters.forEach(boundSql::setAdditionalParameter);
	}

	public static MPBoundSql mpBoundSql(BoundSql boundSql) {
		return new MPBoundSql(boundSql);
	}

	public static MPStatementHandler mpStatementHandler(StatementHandler statementHandler) {
		statementHandler = realTarget(statementHandler);
		MetaObject object = SystemMetaObject.forObject(statementHandler);
		return new MPStatementHandler(SystemMetaObject.forObject(object.getValue("delegate")));
	}

	/**
	 * {@link org.apache.ibatis.executor.statement.BaseStatementHandler}
	 */
	public static class MPStatementHandler {
		private final MetaObject statementHandler;

		MPStatementHandler(MetaObject statementHandler) {
			this.statementHandler = statementHandler;
		}

		public ParameterHandler parameterHandler() {
			return get("parameterHandler");
		}

		public MappedStatement mappedStatement() {
			return get("mappedStatement");
		}

		public Executor executor() {
			return get("executor");
		}

		public MPBoundSql mPBoundSql() {
			return new MPBoundSql(boundSql());
		}

		public BoundSql boundSql() {
			return get("boundSql");
		}

		public Configuration configuration() {
			return get("configuration");
		}

		@SuppressWarnings("unchecked")
		private  T get(String property) {
			return (T) statementHandler.getValue(property);
		}
	}

	/**
	 * {@link BoundSql}
	 */
	public static class MPBoundSql {
		private final MetaObject boundSql;
		private final BoundSql delegate;

		MPBoundSql(BoundSql boundSql) {
			this.delegate = boundSql;
			this.boundSql = SystemMetaObject.forObject(boundSql);
		}

		public String sql() {
			return delegate.getSql();
		}

		public void sql(String sql) {
			boundSql.setValue("sql", sql);
		}

		public List parameterMappings() {
			List parameterMappings = delegate.getParameterMappings();
			return new ArrayList<>(parameterMappings);
		}

		public void parameterMappings(List parameterMappings) {
			boundSql.setValue("parameterMappings", Collections.unmodifiableList(parameterMappings));
		}

		public Object parameterObject() {
			return get("parameterObject");
		}

		public Map additionalParameters() {
			return get("additionalParameters");
		}

		@SuppressWarnings("unchecked")
		private  T get(String property) {
			return (T) boundSql.getValue(property);
		}
	}
}