io.quarkus.load.shedding.runtime.HttpLoadShedding Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quarkus-load-shedding Show documentation

Shed excess load and keep your service available

There is a newer version: 3.17.0

Show newest version

package io.quarkus.load.shedding.runtime;

import jakarta.annotation.Priority;
import jakarta.enterprise.event.Observes;
import jakarta.inject.Singleton;

import io.netty.handler.codec.http.HttpHeaderNames;
import io.netty.handler.codec.http.HttpResponseStatus;
import io.vertx.core.AsyncResult;
import io.vertx.core.Handler;
import io.vertx.core.http.HttpServerResponse;
import io.vertx.ext.web.Router;

@Singleton
public class HttpLoadShedding {
    public void init(@Observes @Priority(-1_000_000_000) Router router, OverloadDetector detector,
            PriorityLoadShedding priority, LoadSheddingRuntimeConfig config) {

        if (!config.enabled()) {
            return;
        }

        router.route().order(-1_000_000_000).handler(ctx -> {
            if (detector.isOverloaded() && priority.shedLoad(ctx.request())) {
                HttpServerResponse response = ctx.response();
                response.setStatusCode(HttpResponseStatus.SERVICE_UNAVAILABLE.code());
                response.headers().add(HttpHeaderNames.CONNECTION, "close");
                response.endHandler(new Handler() {
                    @Override
                    public void handle(Void ignored) {
                        ctx.request().connection().close();
                    }
                });
                response.end();
            } else {
                detector.requestBegin();
                long start = System.nanoTime();
                ctx.addEndHandler(new Handler>() {
                    @Override
                    public void handle(AsyncResult ignored) {
                        long end = System.nanoTime();
                        detector.requestEnd((end - start) / 1_000);
                    }
                });
                ctx.next();
            }
        });
    }
}