de.malkusch.telgrambot.api.TelegramRateLimitedApi Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of telegrambot Show documentation

Telegram Bot

There is a newer version: 0.0.23

package de.malkusch.telgrambot.api;

import de.malkusch.telgrambot.MessageId;
import de.malkusch.telgrambot.PinnedMessage;
import de.malkusch.telgrambot.TelegramApi;
import de.malkusch.telgrambot.UpdateReceiver;

import java.time.Duration;
import java.util.function.Function;

import static java.util.Objects.requireNonNull;

final class TelegramRateLimitedApi extends AbstractTelegramApiProxy {

    private final RateLimiter groupLimit;
    private final RateLimiter messageLimit;
    private final RateLimiter pinLimit;

    public TelegramRateLimitedApi(TelegramApi api, Timeouts timeouts) {
        super(api);
        requireNonNull(timeouts);

        groupLimit = new RateLimiter(Duration.ofMinutes(1), 19, timeouts.groupThrottle(), "group");
        messageLimit = new RateLimiter(Duration.ofSeconds(1), 29, timeouts.messageThrottle(), "message");
        pinLimit = new RateLimiter(Duration.ofSeconds(1), 2, timeouts.pinThrottle(), "pin");
    }

    @Override
    public void receiveUpdates(UpdateReceiver... receivers) {
        api.receiveUpdates(receivers);
    }

    @Override
    protected  R delegate(Function call) {
        messageLimit.acquire();
        groupLimit.acquire();
        return super.delegate(call);
    }

    @Override
    public void pin(MessageId message) {
        pinLimit.acquire();
        super.pin(message);
    }

    @Override
    public PinnedMessage pinned() {
        pinLimit.acquire();
        return super.pinned();
    }

    @Override
    public void unpin() {
        pinLimit.acquire();
        super.unpin();
    }

    @Override
    public void unpin(MessageId message) {
        pinLimit.acquire();
        super.unpin(message);
    }

    @Override
    public void close() throws Exception {
        try (pinLimit; groupLimit; messageLimit) {
            super.close();
        }
    }
}