com.alibaba.boot.dubbo.ratelimiting.RateLimitingMapRepository Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-boot-starter-dubbo Show documentation

Spring Boot Dubbo Rpc

There is a newer version: 1.5.33

Show newest version

package com.alibaba.boot.dubbo.ratelimiting;

import java.util.Date;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * Repository storing data used by the gateway's rate limiting filter.
 */
public class RateLimitingMapRepository implements RateLimitingRepository {

    private Map store = new ConcurrentHashMap<>();

    public RateLimitingMapRepository() {
    }

    public RateLimitingMapRepository(Map store) {
        this.store = store;
    }

    public void incrementCounter(String id, String timeUnit, Date time) {
        Rate rateLimiting = store.get(id);
        if (rateLimiting == null) {
            rateLimiting = new Rate()
                    .setTime(time)
                    .setTimeUnit(timeUnit);
            store.put(id, rateLimiting);
        }

        rateLimiting.getAtomicLong().incrementAndGet();
    }

    public long getCounter(String id, String timeUnit, Date time) {
        Rate rateLimiting = store.get(id);

        if (rateLimiting == null) {
            rateLimiting = new Rate()
                    .setTime(time)
                    .setTimeUnit(timeUnit);
            store.put(id, rateLimiting);
        }

        if ((time.getTime() - rateLimiting.getTime().getTime()) > toMilliSeconds(timeUnit)) {
            rateLimiting = new Rate()
                    .setTime(time)
                    .setTimeUnit(timeUnit);
            store.put(id, rateLimiting);
        }

        return rateLimiting.getAtomicLong().get();
    }

    private long toMilliSeconds(String timeUnit) {
        if ("hours".equalsIgnoreCase(timeUnit)) {
            return 60 * 60 * 1000;
        } else if ("seconds".equalsIgnoreCase(timeUnit)) {
            return 1000;
        } else if ("days".equalsIgnoreCase(timeUnit)) {
            return 60 * 60 * 1000 * 24;
        } else if ("minutes".equalsIgnoreCase(timeUnit)) {
            return 60 * 1000;
        } else {
            return 0;
        }
    }

}