mirror of https://github.com/apache/kafka.git
88 lines
3.5 KiB
Bash
Executable File
88 lines
3.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# 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.
|
|
|
|
# This script automates the process of setting up a local machine for running Kafka system tests
|
|
export GREP_OPTIONS='--color=never'
|
|
|
|
# Helper function which prints version numbers so they can be compared lexically or numerically
|
|
function version { echo "$@" | awk -F. '{ printf("%03d%03d%03d%03d\n", $1,$2,$3,$4); }'; }
|
|
|
|
base_dir=`dirname $0`/..
|
|
cd $base_dir
|
|
|
|
echo "Checking Virtual Box installation..."
|
|
bad_vb=false
|
|
if [ -z `vboxmanage --version` ]; then
|
|
echo "It appears that Virtual Box is not installed. Please install and try again (see https://www.virtualbox.org/ for details)"
|
|
bad_vb=true
|
|
else
|
|
echo "Virtual Box looks good."
|
|
fi
|
|
|
|
echo "Checking Vagrant installation..."
|
|
vagrant_version=`vagrant --version | egrep -o "[0-9]+\.[0-9]+\.[0-9]+"`
|
|
bad_vagrant=false
|
|
if [ "$(version $vagrant_version)" -lt "$(version 1.6.4)" ]; then
|
|
echo "Found Vagrant version $vagrant_version. Please upgrade to 1.6.4 or higher (see https://www.vagrantup.com for details)"
|
|
bad_vagrant=true
|
|
else
|
|
echo "Vagrant installation looks good."
|
|
fi
|
|
|
|
if [ "x$bad_vagrant" == "xtrue" -o "x$bad_vb" == "xtrue" ]; then
|
|
exit 1
|
|
fi
|
|
|
|
echo "Checking for necessary Vagrant plugins..."
|
|
hostmanager_version=`vagrant plugin list | grep vagrant-hostmanager | egrep -o "[0-9]+\.[0-9]+\.[0-9]+"`
|
|
if [ -z "$hostmanager_version" ]; then
|
|
vagrant plugin install vagrant-hostmanager
|
|
fi
|
|
|
|
echo "Creating and packaging a reusable base box for Vagrant..."
|
|
vagrant/package-base-box.sh
|
|
|
|
# Set up Vagrantfile.local if necessary
|
|
if [ ! -e Vagrantfile.local ]; then
|
|
echo "Creating Vagrantfile.local..."
|
|
cp vagrant/system-test-Vagrantfile.local Vagrantfile.local
|
|
else
|
|
echo "Found an existing Vagrantfile.local. Keeping without overwriting..."
|
|
fi
|
|
|
|
# Sanity check contents of Vagrantfile.local
|
|
echo "Checking Vagrantfile.local..."
|
|
vagrantfile_ok=true
|
|
num_brokers=`egrep -o "num_brokers\s*=\s*[0-9]+" Vagrantfile.local | cut -d '=' -f 2 | xargs`
|
|
num_zookeepers=`egrep -o "num_zookeepers\s*=\s*[0-9]+" Vagrantfile.local | cut -d '=' -f 2 | xargs`
|
|
num_workers=`egrep -o "num_workers\s*=\s*[0-9]+" Vagrantfile.local | cut -d '=' -f 2 | xargs`
|
|
if [ "x$num_brokers" == "x" -o "$num_brokers" != 0 ]; then
|
|
echo "Vagrantfile.local: bad num_brokers. Update to: num_brokers = 0"
|
|
vagrantfile_ok=false
|
|
fi
|
|
if [ "x$num_zookeepers" == "x" -o "$num_zookeepers" != 0 ]; then
|
|
echo "Vagrantfile.local: bad num_zookeepers. Update to: num_zookeepers = 0"
|
|
vagrantfile_ok=false
|
|
fi
|
|
if [ "x$num_workers" == "x" -o "$num_workers" == 0 ]; then
|
|
echo "Vagrantfile.local: bad num_workers (size of test cluster). Set num_workers high enough to run your tests."
|
|
vagrantfile_ok=false
|
|
fi
|
|
|
|
if [ "$vagrantfile_ok" == "true" ]; then
|
|
echo "Vagrantfile.local looks good."
|
|
fi
|