rabbitmq 短信验证码

1**.生成的验证码用redis存减少数据库压力**

2.通知运营商发送短信的事情交给rabbitmq的队列去做，无论成功或者是失败，用户那边都不知道。没有收到验证码（监听失败）用户只会觉得是运营商的问题，而不会怀疑是我们的系统有问题。

（ps：如果没有用rabbitmq来监听，通知运营商的业务失败了，用户就直接知道系统爆了，然而发送验证码这个业务失败又能怎么样呢？多发几次能成功就行，但是不能让用户知道我们的系统很垃圾，这就是使用mq实现异步调用的原因）

新建一个springboot项目,引入web，redis，rabbitmq

复制代码

# 应用服务 WEB 访问端口
server.port=8080

# redis配置
spring.data.redis.host=192.168.168.168
spring.data.redis.port=6379
# spring.data.redis.password=123456
spring.data.redis.database=0
# lettuce连接池
# 最大活跃连接数
spring.data.redis.lettuce.pool.max-active=8
# 最大空闲连接数
spring.data.redis.lettuce.pool.max-idle=8
# 最小空闲连接数
spring.data.redis.lettuce.pool.min-idle=0
# 当所有连接都忙时，客户端等待可用连接的最大时间
spring.data.redis.lettuce.pool.max-wait=100ms


# rabbitmq配置
# 主机ip
spring.rabbitmq.host=192.168.168.168
# rabbitmq的编程端口，默认5672
spring.rabbitmq.port=5672
# 账号和密码
spring.rabbitmq.username=chen
spring.rabbitmq.password=123456
# 虚拟主机
spring.rabbitmq.virtual-host=/test
# 通过设置prefetch来控制消费者预取的消息数量。这条配置告诉RabbitMQ的消费者一次只从队列中拉取一条消息进行处理。
spring.rabbitmq.listener.simple.prefetch=1

复制代码

package com.gmgx.demorabbitmqsms.controller;

import ch.qos.logback.core.util.TimeUtil;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.Random;
import java.util.concurrent.TimeUnit;

@Tag(name = "验证码控制器")
@RequestMapping("sms")
@RestController
public class SmsController {
    @Resource
    private RabbitTemplate rabbitTemplate;

    @Resource
    private StringRedisTemplate stringRedisTemplate;


    @Operation(summary = "发送短信验证码")
    @GetMapping("send")
    public String send() {
        //生成4位的数字验证码
        Random rand = new Random();
        String code = (rand.nextInt(8999) + 1000) + "";

        //验证码存在redis，设置1分钟过期  校验时查redis
        stringRedisTemplate.opsForValue().set("smsCode", code, 60, TimeUnit.SECONDS);

        //发送消息到rabbitmq的sms.exchange  这个队列的任务是通知运营商发送短信给用户（模拟）
        rabbitTemplate.convertAndSend("sms.exchange", "", code);
        return "发送成功";//无论怎样都返回发送成功，欺骗用户
    }

    @Operation(summary = "验证")
    @GetMapping("valify/{code}")
    public String valify(@PathVariable String code) {
        if (!stringRedisTemplate.hasKey("smsCode")) {
            return "验证失败";
        }
        if (!code.equals(stringRedisTemplate.opsForValue().get("smsCode"))) {
            return "验证失败";
        }
        stringRedisTemplate.delete("smsCode");
        return "验证成功";
    }
}

复制代码

package com.gmgx.demorabbitmqsms.listener;


import org.springframework.amqp.core.ExchangeTypes;
import org.springframework.amqp.rabbit.annotation.Exchange;
import org.springframework.amqp.rabbit.annotation.Queue;
import org.springframework.amqp.rabbit.annotation.QueueBinding;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.stereotype.Component;

@Component
public class SmsListener {
    @RabbitListener(bindings = @QueueBinding(
            value = @Queue(name = "sms.queue"),
            exchange = @Exchange(name = "sms.exchange", type = ExchangeTypes.DIRECT)
    ))
    public void listen(String msg) {
        try {
            //休眠5秒，模拟通知运营商发送短信给用户
            Thread.sleep(5000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        System.out.println("【中国移动】 您的验证码为" + msg + " 请在1分钟内校验");
    }
}

测试