File size: 2,520 Bytes
fe5d4e8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 | #!/usr/bin/env bash
#
#/**
# * Copyright 2007 The Apache Software Foundation
# *
# * 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.
# */
# zk env vars to replace values in config files
export ZK_dataDir=${ZK_dataDir:-"data/zookeeper/data"}
export ZK_dataLogDir=${ZK_dataLogDir:-"data/zookeeper/txlog"}
export ZK_standaloneEnabled=${ZK_standaloneEnabled:-"false"}
export ZK_dynamicConfigFile=${ZK_dynamicConfigFile:-"${BK_HOME}/conf/zookeeper.conf.dynamic"}
export ZK_SERVERS=${ZK_SERVERS:-"server.1=127.0.0.1:2888:3888:participant;0.0.0.0:2181"}
export ZK_ID=${ZK_ID:-"1"}
echo "Environment Vars for zookeeper:"
echo " ZK_dataDir = ${ZK_dataDir}"
echo " ZK_dataLogDir = ${ZK_dataLogDir}"
echo " ZK_ID = ${ZK_ID}"
echo " ZK_SERVERS = ${ZK_SERVERS}"
echo " ZK_standaloneEnabled = ${ZK_standaloneEnabled}"
echo " ZK_dynamicConfigFile = ${ZK_dynamicConfigFile}"
function create_zk_dirs() {
mkdir -p "${ZK_dataDir}" "${ZK_dataLogDir}"
echo "Created zookeeper dirs : "
echo " data = ${ZK_dataDir}"
echo " txnlog = ${ZK_dataLogDir}"
if [[ ! -f "${ZK_dataDir}/myid" ]]; then
echo "${ZK_ID}" > "${ZK_dataDir}/myid"
fi
# -------------- #
# Allow the container to be started with `--user`
if [ "$(id -u)" = '0' ]; then
chown -R "${BK_USER}:${BK_USER}" "${ZK_dataDir}" "${ZK_dataLogDir}"
fi
# -------------- #
}
function create_zk_dynamic_conf() {
for server in ${ZK_SERVERS}; do
echo "$server" >> "${ZK_dynamicConfigFile}"
done
}
function init_zookeeper() {
# apply zookeeper envs
python scripts/apply-config-from-env.py ${BK_HOME}/conf/zookeeper.conf
# create dirs if they don't exist
create_zk_dirs
# create dynamic config
create_zk_dynamic_conf
} |