mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-04 15:50:14 +00:00
80 lines
2.4 KiB
Bash
80 lines
2.4 KiB
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
# Add default URI (http) scheme if needed
|
||
|
|
if ! echo $NEO4J_HOST | grep -q "://" ; then
|
||
|
|
NEO4J_HOST="http://$NEO4J_HOST"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ ! -z $ELASTICSEARCH_USERNAME ]] && [[ -z $ELASTICSEARCH_AUTH_HEADER ]]; then
|
||
|
|
AUTH_TOKEN=$(echo -ne "$ELASTICSEARCH_USERNAME:$ELASTICSEARCH_PASSWORD" | base64 --wrap 0)
|
||
|
|
ELASTICSEARCH_AUTH_HEADER="Authorization:Basic $AUTH_TOKEN"
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Add default header if needed
|
||
|
|
if [[ -z $ELASTICSEARCH_AUTH_HEADER ]]; then
|
||
|
|
ELASTICSEARCH_AUTH_HEADER="Accept: */*"
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ $ELASTICSEARCH_USE_SSL == true ]]; then
|
||
|
|
ELASTICSEARCH_PROTOCOL=https
|
||
|
|
else
|
||
|
|
ELASTICSEARCH_PROTOCOL=http
|
||
|
|
fi
|
||
|
|
|
||
|
|
WAIT_FOR_EBEAN=""
|
||
|
|
if [[ $SKIP_EBEAN_CHECK != true ]]; then
|
||
|
|
if [[ $ENTITY_SERVICE_IMPL == ebean ]] || [[ -z $ENTITY_SERVICE_IMPL ]]; then
|
||
|
|
WAIT_FOR_EBEAN=" -wait tcp://$EBEAN_DATASOURCE_HOST "
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
WAIT_FOR_CASSANDRA=""
|
||
|
|
if [[ $ENTITY_SERVICE_IMPL == cassandra ]] && [[ $SKIP_CASSANDRA_CHECK != true ]]; then
|
||
|
|
WAIT_FOR_CASSANDRA=" -wait tcp://$CASSANDRA_DATASOURCE_HOST "
|
||
|
|
fi
|
||
|
|
|
||
|
|
WAIT_FOR_KAFKA=""
|
||
|
|
if [[ $SKIP_KAFKA_CHECK != true ]]; then
|
||
|
|
WAIT_FOR_KAFKA=" -wait tcp://$(echo $KAFKA_BOOTSTRAP_SERVER | sed 's/,/ -wait tcp:\/\//g') "
|
||
|
|
fi
|
||
|
|
|
||
|
|
WAIT_FOR_NEO4J=""
|
||
|
|
if [[ $GRAPH_SERVICE_IMPL != elasticsearch ]] && [[ $SKIP_NEO4J_CHECK != true ]]; then
|
||
|
|
WAIT_FOR_NEO4J=" -wait $NEO4J_HOST "
|
||
|
|
fi
|
||
|
|
|
||
|
|
OTEL_AGENT=""
|
||
|
|
if [[ $ENABLE_OTEL == true ]]; then
|
||
|
|
OTEL_AGENT="-javaagent:/datahub/datahub-upgrade/lib/opentelemetry-javaagent.jar "
|
||
|
|
if [ -z "${OTEL_EXPORTER_OTLP_MAX_PAYLOAD_SIZE}" ]; then
|
||
|
|
export OTEL_EXPORTER_OTLP_MAX_PAYLOAD_SIZE=4194304
|
||
|
|
fi
|
||
|
|
if [ -z "${OTEL_EXPORTER_OTLP_HTTP_HTTP2_MAX_FRAME_SIZE}" ]; then
|
||
|
|
export OTEL_EXPORTER_OTLP_HTTP_HTTP2_MAX_FRAME_SIZE=16777215
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
|
||
|
|
PROMETHEUS_AGENT=""
|
||
|
|
if [[ $ENABLE_PROMETHEUS == true ]]; then
|
||
|
|
PROMETHEUS_AGENT="-javaagent:/datahub/datahub-upgrade/lib/jmx_prometheus_javaagent.jar=4318:/datahub/datahub-gms/scripts/prometheus-config.yaml "
|
||
|
|
fi
|
||
|
|
|
||
|
|
COMMON="
|
||
|
|
$WAIT_FOR_EBEAN \
|
||
|
|
$WAIT_FOR_CASSANDRA \
|
||
|
|
$WAIT_FOR_KAFKA \
|
||
|
|
$WAIT_FOR_NEO4J \
|
||
|
|
-timeout 240s \
|
||
|
|
java $JAVA_OPTS $JMX_OPTS \
|
||
|
|
$OTEL_AGENT \
|
||
|
|
-Dstats=unsecure \
|
||
|
|
-Dloader.path=/datahub/datahub-upgrade/lib \
|
||
|
|
-jar /datahub/datahub-upgrade/bin/datahub-upgrade.jar"
|
||
|
|
|
||
|
|
if [[ $SKIP_ELASTICSEARCH_CHECK != true ]]; then
|
||
|
|
exec /usr/local/bin/dockerize \
|
||
|
|
-wait $ELASTICSEARCH_PROTOCOL://$ELASTICSEARCH_HOST:$ELASTICSEARCH_PORT -wait-http-header "$ELASTICSEARCH_AUTH_HEADER" \
|
||
|
|
$COMMON "$@"
|
||
|
|
else
|
||
|
|
exec /usr/local/bin/dockerize $COMMON "$@"
|
||
|
|
fi
|