提交外包代码

puzhibing

2023-10-08 22199bbdda579861736420fe26c2873ab0f5d21c

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
package com.sinata.rest.modular.auth.util;
 
import com.alibaba.fastjson.JSON;
import com.sinata.common.enums.EnumRedisKey;
import com.sinata.common.model.AccountCheckNumVo;
import com.sinata.rest.core.util.DateUtils;
import com.sinata.rest.modular.system.service.RedisTemplateService;
import org.springframework.stereotype.Component;
import org.springframework.stereotype.Service;
 
import javax.annotation.Resource;
import java.util.Date;
 
/**
 * 验证密码/验证码次数限制
 */
@Service
@Component
public class AccountCheckUtil {
 
    @Resource
    public RedisTemplateService redisTemplateService;
 
    /**
     * 账号验证码-验证次数
     *
     * @param account 账号
     * @return vo为null则超次数逻辑终止，不为空逻辑继续
     */
    public AccountCheckNumVo accountCheckCodeNum(String account) {
        // 获取账户验证次数
        Object chMap = redisTemplateService.hashGet(EnumRedisKey.MAP_ACCOUNT_CHECK.index, account);
        String today = DateUtils.formatDate(new Date(), "yyyy-MM-dd");
        AccountCheckNumVo ch;
        if (chMap != null) {
            ch = JSON.parseObject(chMap.toString(), AccountCheckNumVo.class);
            if (today.equals(ch.getCheckCodeDate())) {
                ch.setCheckCodeNum(ch.getCheckCodeNum() + 1);
            } else {
                ch.setCheckCodeNum(1);
            }
        } else {
            ch = new AccountCheckNumVo();
            ch.setAccount(account);
        }
        ch.setCheckCodeDate(today);
 
        // 判断验证次数
        if (ch.getTotalCodeNum() < ch.getCheckCodeNum()) {
            redisTemplateService.hashPushHashMap(EnumRedisKey.MAP_ACCOUNT_CHECK.index, account, JSON.toJSONString(ch));
            return null;
        } else {
            redisTemplateService.hashPushHashMap(EnumRedisKey.MAP_ACCOUNT_CHECK.index, account, JSON.toJSONString(ch));
        }
        return ch;
    }
 
    public void resetAccountCheckCodeNum(AccountCheckNumVo vo) {
        // 重置-账户验证次数
        vo.setCheckCodeNum(1);
        redisTemplateService.hashPushHashMap(EnumRedisKey.MAP_ACCOUNT_CHECK.index, vo.getAccount(), JSON.toJSONString(vo));
    }
 
    /**
     * 账号密码-验证次数
     */
    public AccountCheckNumVo accountCheckPwdNum(String account) {
        // 获取账户验证次数
        Object chMap = redisTemplateService.hashGet(EnumRedisKey.MAP_ACCOUNT_CHECK.index, account);
        String today = DateUtils.formatDate(new Date(), "yyyy-MM-dd");
        AccountCheckNumVo ch;
        if (chMap != null) {
            ch = JSON.parseObject(chMap.toString(), AccountCheckNumVo.class);
            if (today.equals(ch.getCheckPwdDate())) {
                ch.setCheckPwdNum(ch.getCheckPwdNum() + 1);
            } else {
                ch.setCheckPwdNum(1);
            }
        } else {
            ch = new AccountCheckNumVo();
            ch.setAccount(account);
        }
        ch.setCheckPwdDate(today);
 
        if (ch.getTotalPwdNum() < ch.getCheckPwdNum()) {
            redisTemplateService.hashPushHashMap(EnumRedisKey.MAP_ACCOUNT_CHECK.index, account, JSON.toJSONString(ch));
            return null;
        } else {
            redisTemplateService.hashPushHashMap(EnumRedisKey.MAP_ACCOUNT_CHECK.index, account, JSON.toJSONString(ch));
        }
 
        return ch;
    }
 
    public void resetAccountCheckPwdNum(AccountCheckNumVo vo) {
        //重置-账户验证次数
        vo.setCheckPwdNum(1);
        redisTemplateService.hashPushHashMap(EnumRedisKey.MAP_ACCOUNT_CHECK.index, vo.getAccount(), JSON.toJSONString(vo));
    }
 
}