#!/bin/bash
#
#***************************************************************************
# Author: liwanggui
# Email: liwanggui@163.com
# Date: 2021-09-30
# FileName: mongodb.sh
# Description: This is a test script
# Copyright (C): 2021 All rights reserved
#***************************************************************************
#

SourceDir="/usr/local/src"
InstallDir="/usr/local"

MongoDownloadUrl="https://fastdl.mongodb.org/linux/mongodb-linux-x86_64-rhel70-3.6.23.tgz"
MongoPkgName=$(basename "$MongoDownloadUrl")
MongoName="${MongoPkgName%.tgz}"

MongoPort=27017


# 分片
read -p "Do you want to configure a sharded cluster? [y/N] " -n 1 -r IsConfig
echo 

if [[ "${IsConfig}" == "y" ]] || [[ "${IsConfig}" == "Y" ]]; then
    echo "Please select the role of the sharded cluster"
    select clusterRole in shardsvr configsvr
    do
        if [[ "${clusterRole}" ]]; then
            break 
        fi
    done

    read -p "Please specify a replication set name:" -r replSetName
else
  # 复制集
  read -p "Do you want to configure a replication set? [y/N] " -n 1 -r IsConfig
  echo 

  if [[ "${IsConfig}" == "y" ]] || [[ "${IsConfig}" == "Y" ]]; then
      read -p "Please specify a replication set name:" -r replSetName
  fi

  unset IsConfig
fi

# 错误检测
if [[ "$clusterRole" == "configsvr" ]] && [[ $replSetName != "configReplSet"  ]]; then
    echo "The name of the mongo config replication set can only be configReplSet"
    echo "Set the name of the replication set to configReplSet"
    replSetName="configReplSet"
fi

if [[ "$clusterRole" == "shardsvr" ]] && [[ $replSetName == "configReplSet"  ]]; then
    echo "The name of the shard replication set cannot be configured as configReplSet"
    echo "Please reset."
    exit
fi


## 获取本地网卡 ip 
echo "Please select your local network interface: "
select inter in $(sed -n '3,$p' /proc/net/dev | awk -F':' '{print $1}' | grep -v 'lo'); do 
    if [[ "${inter}" ]]; then
        break 
    fi
done

lanIP=$(ip addr show "${inter}" | grep "${inter}\$" | grep -E -o '[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}' | head -n1)
echo "${inter}: $lanIP"


## 下载 MongoDB
cd "$SourceDir" || exit

if [[ ! -f "$MongoPkgName" ]]; then
    curl -o "$MongoPkgName" "$MongoDownloadUrl"
fi

## 安装

if [[ ! -d "$(readlink -f ${InstallDir}/mongodb)" ]]; then
    tar xzf "$MongoPkgName" -C "$InstallDir"
    cd "$InstallDir" || exit
    ln -sfn "$MongoName" mongodb
    ln -s "${InstallDir}/mongodb/bin/*" /usr/local/bin/
fi

## 创建 mongod 用户
if ! /usr/bin/id -g mongod &>/dev/null; then
    /usr/sbin/groupadd -r mongod
fi

if ! /usr/bin/id mongod &>/dev/null; then
    /usr/sbin/useradd -M -r -g mongod -d /data/mongo -s /bin/false -c mongod mongod > /dev/null 2>&1
fi

## 创建相应目录
mkdir -p /etc/mongo
mkdir -p /var/log/mongodb
mkdir -p /var/run/mongodb
mkdir -p /data/mongo/mongod

chown -R mongod.mongod /var/log/mongodb
chown -R mongod.mongod /var/run/mongodb
chown -R mongod.mongod /data/mongo


## 生成 mongod 配置文件
cat >/etc/mongo/mongod.conf<<EOF
# mongod.conf
# for documentation of all options, see:
#   http://docs.mongodb.org/manual/reference/configuration-options/

# where to write logging data.
systemLog:
  destination: file
  logAppend: true
  path: /var/log/mongodb/mongod.log

# Where and how to store data.
storage:
  journal:
    enabled: true
  dbPath: /data/mongo/mongod
  directoryPerDB: true
  #engine: wiredTiger
  wiredTiger:
    engineConfig:
      cacheSizeGB: 1
      directoryForIndexes: true
    collectionConfig:
      blockCompressor: zlib
    indexConfig:
      prefixCompression: true

# how the process runs
processManagement:
  fork: true  # fork and run in background
  pidFilePath: /var/run/mongodb/mongod.pid  # location of pidfile
  timeZoneInfo: /usr/share/zoneinfo

# network interfaces
net:
  port: ${MongoPort}
  bindIp: 127.0.0.1,${lanIP} # Listen to local interface only, comment to listen on all interfaces.

security:
  keyFile: /etc/mongo/mongod.key

EOF

## 复制集配置
if [[ "$replSetName" ]] && [[ "${replSetName}x" != "x" ]]; then
  cat >/etc/mongo/mongod.conf<<EOF

replication:
  oplogSizeMB: 2048
  # config 复制集名称为: configReplSet
  replSetName: configReplSet

EOF

fi

## 分片配置
if [[ "$clusterRole" ]] && [[ "${clusterRole}x" != "x" ]]; then
  cat >/etc/mongo/mongod.conf<<EOF
sharding:
  clusterRole: configsvr
EOF

fi

## 生成验证 key
# openssl rand -base64 756
##

cat >/etc/mongo/mongod.key<<EOF
ry2cCv+u8AuoVsKmQqmwOXeP2nfH+CG+VD9dMpUUHcggztWQ09VDyXJTXhGgBnYu
a9UH3jvfbmT7mqRV2esOxSnTazcY5mXjyMCMzgaSvpN8OtfuDZXX20v5qpYCqjCj
xlLyyDkeFhOeTqWJcQKvwhZnh0SXC65QwGxeEL8DRnLXziBAooR4EYkRGgZnGvmP
23KRhagRgQ7iwivwiXoepf+PwtZ3DKi2wudUTcojvFPa+ls5rhXSnEGChMPlL8zV
b+xGDdOe4YDOiDoAATYLRkDUfUpG8DF921azQ4WH1g7BJU/zY2CpAX8nd4lhBA7o
6YD51kR6odBGUGGarZOvk7tW4DY+sWSK5pbc7nttFICUcw87K1R9Hxe1wEVNqo/m
xcQ0vtgJ7IvHRwIbvYxmEoEfnBzqOkfZLXKgE/tgEvphdoiUCCyJ89JrkN3W8o2Q
yb1RuvEOPE+8MjyE1vKkwuwnn95ePle8AJ8+fb3W5p5n2lNpBYrbe2CfWjRVMqri
bnnb93DsG5Qd4U+QYJ/ZSnAofPdSLRxQn54MLIPHRuPlpJuCQXeB0bnQeGbTMjaO
2aaB+kOpwfWMcQBqvdErdQqeFgITFEGTG1sa92Nc+9zO07R6FQ7qveKTOMoH1hLB
chfBPAt5pck50h3DtGMZgmhCzR2606VOVgiOw4s4UQoG28fvA9Fhj/xA2qIFcloz
Yz9urc0bFMoOQOVPCmClFbUTbG+Jp1O1yc0RbXyBvecPs4BQNdSd+K1KE5BcWGsy
RkgmsNUEaPMIqiT7SVYxny59UmegW1uiict3orJtOEAuRlEDAWn6E0E6FYWoy9Qk
CYk+X7kB1CF15/KYGCsADxPf5YZy1UpWjiJtWYozKGwH/Ri1dLs3CMUxbNPArcBV
THh3GHz4gE31/cZpRSux3LN/uLOEyCQvwHmdwngDantN52Ma/KwQBcbqBGeoTUq3
4VhBTVzaOng/F1t+XAZtrXB0XM6MsztAvYP2Brpw/54DSBTM
EOF

chmod 400 /etc/mongo/mongod.key
chown mongod.mongod /etc/mongo/mongod.key

## 生成 systemd 服务单元配置
cat > /usr/lib/systemd/system/mongod.service <<EOF
[Unit]
Description=MongoDB Database Server
Documentation=https://docs.mongodb.org/manual
After=network.target

[Service]
User=mongod
Group=mongod
Environment="OPTIONS=-f /etc/mongo/mongod.conf"
ExecStart=/usr/local/mongodb/bin/mongod \$OPTIONS
ExecStartPre=/usr/bin/mkdir -p /var/run/mongodb
ExecStartPre=/usr/bin/chown mongod:mongod /var/run/mongodb
ExecStartPre=/usr/bin/chmod 0755 /var/run/mongodb
PermissionsStartOnly=true
PIDFile=/var/run/mongodb/mongod.pid
Type=forking
# file size
LimitFSIZE=infinity
# cpu time
LimitCPU=infinity
# virtual memory size
LimitAS=infinity
# open files
LimitNOFILE=65535
# processes/threads
LimitNPROC=65535
# locked memory
LimitMEMLOCK=infinity
# total threads (user+kernel)
TasksMax=infinity
TasksAccounting=false
# Recommended limits for for mongod as specified in
# http://docs.mongodb.org/manual/reference/ulimit/#recommended-settings

[Install]
WantedBy=multi-user.target
EOF

## 设置为开机自启并启动
systemctl enable mongod.service --now