rebue.wheel.vertx.web.BlackListHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of wheel-vertx Show documentation

wheel利用vert.x框架进行开发常用的工具库

The newest version!

package rebue.wheel.vertx.web;

import org.apache.commons.lang3.StringUtils;

import io.vertx.core.http.HttpServerRequest;
import io.vertx.ext.web.RoutingContext;
import io.vertx.ext.web.handler.SecurityPolicyHandler;
import io.vertx.redis.client.RedisAPI;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import rebue.wheel.api.dic.HttpStatusCodeDic;
import rebue.wheel.vertx.config.WebProperties;

@Slf4j
public class BlackListHandler implements SecurityPolicyHandler {
    private final RedisAPI redisApi;
    private final String   redisPrefix;

    @SneakyThrows
    public BlackListHandler(WebProperties.BlackListProperties blackListProperties, RedisAPI redisApi) {
        log.info("初始化黑名单处理器");
        if (redisApi == null) {
            throw new IllegalArgumentException("黑名单处理器依赖 Redis，必须保证 Redis 正确加载");
        }
        this.redisApi = redisApi;

        if (StringUtils.isBlank(blackListProperties.getRedisPrefix())) {
            throw new IllegalArgumentException("必须配置 redisPrefix 参数");
        }
        this.redisPrefix = blackListProperties.getRedisPrefix();
    }

    @Override
    public void handle(RoutingContext routingContext) {
        log.debug("进入黑名单处理器");
        HttpServerRequest request = routingContext.request();
        if (!request.isEnded()) {
            log.debug("暂停请求");
            request.pause();
        }

        String srcIp = request.remoteAddress().host();
        this.redisApi.get(this.redisPrefix + srcIp + "{" + srcIp + "}").onSuccess(result -> {
            log.debug("获取黑名单结果：{}", result);
            if (!request.isEnded()) {
                log.debug("恢复请求");
                request.resume();
            }
            if (result == null) {
                routingContext.next();
            } else {
                routingContext.fail(HttpStatusCodeDic.FORBIDDEN.getCode());
            }
        }).onFailure(err -> {
            log.error("Redis获取黑名单出错", err);
            if (!request.isEnded()) {
                log.debug("恢复请求");
                request.resume();
            }
            routingContext.fail(HttpStatusCodeDic.BAD_GATEWAY.getCode());
        });

    }
}