#!/usr/bin/env bash

# auth:kaliarch

# version:v1.0

# func:elasticsearch 5.4.1/6.0.1/6.3.1安装

# 定义安装目录、及日志信息

. /etc/init.d/functions

[ $(id -u) != "0" ] && echo "Error: You must be root to run this script" && exit 1

export PATH=/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin

download_path=/tmp/tmpdir/

install_log_name=install_elasticsearch.log

env_file=/etc/profile.d/elasticsearch.sh

install_log_path=/var/log/appinstall/

install_path=/usr/local/

software_config_file=${install_path}elasticsearch/config/elasticsearch.yml

sysversion=$(rpm -q centos-release | cut -d- -f3)

jvm_conf="/usr/local/elasticsearch/config/jvm.options"

sys_mem=`free -m | grep Mem: | awk '{print $2}' | awk '{sum+=$1} END {print sum/1024}' | cut -d. -f1`

hostname=elk-server

clear

echo "##########################################"

echo "# #"

echo "# 安装 elasticsearch 5.4.1/6.0.1/6.3.1 #"

echo "# #"

echo "##########################################"

echo "1: Install elasticsearch 5.4.1"

echo "2: Install elasticsearch 6.0.1"

echo "3: Install elasticsearch 6.3.1"

echo "4: EXIT"

# 选择安装软件版本

read -p "Please input your choice:" softversion

if [ "${softversion}" == "1" ]; then

URL="https://anchnet-script.oss-cn-shanghai.aliyuncs.com/elasticsearch/elasticsearch-5.4.1.tar.gz"

elif [ "${softversion}" == "2" ]; then

URL="https://anchnet-script.oss-cn-shanghai.aliyuncs.com/elasticsearch/elasticsearch-6.0.1.tar.gz"

elif [ "${softversion}" == "3" ]; then

URL="https://anchnet-script.oss-cn-shanghai.aliyuncs.com/elasticsearch/elasticsearch-6.3.1.tar.gz"

elif [ "${softversion}" == "4" ]; then

echo "you choce channel!"

exit 1;

else

echo "input Error! Place input{1|2|3|4|5}"

exit 0;

fi

# 传入内容,格式化内容输出,可以传入多个参数,用空格隔开

output_msg() {

for msg in $*; do

action $msg /bin/true

done

}

# 判断命令是否存在,第一个参数 $1 为判断的命令,第二个参数为提供该命令的yum 软件包名称

check_yum_command() {

output_msg "命令检查:$1"

hash $1 > /dev/null 2>&1

if [ $? -eq 0 ]; then

echo "`date +%F' '%H:%M:%S` check command $1 " >> ${install_log_path}${install_log_name} && return 0

else

yum -y install $2 > /dev/null 2>&1

# hash $Command || { echo "`date +%F' '%H:%M:%S` $2 is installed fail">>${install_log_path}${install_log_name} ; exit 1 }

fi

}

# 判断目录是否存在,传入目录绝对路径,可以传入多个目录

check_dir() {

output_msg "目录检查"

for dirname in $*; do

[ -d $dirname ] || mkdir -p $dirname > /dev/null 2>&1

echo "`date +%F' '%H:%M:%S` $dirname check success!" >> ${install_log_path}${install_log_name}

done

}

# 下载文件并解压至安装目录,传入url链接地址

download_file() {

output_msg "下载源码包"

mkdir -p $download_path

for file in $*; do

wget $file -c -P $download_path &> /dev/null

if [ $? -eq 0 ]; then

echo "`date +%F' '%H:%M:%S` $file download success!" >> ${install_log_path}${install_log_name}

else

echo "`date +%F' '%H:%M:%s` $file download fail!" >> ${install_log_path}${install_log_name} && exit 1

fi

done

}

# 解压文件,可以传入多个压缩文件绝对路径,用空格隔开,解压至安装目录

extract_file() {

output_msg "解压源码"

for file in $*; do

if [ "${file##*.}" == "gz" ]; then

tar -zxf $file -C $install_path && echo "`date +%F' '%H:%M:%S` $file extrac success!,path is $install_path" >> ${install_log_path}${install_log_name}

elif [ "${file##*.}" == "zip" ]; then

unzip -q $file -d $install_path && echo "`date +%F' '%H:%M:%S` $file extrac success!,path is $install_path" >> ${install_log_path}${install_log_name}

else

echo "`date +%F' '%H:%M:%S` $file type error, extrac fail!" >> ${install_log_path}${install_log_name} && exit 1

fi

done

}

# 配置环境变量,第一个参数为添加环境变量的绝对路径

config_env() {

output_msg "环境变量配置"

echo "export PATH=\$PATH:$1" > ${env_file}

source ${env_file} && echo "`date +%F' '%H:%M:%S` 软件安装完成!" >> ${install_log_path}${install_log_name}

}

# 配置主机名，第一个为主机名

config_hostname() {

if [ ${sysversion} -eq 6 ]; then

hostname $1

elif [ ${sysversion} -eq 7 ]; then

hostnamectl set-hostname $1

else

echo "`date +%F' '%H:%M:%S` hostname $1 config fail" >> ${install_log_path}${install_log_name}

fi

}

config_limits() {

output_msg "配置limits"

cat >> /etc/security/limits.conf << EOF

* soft nofile 65536

* hard nofile 65536

* soft nproc 65536

* hard nproc 65536

EOF

echo "vm.max_map_count = 655360" >> /etc/sysctl.conf

sysctl -p > /dev/null 2>&1

}

# 添加配置文件

add_config() {

cat > $1 << EOF

cluster.name: my-application

node.name: ${hostname}

path.data: /usr/local/elasticsearch/data

path.logs: /usr/local/elasticsearch/logs

network.host: 127.0.0.1

http.port: 9200

discovery.zen.ping.unicast.hosts: ["$hostname"]

EOF

}

config_user() {

useradd $1 > /dev/null 2>&1

if [ $? -eq 0 ]; then

echo "`date +%F' '%H:%M:%S` $1 user add success" >> ${install_log_path}${install_log_name}

else

echo "`date +%F' '%H:%M:%S` $1 user add fail" >> ${install_log_path}${install_log_name} && exit 1

fi

chown ${1}.${1} ${install_path}elasticsearch/ -R

}

config_jvm() {

if [ ${sys_mem} -eq 0 ]; then

sed -i "s#`grep "^-Xmx" ${jvm_conf}`#"-Xmx512m"#g" ${jvm_conf}

sed -i "s#`grep "^-Xms" ${jvm_conf}`#"-Xms512m"#g" ${jvm_conf}

else

sed -i "s#`grep "^-Xmx" ${jvm_conf}`#"-Xmx${sys_mem}g"#g" ${jvm_conf}

sed -i "s#`grep "^-Xms" ${jvm_conf}`#"-Xms${sys_mem}g"#g" ${jvm_conf}

fi

}

main() {

check_dir $install_log_path $install_path

check_yum_command wget wget

download_file $URL

config_hostname $hostname

software_name=$(echo $URL | awk -F'/' '{print $NF}' | awk -F'.tar.gz' '{print $1}')

for filename in `ls $download_path`; do

extract_file ${download_path}$filename

done

rm -fr ${download_path}

ln -s $install_path$software_name ${install_path}elasticsearch

add_config $software_config_file

check_dir ${install_path}elasticsearch/{data,logs}

config_user elasticsearch

config_env ${install_path}elasticsearch/bin

config_limits

config_jvm

echo "请使用一下命令启动服务:'su - elasticsearch -c 'nohup /usr/local/elasticsearch/bin/elasticsearch &'"

}

main