[

  {rabbit,

    [

      {cluster_partition_handling, pause_minority}

    ]

  }

].

ERLANG_EL6_RPM=erlang-18.3-.el6.x86_64.rpm

ERLANG_EL6_RPM_URL=http://www.rabbitmq.com/releases/erlang/erlang-18.3-1.el6.x86_64.rpm

ERLANG_EL7_RPM=erlang-18.3-.el7.centos.x86_64.rpm

ERLANG_EL7_RPM_URL=http://www.rabbitmq.com/releases/erlang/erlang-18.3-1.el7.centos.x86_64.rpm

RABBITMQ_RPM=rabbitmq-server-3.6.-.noarch.rpm

RABBITMQ_RPM_URL=http://www.rabbitmq.com/releases/rabbitmq-server/v3.6.1/rabbitmq-server-3.6.1-1.noarch.rpm

RABBITMQ_KEY=rabbitmq-signing-key-public.asc

RABBITMQ_KEY_URL=https://www.rabbitmq.com/rabbitmq-signing-key-public.asc

RABBITMQ_CONFIG_FILE=rabbitmq.config

ERLANG_COOKIE=AZVOCZYZZBVFLBPTBXUD

RABBITMQ_HTTP_PORT=

ERL_EPMD_PORT=

RABBITMQ_DIST_PORT=

ADMIN_PWD=admin

#!/bin/bash

#====================================================================

# install_rabbitmq.sh

#

# Linux RabbitMQ Auto Install Script

#

# Copyright (c) , Edward Guan <edward.guan@mkcorp.com>

# All rights reserved.

# Distributed under the GNU General Public License, version 3.0.

#

# Intro: http://www.cnblogs.com/edward2013/p/5061511.html

#

#====================================================================

echo "Note: This tiny script has been hardcoded specifically for RHEL/CentOS"

echo ""

if [ $(id -u) != "" ]; then

    echo "Error: You must be root to run this script!"

    exit

fi

# defind functions

msg() {

    printf '%b\n' "$1" >&

}

success() {

    msg "\33[32m[✔]\33[0m ${1}${2}"

}

error() {

    msg "\33[31m[✘]\33[0m ${1}${2}"

    exit

}

program_exists() {

    command -v $ >/dev/null >&

}

function open_rabbitmq_ports() {

    if program_exists firewall-cmd; then

        firewall-cmd -q --permanent \

            --add-port={$RABBITMQ_HTTP_PORT/tcp,$ERL_EPMD_PORT/tcp,$RABBITMQ_DIST_PORT/tcp}

        firewall-cmd -q --reload

        firewall-cmd -q --query-port $RABBITMQ_HTTP_PORT/tcp || return

        firewall-cmd -q --query-port $ERL_EPMD_PORT/tcp || return

        firewall-cmd -q --query-port $RABBITMQ_DIST_PORT/tcp || return

    fi

}

function erase_rabbitmq() {

    rpm -e rabbitmq-server

}

function update_rabbitmq_package() {

    local TMP_FILE=/tmp/.$(basename "$0").tmp

    # install required package

    program_exists wget || yum -y install wget >/dev/null >&

    # install erlang rpm

    if [ ! -f "$ERLANG_RPM_PATH" ]; then

        msg "WARNING: $ERLANG_RPM_PATH not found."

        msg "Try to download and install from $ERLANG_RPM_URL..."

        wget $ERLANG_RPM_URL -P $SCRIPT_PATH || return

    fi

    rpm -U $ERLANG_RPM_PATH >$TMP_FILE && success "Installed $ERLANG_RPM_PATH" || {

        sed 's/^[ |\t]*//' $TMP_FILE | grep "already installed" || return

    }

    # import rabbitmq key

    if [ ! -f "$RABBITMQ_KEY_PATH" ]; then

        msg "WARNING: $RABBITMQ_KEY_PATH not found."

        msg "Try to download and import from $RABBITMQ_KEY_URL..."

        wget $RABBITMQ_KEY_URL -P $SCRIPT_PATH || return

    fi

    rpm --import $RABBITMQ_KEY_PATH && success "Imported $RABBITMQ_KEY_PATH"

    # install rabbitmq rpm

    if [ ! -f "$RABBITMQ_RPM_PATH" ]; then

        msg "WARNING: $RABBITMQ_RPM_PATH not found."

        msg "Try to download and install from $RABBITMQ_RPM_URL..."

        wget $RABBITMQ_RPM_URL -P $SCRIPT_PATH || return

    fi

    rpm -U $RABBITMQ_RPM_PATH >$TMP_FILE && success "Installed $RABBITMQ_RPM_PATH" || {

        sed 's/^[ |\t]*//' $TMP_FILE | grep "already installed" || return

    }

}

function set_rabbitmq_users() {

    if program_exists rabbitmqctl; then

        rabbitmqctl list_users | grep -q guest && \

            rabbitmqctl delete_user guest >/dev/null && \

            success "Deleted user [guest]"

        rabbitmqctl list_users | grep -q admin || {

            rabbitmqctl add_user admin $ADMIN_PWD >/dev/null && \

            rabbitmqctl set_user_tags admin administrator >/dev/null && \

            rabbitmqctl set_permissions -p / admin ".*" ".*" ".*" >/dev/null && \

            success "Added user [admin]"

        }

    else

        msg "Command not find: rabbitmqctl" && return

    fi

}

function set_rabbitmq_policies() {

    if program_exists rabbitmqctl; then

        rabbitmqctl list_policies | grep -q ha-all || {

            rabbitmqctl set_policy ha-all "^" '{"ha-mode":"all","ha-sync-mode":"automatic"}' >/dev/null && \

            success "Added policy [ha-all]"

        }

    else

        msg "Command not find: rabbitmqctl" && return

    fi

}

function install_rabbitmq() {

    rpm -q rabbitmq-server >/dev/null && \

        error "$(rpm -q rabbitmq-server) is already installed"

    update_rabbitmq_package || return 

    # enable rabbitmq plugin

    rabbitmq-plugins enable rabbitmq_management >/dev/null && \

        success "Enabled rabbitmq plugin [rabbitmq_management]" || return 

    # open firewall ports

    open_rabbitmq_ports && \

        success "Opened rabbitmq ports [$RABBITMQ_HTTP_PORT/tcp,$ERL_EPMD_PORT/tcp,$RABBITMQ_DIST_PORT/tcp]" || \

        return 

    # deploy rabbitmq.config

    RABBITMQ_CONFIG_PATH="$SCRIPT_PATH/$RABBITMQ_CONFIG_FILE"

    RABBITMQ_CONFIG_LOCATION=/etc/rabbitmq/rabbitmq.config

    cp $RABBITMQ_CONFIG_PATH $RABBITMQ_CONFIG_LOCATION && \

        success "Copyed $RABBITMQ_CONFIG_PATH to $RABBITMQ_CONFIG_LOCATION" || return 

    # add host entry into /etc/hosts

    HOST_ENTRY=$(egrep "127.0.0.1 +$(hostname -s)" /etc/hosts)

    if [ "x$HOST_ENTRY" == "x" ]; then

        echo "127.0.0.1   $(hostname -s)" >> /etc/hosts && \

        success "Added [127.0.0.1   $(hostname -s)] into /etc/hosts" || return

    else

        msg "[$HOST_ENTRY] is already added into /etc/hosts"

    fi

    # set erlang cookie

    [ "x$COOKIE" == "x" ] && COOKIE="$ERLANG_COOKIE"

    ERLANG_COOKIE_PATH=/var/lib/rabbitmq/.erlang.cookie

    if [ ! -f "$ERLANG_COOKIE_PATH" ]; then

        msg "WARNING: $ERLANG_COOKIE_PATH is not exist, create new one"

        touch $ERLANG_COOKIE_PATH || return

    fi

    chmod  $ERLANG_COOKIE_PATH

    echo -n "$COOKIE" >$ERLANG_COOKIE_PATH

    chmod  $ERLANG_COOKIE_PATH

    chown rabbitmq:rabbitmq $ERLANG_COOKIE_PATH

    if [ $(cat $ERLANG_COOKIE_PATH) = "$COOKIE" ]; then

            success "Set erlang cookie value to $COOKIE"

    fi

}

function join_rabbitmq_cluster() {

    local ret=

    if [ "x$SERVER" != "x" ] && [ "$SERVER" != $(hostname -s) ]; then

        if program_exists rabbitmqctl; then

            rabbitmqctl stop_app >/dev/null

            rabbitmqctl reset >/dev/null

            if [ "$NODE" == "ram" ]; then

                rabbitmqctl join_cluster --ram rabbit@$SERVER >/dev/null || let ret++

            else

                rabbitmqctl join_cluster rabbit@$SERVER >/dev/null || let ret++

            fi

            rabbitmqctl start_app >/dev/null

        else

            msg "Command not find: rabbitmqctl" && return

        fi

        if [ "$ret" -eq  ]; then

            success "Joined rabbit@$SERVER"

            return

        else

            return

        fi

    fi

}

function restart_rabbitmq() {

    # kill all rabbitmq server processes

    local RABBITMQ_PIDS=$(ps -ef | grep ^rabbitmq | cut -c - | tr -s "\n" " ")

    if [ "x$RABBITMQ_PIDS" != "x" ]; then

        kill - $RABBITMQ_PIDS && success "Killed all rabbitmq server processes"

    fi

    # enable and start rabbitmq server

    chkconfig rabbitmq-server on

    service rabbitmq-server start

}

function print_usage() {

    echo "Usage: $(basename "$") [OPTIONS...]"

    echo ""

    echo "Options"

    echo "  [-h|--help]                 Prints a short help text and exists"

    echo "  [-i|--install]              Install rabbitmq server"

    echo "  [-u|--update]               Update rabbitmq server"

    echo "  [-e|--erase]                Erase (uninstall) rabbitmq server"

    echo "  [-c|--cookie] <cookie>      Set erlang cookie"

    echo "  [-j|--join] <server>        Join rabbitmq cluster"

    echo "  [-n|--node] <disc|ram>      Set cluster node type"

}

# read the options

TEMP=`getopt -o hiuec:j:n: --long help,install,update,erase,cookie:,join:,node: -n $(basename "$0") -- "$@"`

eval set -- "$TEMP"

# extract options and their arguments into variables.

while true; do

    case "$1" in

        -h|--help) print_usage ; exit  ;;

        -i|--install) ACTION=install ; shift ;;

        -u|--update) ACTION=update ; shift ;;

        -e|--erase) ACTION=erase ; shift ;;

        -c|--cookie) COOKIE=$ ; shift  ;;

        -j|--join) SERVER=$ ; shift  ;;

        -n|--node) NODE=$ ; shift  ;;

        --) shift ; break ;;

        *) error "Internal error!" ;;

    esac

done

# get script path

SCRIPT_PATH="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"

# load settings

source "$SCRIPT_PATH/settings.conf" || exit 

# set erlang and rabbitmq rpm & key path

case $(uname -r) in

    *el7*)

        ERLANG_RPM_PATH="$SCRIPT_PATH/$ERLANG_EL7_RPM";

        ERLANG_RPM_URL="$ERLANG_EL7_RPM_URL" ;;

    *el6*)

        ERLANG_RPM_PATH="$SCRIPT_PATH/$ERLANG_EL6_RPM";

        ERLANG_RPM_URL="$ERLANG_EL6_RPM_URL" ;;

    *) error "Your system is not RHEL/CentOS" ;;

esac

RABBITMQ_RPM_PATH="$SCRIPT_PATH/$RABBITMQ_RPM"

RABBITMQ_KEY_PATH="$SCRIPT_PATH/$RABBITMQ_KEY"

if [ "x$ACTION" == "x" ] && [ "x$SERVER" == "x" ]; then

    error "$(basename "$"): missing operand\n" \

    "Try '$(basename "$") --help' for more information."

fi

echo "$NODE" | grep -qE "^disk$|^ram$|^$" || {

    error "$(basename "$"): -n: must be disk or ram"

}

# if ACTION is erase, erase rabbitmq server

if [ "$ACTION" == "erase" ]; then

    erase_rabbitmq && {

        msg "\nThanks for erasing rabbitmq-server."

        msg "© `date +%Y` marykay company"

    } || error "Failed erase rabbitmq server"

    exit

fi

# if ACTION is update, update rabbitmq server

if [ "$ACTION" == "update" ]; then

    update_rabbitmq_package && restart_rabbitmq && {

        msg "\nThanks for updating rabbitmq-server."

        msg "© `date +%Y` marykay company"

    } || error "Failed update rabbitmq server"

    exit

fi

# if ACTION is install, install rabbitmq server

if [ "$ACTION" == "install" ]; then

    install_rabbitmq && restart_rabbitmq && set_rabbitmq_users && set_rabbitmq_policies && join_rabbitmq_cluster && {

        msg "\nThanks for installing rabbitmq-server."

        msg "© `date +%Y` marykay company"

    } || error "Failed install rabbitmq server"

else

    if [ "x$SERVER" != "x" ]; then

        join_rabbitmq_cluster && {

            msg "\nThanks for joining rabbitmq-server."

            msg "© `date +%Y` marykay company"

        } || error "Failed join rabbitmq server"

    fi

fi