com.jetdrone.vertx.yoke.middleware.TooBusy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of yoke.middleware.toobusy Show documentation

There is a newer version: 3.0.0

Show newest version

package com.jetdrone.vertx.yoke.middleware;

import com.jetdrone.vertx.yoke.Middleware;
import com.jetdrone.vertx.yoke.Yoke;
import org.jetbrains.annotations.NotNull;
import org.vertx.java.core.Handler;
import org.vertx.java.core.Vertx;

public final class TooBusy extends Middleware {

    private final long highWaterMark;
    private final String message;

    private long timerID;
    private long t0;
    private long dt;

    public TooBusy() {
        this(70);
    }

    public TooBusy(long highWaterMark) {
        this(highWaterMark, "Server is too busy. Please, try again later.");
    }

    public TooBusy(long highWaterMark, String message) {
        this.highWaterMark = highWaterMark * 1000000;
        this.message = message;
    }

    @Override
    public Middleware init(@NotNull final Yoke yoke, @NotNull final String mount) {
        super.init(yoke, mount);

        t0 = System.nanoTime();
        dt = 500 * 1000000; // nano time

        timerID = yoke.vertx().setPeriodic(500, new Handler() {
            @Override
            public void handle(Long timerID) {
                final long t1 = System.nanoTime();
                dt = t1 - t0;
                t0 = t1;
            }
        });

        return this;
    }

    public long getLag() {
        return dt - 500 * 1000000;
    }

    public void shutdown() {
        yoke.vertx().cancelTimer(timerID);
    }

    @Override
    public void handle(@NotNull YokeRequest request, @NotNull Handler