fix(nocode): fix no-code upgrade (#12494)

datahub-upgrade/build.gradle

@@ -152,6 +152,18 @@ task runRestoreIndicesUrn(type: Exec) {
           bootJar.getArchiveFile().get(), "-u", "RestoreIndices", "-a", "batchSize=100", "-a", "urnBasedPagination=true"
 }
 
+task runNoCode(type: Exec) {
+  dependsOn bootJar
+  group = "Execution"
+  description = "Run the NoCodeDataMigration process locally."
+  environment "ENTITY_REGISTRY_CONFIG_PATH", "../metadata-models/src/main/resources/entity-registry.yml"
+  commandLine "java", "-agentlib:jdwp=transport=dt_socket,address=5003,server=y,suspend=n",
+          "-jar",
+          "-Dkafka.schemaRegistry.url=http://localhost:8080/schema-registry/api",
+          "-Dserver.port=8083",
+          bootJar.getArchiveFile().get(), "-u", "NoCodeDataMigration", "-a", "batchSize=100", "-a", "dbType=MYSQL"
+}
+
 docker {
   name "${docker_registry}/${docker_repo}:v${version}"
   version "v${version}"

datahub-upgrade/src/main/java/com/linkedin/datahub/upgrade/config/OpenTelemetryConfig.java

@@ -0,0 +1,21 @@
+package com.linkedin.datahub.upgrade.config;
+
+import com.linkedin.gms.factory.system_telemetry.OpenTelemetryBaseFactory;
+import io.datahubproject.metadata.context.TraceContext;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+public class OpenTelemetryConfig extends OpenTelemetryBaseFactory {
+
+  @Override
+  protected String getApplicationComponent() {
+    return "datahub-upgrade";
+  }
+
+  @Bean
+  @Override
+  protected TraceContext traceContext() {
+    return super.traceContext();
+  }
+}

datahub-upgrade/src/test/java/com/linkedin/datahub/upgrade/UpgradeCliApplicationTest.java

@@ -10,6 +10,7 @@ import com.linkedin.datahub.upgrade.system.BlockingSystemUpgrade;
 import com.linkedin.metadata.dao.throttle.NoOpSensor;
 import com.linkedin.metadata.dao.throttle.ThrottleSensor;
 import com.linkedin.metadata.search.elasticsearch.indexbuilder.ESIndexBuilder;
+import io.datahubproject.metadata.context.TraceContext;
 import javax.inject.Named;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Qualifier;
@@ -38,6 +39,8 @@ public class UpgradeCliApplicationTest extends AbstractTestNGSpringContextTests
   @Autowired
   private ThrottleSensor kafkaThrottle;
 
+  @Autowired private TraceContext traceContext;
+
   @Test
   public void testRestoreIndicesInit() {
     /*
@@ -62,4 +65,9 @@ public class UpgradeCliApplicationTest extends AbstractTestNGSpringContextTests
     assertEquals(
         new NoOpSensor(), kafkaThrottle, "No kafka throttle controls expected in datahub-upgrade");
   }
+
+  @Test
+  public void testTraceContext() {
+    assertNotNull(traceContext);
+  }
 }