improve script

init.sh → machine-init.sh

@@ -2,12 +2,14 @@
 #王庭威
 #2020/1/7
 local_ip=192.168.137.200
-disk=
-##################################################不常改变的变量####################
+nfs_disk=
+data_disk=/dev/sdb
+data_mount_path=/data
+##################################################不常改变的变量##############################
 repo_ip=192.168.1.145
 repo_host=apt.offline.com
 end_ip=`echo $repo_ip |awk -F "." '{print $4}'`
-##################################################基础函数#################################
+##################################################基础函数###################################3
 base() {
 echo "##################设置修改内网apt源############################################"
 mv /etc/apt/sources.list /etc/apt/sources.list.bak
@@ -23,14 +25,47 @@ passwd root
 echo "设置节点dns"
 rm -rf /etc/resolv.conf && echo "nameserver 114.114.114.114" >>/etc/resolv.conf
 }
-###########################################################################################
+#############################################data数据盘挂载基础函数##################
+mount-data(){
+read -p "挂载data的数据盘如果超过2T、请输入parted、否则输入fdisk" partition_method
+if [ $partition_method = "fdisk"];then
+echo "使用fdisk方法分区"
+fdisk $data_disk
+else 
+echo "使用parted方法分区"
+parted $data_disk
+echo "######################开始数据盘分区##################################"
+mkfs.ext4 ${data_disk}1
+echo "#####################创建数据盘挂载点目录############################"
+mkdir $data_mount_path
+echo "#########################挂载数据盘到挂载点#########################"
+mount ${data_disk}1 $data_mount_path
+echo "##########################根据UUID设置自动挂载######################"
+uuid=`blkid |grep {data_disk}1 |awk -F '"' '{print $2}'`
+echo "UUID=$uuid $data_mount_path ext4 defaults 0 0" >>/etc/fstab 
+}
+
+#########################################上层函数##################################
 cpu() {
+echo "#####################执行base底层函数###########################"
 base
 echo "#################设置本机主机名#############################################"
 hostname seetaas-cpu-$end_ip
 echo "seetaas-cpu-$end_ip" >>/etc/hostname
 echo "#################设置本机hosts#############################################"
 echo "$local_ip seetaas-cpu-$end_ip" >>/etc/hosts
+echo "#########################执行挂载数据盘操作############################"
+mount-data
+}
+gpu() {
+base
+echo "#################设置本机主机名#############################################"
+hostname seetaas-gpu-$end_ip
+echo "seetaas-gpu-$end_ip" >>/etc/hostname
+echo "#################设置本机hosts#############################################"
+echo "$local_ip seetaas-gpu-$end_ip" >>/etc/hosts
+echo "#########################执行挂载数据盘操作#######################"
+mount-data
 }
 gpu() {
 base
@@ -40,9 +75,6 @@ echo "seetaas-gpu-$end_ip" >>/etc/hostname
 echo "#################设置本机hosts#############################################"
 echo "$local_ip seetaas-gpu-$end_ip" >>/etc/hosts
 }
-
-
-
 storage() {
 base
 echo "#################设置本机主机名#############################################"
@@ -50,7 +82,31 @@ hostname seetaas-storage-$end_ip
 echo "seetaas-storage-$end_ip" >>/etc/hostname
 echo "#################设置本机hosts#############################################"
 echo "$local_ip seetaas-storage-$end_ip" >>/etc/hosts
-
-
-
 }
+####################################################################################
+case $1 in
+cpu)
+}
+storage() {
+base
+echo "#################设置本机主机名#############################################"
+hostname seetaas-storage-$end_ip
+echo "seetaas-storage-$end_ip" >>/etc/hostname
+echo "#################设置本机hosts#############################################"
+echo "$local_ip seetaas-storage-$end_ip" >>/etc/hosts
+}
+####################################################################################
+case $1 in
+cpu)
+  cpu
+  ;;
+gpu)
+  gpu
+  ;;
+storage)
+  storage
+  ;;
+*)
+  echo "please input (gpu|cpu|storage)"
+esac
+