Java Redis发送短信验证码

package com.redis.demo;

import redis.clients.jedis.Jedis;

import java.util.Random;

public class PhoneCode {
    public static void main(String[] args) {
      // 模拟验证码发送
        verifyCode("13899996666");
    }

    // 3.验证码校验
    public static void getRedisCode(String phone, String code) {
        // 连接redis
        Jedis jedis = new Jedis("127.0.0.1", 6379);
        // 验证码key
        String codeKey = "VerifyCode" + phone + ":code";
        // Redis中验证码
        String redisCode = jedis.get(codeKey);
        // 判断
        if (redisCode.equals(code)) {
            System.out.println("成功");
        } else {
            System.out.println("失败");
        }
        jedis.close();
    }

    // 2.每个手机每天只能发送3次，验证码放到redis中，设置过期时间
    public static void verifyCode(String phone) {
        // 连接redis
        Jedis jedis = new Jedis("127.0.0.1", 6379);

        // 手机发送次数key
        String countKey = "VerifyCode" + phone + ":count";

        // 验证码key
        String codeKey = "VerifyCode" + phone + ":code";
        String count = jedis.get(countKey);
        System.out.println(count);
        if (count == null) {
            // 第一次发送，设置发送次数为1
            jedis.setex(countKey, 24*60*60, "1");
        } else if(Integer.parseInt(count) <= 2) {
            // 发送次数加1
            jedis.incr(countKey);
        } else if(Integer.parseInt(count) > 3){
            System.out.println("今天的发送次数已经超过3次");
            jedis.close();
        }

        // 1.发送验证码放到redis
        String vcode = getCode();
        jedis.setex(codeKey, 120, vcode);
        String s = jedis.get(codeKey);
        System.out.println(s);
        jedis.close();
    }

    // 生成6位数字验证码
    public static String getCode() {
        Random random = new Random();
        String code = "";
        for (int i = 0; i < 6; i++) {
            int rand = random.nextInt(10);
            code += rand;
        }
        return code;
    }
}