110 lines
3.0 KiB
Bash
Executable File
110 lines
3.0 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
##################################################
|
|
# Mount dir #
|
|
# - /opt/kafka/data #
|
|
# - /opt/kafka/logs #
|
|
# ENV #
|
|
# - CONF #
|
|
# - CLUSTER_ID #
|
|
# - LEAD_CONTROLLER #
|
|
# - KAFKA_HEAP_OPTS #
|
|
# - EXTRA_ARGS #
|
|
# - _CONF_* #
|
|
##################################################
|
|
|
|
set -euo pipefail
|
|
export LANG=en_US.UTF-8
|
|
trap Quit EXIT
|
|
|
|
PIDS=
|
|
GOT_SIGTERM=
|
|
CONF=${CONF:?}
|
|
CONF_FILE=config/$CONF.properties
|
|
CLUSTER_ID=${CLUSTER_ID:?}
|
|
LEAD_CONTROLLER=${LEAD_CONTROLLER:-}
|
|
|
|
function Print {
|
|
local file=/dev/null
|
|
[ '-f' = "$1" ] && file=$2 && shift && shift
|
|
date +"[%F %T] $*" | tee -a $file
|
|
}
|
|
|
|
function Quit {
|
|
while :; do
|
|
pkill -f java && Print killing java ... || break
|
|
sleep 1
|
|
done
|
|
Print Container stopped.
|
|
test -n "$GOT_SIGTERM"
|
|
}
|
|
|
|
function Usage {
|
|
Print 'This container should run with
|
|
**root user**
|
|
**/opt/kafka/{data,logs} mounted from host**
|
|
'
|
|
}
|
|
|
|
function ModifyConf {
|
|
local kv=
|
|
! cp -f $CONF_FILE.origin $CONF_FILE \
|
|
&& Print Skipped modifying $CONF_FILE ... \
|
|
&& return 0
|
|
Print Modifying $CONF_FILE ...
|
|
echo -e "\n#### Docker" >> $CONF_FILE
|
|
while read kv; do
|
|
[ -z "$kv" ] && break
|
|
Print Modifying property: ${kv%%=*} ...
|
|
sed -i "/^${kv%%=*} *=/d" $CONF_FILE
|
|
echo "$kv" >> $CONF_FILE
|
|
done <<< "$(env | grep '^_CONF_' | sed 's/_CONF_//')"
|
|
}
|
|
|
|
function StartProc {
|
|
[[ ! "$CONF" =~ ^broker|controller|server$ ]] \
|
|
&& Print Unknown conf: $CONF! \
|
|
&& exit 1
|
|
Print Formatting storage ...
|
|
if [ -z "$LEAD_CONTROLLER" -a 'broker' != "$CONF" ]; then
|
|
kafka-storage.sh format -g -t $CLUSTER_ID -c $CONF_FILE -s
|
|
else
|
|
kafka-storage.sh format -g -t $CLUSTER_ID -c $CONF_FILE -N
|
|
fi
|
|
Print Starting kafka ...
|
|
kafka-server-start.sh $CONF_FILE &>> logs/kafka.out &
|
|
PIDS="$PIDS $!"
|
|
Print Kafka started.
|
|
[ -e data/$CONF.lock ] && return 0
|
|
[ -z "$LEAD_CONTROLLER" -o 'broker' == "$CONF" ] \
|
|
&& touch data/$CONF.lock \
|
|
&& return 0
|
|
Print Join in kraft cluster with $LEAD_CONTROLLER after 10s ...
|
|
sleep 10
|
|
[ ! -e /proc/$! ] && Print Unexpected error! && exit
|
|
timeout 10 kafka-metadata-quorum.sh --command-config $CONF_FILE \
|
|
--bootstrap-controller $LEAD_CONTROLLER add-controller \
|
|
&& touch data/$CONF.lock \
|
|
&& return 0
|
|
Print Failed to join cluster with $LEAD_CONTROLLER!
|
|
exit 1
|
|
}
|
|
|
|
function Main {
|
|
local pid=
|
|
cd /opt/kafka
|
|
Usage
|
|
ModifyConf
|
|
StartProc
|
|
trap "GOT_SIGTERM=1; Print Got SIGTERM ..." SIGTERM
|
|
while [ -z "$GOT_SIGTERM" ] && sleep 2; do
|
|
for pid in $PIDS; do
|
|
[ ! -e /proc/$pid ] && Print Unexpected error! && exit
|
|
done
|
|
done
|
|
}
|
|
|
|
# Start here
|
|
Main
|
|
|