Optimize shell scripts for one click account function

Author: HanTang1

dss-server/src/main/resources/linkis.properties

@@ -67,5 +67,5 @@ wds.linkis.kerberos.kdc.user.name=
 wds.linkis.kerberos.kdc.user.password=
 wds.linkis.kerberos.ssh.port=22
 wds.dss.deploy.path=/usr/local/dss_linkis
-wds.dss.user.account.command.class=com.webank.wedatasphpere.dss.user.service.impl.LinuxUserCommand,com.webank.wedatasphpere.dss.user.service.impl.LdapCommand,com.webank.wedatasphpere.dss.user.service.impl.WorkspaceCommand,com.webank.wedatasphpere.dss.user.service.impl.KerberosCommand,com.webank.wedatasphpere.dss.user.service.impl.MetastoreCommand,com.webank.wedatasphpere.dss.user.service.impl.AzkabanCommand
+wds.dss.user.account.command.class=com.webank.wedatasphpere.dss.user.service.impl.LinuxUserCommand,com.webank.wedatasphpere.dss.user.service.impl.KerberosCommand,com.webank.wedatasphpere.dss.user.service.impl.LdapCommand,com.webank.wedatasphpere.dss.user.service.impl.WorkspaceCommand,com.webank.wedatasphpere.dss.user.service.impl.MetastoreCommand,com.webank.wedatasphpere.dss.user.service.impl.AzkabanCommand
 

dss-user-manager/src/main/resources/default/AddschedulerUser.sh

@@ -15,4 +15,4 @@ if grep -i "^${user}=" $installDir/token.properties;
       fi
 else
     echo "$user=$password" >> $installDir/token.properties
-fi
+fi

dss-user-manager/src/main/resources/default/HdfsPath.sh

@@ -1,7 +1,13 @@
 #!/bin/bash
-
+source /etc/profile
 user=$1
 dir=$2
 echo $1 $2
-hdfs dfs -mkdir $dir
-hdfs dfs -chown $user:$user $dir
+id $user
+if [ $? -ne 0 ]; then
+  sudo useradd $user -s /sbin/nologin
+  echo "create user successfully"
+fi
+
+hdfs dfs -mkdir -p $dir
+hdfs dfs -chown $user:$user $dir

dss-user-manager/src/main/resources/default/Kerberos.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-
+source /etc/profile
 #需要将当前登录用户，如dss加入到sudoers
 
 #函数
@@ -62,7 +62,8 @@ gen_keytab(){
            generate_user $user
            sudo chown $user $CENTER_KEYTAB_PATH/$user.keytab
            sudo su - $user -c "kinit -kt $CENTER_KEYTAB_PATH/$user.keytab $principal"
-           sudo su - op -c "crontab -l > conf && echo '* */12 * * *  sudo -u $user kinit -kt $CENTER_KEYTAB_PATH/$user.keytab $principal' >> conf && crontab conf && rm -f conf"
+           deployUser=`whoami`
+           sudo su - $deployUser -c "crontab -l > conf && echo '* */12 * * *  sudo -u $user kinit -kt $CENTER_KEYTAB_PATH/$user.keytab $principal' >> conf && crontab conf && rm -f conf"
        fi
     else
        echo "the $user.keytab does not exist, please check your previous steps!"

dss-user-manager/src/main/resources/default/LinuxPath.sh

@@ -1,7 +1,12 @@
 #!/bin/bash
-
+source /etc/profile
 user=$1
 dir=$2
 echo $1 $2;
-sudo mkdir $dir
-sudo chown $user:$user $dir
+id $user
+if [ $? -ne 0 ]; then
+  useradd $user -s /sbin/nologin
+  echo "create user successfully"
+fi
+sudo mkdir -p $dir
+sudo chown $user:$user $dir

dss-user-manager/src/main/resources/default/Metastore.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-
+source /etc/profile
 user_name=$1
 db_name=$2
 path=$3
@@ -22,16 +22,20 @@ if [[ $? -ne 0 ]]; then
     echo "create database failed!"
 else
      #修改数据库所属，将principal用户添加到metastore侧hive-site.xml hive.users.in.admin.role中
-   hive -e "set role admin ; grant all on database $db_name to user $user_name"
-   echo "grant database $db_name successfully!"
+   if [ $kerberos_enable = "0" ]; then
+       hive -e "grant all on database $db_name to user $user_name"
+   else
+     hive -e "set role admin; grant all on database $db_name to user $user_name"
+   fi
 fi
 
   #三、hdfs操作
 if [[ $? -ne 0 ]]; then
     #回滚
   hive -e "drop database $db_name"
-  echo "rollback finished!"
+  echo "grant database failed,rollback finished!"
 else
+  echo "grant database $db_name successfully!"
     #修改hdfs路径所属
   hdfs dfs -chown $user_name:$user_name $path
     #修改hdfs路径权限