mirror of
https://github.com/clearlinux/dockerfiles.git
synced 2026-10-17 18:02:38 +00:00
91 lines
3.7 KiB
Bash
91 lines
3.7 KiB
Bash
#!/bin/sh
|
|
|
|
###############################################################################
|
|
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
###############################################################################
|
|
|
|
# If unspecified, the hostname of the container is taken as the JobManager address
|
|
JOB_MANAGER_RPC_ADDRESS=${JOB_MANAGER_RPC_ADDRESS:-$(hostname -f)}
|
|
CONF_FILE="/usr/share/defaults/flink/flink-conf.yaml"
|
|
|
|
drop_privs_cmd() {
|
|
return
|
|
}
|
|
|
|
if [ "$1" = "help" ]; then
|
|
echo "Usage: $(basename "$0") (jobmanager|taskmanager|help)"
|
|
exit 0
|
|
elif [ "$1" = "jobmanager" ]; then
|
|
shift 1
|
|
echo "Starting Job Manager"
|
|
|
|
if grep -E "^jobmanager\.rpc\.address:.*" "${CONF_FILE}" > /dev/null; then
|
|
sed -i -e "s/jobmanager\.rpc\.address:.*/jobmanager.rpc.address: ${JOB_MANAGER_RPC_ADDRESS}/g" "${CONF_FILE}"
|
|
else
|
|
echo "jobmanager.rpc.address: ${JOB_MANAGER_RPC_ADDRESS}" >> "${CONF_FILE}"
|
|
fi
|
|
|
|
if grep -E "^blob\.server\.port:.*" "${CONF_FILE}" > /dev/null; then
|
|
sed -i -e "s/blob\.server\.port:.*/blob.server.port: 6124/g" "${CONF_FILE}"
|
|
else
|
|
echo "blob.server.port: 6124" >> "${CONF_FILE}"
|
|
fi
|
|
|
|
if grep -E "^query\.server\.port:.*" "${CONF_FILE}" > /dev/null; then
|
|
sed -i -e "s/query\.server\.port:.*/query.server.port: 6125/g" "${CONF_FILE}"
|
|
else
|
|
echo "query.server.port: 6125" >> "${CONF_FILE}"
|
|
fi
|
|
|
|
echo "config file: " && grep '^[^\n#]' "${CONF_FILE}"
|
|
exec $(drop_privs_cmd) "$FLINK_HOME/bin/jobmanager.sh" start-foreground "$@"
|
|
elif [ "$1" = "taskmanager" ]; then
|
|
shift 1
|
|
echo "Starting Task Manager"
|
|
|
|
TASK_MANAGER_NUMBER_OF_TASK_SLOTS=${TASK_MANAGER_NUMBER_OF_TASK_SLOTS:-$(grep -c ^processor /proc/cpuinfo)}
|
|
|
|
if grep -E "^jobmanager\.rpc\.address:.*" "${CONF_FILE}" > /dev/null; then
|
|
sed -i -e "s/jobmanager\.rpc\.address:.*/jobmanager.rpc.address: ${JOB_MANAGER_RPC_ADDRESS}/g" "${CONF_FILE}"
|
|
else
|
|
echo "jobmanager.rpc.address: ${JOB_MANAGER_RPC_ADDRESS}" >> "${CONF_FILE}"
|
|
fi
|
|
|
|
if grep -E "^taskmanager\.numberOfTaskSlots:.*" "${CONF_FILE}" > /dev/null; then
|
|
sed -i -e "s/taskmanager\.numberOfTaskSlots:.*/taskmanager.numberOfTaskSlots: ${TASK_MANAGER_NUMBER_OF_TASK_SLOTS}/g" "${CONF_FILE}"
|
|
else
|
|
echo "taskmanager.numberOfTaskSlots: ${TASK_MANAGER_NUMBER_OF_TASK_SLOTS}" >> "${CONF_FILE}"
|
|
fi
|
|
|
|
if grep -E "^blob\.server\.port:.*" "${CONF_FILE}" > /dev/null; then
|
|
sed -i -e "s/blob\.server\.port:.*/blob.server.port: 6124/g" "${CONF_FILE}"
|
|
else
|
|
echo "blob.server.port: 6124" >> "${CONF_FILE}"
|
|
fi
|
|
|
|
if grep -E "^query\.server\.port:.*" "${CONF_FILE}" > /dev/null; then
|
|
sed -i -e "s/query\.server\.port:.*/query.server.port: 6125/g" "${CONF_FILE}"
|
|
else
|
|
echo "query.server.port: 6125" >> "${CONF_FILE}"
|
|
fi
|
|
|
|
echo "config file: " && grep '^[^\n#]' "${CONF_FILE}"
|
|
exec $(drop_privs_cmd) "$FLINK_HOME/bin/taskmanager.sh" start-foreground "$@"
|
|
fi
|
|
|
|
exec "$@"
|