175 lines
4.9 KiB
Bash
175 lines
4.9 KiB
Bash
#!/bin/bash
|
||
|
||
# 脚本描述：安装Golang并设置国内代理源
|
||
# 适用环境：Linux系统（支持Ubuntu、CentOS等主流发行版）
|
||
# Go版本：1.24.2
|
||
# 下载源：阿里云镜像
|
||
|
||
# 变量定义
|
||
GO_VERSION="go1.24.2"
|
||
GO_TAR="${GO_VERSION}.linux-amd64.tar.gz"
|
||
DOWNLOAD_URL="https://mirrors.aliyun.com/golang/${GO_TAR}"
|
||
INSTALL_DIR="/usr/local"
|
||
GO_DIR="${INSTALL_DIR}/go"
|
||
TMP_DIR="/tmp"
|
||
GOPROXY_URL="https://goproxy.cn,direct"
|
||
|
||
# 颜色输出函数
|
||
print_info() {
|
||
echo -e "\033[32m[INFO]\033[0m $1"
|
||
}
|
||
|
||
print_error() {
|
||
echo -e "\033[31m[ERROR]\033[0m $1"
|
||
}
|
||
|
||
print_warning() {
|
||
echo -e "\033[33m[WARNING]\033[0m $1"
|
||
}
|
||
|
||
# 检查是否为root用户或有sudo权限
|
||
check_privileges() {
|
||
if [ "$EUID" -ne 0 ]; then
|
||
if ! command -v sudo &> /dev/null; then
|
||
print_error "需要root权限或sudo命令来执行安装，请以root用户运行或安装sudo。"
|
||
exit 1
|
||
fi
|
||
print_warning "当前非root用户，将使用sudo执行部分命令。"
|
||
fi
|
||
}
|
||
|
||
# 检查依赖工具
|
||
check_dependencies() {
|
||
local deps=("wget" "tar")
|
||
for dep in "${deps[@]}"; do
|
||
if ! command -v "$dep" &> /dev/null; then
|
||
print_info "安装依赖工具 $dep..."
|
||
if [ "$EUID" -eq 0 ]; then
|
||
apt update && apt install -y "$dep" || yum install -y "$dep" || dnf install -y "$dep"
|
||
else
|
||
sudo apt update && sudo apt install -y "$dep" || sudo yum install -y "$dep" || sudo dnf install -y "$dep"
|
||
fi
|
||
if [ $? -ne 0 ]; then
|
||
print_error "安装 $dep 失败，请手动安装后再运行脚本。"
|
||
exit 1
|
||
fi
|
||
fi
|
||
done
|
||
}
|
||
|
||
# 下载Go安装包
|
||
download_go() {
|
||
print_info "开始下载Golang安装包..."
|
||
if [ -f "${TMP_DIR}/${GO_TAR}" ]; then
|
||
print_info "安装包已存在，跳过下载。"
|
||
return 0
|
||
fi
|
||
if ! wget -O "${TMP_DIR}/${GO_TAR}" "$DOWNLOAD_URL"; then
|
||
print_error "下载Golang失败，请检查网络连接或URL是否有效。"
|
||
exit 1
|
||
fi
|
||
print_info "下载完成：${TMP_DIR}/${GO_TAR}"
|
||
}
|
||
|
||
# 安装Go
|
||
install_go() {
|
||
print_info "开始安装Golang到${INSTALL_DIR}..."
|
||
if [ -d "$GO_DIR" ]; then
|
||
print_warning "目录${GO_DIR}已存在，将被覆盖。"
|
||
if [ "$EUID" -eq 0 ]; then
|
||
rm -rf "$GO_DIR"
|
||
else
|
||
sudo rm -rf "$GO_DIR"
|
||
fi
|
||
fi
|
||
if [ "$EUID" -eq 0 ]; then
|
||
tar -C "$INSTALL_DIR" -xzf "${TMP_DIR}/${GO_TAR}"
|
||
else
|
||
sudo tar -C "$INSTALL_DIR" -xzf "${TMP_DIR}/${GO_TAR}"
|
||
fi
|
||
if [ $? -ne 0 ]; then
|
||
print_error "解压安装包失败，请检查文件完整性。"
|
||
exit 1
|
||
fi
|
||
print_info "Golang安装完成。"
|
||
}
|
||
|
||
# 设置环境变量
|
||
setup_env() {
|
||
print_info "设置环境变量..."
|
||
local profile_file=""
|
||
if [ -f "$HOME/.bashrc" ]; then
|
||
profile_file="$HOME/.bashrc"
|
||
elif [ -f "$HOME/.bash_profile" ]; then
|
||
profile_file="$HOME/.bash_profile"
|
||
elif [ -f "$HOME/.zshrc" ]; then
|
||
profile_file="$HOME/.zshrc"
|
||
else
|
||
print_warning "未找到常见的shell配置文件，将使用/etc/profile（需要root权限）。"
|
||
profile_file="/etc/profile"
|
||
fi
|
||
|
||
# 检查是否已设置环境变量，避免重复添加
|
||
if ! grep -q "/usr/local/go/bin" "$profile_file"; then
|
||
cat >> "$profile_file" << 'EOF'
|
||
|
||
# Go environment variables
|
||
export PATH=$PATH:/usr/local/go/bin
|
||
export GOPATH=$HOME/go
|
||
export PATH=$PATH:$GOPATH/bin
|
||
EOF
|
||
print_info "环境变量已写入${profile_file}。"
|
||
else
|
||
print_info "环境变量已存在，跳过写入。"
|
||
fi
|
||
|
||
# 立即生效（仅对当前会话）
|
||
export PATH=$PATH:/usr/local/go/bin
|
||
export GOPATH=$HOME/go
|
||
export PATH=$PATH:$GOPATH/bin
|
||
}
|
||
|
||
# 设置Go代理
|
||
setup_proxy() {
|
||
print_info "设置Go国内代理源..."
|
||
if command -v go &> /dev/null; then
|
||
go env -w GOPROXY="$GOPROXY_URL"
|
||
if [ $? -eq 0 ]; then
|
||
print_info "代理源已设置为：$GOPROXY_URL"
|
||
else
|
||
print_error "设置代理源失败，请手动运行 'go env -w GOPROXY=$GOPROXY_URL'"
|
||
fi
|
||
else
|
||
print_error "Go命令未找到，代理设置失败。请检查安装是否成功。"
|
||
fi
|
||
}
|
||
|
||
# 验证安装
|
||
verify_installation() {
|
||
print_info "验证Golang安装..."
|
||
if command -v go &> /dev/null; then
|
||
go_version=$(go version)
|
||
print_info "Golang版本：$go_version"
|
||
goproxy=$(go env GOPROXY)
|
||
print_info "GOPROXY设置：$goproxy"
|
||
else
|
||
print_error "Golang未正确安装，请检查错误信息。"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
# 主函数
|
||
main() {
|
||
check_privileges
|
||
check_dependencies
|
||
download_go
|
||
install_go
|
||
setup_env
|
||
setup_proxy
|
||
verify_installation
|
||
print_info "Golang安装和配置完成！请运行 'source ~/.bashrc' 或重新登录以应用环境变量。"
|
||
}
|
||
|
||
# 执行主函数
|
||
main
|