import redis.clients.jedis.Jedis;

import java.util.Random;

public class ValidationTest {
  public static void main(String[] args) {
    //getValidation("15005076571");
    //checkValidation("769897","15005076571");
  }

  static void getValidation(String tel) {
    //主機(jī)、端口
    Jedis jedis = new Jedis("myhost", 6379);
    //密碼
    jedis.auth("mypassword");
    try {
      //獲取電話(huà)號(hào)碼
      String phoneNo = tel;
      //本人用1庫(kù)進(jìn)行測(cè)試
      jedis.select(1);
      String countKey = phoneNo + ":count";
      String codeKey = phoneNo + ":code";
      //獲取指定的電話(huà)號(hào)碼發(fā)送的驗(yàn)證碼次數(shù)
      String cnt = jedis.get(countKey);
      //對(duì)次數(shù)進(jìn)行判斷
      if (cnt == null) {
        //沒(méi)有發(fā)送過(guò)驗(yàn)證碼
        jedis.setex(countKey, 60 * 60 * 24, "1");
        //發(fā)送驗(yàn)證碼，假設(shè)生成的驗(yàn)證碼
        StringBuffer code = new StringBuffer();
        for (int i = 0; i < 6; i++) {
          code.append(new Random().nextInt(10));
        }
        System.out.println("code:" + code);
        //緩存中添加驗(yàn)證碼
        jedis.setex(codeKey, 60 * 2, code.toString());
      } else {
        if (Integer.parseInt(cnt) < 3) {
          //發(fā)送驗(yàn)證碼，假設(shè)生成的驗(yàn)證碼
          StringBuffer code = new StringBuffer();
          for (int i = 0; i < 6; i++) {
            code.append(new Random().nextInt(10));
          }
          System.out.println("code:" + code);
          //緩存中添加驗(yàn)證碼
          jedis.setex(codeKey, 60 * 2, code.toString());
          //遞增手機(jī)發(fā)送數(shù)量
          jedis.incr(countKey);
        } else {
          //返回超出3次，禁止發(fā)送
          System.out.println("超出3次，禁止發(fā)送");
        }
      }
    } catch (Exception e) {
      //這邊其實(shí)是需要回滾下redis
      e.printStackTrace();
    } finally {
      //關(guān)閉redis
      if (jedis != null) {
        jedis.close();
      }
    }
  }

  static void checkValidation(String code, String tel) {
    Jedis jedis = null;
    try {
      jedis = new Jedis("myhost", 6379);
      //密碼
      jedis.auth("mypassword");
      jedis.select(1);
      String codeKey = tel + ":code";
      String validation = jedis.get(codeKey);
      if (validation == null) {
        System.out.println("驗(yàn)證碼未發(fā)送或者失效");
      } else {
        if (validation.equals(code)) {
          System.out.println("驗(yàn)證成功");
        } else {
          System.out.println("驗(yàn)證失敗");
        }
      }
    } catch (Exception e) {
      e.printStackTrace();
    } finally {
      if (jedis != null) {
        jedis.close();
      }
    }
  }
}