美文网首页web
短信限制

短信限制

作者: wangxiaoda | 来源:发表于2017-02-19 17:05 被阅读20次

    获取用户真实IP地址

    import javax.servlet.http.HttpServletRequest;
    import java.net.InetAddress;
    import java.net.UnknownHostException;
    
    public class IPUtil {
    
        /**
         * 获取请求的ip
         *
         * @return
         */
        public static String getIp(HttpServletRequest request) {
    
            String ipAddress = null;
    
            ipAddress = request.getHeader("x-forwarded-for");
            if (ipAddress == null || ipAddress.length() == 0
                    || "unknown".equalsIgnoreCase(ipAddress)) {
                ipAddress = request.getHeader("Proxy-Client-IP");
            }
            if (ipAddress == null || ipAddress.length() == 0
                    || "unknown".equalsIgnoreCase(ipAddress)) {
                ipAddress = request.getHeader("WL-Proxy-Client-IP");
            }
            if (ipAddress == null || ipAddress.length() == 0
                    || "unknown".equalsIgnoreCase(ipAddress)) {
                ipAddress = request.getRemoteAddr();
                if (ipAddress.equals("127.0.0.1")) {
                    // 根据网卡取本机配置的IP
                    InetAddress inet = null;
                    try {
                        inet = InetAddress.getLocalHost();
                    } catch (UnknownHostException e) {
                        e.printStackTrace();
                    }
                    ipAddress = inet.getHostAddress();
                }
    
            }
            //对于通过多个代理的情况,第一个IP为客户端真实IP,多个IP按照','分割
            if(ipAddress!=null && ipAddress.length()>15){ //"***.***.***.***".length() = 15
                if(ipAddress.indexOf(",")>0){
                    ipAddress = ipAddress.substring(0,ipAddress.indexOf(","));
                }
            }
            return ipAddress;
        }
    }
    
    

    获取配置文件属性值

    //application中IP数量限制
    @Value("${phone.msg.ip.limit}")
    private String ipLimitNum;
    
    //application中系统数量限制
    @Value("${phone.msg.sys.limit}")
    private String sysLimitNum;
    

    redis目录设置

    //IP地址次数限制key(1自然天10次)
    private static final String USER_PHONEMSG_IP_LIMIT = "USER_PHONEMSG_IP_LIMIT" + ":";
    
    //系统次数限制key(1小时内50条)
    private static final String USER_PHONEMSG_SYSTEM_LIMIT = "USER_PHONEMSG_SYSTEM_LIMIT" + ":";
    

    判断当前IP是否达到发送限制

    /**
         * 判断当前IP是否达到发送限制
         *
         * @param ip
         * @return
         */
        public boolean checkIPLimit(String ip) {
            Jedis jedis = jedisPool.getResource();
            Calendar c = Calendar.getInstance();
            //获取当前几号
            int datenum = c.get(Calendar.DATE);
            //今天是否有发送记录
            String value = jedis.get(USER_PHONEMSG_IP_LIMIT + ip + ":" + datenum);
            jedis.close();
            if (StringUtils.isNotBlank(value)) {
                int num = Integer.valueOf(value);
                return num < Integer.valueOf(ipLimitNum) ? true : false;
            } else {
                return true;
            }
        }
    

    判断当前系统是否达到发送限制

    /**
         * 判断当前系统是否达到发送限制
         *
         * @return
         */
        public boolean checkSysLimit() {
            Jedis jedis = jedisPool.getResource();
            Calendar c = Calendar.getInstance();
            //获取当前是几时
            int hournum = c.get(Calendar.HOUR_OF_DAY);
            //当前小时是否有发送记录
            String value = jedis.get(USER_PHONEMSG_SYSTEM_LIMIT + hournum);
            jedis.close();
            if (StringUtils.isNotBlank(value)) {
                int num = Integer.valueOf(value);
                return num <  Integer.valueOf(sysLimitNum) ? true : false;
            } else {
                return true;
            }
        }
    

    当前IP发送短信数加1

    /**
         * 当前IP发送短信数加1
         *
         * @param ip
         * @return
         */
        public void setIPLimit(String ip) {
            Jedis jedis = jedisPool.getResource();
            Calendar c = Calendar.getInstance();
            //获取当前几号
            int datenum = c.get(Calendar.DATE);
            String key = USER_PHONEMSG_IP_LIMIT + ip + ":" + datenum;
            //今天是否有发送记录
            String value = jedis.get(key);
            //更新发送记录
            if (StringUtils.isNotBlank(value) /*&& Integer.valueOf(value) <  Integer.valueOf(ipLimitNum)*/) {//不需要判断是否超数
                int num = Integer.valueOf(value);
                //计数加1
                num += 1;
                //获取上次所剩时间
                int time = Integer.parseInt(String.valueOf(jedis.ttl(key)));
                //此次赋值会使生命周期丢失
                jedis.set(key, String.valueOf(num));
                //延续上次的生命周期
                jedis.expire(key, time);
            } else {
                jedis.set(key, String.valueOf(1));
                //生命周期为60*60*24秒,24小时为间隔,保证业务及释放
                jedis.expire(key, 60 * 60 * 24);
            }
            jedis.close();
        }
    

    判断当前系统是否达到发送限制

    /**
         * 判断当前系统是否达到发送限制
         *
         * @return
         */
        public void setSysLimit() {
            Jedis jedis = jedisPool.getResource();
            Calendar c = Calendar.getInstance();
            //获取当前时刻
            int hournum = c.get(Calendar.HOUR_OF_DAY);
            String key = USER_PHONEMSG_SYSTEM_LIMIT + hournum;
            //当前小时是否有发送记录
            String value = jedis.get(key);
            //更新发送记录
            if (StringUtils.isNotBlank(value) /*&& Integer.valueOf(value) <  Integer.valueOf(sysLimitNum)*/) {//不需要判断是否超数
                int num = Integer.valueOf(value);
                //计数加1
                num += 1;
                //获取上次所剩时间
                int time = Integer.parseInt(String.valueOf(jedis.ttl(key)));
                //此次赋值会使生命周期丢失
                jedis.set(key, String.valueOf(num));
                //延续上次的生命周期
                jedis.expire(key, time);
            } else {
                jedis.set(key, String.valueOf(1));
                //生命周期为60*60秒,最小1小时为间隔,保证业务及释放
                jedis.expire(key, 60 * 60);
            }
            jedis.close();
        }
    

    发送短信方法中调用

    //判断是否达到该时间段短信发送限制(系统、IP限制)
    if(!(redisService.checkIPLimit(ip) && redisService.checkSysLimit())){
            responseEntity.setting(CodeEnum.PHONE_LIMIT);
            RequestContext.setResponseEntity(responseEntity);
            return;
    }
    redisService.setIPLimit(ip);
    redisService.setSysLimit();
    

    相关文章

      网友评论

        本文标题:短信限制

        本文链接:https://www.haomeiwen.com/subject/hmyjwttx.html