276 lines
7.1 KiB
Bash
276 lines
7.1 KiB
Bash
#!/bin/bash
|
||
|
||
# 定义颜色
|
||
readonly RED='\033[0;31m'
|
||
readonly GREEN='\033[0;32m'
|
||
readonly YELLOW='\033[1;33m'
|
||
readonly BLUE='\033[0;34m'
|
||
readonly NC='\033[0m' # No Color
|
||
|
||
# 定义常量
|
||
readonly GO_VERSION=${1:-"1.23.4"} # 默认版本为 1.23.4
|
||
readonly GO_ROOT="/usr/local/go"
|
||
readonly GO_WORKSPACE="/usr/local/go_workspace"
|
||
readonly GO_URL="https://dl.google.com/go"
|
||
readonly TEMP_DIR="/tmp/go_install_$$" # 使用 PID 创建唯一临时目录
|
||
readonly LOG_FILE="/tmp/go_install_$$.log"
|
||
|
||
# 日志函数
|
||
log_info() {
|
||
echo -e "${GREEN}[INFO]${NC} $1"
|
||
echo "[$(date '+%Y-%m-%d %H:%M:%S')] [INFO] $1" >> "$LOG_FILE"
|
||
}
|
||
|
||
log_warn() {
|
||
echo -e "${YELLOW}[WARN]${NC} $1"
|
||
echo "[$(date '+%Y-%m-%d %H:%M:%S')] [WARN] $1" >> "$LOG_FILE"
|
||
}
|
||
|
||
log_error() {
|
||
echo -e "${RED}[ERROR]${NC} $1"
|
||
echo "[$(date '+%Y-%m-%d %H:%M:%S')] [ERROR] $1" >> "$LOG_FILE"
|
||
}
|
||
|
||
# 清理函数
|
||
cleanup() {
|
||
log_info "清理临时文件..."
|
||
rm -rf "$TEMP_DIR"
|
||
if [ $? -ne 0 ]; then
|
||
log_warn "清理临时文件失败，请手动删除: $TEMP_DIR"
|
||
fi
|
||
}
|
||
|
||
# 错误处理
|
||
handle_error() {
|
||
log_error "安装过程中发生错误，请检查日志文件: $LOG_FILE"
|
||
cleanup
|
||
exit 1
|
||
}
|
||
|
||
# 设置错误处理
|
||
trap 'handle_error' ERR
|
||
|
||
# 检查是否为 root 用户
|
||
check_root() {
|
||
if [ "$EUID" -ne 0 ]; then
|
||
log_error "请使用 root 权限运行此脚本"
|
||
echo "使用: sudo bash $0 [version]"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
# 检查版本格式
|
||
check_version() {
|
||
if [ -z "$1" ]; then
|
||
log_info "未指定版本，将安装默认版本: $GO_VERSION"
|
||
return
|
||
fi
|
||
|
||
if ! echo "$GO_VERSION" | grep -qE '^[0-9]+\.[0-9]+\.[0-9]+$'; then
|
||
log_error "无效的版本格式: $GO_VERSION"
|
||
echo "正确的格式例如: 1.23.4, 1.22.0"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
# 检测操作系统
|
||
detect_os() {
|
||
if [ -f /etc/os-release ]; then
|
||
. /etc/os-release
|
||
OS_NAME=$NAME
|
||
OS_VERSION=$VERSION_ID
|
||
log_info "检测到系统: $OS_NAME $OS_VERSION"
|
||
else
|
||
log_error "不支持的系统类型"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
# 安装依赖
|
||
install_dependencies() {
|
||
log_info "安装必要的依赖..."
|
||
case "$OS_NAME" in
|
||
*"Ubuntu"*|*"Debian"*)
|
||
apt update -qq
|
||
apt install -y wget tar curl
|
||
;;
|
||
*"CentOS"*|*"Red Hat"*)
|
||
yum install -y wget tar curl
|
||
;;
|
||
*"Fedora"*)
|
||
dnf install -y wget tar curl
|
||
;;
|
||
*"openSUSE"*)
|
||
zypper install -y wget tar curl
|
||
;;
|
||
*"Arch"*)
|
||
pacman -Sy --noconfirm wget tar curl
|
||
;;
|
||
*)
|
||
log_error "不支持的系统类型: $OS_NAME"
|
||
exit 1
|
||
;;
|
||
esac
|
||
}
|
||
|
||
# 下载 Go
|
||
download_go() {
|
||
log_info "准备下载 Go $GO_VERSION..."
|
||
|
||
# 检查版本是否存在
|
||
local check_url="$GO_URL/go$GO_VERSION.linux-amd64.tar.gz"
|
||
if ! curl --output /dev/null --silent --head --fail "$check_url"; then
|
||
log_error "版本 $GO_VERSION 不存在或无法访问"
|
||
echo "请检查版本号是否正确，或访问 https://golang.org/dl/ 查看可用版本"
|
||
exit 1
|
||
fi
|
||
|
||
mkdir -p "$TEMP_DIR"
|
||
cd "$TEMP_DIR" || exit 1
|
||
|
||
local archive="go$GO_VERSION.linux-amd64.tar.gz"
|
||
log_info "正在下载 $archive ..."
|
||
if ! wget -q "$GO_URL/$archive"; then
|
||
log_error "下载失败，请检查网络连接或版本号是否正确"
|
||
exit 1
|
||
fi
|
||
|
||
# 验证下载文件
|
||
if [ ! -f "$archive" ]; then
|
||
log_error "下载文件不存在"
|
||
exit 1
|
||
fi
|
||
|
||
log_info "下载完成"
|
||
}
|
||
|
||
# 在主要函数之前添加检测函数
|
||
check_existing_go() {
|
||
if command -v go &> /dev/null; then
|
||
local current_version
|
||
current_version=$(go version | awk '{print $3}' | sed 's/go//')
|
||
echo -e "${YELLOW}检测到系统已安装 Go ${current_version}${NC}"
|
||
echo -n "是否继续安装 Go ${GO_VERSION}？[y/N] "
|
||
read -r answer
|
||
if [[ ! "$answer" =~ ^[Yy]$ ]]; then
|
||
echo -e "${GREEN}安装已取消${NC}"
|
||
exit 0
|
||
fi
|
||
|
||
echo -e "${YELLOW}将备份现有安装到: ${GO_ROOT}_backup_$(date '+%Y%m%d_%H%M%S')${NC}"
|
||
echo -n "是否继续？[y/N] "
|
||
read -r answer
|
||
if [[ ! "$answer" =~ ^[Yy]$ ]]; then
|
||
echo -e "${GREEN}安装已取消${NC}"
|
||
exit 0
|
||
fi
|
||
fi
|
||
}
|
||
|
||
# 修改安装函数
|
||
install_go() {
|
||
log_info "安装 Go..."
|
||
|
||
# 备份已存在的 Go 安装
|
||
if [ -d "$GO_ROOT" ]; then
|
||
local backup_dir="${GO_ROOT}_backup_$(date '+%Y%m%d_%H%M%S')"
|
||
log_info "备份现有安装到: $backup_dir"
|
||
if ! mv "$GO_ROOT" "$backup_dir"; then
|
||
log_error "备份失败，无法继续安装"
|
||
exit 1
|
||
fi
|
||
log_info "备份完成"
|
||
fi
|
||
|
||
# 解压安装
|
||
log_info "正在解压安装..."
|
||
tar -C /usr/local -xzf "go$GO_VERSION.linux-amd64.tar.gz"
|
||
if [ ! -d "$GO_ROOT" ]; then
|
||
log_error "安装失败，目录不存在: $GO_ROOT"
|
||
exit 1
|
||
fi
|
||
|
||
# 创建工作空间
|
||
mkdir -p "$GO_WORKSPACE"/{src,pkg,bin}
|
||
chmod -R 777 "$GO_WORKSPACE"
|
||
}
|
||
|
||
# 配置环境变量
|
||
setup_environment() {
|
||
log_info "配置环境变量..."
|
||
|
||
# 创建环境变量配置文件
|
||
cat > /etc/profile.d/go.sh << EOF
|
||
export GOROOT=$GO_ROOT
|
||
export GOPATH=$GO_WORKSPACE
|
||
export PATH=\$PATH:\$GOROOT/bin:\$GOPATH/bin
|
||
EOF
|
||
|
||
chmod 644 /etc/profile.d/go.sh
|
||
|
||
# 立即生效
|
||
source /etc/profile.d/go.sh
|
||
}
|
||
|
||
# 验证安装
|
||
verify_installation() {
|
||
log_info "验证安装..."
|
||
|
||
if ! command -v go &> /dev/null; then
|
||
log_error "Go 命令不可用，安装似乎失败了"
|
||
exit 1
|
||
fi
|
||
|
||
local installed_version
|
||
installed_version=$(go version | awk '{print $3}' | sed 's/go//')
|
||
if [ "$installed_version" != "$GO_VERSION" ]; then
|
||
log_error "版本不匹配: 预期 $GO_VERSION, 实际 $installed_version"
|
||
exit 1
|
||
fi
|
||
|
||
log_info "Go $GO_VERSION 安装成功!"
|
||
}
|
||
|
||
# 显示完成信息
|
||
show_completion() {
|
||
echo
|
||
echo -e "${GREEN}===============================================${NC}"
|
||
echo -e "${BLUE}Go $GO_VERSION 安装完成！${NC}"
|
||
echo -e "${GREEN}-----------------------------------------------${NC}"
|
||
echo -e "请执行以下步骤完成配置："
|
||
echo
|
||
echo -e "1. 重新加载环境变量:"
|
||
echo -e " ${YELLOW}source /etc/profile${NC}"
|
||
echo
|
||
echo -e "2. 配置 Go 模块和代理 (推荐):"
|
||
echo -e " ${YELLOW}go env -w GO111MODULE=on${NC}"
|
||
echo -e " ${YELLOW}go env -w GOPROXY=https://goproxy.cn,direct${NC}"
|
||
echo
|
||
echo -e "3. 验证安装:"
|
||
echo -e " ${YELLOW}go version${NC}"
|
||
echo -e "${GREEN}===============================================${NC}"
|
||
|
||
log_info "安装日志已保存至: $LOG_FILE"
|
||
}
|
||
|
||
# 主函数
|
||
main() {
|
||
log_info "开始安装 Go $GO_VERSION..."
|
||
|
||
check_root
|
||
check_version
|
||
check_existing_go
|
||
detect_os
|
||
install_dependencies
|
||
download_go
|
||
install_go
|
||
setup_environment
|
||
verify_installation
|
||
cleanup
|
||
show_completion
|
||
}
|
||
|
||
# 执行主函数
|
||
main
|
||
|