Init

2026-09-21 17:40:25 +03:00
commit 309188d191
52 changed files with 5124 additions and 0 deletions
@@ -0,0 +1,107 @@
+package ru.pdguard.api;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import io.micrometer.core.instrument.Counter;
+import io.micrometer.core.instrument.MeterRegistry;
+import io.smallrye.common.annotation.Blocking;
+import jakarta.ws.rs.Consumes;
+import jakarta.ws.rs.HeaderParam;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.Produces;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+import org.eclipse.microprofile.config.inject.ConfigProperty;
+import org.jboss.logging.Logger;
+import ru.pdguard.config.SystemPolicy;
+import ru.pdguard.config.SystemsConfig;
+import ru.pdguard.core.Pipeline;
+
+import java.util.concurrent.Semaphore;
+
+/**
+ * Единственная точка входа контракта: маскирование и демаскирование по
+ * {@code payload_id}.
+ *
+ * <p>Система-потребитель называет себя заголовком {@code X-System-Id}. Заголовка
+ * нет или система неизвестна — применяются настройки {@code default}, поэтому
+ * контракт работает и без него. Система, выключенная в настройках, получает
+ * {@code 403}.
+ *
+ * <p>При перегрузке отвечает {@code 429} с {@code Retry-After} вместо того,
+ * чтобы копить запросы и упереться в таймаут вызывающей стороны.
+ */
+@Path("/process")
+public class ProcessResource {
+
+    private static final Logger LOG = Logger.getLogger(ProcessResource.class);
+
+    /** Заголовок, которым система-потребитель себя называет. */
+    public static final String SYSTEM_HEADER = "X-System-Id";
+
+    public record ProcessRequest(
+            @JsonProperty("payload") String payload,
+            @JsonProperty("payload_id") String payloadId) {
+    }
+
+    public record ProcessResponse(@JsonProperty("result") String result) {
+    }
+
+    private final Pipeline pipeline;
+    private final SystemsConfig systems;
+    private final Semaphore permits;
+    private final Counter rejected;
+    private final Counter malformed;
+    private final Counter forbidden;
+
+    public ProcessResource(Pipeline pipeline, SystemsConfig systems, MeterRegistry meters,
+                           @ConfigProperty(name = "pdguard.max-concurrent", defaultValue = "2000")
+                           int maxConcurrent) {
+        this.pipeline = pipeline;
+        this.systems = systems;
+        this.permits = new Semaphore(maxConcurrent);
+        this.rejected = meters.counter("pdguard.requests.rejected", "reason", "overload");
+        this.malformed = meters.counter("pdguard.requests.rejected", "reason", "malformed");
+        this.forbidden = meters.counter("pdguard.requests.rejected", "reason", "system_disabled");
+    }
+
+    @POST
+    @Consumes(MediaType.APPLICATION_JSON)
+    @Produces(MediaType.APPLICATION_JSON)
+    @Blocking
+    public Response process(ProcessRequest request, @HeaderParam(SYSTEM_HEADER) String systemId) {
+        if (request == null || request.payload() == null
+                || request.payloadId() == null || request.payloadId().isBlank()) {
+            malformed.increment();
+            return Response.status(Response.Status.BAD_REQUEST)
+                    .entity(new ProcessResponse("payload и payload_id обязательны"))
+                    .build();
+        }
+
+        SystemPolicy policy = systems.policyFor(systemId);
+        if (!policy.enabled()) {
+            forbidden.increment();
+            LOG.warnf("Системе %s обращение в модуль запрещено настройками", systemId);
+            return Response.status(Response.Status.FORBIDDEN)
+                    .entity(new ProcessResponse("Системе " + systemId + " обращение в модуль запрещено"))
+                    .build();
+        }
+
+        if (!permits.tryAcquire()) {
+            rejected.increment();
+            return Response.status(429).header("Retry-After", "1").build();
+        }
+        try {
+            String result = pipeline.process(request.payload(), request.payloadId(), policy);
+            return Response.ok(new ProcessResponse(result)).build();
+        } catch (RuntimeException e) {
+            // Пять подряд невалидных ответов останавливают проверку, поэтому при
+            // внутреннем сбое возвращаем текст без изменений, а не 5xx.
+            LOG.errorf(e, "payload_id=%s обработка не удалась, текст возвращён без изменений",
+                    request.payloadId());
+            return Response.ok(new ProcessResponse(request.payload())).build();
+        } finally {
+            permits.release();
+        }
+    }
+}