Bmoric/extract scheduler api

airbyte-server/src/main/java/io/airbyte/server/ServerFactory.java

@@ -28,6 +28,7 @@
 import io.airbyte.server.apis.NotificationsApiController;
 import io.airbyte.server.apis.OpenapiApiController;
 import io.airbyte.server.apis.OperationApiController;
+import io.airbyte.server.apis.SchedulerApiController;
 import io.airbyte.server.apis.binders.AttemptApiBinder;
 import io.airbyte.server.apis.binders.ConnectionApiBinder;
 import io.airbyte.server.apis.binders.DbMigrationBinder;
@@ -41,6 +42,7 @@
 import io.airbyte.server.apis.binders.NotificationApiBinder;
 import io.airbyte.server.apis.binders.OpenapiApiBinder;
 import io.airbyte.server.apis.binders.OperationApiBinder;
+import io.airbyte.server.apis.binders.SchedulerApiBinder;
 import io.airbyte.server.apis.binders.SourceOauthApiBinder;
 import io.airbyte.server.apis.factories.AttemptApiFactory;
 import io.airbyte.server.apis.factories.ConnectionApiFactory;
@@ -55,6 +57,7 @@
 import io.airbyte.server.apis.factories.NotificationsApiFactory;
 import io.airbyte.server.apis.factories.OpenapiApiFactory;
 import io.airbyte.server.apis.factories.OperationApiFactory;
+import io.airbyte.server.apis.factories.SchedulerApiFactory;
 import io.airbyte.server.apis.factories.SourceOauthApiFactory;
 import io.airbyte.server.handlers.AttemptHandler;
 import io.airbyte.server.handlers.ConnectionsHandler;
@@ -197,6 +200,8 @@ public ServerRunnable create(final SynchronousSchedulerClient synchronousSchedul
 
       OpenapiApiFactory.setValues(openApiConfigHandler);
 
+      SchedulerApiFactory.setValues(schedulerHandler);
+
       // server configurations
       final Set<Class<?>> componentClasses = Set.of(
           ConfigurationApi.class,
@@ -213,6 +218,7 @@ public ServerRunnable create(final SynchronousSchedulerClient synchronousSchedul
           NotificationsApiController.class,
           OpenapiApiController.class,
           OperationApiController.class,
+          SchedulerApiController.class,
           SourceOauthApiFactory.class);
 
       final Set<Object> components = Set.of(
@@ -231,6 +237,7 @@ public ServerRunnable create(final SynchronousSchedulerClient synchronousSchedul
           new NotificationApiBinder(),
           new OpenapiApiBinder(),
           new OperationApiBinder(),
+          new SchedulerApiBinder(),
           new SourceOauthApiBinder());
 
       // construct server

airbyte-server/src/main/java/io/airbyte/server/apis/ConfigurationApi.java

@@ -900,19 +900,31 @@ public ConnectionState getState(final ConnectionIdRequestBody connectionIdReques
   }
 
   // SCHEDULER
+  /**
+   * This implementation has been moved to {@link SchedulerApiController}. Since the path of
+   * {@link SchedulerApiController} is more granular, it will override this implementation
+   */
   @Override
   public CheckConnectionRead executeSourceCheckConnection(final SourceCoreConfig sourceConfig) {
-    return execute(() -> schedulerHandler.checkSourceConnectionFromSourceCreate(sourceConfig));
+    throw new NotImplementedException();
   }
 
+  /**
+   * This implementation has been moved to {@link SchedulerApiController}. Since the path of
+   * {@link SchedulerApiController} is more granular, it will override this implementation
+   */
   @Override
   public CheckConnectionRead executeDestinationCheckConnection(final DestinationCoreConfig destinationConfig) {
-    return execute(() -> schedulerHandler.checkDestinationConnectionFromDestinationCreate(destinationConfig));
+    throw new NotImplementedException();
   }
 
+  /**
+   * This implementation has been moved to {@link SchedulerApiController}. Since the path of
+   * {@link SchedulerApiController} is more granular, it will override this implementation
+   */
   @Override
   public SourceDiscoverSchemaRead executeSourceDiscoverSchema(final SourceCoreConfig sourceCreate) {
-    return execute(() -> schedulerHandler.discoverSchemaForSourceFromSourceCreate(sourceCreate));
+    throw new NotImplementedException();
   }
 
   /**

airbyte-server/src/main/java/io/airbyte/server/apis/SchedulerApiController.java

@@ -0,0 +1,35 @@
+/*
+ * Copyright (c) 2022 Airbyte, Inc., all rights reserved.
+ */
+
+package io.airbyte.server.apis;
+
+import io.airbyte.api.generated.SchedulerApi;
+import io.airbyte.api.model.generated.CheckConnectionRead;
+import io.airbyte.api.model.generated.DestinationCoreConfig;
+import io.airbyte.api.model.generated.SourceCoreConfig;
+import io.airbyte.api.model.generated.SourceDiscoverSchemaRead;
+import io.airbyte.server.handlers.SchedulerHandler;
+import lombok.AllArgsConstructor;
+
+@AllArgsConstructor
+public class SchedulerApiController implements SchedulerApi {
+
+  private final SchedulerHandler schedulerHandler;
+
+  @Override
+  public CheckConnectionRead executeDestinationCheckConnection(final DestinationCoreConfig destinationCoreConfig) {
+    return ConfigurationApi.execute(() -> schedulerHandler.checkDestinationConnectionFromDestinationCreate(destinationCoreConfig));
+  }
+
+  @Override
+  public CheckConnectionRead executeSourceCheckConnection(final SourceCoreConfig sourceCoreConfig) {
+    return ConfigurationApi.execute(() -> schedulerHandler.checkSourceConnectionFromSourceCreate(sourceCoreConfig));
+  }
+
+  @Override
+  public SourceDiscoverSchemaRead executeSourceDiscoverSchema(final SourceCoreConfig sourceCoreConfig) {
+    return ConfigurationApi.execute(() -> schedulerHandler.discoverSchemaForSourceFromSourceCreate(sourceCoreConfig));
+  }
+
+}

airbyte-server/src/main/java/io/airbyte/server/apis/binders/SchedulerApiBinder.java

@@ -0,0 +1,21 @@
+/*
+ * Copyright (c) 2022 Airbyte, Inc., all rights reserved.
+ */
+
+package io.airbyte.server.apis.binders;
+
+import io.airbyte.server.apis.SchedulerApiController;
+import io.airbyte.server.apis.factories.SchedulerApiFactory;
+import org.glassfish.hk2.utilities.binding.AbstractBinder;
+import org.glassfish.jersey.process.internal.RequestScoped;
+
+public class SchedulerApiBinder extends AbstractBinder {
+
+  @Override
+  protected void configure() {
+    bindFactory(SchedulerApiFactory.class)
+        .to(SchedulerApiController.class)
+        .in(RequestScoped.class);
+  }
+
+}

airbyte-server/src/main/java/io/airbyte/server/apis/factories/SchedulerApiFactory.java

@@ -0,0 +1,29 @@
+/*
+ * Copyright (c) 2022 Airbyte, Inc., all rights reserved.
+ */
+
+package io.airbyte.server.apis.factories;
+
+import io.airbyte.server.apis.SchedulerApiController;
+import io.airbyte.server.handlers.SchedulerHandler;
+import org.glassfish.hk2.api.Factory;
+
+public class SchedulerApiFactory implements Factory<SchedulerApiController> {
+
+  private static SchedulerHandler schedulerHandler;
+
+  public static void setValues(final SchedulerHandler schedulerHandler) {
+    SchedulerApiFactory.schedulerHandler = schedulerHandler;
+  }
+
+  @Override
+  public SchedulerApiController provide() {
+    return new SchedulerApiController(SchedulerApiFactory.schedulerHandler);
+  }
+
+  @Override
+  public void dispose(final SchedulerApiController instance) {
+    /* no op */
+  }
+
+}