hamfisted cleanup

Author: yaauie

libs/logstash-bridge/src/main/java/org/elasticsearch/logstashbridge/core/CheckedBiFunctionBridge.java

@@ -0,0 +1,22 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the "Elastic License
+ * 2.0", the "GNU Affero General Public License v3.0 only", and the "Server Side
+ * Public License v 1"; you may not use this file except in compliance with, at
+ * your election, the "Elastic License 2.0", the "GNU Affero General Public
+ * License v3.0 only", or the "Server Side Public License, v 1".
+ */
+
+package org.elasticsearch.logstashbridge.core;
+
+import org.elasticsearch.common.CheckedBiFunction;
+import org.elasticsearch.logstashbridge.StableBridgeAPI;
+
+public interface CheckedBiFunctionBridge<T, U, R, E extends Exception> extends StableBridgeAPI<CheckedBiFunction<T, U, R, E>> {
+    R apply(T t, U u) throws E;
+
+    @Override
+    default CheckedBiFunction<T, U, R, E> toInternal() {
+        return this::apply;
+    }
+}

libs/logstash-bridge/src/main/java/org/elasticsearch/logstashbridge/geoip/GeoIpProcessorBridge.java

@@ -9,18 +9,16 @@
 package org.elasticsearch.logstashbridge.geoip;
 
 import org.elasticsearch.ingest.geoip.GeoIpProcessor;
-import org.elasticsearch.ingest.geoip.IpDatabaseProvider;
-import org.elasticsearch.logstashbridge.StableBridgeAPI;
+import org.elasticsearch.logstashbridge.ingest.ProcessorBridge;
 
 /**
  * An external bridge for {@link GeoIpProcessor}
  */
 public interface GeoIpProcessorBridge {
 
-    class Factory extends StableBridgeAPI.ProxyInternal<GeoIpProcessor.Factory> {
-
-        public Factory(String type, IpDatabaseProvider ipDatabaseProvider) {
-            super(new GeoIpProcessor.Factory(type, ipDatabaseProvider));
+    class Factory extends ProcessorBridge.Factory.ProxyInternal {
+        public Factory(final String type, final IpDatabaseProviderBridge ipDatabaseProviderBridge) {
+            super(new GeoIpProcessor.Factory(type, ipDatabaseProviderBridge.toInternal()));
         }
     }
 }

libs/logstash-bridge/src/main/java/org/elasticsearch/logstashbridge/geoip/IpDatabaseBridge.java

@@ -10,11 +10,11 @@
 
 import com.maxmind.db.Reader;
 
-import org.elasticsearch.ExceptionsHelper;
 import org.elasticsearch.common.CheckedBiFunction;
 import org.elasticsearch.core.Nullable;
 import org.elasticsearch.ingest.geoip.IpDatabase;
 import org.elasticsearch.logstashbridge.StableBridgeAPI;
+import org.elasticsearch.logstashbridge.core.CheckedBiFunctionBridge;
 
 import java.io.IOException;
 
@@ -25,26 +25,11 @@ public interface IpDatabaseBridge extends StableBridgeAPI<IpDatabase> {
 
     String getDatabaseType() throws IOException;
 
-    MaxMindDbBridge.Reader getDatabaseReader() throws IOException;
-
     @Nullable
-    default <RESPONSE> RESPONSE getResponse(String ipAddress, CheckedBiFunction<Reader, String, RESPONSE, Exception> responseProvider) {
-        try {
-            return responseProvider.apply(this.getDatabaseReader().toInternal(), ipAddress);
-        } catch (Exception e) {
-            throw ExceptionsHelper.convertToRuntime(e);
-        }
-    }
+    <RESPONSE> RESPONSE getResponse(String ipAddress, CheckedBiFunctionBridge<Reader, String, RESPONSE, Exception> responseProvider);
 
     void close() throws IOException;
 
-    static IpDatabaseBridge fromInternal(final IpDatabase internalDatabase) {
-        if (internalDatabase instanceof AbstractExternal.ProxyExternal externalProxy) {
-            return externalProxy.getIpDatabaseBridge();
-        }
-        return new ProxyInternal(internalDatabase);
-    }
-
     /**
      * The {@code IpDatabaseBridge.AbstractExternal} is an abstract base class for implementing
      * the {@link IpDatabaseBridge} externally to the Elasticsearch code-base. It takes care of
@@ -64,10 +49,6 @@ public IpDatabase toInternal() {
 
         private class ProxyExternal implements IpDatabase {
 
-            private AbstractExternal getIpDatabaseBridge() {
-                return AbstractExternal.this;
-            }
-
             @Override
             public String getDatabaseType() throws IOException {
                 return AbstractExternal.this.getDatabaseType();
@@ -78,7 +59,7 @@ public <RESPONSE> RESPONSE getResponse(
                 String ipAddress,
                 CheckedBiFunction<Reader, String, RESPONSE, Exception> responseProvider
             ) {
-                return AbstractExternal.this.getResponse(ipAddress, responseProvider);
+                return AbstractExternal.this.getResponse(ipAddress, responseProvider::apply);
             }
 
             @Override
@@ -87,34 +68,4 @@ public void close() throws IOException {
             }
         }
     }
-
-    /**
-     * An implementation of {@link IpDatabaseBridge} that proxies to an internal {@link IpDatabase}
-     */
-    class ProxyInternal extends StableBridgeAPI.ProxyInternal<IpDatabase> implements IpDatabaseBridge {
-
-        public ProxyInternal(final IpDatabase delegate) {
-            super(delegate);
-        }
-
-        @Override
-        public String getDatabaseType() throws IOException {
-            return toInternal().getDatabaseType();
-        }
-
-        @Override
-        public MaxMindDbBridge.Reader getDatabaseReader() throws IOException {
-            return null;
-        }
-
-        @Override
-        public <RESPONSE> RESPONSE getResponse(String ipAddress, CheckedBiFunction<Reader, String, RESPONSE, Exception> responseProvider) {
-            return toInternal().getResponse(ipAddress, responseProvider);
-        }
-
-        @Override
-        public void close() throws IOException {
-            toInternal().close();
-        }
-    }
 }

libs/logstash-bridge/src/main/java/org/elasticsearch/logstashbridge/geoip/IpDatabaseProviderBridge.java

@@ -14,7 +14,7 @@
 import org.elasticsearch.ingest.geoip.IpDatabaseProvider;
 import org.elasticsearch.logstashbridge.StableBridgeAPI;
 
-import java.util.Objects;
+import static org.elasticsearch.logstashbridge.StableBridgeAPI.toInternalNullable;
 
 /**
  * An external bridge for {@link Processor}
@@ -25,13 +25,6 @@ public interface IpDatabaseProviderBridge extends StableBridgeAPI<IpDatabaseProv
 
     IpDatabaseBridge getDatabase(String name);
 
-    static IpDatabaseProviderBridge fromInternal(final IpDatabaseProvider internalProvider) {
-        if (internalProvider instanceof IpDatabaseProviderBridge.AbstractExternal.ProxyExternal externalProxy) {
-            return externalProxy.getIpDatabaseProviderBridge();
-        }
-        return new IpDatabaseProviderBridge.ProxyInternal(internalProvider);
-    }
-
     /**
      * The {@code IpDatabaseProviderBridge.AbstractExternal} is an abstract base class for implementing
      * the {@link IpDatabaseProviderBridge} externally to the Elasticsearch code-base. It takes care of
@@ -56,34 +49,13 @@ private AbstractExternal getIpDatabaseProviderBridge() {
 
             @Override
             public Boolean isValid(ProjectId projectId, String name) {
-                return IpDatabaseProviderBridge.AbstractExternal.this.isValid(name);
+                return AbstractExternal.this.isValid(name);
             }
 
             @Override
             public IpDatabase getDatabase(ProjectId projectId, String name) {
-                IpDatabaseBridge bridge = IpDatabaseProviderBridge.AbstractExternal.this.getDatabase(name);
-                return Objects.isNull(bridge) ? null : bridge.toInternal();
+                return toInternalNullable(AbstractExternal.this.getDatabase(name));
             }
         }
     }
-
-    /**
-     * An implementation of {@link IpDatabaseProviderBridge} that proxies to an internal {@link IpDatabaseProvider}
-     */
-    class ProxyInternal extends StableBridgeAPI.ProxyInternal<IpDatabaseProvider> implements IpDatabaseProviderBridge {
-        public ProxyInternal(final IpDatabaseProvider delegate) {
-            super(delegate);
-        }
-
-        @Override
-        public Boolean isValid(String name) {
-            return toInternal().isValid(ProjectId.DEFAULT, name);
-        }
-
-        @Override
-        public IpDatabaseBridge getDatabase(String name) {
-            IpDatabase ipDatabase = toInternal().getDatabase(ProjectId.DEFAULT, name);
-            return new IpDatabaseBridge.ProxyInternal(ipDatabase);
-        }
-    }
 }

libs/logstash-bridge/src/main/java/org/elasticsearch/logstashbridge/geoip/MaxMindDbBridge.java

@@ -7,7 +7,7 @@
  * License v3.0 only", or the "Server Side Public License, v 1".
  */
 
-package org.elasticsearch.logstashbridge.core;
+package org.elasticsearch.logstashbridge.ingest;
 
 import org.elasticsearch.ingest.common.FailProcessorException;
 import org.elasticsearch.logstashbridge.StableBridgeAPI;

libs/logstash-bridge/src/main/java/org/elasticsearch/logstashbridge/core/FailProcessorExceptionBridge.java renamed to libs/logstash-bridge/src/main/java/org/elasticsearch/logstashbridge/ingest/FailProcessorExceptionBridge.java

@@ -204,7 +204,7 @@ default Processor.Factory toInternal() {
          * An implementation of {@link ProcessorBridge.Factory} that proxies to an internal {@link Processor.Factory}
          */
         class ProxyInternal extends StableBridgeAPI.ProxyInternal<Processor.Factory> implements Factory {
-            private ProxyInternal(final Processor.Factory delegate) {
+            protected ProxyInternal(final Processor.Factory delegate) {
                 super(delegate);
             }