[5889] Bmoric/add application name [2/3]

airbyte-workers/src/main/java/io/airbyte/workers/Application.java

@@ -0,0 +1,15 @@
+/*
+ * Copyright (c) 2021 Airbyte, Inc., all rights reserved.
+ */
+
+package io.airbyte.workers;
+
+public interface Application {
+
+  default String getApplicationName() {
+    // This value should only be used in the U-Test, it is an empty string instead of airbyte-test in
+    // order to avoid displaying airbyte-test in prod
+    return "";
+  }
+
+}

airbyte-workers/src/main/java/io/airbyte/workers/DbtTransformationRunner.java

@@ -48,7 +48,7 @@ public void start() throws Exception {
    * transform-config scripts (to translate Airbyte Catalogs into Dbt profiles file). Thus, we depend
    * on the NormalizationRunner to configure the dbt project with the appropriate destination settings
    * and pull the custom git repository into the workspace.
-   *
+   * <p>
    * Once the workspace folder/files is setup to run, we invoke the custom transformation command as
    * provided by the user to execute whatever extra transformation has been implemented.
    */

airbyte-workers/src/main/java/io/airbyte/workers/DbtTransformationWorker.java

@@ -13,7 +13,7 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-public class DbtTransformationWorker implements Worker<OperatorDbtInput, Void> {
+public class DbtTransformationWorker implements Worker<OperatorDbtInput, Void>, Application {
 
   private static final Logger LOGGER = LoggerFactory.getLogger(DbtTransformationWorker.class);
 
@@ -77,4 +77,9 @@ public void cancel() {
     }
   }
 
+  @Override
+  public String getApplicationName() {
+    return "normalization-worker";
+  }
+
 }

airbyte-workers/src/main/java/io/airbyte/workers/DefaultCheckConnectionWorker.java

@@ -86,4 +86,9 @@ public void cancel() {
     WorkerUtils.cancelProcess(process);
   }
 
+  @Override
+  public String getApplicationName() {
+    return "check-connection-worker";
+  }
+
 }

airbyte-workers/src/main/java/io/airbyte/workers/DefaultDiscoverCatalogWorker.java

@@ -81,4 +81,9 @@ public void cancel() {
     WorkerUtils.cancelProcess(process);
   }
 
+  @Override
+  public String getApplicationName() {
+    return "discover-worker";
+  }
+
 }

airbyte-workers/src/main/java/io/airbyte/workers/DefaultGetSpecWorker.java

@@ -81,4 +81,9 @@ public void cancel() {
     WorkerUtils.cancelProcess(process);
   }
 
+  @Override
+  public String getApplicationName() {
+    return "get-spec-worker";
+  }
+
 }

airbyte-workers/src/main/java/io/airbyte/workers/DefaultNormalizationWorker.java

@@ -82,4 +82,9 @@ public void cancel() {
     }
   }
 
+  @Override
+  public String getApplicationName() {
+    return "normalization-worker";
+  }
+
 }

airbyte-workers/src/main/java/io/airbyte/workers/DefaultReplicationWorker.java

@@ -271,4 +271,9 @@ public void cancel() {
 
   }
 
+  @Override
+  public String getApplicationName() {
+    return "sync-worker";
+  }
+
 }

airbyte-workers/src/main/java/io/airbyte/workers/EchoWorker.java

@@ -25,4 +25,9 @@ public void cancel() {
     // no-op
   }
 
+  @Override
+  public String getApplicationName() {
+    return "airbyte-echo-worker";
+  }
+
 }

airbyte-workers/src/main/java/io/airbyte/workers/Worker.java

@@ -6,7 +6,7 @@
 
 import java.nio.file.Path;
 
-public interface Worker<InputType, OutputType> {
+public interface Worker<InputType, OutputType> extends Application {
 
   /**
    * Blocking call to run the worker's workflow. Once this is complete, getStatus should return either

airbyte-workers/src/main/java/io/airbyte/workers/protocols/MessageTracker.java

@@ -5,10 +5,11 @@
 package io.airbyte.workers.protocols;
 
 import io.airbyte.config.State;
+import io.airbyte.workers.Application;
 import java.util.Optional;
 import java.util.function.Consumer;
 
-public interface MessageTracker<T> extends Consumer<T> {
+public interface MessageTracker<T> extends Consumer<T>, Application {
 
   @Override
   void accept(T message);

airbyte-workers/src/main/java/io/airbyte/workers/protocols/airbyte/AirbyteMessageTracker.java

@@ -52,4 +52,9 @@ public Optional<State> getOutputState() {
     return Optional.ofNullable(outputState.get());
   }
 
+  @Override
+  public String getApplicationName() {
+    return "airbyte-message-tracker";
+  }
+
 }