build.sh

#!/bin/bash
set -eo pipefail

mode=''

# The argument must be either 'zookeeper' or 'clickhouse'
if [ "$#" -lt 0 ] && [ "$1" != "zookeeper" ] && [ "$1" != "clickhouse" ]; then
  echo "Usage: $0 [zookeeper|clickhouse]"
  exit 1
fi

if [ "$1" != "clickhouse" ]; then
  mode='zookeeper'
else
  mode='clickhouse'
fi

ROOT=$(dirname "$0")
ROOT=$(cd "$ROOT"; pwd)
mkdir -p $ROOT/build

function version()
{
  grep VERSION_STRING $ROOT/version.txt | awk '{print $2}' | awk  -F')' '{print $1}'
}

# build binary
function build()
{
  echo "Building in $1 compatible mode."
  cd $ROOT/build && find $ROOT/build -type f ! -name '*.tar.gz' -exec rm -f {} +

  COMPATIBLE_MODE_ZOOKEEPER="ON"
  if [ -n "$1" ] && [ "$1" == "clickhouse" ]; then
    COMPATIBLE_MODE_ZOOKEEPER="OFF"
  fi
  cmake .. -DCMAKE_BUILD_TYPE=RelWithDebInfo -DCOMPATIBLE_MODE_ZOOKEEPER=$COMPATIBLE_MODE_ZOOKEEPER

  PARALLEL="$(($(nproc) / 4 + 1))"
  make -j $PARALLEL

  echo "Build complete!"
  echo ""
}

# make tarball
function package()
{
  echo "Packaging in $1 compatible mode."
  cd "$ROOT"
  mkdir -p build/RaftKeeper
  mkdir -p build/RaftKeeper/bin
  mkdir -p build/RaftKeeper/lib
  mkdir -p build/RaftKeeper/conf
  mkdir -p build/RaftKeeper/log
  mkdir -p build/RaftKeeper/data

  cp programs/server/bin/start.sh build/RaftKeeper/bin/.
  cp programs/server/bin/stop.sh build/RaftKeeper/bin/.
  cp build/programs/raftkeeper build/RaftKeeper/lib/.
  cp programs/server/config.xml build/RaftKeeper/conf/.

  cd $ROOT/build

  if [ -n "$1" ] && [ "$1" == "clickhouse" ]; then
    file_name=RaftKeeper-`version`-linux-x86_64-clickhouse.tar.gz
  else
    file_name=RaftKeeper-`version`-linux-x86_64.tar.gz
  fi
  tar -czvf $file_name RaftKeeper

  rm -rf RaftKeeper

  echo "RaftKeeper is built into build/${file_name}."
}

build "$mode"
package "$mode"
echo "Done!"