jeesite mybatis拦截器sql语句与activemq的使用

拦截器

import java.text.DateFormat;
import java.util.Date;
import java.util.List;
import java.util.Locale;
import java.util.Properties;

import javax.jms.Destination;
import javax.servlet.ServletContext;

import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.ParameterMapping;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.plugin.Plugin;
import org.apache.ibatis.plugin.Signature;
import org.apache.ibatis.reflection.MetaObject;
import org.apache.ibatis.session.Configuration;
import org.apache.ibatis.type.TypeHandlerRegistry;
import org.slf4j.LoggerFactory;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Component;
import org.springframework.web.context.ContextLoader;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.context.support.WebApplicationContextUtils;

/**
 * 注解拦截接口的方法 
 * Executor (update, query, flushStatements, commit, rollback,getTransaction, close, isClosed) 
 * ParameterHandler (getParameterObject,setParameters) 
 * ResultSetHandler (handleResultSets, handleOutputParameters)
 * StatementHandler (prepare, parameterize, batch, update, query)
 */
@Component
@Intercepts(@Signature(type = Executor.class, method = "update", args = { MappedStatement.class, Object.class }))
public class MsSQLUpdateInterceptor  implements Interceptor {

    public Object intercept(Invocation invocation) throws Throwable {
        try {
            return invocation.proceed();
        } finally {
            MappedStatement mappedStatement = (MappedStatement) invocation.getArgs()[0];
            String methodName = invocation.getMethod().getName();
            Object parameter = null;
            if (invocation.getArgs().length > 1) {
                parameter = invocation.getArgs()[1];// 方法参数
            }
            BoundSql boundSql = mappedStatement.getBoundSql(parameter);
            Configuration configuration = mappedStatement.getConfiguration();
            if (methodName.equals("update")) {
                String sql = getSql(configuration, boundSql);
                //用来判断获取的sql语句是否符合自己想要的
                if (StringUtils.contains(sql, "sender")) {
                     if (StringUtils.isNotBlank(sql)) {
                         WebApplicationContext webApplicationContext = ContextLoader.getCurrentWebApplicationContext();
                         ServletContext servletContext = webApplicationContext.getServletContext();
                         ApplicationContext ac = WebApplicationContextUtils.getWebApplicationContext(servletContext);
                         SenderService senderService = (SenderService) ac.getBean("senderService");
                         Destination destination=(Destination) ac.getBean("demoQueueDestination");
                         senderService.sendMessage(destination,sql);
                    }
                }
            }
        }
    }

    public Object plugin(Object target) {
        return Plugin.wrap(target, this);
    }

    public void setProperties(Properties properties) {
    }

    /**
     * 参数的处理,日期格式化,添加单引号
     * @param obj
     * @return
     */
    private static String getParameterValue(Object obj) {
        String value = null;
        if (obj instanceof String) {
            value = "'" + obj.toString() + "'";
        } else if (obj instanceof Date) {
            DateFormat formatter = DateFormat.getDateTimeInstance(DateFormat.DEFAULT, DateFormat.DEFAULT, Locale.CHINA);
            value = "'" + formatter.format(new Date()) + "'";
        } else {
            if (obj != null) {
                value = obj.toString();
            } else {
                value = "";
            }

        }
        return value;
    }

    /**
     * 拼接sql语句
     * @param configuration
     * @param boundSql
     * @return
     */
    public static String getSql(Configuration configuration, BoundSql boundSql) {
        Object parameterObject = boundSql.getParameterObject();
        List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();
        String sql = boundSql.getSql().replaceAll("[\\s]+", " ");
        if (parameterMappings.size() > 0 && parameterObject != null) {
            TypeHandlerRegistry typeHandlerRegistry = configuration.getTypeHandlerRegistry();
            if (typeHandlerRegistry.hasTypeHandler(parameterObject.getClass())) {
                sql = sql.replaceFirst("\\?", getParameterValue(parameterObject));

            } else {
                MetaObject metaObject = configuration.newMetaObject(parameterObject);
                for (ParameterMapping parameterMapping : parameterMappings) {
                    String propertyName = parameterMapping.getProperty();
                    if (metaObject.hasGetter(propertyName)) {
                        Object obj = metaObject.getValue(propertyName);
                        sql = sql.replaceFirst("\\?", getParameterValue(obj));
                    } else if (boundSql.hasAdditionalParameter(propertyName)) {
                        Object obj = boundSql.getAdditionalParameter(propertyName);
                        sql = sql.replaceFirst("\\?", getParameterValue(obj));
                    }
                }
            }
        }
        return sql;
    }

}

发送者

import javax.jms.Destination;
import javax.jms.JMSException;
import javax.jms.Message;
import javax.jms.Session;
import javax.servlet.ServletContext;

import org.springframework.context.ApplicationContext;
import org.springframework.jms.core.JmsTemplate;
import org.springframework.jms.core.MessageCreator;
import org.springframework.stereotype.Service;
import org.springframework.web.context.ContextLoader;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.context.support.WebApplicationContextUtils;

/**
 * activemq的消息生产者做成一个服务,当我们需要发送消息的时候,
 * 只需要调用SenderService实例中的sendMessage方法就可以向默认目的发送一个消息。
 * 
 * 这里提供了两个发送方式,一个是发送到默认的目的地,一个是根据目的地发送消息
 * 
 * @author kzy
 * 
 */
@Service
public class SenderService{


    public void sendMessage(Destination destination,final String msg){
    //用着方法的原因是因为我这个是mybatis的来拦截器,所以spring的注入是无效的,所以用这种方法去注入对象
        WebApplicationContext webApplicationContext = ContextLoader.getCurrentWebApplicationContext();
        ServletContext servletContext = webApplicationContext.getServletContext();
        ApplicationContext ac = WebApplicationContextUtils.getWebApplicationContext(servletContext);
        JmsTemplate jmsTemplate = (JmsTemplate) ac.getBean("jmsTemplate");
        jmsTemplate.send(destination, new MessageCreator() {
            public Message createMessage(Session session) throws JMSException {
                return session.createTextMessage(msg);
            }
        });
    }

}

接受者

import javax.jms.Destination;
import javax.jms.TextMessage;
import javax.servlet.ServletContext;

import org.springframework.context.ApplicationContext;
import org.springframework.jms.core.JmsTemplate;
import org.springframework.stereotype.Service;
import org.springframework.web.context.ContextLoader;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.context.support.WebApplicationContextUtils;

/**
 * 消息接受者
 * @author kzy
 *
 */
@Service
public class ReceiverService {

    public TextMessage receive(Destination destination) {

        WebApplicationContext webApplicationContext = ContextLoader.getCurrentWebApplicationContext();
        ServletContext servletContext = webApplicationContext.getServletContext();
        ApplicationContext ac = WebApplicationContextUtils.getWebApplicationContext(servletContext);
        JmsTemplate jmsTemplate= (JmsTemplate) ac.getBean("jmsTemplate");

        TextMessage textMessage = (TextMessage) jmsTemplate.receive(destination);
        return textMessage;
    }

}

配置文件

配置文件在两个项目都必须有
spring-context-activemq.xml

<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
       xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
       xmlns:amq="http://activemq.apache.org/schema/core"
       xmlns:jms="http://www.springframework.org/schema/jms"
       xmlns:context="http://www.springframework.org/schema/context"
       xmlns:mvc="http://www.springframework.org/schema/mvc"
       xsi:schemaLocation="
        http://www.springframework.org/schema/beans
        http://www.springframework.org/schema/beans/spring-beans-4.1.xsd
        http://www.springframework.org/schema/context
        http://www.springframework.org/schema/context/spring-context-4.1.xsd
        http://www.springframework.org/schema/mvc
        http://www.springframework.org/schema/mvc/spring-mvc-4.1.xsd
        http://www.springframework.org/schema/jms
        http://www.springframework.org/schema/jms/spring-jms-4.1.xsd
        http://activemq.apache.org/schema/core
        http://activemq.apache.org/schema/core/activemq-core-5.12.1.xsd"
>

    <context:component-scan base-package="com.spingcn" />
    <mvc:annotation-driven />

    <amq:connectionFactory id="amqConnectionFactory"  brokerURL="tcp://localhost:61616" <!-- 需要替换成服务器上的地址 -->
        userName="admin" password="admin" />

    <!-- 配置JMS连接工长 -->
    <bean id="connectionFactory"
          class="org.springframework.jms.connection.CachingConnectionFactory">
        <constructor-arg ref="amqConnectionFactory" />
        <property name="sessionCacheSize" value="100" />
    </bean>

    <!-- 定义消息队列(Queue) -->
    <bean id="demoQueueDestination" class="org.apache.activemq.command.ActiveMQQueue">
        <!-- 设置消息队列的名字 -->
        <constructor-arg>
            <value>test</value>
        </constructor-arg>
    </bean>

    <!-- 配置JMS模板(Queue),Spring提供的JMS工具类,它发送、接收消息。 -->
    <bean id="jmsTemplate" class="org.springframework.jms.core.JmsTemplate">
        <property name="connectionFactory" ref="connectionFactory" />
        <property name="defaultDestination" ref="demoQueueDestination" />
        <property name="receiveTimeout" value="10000" />
        <!-- true是topic,false是queue,默认是false,此处显示写出false -->
        <property name="pubSubDomain" value="false" />
    </bean>

</beans>

web.xml文件

<context-param>
    <param-name>contextConfigLocation</param-name>
    <param-value>
        classpath:spring.xml;
        classpath:spring-context-activemq.xml;<!-- 这里需要跟上方的配置文件同名 -->
    </param-value>
  </context-param>

拦截器的配置

mybatis-config.xml

    <!-- 插件配置 -->
    <plugins>
        <plugin interceptor="com.spingcn.modules.sys.interceptor.MsSQLUpdateInterceptor" />
    </plugins>

</configuration>

导入的包

activemq-all-5.14.0.jar
activemq-broker-5.14.0.jar
activemq-client-5.14.0.jar
geronimo-j2ee-management_1.1_spec-1.0.1.jar
geronimo-jms_1.1_spec-1.1.1.jar
slf4j-api-1.7.13.jar

猜你喜欢

转载自blog.csdn.net/k_clound/article/details/77679825