当前位置：首页 > news >正文

rabbitmq 短信验证码

news 来源：原创 2024/9/25 21:31:47

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 {@Resourceprivate RabbitTemplate rabbitTemplate;@Resourceprivate StringRedisTemplate stringRedisTemplate;@Operation(summary = "发送短信验证码")@GetMapping("send")public String send() {//生成4位的数字验证码Random rand = new Random();String code = (rand.nextInt(8999) + 1000) + "";//验证码存在redis，设置1分钟过期  校验时查redisstringRedisTemplate.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分钟内校验");}
}

测试