Linux自动安装JDK的shell脚本

Linux自动安装JDK的shell脚本

A：本脚本运行的机器，Linux

B：待安装JDK的机器, Linux

首先在脚本运行的机器A上确定可以ssh无密码登录到待安装jdk的机器B上，然后就可以在A上运行本脚本:

$ ./install-jdk.sh B的IP

or:

$ ./install-jdk.sh "B的IP" "JDK的URI"

就可以在机器B上安装JDK。jdk使用的tar包需要用户自己设定DEFAULT_JDK_SRC=？，保证可以wget得到即可。下面是全部脚本内容：

#!/bin/bash
#
# @file
#   install-jdk.sh
#
# @date
#   2013-12-19
#
# @author
#   cheungmine
#
# @version
#   0.0.1pre
#
# @usage:
#   ./install-jdk.sh 192.168.122.206
#
################################################################################
. common.sh

#***********************************************************
# install_jdk
#   install jdk on machine: /usr/local/lib
#
# Parameters:
#   machine - root@ipaddr
#   jdkUri  - uri for fetching tarball
#
# Example:
#
#   install_jdk root@192.168.122.206 ftp://vm-ftp/pub/tarball/jdk-7u67-linux-x64.tar.gz
#
#***********************************************************
. common.sh

# YOU MIGHT CHANGE BELOW LINE TO GET YOUR JDK TARBALL:
DEFAULT_JDK_SRC="ftp://vm-ftp/pub/tarball/jdk-7u67-linux-x64.tar.gz"

# DO NOT CHANGE BELOW TWO LINES:
INSTALL_DIR="/usr/local/lib/java"
LOCAL_DIR="./.tmp"

function install_jdk() {
    echo -e "<INFO> install jdk on machine: $1"

    local DEST_LOGIN=$1
    local JDK_URI=$2
    local TAR=$(basename $JDK_URI)

    echo -e "<INFO> jdk: '$JDK_URI'"

    wget -c $JDK_URI -P $LOCAL_DIR -O $LOCAL_DIR/$TAR

    $(is_empty_dir "$LOCAL_DIR/jdk_untar")
    local ret=$?

    case $ret in
    $DIR_NOT_EXISTED)
        mkdir -p $LOCAL_DIR/jdk_untar
        ;;
    $DIR_IS_EMPTY)
        ;;
    $DIR_NOT_EMPTY)
        rm -rf $LOCAL_DIR/jdk_untar/*
        ;;
    *)
        exit $ERR_FATAL_ERROR
        ;;
    esac

    # untar to jdk_untar
    tar -zxf $LOCAL_DIR/$TAR -C $LOCAL_DIR/jdk_untar

    $(is_empty_dir "$LOCAL_DIR/jdk_untar")
    local ret=$?

    if [ "$ret" -eq "$DIR_NOT_EMPTY" ]; then
        local jdk_home=`ls $LOCAL_DIR/jdk_untar 2>/dev/null`
        echo $jdk_home
    else
        exit $ERR_FATAL_ERROR
    fi

    echo -e "<INFO> create folder on: $DEST_LOGIN:$INSTALL_DIR"
    local ret=`ssh $DEST_LOGIN "mkdir $INSTALL_DIR"`

    echo -e "<INFO> copy $jdk_home/ to: $DEST_LOGIN:$INSTALL_DIR/"
    local ret=`scp -r $LOCAL_DIR/jdk_untar/$jdk_home $DEST_LOGIN:$INSTALL_DIR`

    # remove local tar
    rm -rf $LOCAL_DIR/jdk_untar

    local DEST_JAVA_HOME=$INSTALL_DIR/$jdk_home

    echo -e "<TODO> remove old settings for install_jdk in /etc/profile"

    echo -e "<INFO> set /etc/profile: JAVA_HOME=$DEST_JAVA_HOME"
    local ret=`ssh $DEST_LOGIN "echo '' >> /etc/profile"`
    local ret=`ssh $DEST_LOGIN "echo '#!{{install_jdk@hgdb.net==>' >> /etc/profile"`

    local ret=`ssh $DEST_LOGIN "echo 'export JAVA_HOME=$DEST_JAVA_HOME' >> /etc/profile"`
    local ret=`ssh $DEST_LOGIN "echo 'export CLASSPATH=.:\\$JAVA_HOME/lib/tools.jar:\\$JAVA_HOME/lib/dt.jar' >> /etc/profile"`
    local ret=`ssh $DEST_LOGIN "echo 'export PATH=\\$JAVA_HOME/bin:\\$JAVA_HOME/jre/bin:\\$PATH' >> /etc/profile"`

    local ret=`ssh $DEST_LOGIN "echo '#!<==install_jdk@hgdb.net}}'>> /etc/profile"`

    local ret=`ssh $DEST_LOGIN ". /etc/profile"`
}

function uninstall_jdk() {
    echo -e "<TODO> uninstall jdk from: $1"
}

#=======================================================================
# ---- main() ----
if [ -n $1 ]; then
    DEST_IP=$1
    JDK_SRC=$DEFAULT_JDK_SRC

    if [ $# == 2 ]; then
        JDK_SRC=$2
    fi

    echo -e "<INFO> install jdk on '$DEST_IP', jdk: '$JDK_SRC'"

    install_jdk "root@$DEST_IP" "$JDK_SRC"
fi

此外使用到了common.sh这个脚本：

#!/bin/bash
#
# @file
#   common.sh
#
# @date
#   2013-12-18
#
# @author
#   cheungmine
#
# @version
#   0.0.1pre
#
########################################################################
# Error Codes:
#
ERR_NOERROR=0

ERR_FATAL_ERROR=-100

# current user is not a root
ERR_NOT_ROOT=100

# file is already existed
ERR_FILE_EXISTED=200

#***********************************************************
# chk_root
#   check if is a root user
#
# Example:
#   chk_root
#***********************************************************
function chk_root() {
    if [ $UID -ne 0 ]; then
        echo -e "<chk_root> current user is not a root.\n" \
            "   Because you will run all the steps from this shell with root" \
            "privileges,\n" \
            "   you must become root right by typing:\n" \
            "   $ sudo su"
        exit $ERR_NOT_ROOT
    fi
}

#***********************************************************
# read_cfg
#   read section and key from ini file
#
# Example:
#   value=$(read_cfg $CFG_FILE $SEC_NAME 'key')
#***********************************************************
function read_cfg() {
    CFGFILE=$1
    SECTION=$2
    KEYNAME=$3

    RETVAL=`awk -F '=' '/\['$SECTION'\]/{a=1}a==1&&$1~/'$KEYNAME'/{print $2;exit}' $CFGFILE`
    echo $RETVAL | tr -d '"'
}

#***********************************************************
# write_cfg
#   write section and key to ini file
#
# Example:
#   $(write_cfg $CFG_FILE 'secname' 'key' $value)
#***********************************************************
function write_cfg() {
    CFGFILE=$1
    SECTION=$2
    KEYNAME=$3
    NEWVAL=$4

    RETVAL=`sed -i "/^\[$SECTION\]/,/^\[/ {/^\[$SECTION\]/b;/^\[/b;s/^$KEYNAME*=.*/$KEYNAME=$NEWVAL/g;}" $CFGFILE`
    echo $RETVAL
}

#***********************************************************
# real_path
#   get real path of given relative path
#
# Example:
#   ABS_PATH=$(real_path $(dirname $0))
#***********************************************************
function real_path() {
    \cd "$1"
    /bin/pwd
}

#***********************************************************
# copy_file
#   copy source file to destigation file without overwriting
#
# Example:
#   copy_file $file1 $file2
#***********************************************************
function copy_file() {
    src=$1
    dst=$2

    if [ -a $dst ]
    then
        echo -e "<copy_file> file is already existed: '$dst'"
        exit $ERR_FILE_EXISTED
    fi

    echo -e "<copy_file> '$src' =>\n            '$dst'"

    dstdir=$(dirname $dst)

    if [ -d $dstdir ]
    then
        :
    else
        mkdir -p $dstdir
    fi

    cp -p $src $dst
}

#***********************************************************
# is_empty_dir
#    if a dir is empty
# Returns:
#    0 - not empty dir
#    1 - is empty dir
#    2 - dir not existed
# Example:
#    is_empty_dir ~/workspace
#    echo $?
#
#    $(is_empty_dir './'
#    echo $?
#***********************************************************
DIR_NOT_EXISTED=2
DIR_IS_EMPTY=1
DIR_NOT_EMPTY=0

function is_empty_dir() {
    local olddir=$PWD
    local indir=$1

    if [ ! -d "$indir" ]; then
        return $DIR_NOT_EXISTED
    fi

    cd $indir
    local files=`ls 2>/dev/null`
    cd $olddir

    if [ -n "$files" ]; then
        return $DIR_NOT_EMPTY
    else
        return $DIR_IS_EMPTY
    fi
}