fix(consumer): wait for internal schema registry

datahub-project · Jan 23, 2024 · 7f053ce · 7f053ce
1 parent 4aa26c2
commit 7f053ce
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/docker/datahub-mae-consumer/start.sh b/docker/datahub-mae-consumer/start.sh
@@ -33,6 +33,9 @@ fi
 if [[ ${GRAPH_SERVICE_IMPL:-} != elasticsearch ]] && [[ ${SKIP_NEO4J_CHECK:-false} != true ]]; then
   dockerize_args+=("-wait" "$NEO4J_HOST")
 fi
+if [[ "${KAFKA_SCHEMAREGISTRY_URL:-}" && ${SKIP_SCHEMA_REGISTRY_CHECK:-false} != true ]]; then
+  dockerize_args+=("-wait" "$KAFKA_SCHEMAREGISTRY_URL")
+fi
 
 JAVA_TOOL_OPTIONS="${JDK_JAVA_OPTIONS:-}${JAVA_OPTS:+ $JAVA_OPTS}${JMX_OPTS:+ $JMX_OPTS}"
 if [[ ${ENABLE_OTEL:-false} == true ]]; then

diff --git a/docker/datahub-mce-consumer/start.sh b/docker/datahub-mce-consumer/start.sh
@@ -5,6 +5,11 @@ if [[ $SKIP_KAFKA_CHECK != true ]]; then
   WAIT_FOR_KAFKA=" -wait tcp://$(echo $KAFKA_BOOTSTRAP_SERVER | sed 's/,/ -wait tcp:\/\//g') "
 fi
 
+WAIT_FOR_SCHEMA_REGISTRY=""
+if [[ "$KAFKA_SCHEMAREGISTRY_URL" && $SKIP_SCHEMA_REGISTRY_CHECK != true ]]; then
+  WAIT_FOR_SCHEMA_REGISTRY="-wait $KAFKA_SCHEMAREGISTRY_URL"
+fi
+
 OTEL_AGENT=""
 if [[ $ENABLE_OTEL == true ]]; then
   OTEL_AGENT="-javaagent:opentelemetry-javaagent.jar "
@@ -17,5 +22,6 @@ fi
 
 exec dockerize \
   $WAIT_FOR_KAFKA \
+  $WAIT_FOR_SCHEMA_REGISTRY \
   -timeout 240s \
-  java $JAVA_OPTS $JMX_OPTS $OTEL_AGENT $PROMETHEUS_AGENT -jar /datahub/datahub-mce-consumer/bin/mce-consumer-job.jar
+  java $JAVA_OPTS $JMX_OPTS $OTEL_AGENT $PROMETHEUS_AGENT -jar /datahub/datahub-mce-consumer/bin/mce-consumer-job.jar