小白的web优化之路 三、使用MQ来实现事务异步处理

  在上一篇 小白的web优化之路 二、实战使用redis来缓存列表信息 中，我们讲了使用redis来缓存列表信息，极大的提供了用户访问列表的速度，也减轻了数据库的压力，下面，我们在以一个例子来讲解MQ来实现事务的异步处理。

  在使用了redis解决了列表查询问题之后，小白又遇到了一个问题，刚刚产品经理找他，要他实现一个功能，功能如下：

  当用户完成某一任务时，提高用户的积分，并给此用户的所有朋友发送短信通知，告诉他们此用户完成了某任务，让他们赶紧来玩。

  小白瞬间感觉难住了，提高用户的积分很简单，一个UPDATE就行，但是发送短信通知怎么办呢，假设给一个用户发送短信的时长为50ms，那么串行处理的方式十个用户就是500ms，一百个用户就是5000ms，对于一个接口来说这个时间太长了，是不能接收的，这个问题怎么解决呢？小白陷入了沉思...

 过了一会，小白突然想到，我为什么要马上给用户的好友发送短信呢？就算延迟几秒接收也是可以的啊。我可以在后台创建一个队列，请求来了之后，我先把提高用户的积分，然后把要处理的内容放到队列中，然后直接返回成功就行，后台的处理程序在处理这些队列就可以了。

 小白的想法也就是我们平常常说的MQ，即消息队列。在一些处理时间比较长的任务，我们可以使用MQ进行延时处理，来降低客户的请求等待时间和减轻服务器的压力。

 下面，我们将以activeMQ来解决上述问题。

 首先，先看看没有mq的示例代码：

package com.happyheng;

import com.happyheng.dao.FriendDao;
import com.happyheng.dao.ScoreDao;
import com.happyheng.service.SendMessageService;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;
import java.util.List;

/**
 * 发送信息的Controller
 * Created by happyheng on 2017/6/17.
 */
@RestController
public class MessageController {


    @Resource
    private SendMessageService sendMessageService;


    @Resource
    private ScoreDao scoreDao;
    @Resource
    private FriendDao friendDao;


    @RequestMapping("/sendMessage")
    public String sendMessage() {


        // 首先提高积分
        scoreDao.updateScore();

        // 其次获取用户的好友手机号列表
        List numberList = friendDao.getUserFriendPhoneNumbers();


        // 发送短信
        for (String num:
                numberList) {

            sendMessageService.sendMessage(num, "你的好友完成了xx任务,快来一起玩吧");

        }

        return "success";
    }

}

在 SendMessageService 中，模拟了发送短信的延时：

package com.happyheng.service;

import org.springframework.stereotype.Service;

/**
 *
 * Created by happyheng on 2017/6/17.
 */
@Service
public class SendMessageService {


    public void sendMessage(String phoneNum, String message) {
        
        // 模拟发送短信的耗时操作
        try {
            Thread.sleep(50);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

    }


}

可以看到，如果串行的运行的话，接口的速度太慢。

这时候即可以在接口中使用mq，即先不发送短信，而是直接放到mq中，队列中的消息会有相应的消费者去处理：

package com.happyheng;

import com.happyheng.dao.FriendDao;
import com.happyheng.dao.ScoreDao;
import com.happyheng.service.SendMessageService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jms.core.JmsTemplate;
import org.springframework.jms.core.MessageCreator;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;
import javax.jms.*;
import java.util.List;

/**
 * 发送信息的Controller
 * Created by happyheng on 2017/6/17.
 */
@RestController
public class MessageController {

    @Resource(name = "queueDestination")
    private Destination destination;

    @Autowired
    private JmsTemplate jmsTemplate;

    @Resource
    private SendMessageService sendMessageService;


    @Resource
    private ScoreDao scoreDao;
    @Resource
    private FriendDao friendDao;

  @RequestMapping("/sendMessageByMq")
    public String sendMessageByMq() {

        // 首先提高积分
        scoreDao.updateScore();

        // 其次获取用户的好友手机号列表
        List numberList = friendDao.getUserFriendPhoneNumbers();


        // 将好友手机号直接放到mq中即可, 消费者会发送短信
        for (final String num:
                numberList){

            jmsTemplate.send(destination, new MessageCreator() {
                public Message createMessage(Session session) throws JMSException {

                    MapMessage mapMessage = session.createMapMessage();
                    mapMessage.setString("num", num);
                    mapMessage.setString("text", "你的好友完成了xx任务,快来一起玩吧");
                    return mapMessage;
                }
            });

        }
        return "success";
    }


}

相应的消费者为：

package com.happyheng.consumer;

import com.happyheng.service.SendMessageService;

import javax.annotation.Resource;
import javax.jms.*;

/**
 *
 * Created by happyheng on 2017/6/17.
 */
public class MapMessageListener implements MessageListener {

    @Resource
    private SendMessageService sendMessageService;

    public void onMessage(Message message){

        System.out.println("收到信息");
        MapMessage mapMessage = (MapMessage) message;

        try {
            String num = mapMessage.getString("num");
            String textMessage = mapMessage.getString("text");
            System.out.println("手机号为--" + num + "--信息内容为" + textMessage);

            sendMessageService.sendMessage(num, textMessage);
        } catch (JMSException e) {
            e.printStackTrace();
        }


    }
}

这样会极大的提高接口速度，尤其是在好友数量很多的情况下：

示例代码已经放到github上，此工程为项目中的 web-optimize-mq 工程 ，地址为 项目源码地址

下一篇中，我们将讲解使用Callable与Future来实现异步请求接口优化，地址为 小白的web优化之路 四、使用Callable与Future来实现异步请求