Service Name changes in Ambari

[#1]

Changing Ambari Service name (via Rest API) using a customized single script:

Create a property file (ambServActNmChg.properties) with following parameters :

export AMBARI_ADMIN_USERID=admin
export AMBARI_ADMIN_PASSWORD=admin
export AMBARI_SERVER=<Ambari Host>
export AMBARI_SERVER_PORT=<Ambari port>
export CLUSTER_NAME=<Ambari Cluster>
export NEW_SRV_USER_NAME=<New User Suffix>

Create shell script :

#!/bin/bash

echo " ****** Starting Script to Change Service User names in Ambari ***** "

#######################################################################################################
#
#  This script changes the service account user names in ambari using inputs from properties files.  ##
#
#######################################################################################################
. `dirname ${0}`/ambServActNmChg.properties

curr_date=`date +"%Y%m%d_%H%M%S"`
function check_ambari_server() {
   echo " Check and Start Ambari Server : "

   server_start=`ambari-server status |grep running|awk '{print $3}'`

   echo $server_start

   if [ "$server_start"  == "running" ]; then
       echo "Ambari server running already .. "
   else
       echo " Initiating ambari server start "
       ambari-server start
       sleep 30

       finished=0
       retries=0

       while [ $finished -ne 1 ]

       do
           server_start=`ambari-server status |grep running|awk '{print $3}'`
           echo $server_start

           if [ "$server_start"  == "running" ]; then

             finished=1
           fi
           sleep 5

           let retries=$retries+1

           if [[ $retries == 30 ]]
           then
              echo " Unable to Start Ambari Server. Please check the Ambari Server logs to determine the issue ... "
              exit 1
           fi

           echo " Polling for Ambari Server status $retries "
       done

   fi

}

function change_user_name() {

        echo "   "
        echo " Changing Username in progress .....  The New service users will be suffixed with $NEW_SRV_USER_NAME as follows : "
        echo "   "

        while read line
        do

            #####echo $line |sed 's/://g'
            newuservar=`echo $line |awk -F':' '{print $2}'`
            newuser=`echo $line |awk -F':' '{print $3}'|sed 's/"//g'`
            echo $newuservar ":" $newuser$NEW_SRV_USER_NAME

        done < amb_srv_usr_backup.txt

        echo "   "
        echo " Hit ENTER to update the user names ......: "
        echo "   "
        read input

      while read line
        do

            ###echo $line |sed 's/://g'
            envfile=`echo $line |awk -F':' '{print $1}'`
            newuservar=`echo $line |awk -F':' '{print $2}'`
            newuser=`echo $line |awk -F':' '{print $3}'|sed 's/"//g' |xargs`
            nuser=\"$newuser$NEW_SRV_USER_NAME\"
            echo "  Updating $envfile with " $newuservar "---" $nuser

            setuser=`echo "/var/lib/ambari-server/resources/scripts/configs.sh -u $AMBARI_ADMIN_USERID -p $AMBARI_ADMIN_PASSWORD set  $AMBARI_SERVER ${CLUSTER_NAME} $envfile $newuservar $nuser"`

            eval $setuser

        done < amb_srv_usr_backup.txt

       echo "    "
       echo " Update Completed. Validating new users ... "
       echo "    "

      while read line
        do
            envfile=`echo $line |awk -F':' '{print $1}'`
            /var/lib/ambari-server/resources/scripts/configs.sh -u $AMBARI_ADMIN_USERID -p $AMBARI_ADMIN_PASSWORD get  $AMBARI_SERVER ${CLUSTER_NAME} $envfile |grep user\"

        done < amb_srv_usr_backup.txt
}
function get_user_name() {

        check_ambari_server

	envs=`curl -u $AMBARI_ADMIN_USERID:$AMBARI_ADMIN_PASSWORD -X GET http://$AMBARI_SERVER:$AMBARI_SERVER_PORT/api/v1/clusters/${CLUSTER_NAME}?fields=Clusters/desired_co... |grep -env |awk -F'"' '{print $2}'`

	envvars=`echo "$envs"`

        cluster_env="cluster-env"

        NEWLINE=\n'

        envvars=`echo $envvars ${NEWLINE}  $cluster_env`

        ###   echo $envvars

        rm -f amb_srv_usr_backup.txt
        echo "     "
        echo "     "

        echo " ------------------------------------------------------------------------------------------------ "
        echo " NOTE: Current Ambari User List below: They will be backed up to the file amb_srv_usr_backup.txt  "
        echo " ------------------------------------------------------------------------------------------------ "

	for env in $envvars
	do

	   userlist=`/var/lib/ambari-server/resources/scripts/configs.sh -u $AMBARI_ADMIN_USERID -p $AMBARI_ADMIN_PASSWORD get  $AMBARI_SERVER ${CLUSTER_NAME} $env |grep user\" |grep ':'`

           ####echo $userlist

            if [ "$userlist" != "" ]; then

 	        ulist=$(echo "$userlist" | sed 's/,//g')

                echo "$ulist"
                printf '%s\n' "$ulist"| while IFS= read ul
                do
                   echo $env ": " $ul >> amb_srv_usr_backup.txt
                done

            fi

	done
        echo "     "
        echo "     "

        echo " Backing up the exiting config for furture Restore amb_srv_usr_backup_$curr_date.txt ... "

        cp amb_srv_usr_backup.txt amb_srv_usr_backup_$curr_date.txt

        response=0

        while [ $response -ne 1 ]
        do
            echo  "    "
            echo  " About to Change the Service account user names ... Response is CASE SENSITIVE YES or NO .... Proceed (YES/NO) ??    "
            echo  "    "
            read resp

            if  ([ $resp == "YES" ] || [ $resp == "NO" ]); then
                echo " Response provided is " $resp
                if [ $resp == "YES" ]; then
                      change_user_name
                else
                      echo " Ambari USer Service account change ABORTED ... "
                fi
                response=1

            else
                echo " Response provided is " $resp
                reponse=0
            fi

        done

}
function restore_change_user_name() {

      echo "   "
      echo " Changing Username in progress ..... : "
      echo "   "

      while read line
      do

            newuservar=`echo $line |awk -F':' '{print $2}'`
            newuser=`echo $line |awk -F':' '{print $3}'|sed 's/"//g'`
            echo $newuservar ":" $newuser

      done < amb_srv_usr_backup_RESTORE.txt

      echo "   "
      echo " Hit ENTER to update the user names ......: "
      echo "   "
      read input

      while read line
        do

            ###echo $line |sed 's/://g'
            envfile=`echo $line |awk -F':' '{print $1}'`
            newuservar=`echo $line |awk -F':' '{print $2}'`
            newuser=`echo $line |awk -F':' '{print $3}'|sed 's/"//g' |xargs`
            nuser=\"$newuser\"
            echo "  Updating $envfile with " $newuservar "---" $nuser

            setuser=`echo "/var/lib/ambari-server/resources/scripts/configs.sh -u $AMBARI_ADMIN_USERID -p $AMBARI_ADMIN_PASSWORD set  $AMBARI_SERVER ${CLUSTER_NAME} $envfile $newuservar $nuser"`

            eval $setuser

      done < amb_srv_usr_backup_RESTORE.txt

      echo "    "
      echo " Update Completed. Validating new users ... "
      echo "    "

      while read line
      do
            envfile=`echo $line |awk -F':' '{print $1}'`
            /var/lib/ambari-server/resources/scripts/configs.sh -u $AMBARI_ADMIN_USERID -p $AMBARI_ADMIN_PASSWORD get  $AMBARI_SERVER ${CLUSTER_NAME} $envfile |grep user\"

      done < amb_srv_usr_backup_RESTORE.txt

}
function restore_user_name(){

      echo "Make sure the file to be restore is named as amb_srv_usr_backup_RESTORE.txt ... Enter to proceed "
      read input

      restore_change_user_name

}
#### Main code Starts HERE ####
if [ "$1" == "UPDATE" ]; then

   get_user_name

   echo " Deleting residual .json file on the local folder !!! "

   rm -f *.json
elif [ "$1" == "RESTORE" ]; then

   restore_user_name

   echo " Deleting residual .json file on the local folder !!! "

   rm -f *.json

else

   echo "                                                                        "
   echo "                                                                        "
   echo "Usage: ./ambServActNmChg.sh [UPDATE] [RESTORE]                          "
   echo "                                                                        "
   echo "                                                                        "
   exit 1

fi

Execution:

./ambServActNmChg.sh [UPDATE] [RESTORE]    

The update process backup file with the existing user names. We can use the file created to restore back to original configuration.

Note:

This script needs to be executed on the Ambari Server.