com.github.edgar615.util.vertx.redis.ratelimit.FixedWindowRateLimit Maven / Gradle / Ivy

Go to download

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

The newest version!

package com.github.edgar615.util.vertx.redis.ratelimit;

import com.github.edgar615.util.vertx.redis.AbstractLuaEvaluator;
import io.vertx.core.AsyncResult;
import io.vertx.core.Future;
import io.vertx.core.Handler;
import io.vertx.core.Vertx;
import io.vertx.redis.RedisClient;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.time.Instant;
import java.util.ArrayList;
import java.util.List;

/**
 * Created by edgar on 17-5-29.
 */
public class FixedWindowRateLimit extends AbstractLuaEvaluator {
  private static final Logger LOGGER = LoggerFactory.getLogger(FixedWindowRateLimit.class);


  public FixedWindowRateLimit(Vertx vertx, RedisClient redisClient, Future completed) {
    super(vertx, redisClient, "com/github/edgar615/util/redis/lua/fixed_window_ratelimit.lua", completed);
  }

  /**
   * 限流
   * @param rateLimit 限流设置
   * @param handler　回调
   */
  public void rateLimit(FixedWindowRateLimitRule rateLimit, Handler> handler) {
    List keys = new ArrayList<>();
    List args = new ArrayList<>();
    args.add(rateLimit.getSubject());
    args.add(rateLimit.getLimit() + "");
    args.add(rateLimit.getInterval() + "");
    args.add(Instant.now().getEpochSecond() + "");
    evaluate(keys, args, ar -> {
      if (ar.failed()) {
        LOGGER.error("rateLimit failed", ar.cause());
        handler.handle(Future.failedFuture("rateLimit failed"));;
        return;
      }
      List subjects = new ArrayList<>();
      subjects.add(rateLimit.getSubject());
      RateLimitUtils.createResult(ar.result(), subjects, handler);
    });
  }

}