summaryrefslogtreecommitdiff
path: root/tools/run_zookeeper.sh
blob: 66166c54065989efea6e18165534e39ce456004e (plain)
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
#!/bin/bash

set -e

PKGS=""
PKG_CMD=""
if [ -f "/etc/redhat-release" ]; then
    PKGS="$PKGS java-1.7.0-openjdk-devel"
    PKG_CMD="yum install -y"
elif [ -f "/etc/debian_version" ]; then
    PKGS="$PKGS openjdk-7-jdk"
    PKG_CMD="apt-get install -y"
else
    echo "Unknown operating system!"
    exit 1
fi

JAVA=`which java`
if [ -z "$JAVA" ]; then
    sudo $PKG_CMD $PKGS
fi

function on_exit() {
    local ec="$?"
    if [ -n "$ZOO_DIR" ]; then
        if [ -f "$ZOO_DIR/bin/zkServer.sh" ]; then
            $ZOO_DIR/bin/zkServer.sh stop
        fi
    fi
    return $ec
}

trap "on_exit" EXIT

VERSION="3.4.6"
if [ ! -d "$PWD/.zookeeper" ]; then
    mkdir -p "$PWD/.zookeeper"
fi

ZOO_BASE_DIR="$PWD/.zookeeper/"
if [ ! -f "$PWD/.zookeeper/zookeeper-$VERSION.tar.gz" ]; then
    echo "Downloading: zookeeper-$VERSION -> $PWD/.zookeeper/zookeeper-$VERSION.tar.gz"
    curl -o "$PWD/.zookeeper/zookeeper-$VERSION.tar.gz" \
            "http://www.us.apache.org/dist/zookeeper/zookeeper-$VERSION/zookeeper-$VERSION.tar.gz"
fi

cd "$ZOO_BASE_DIR"
if [ ! -d "zookeeper-$VERSION" ]; then
    echo "Extracting zookeeper-$VERSION.tar.gz -> $ZOO_BASE_DIR/zookeeper-$VERSION"
    mkdir -p $ZOO_BASE_DIR/zookeeper-$VERSION
    tar xzf ./zookeeper-$VERSION.tar.gz -C $ZOO_BASE_DIR/zookeeper-$VERSION --strip-components 1
fi

echo "Adjusting configuration..."
ZOO_DIR="$ZOO_BASE_DIR/zookeeper-$VERSION"
cp "$ZOO_DIR/conf/zoo_sample.cfg" "$ZOO_DIR/conf/zoo.cfg"
ZOO_DATADIR="$ZOO_BASE_DIR/.data/"
mkdir -p "$ZOO_DATADIR"
sed -i -r "s@(dataDir *= *).*@\1$ZOO_DATADIR@" "$ZOO_DIR/conf/zoo.cfg"

export ZOO_DATADIR="$ZOO_DATADIR"
echo "Starting..."
$ZOO_DIR/bin/zkServer.sh start-foreground