Răsfoiți Sursa

国内构建cognio镜像

黄中银 3 luni în urmă
părinte
comite
2fa161f73f

+ 2 - 0
cognio/buildDockerImage_u24_docker.bat

@@ -0,0 +1,2 @@
+@echo off
+powershell -ExecutionPolicy Bypass -File "%~dp0buildDockerImage_u24_docker.ps1" %*

+ 147 - 0
cognio/buildDockerImage_u24_docker.ps1

@@ -0,0 +1,147 @@
+# Cognio 远程构建 Docker 镜像脚本
+
+[Console]::OutputEncoding = [System.Text.Encoding]::UTF8
+
+# ============================================================
+# 配置区
+# ============================================================
+
+# 远程服务器配置
+$remoteUser = "root"
+$remoteHost = "192.168.1.99"
+$remotePort = "22"
+
+# Git 仓库配置
+$repoUrl = "http://ds:39418/AI/Cognio"
+$repoDir = "Cognio"
+
+# Docker 镜像配置
+$imageName = "amwpfiqvy/cognio"
+
+# Docker 镜像加速源(随机选择一个,不带 https:// 前缀)
+$dockerMirrors = @(
+    "cf-workers-docker-io-dfw.pages.dev"
+    "docker.1panel.live"
+    "hub.rat.dev"
+    "docker.m.daocloud.io"
+)
+$dockerMirror = $dockerMirrors | Get-Random
+
+# tmux 会话名
+$tmuxSession = "dbx"
+
+# ============================================================
+# 主程序
+# ============================================================
+
+Write-Host ""
+Write-Host "========================================" -ForegroundColor Cyan
+Write-Host "  Cognio 远程构建 Docker 镜像" -ForegroundColor Cyan
+Write-Host "========================================" -ForegroundColor Cyan
+Write-Host ""
+
+# Docker 镜像标签:提示用户输入(默认包含 latest,-l 排除 latest)
+$imageTagInput = Read-Host "请输入额外的 Docker 镜像标签 (多个用空格分隔,-l=排除latest,留空则仅推送 latest)"
+
+$inputTags = @()
+$excludeLatest = $false
+
+if ($imageTagInput -ne "") {
+    $inputTags = $imageTagInput -split '\s+' | Where-Object { $_ -ne "" }
+    if ($inputTags -contains "-l") {
+        $excludeLatest = $true
+        $inputTags = $inputTags | Where-Object { $_ -ne "-l" }
+    }
+}
+
+# 构建最终标签列表
+if ($excludeLatest) {
+    $imageTags = $inputTags
+} else {
+    $imageTags = @("latest") + $inputTags | Select-Object -Unique
+}
+
+# 构建 -t 参数列表
+$tagParams = ($imageTags | ForEach-Object { "-t ${imageName}:$_" }) -join " "
+
+Write-Host "镜像标签: $($imageTags -join ', ')" -ForegroundColor Yellow
+Write-Host "镜像加速: $dockerMirror" -ForegroundColor Yellow
+Write-Host ""
+
+# 支持命令行参数覆盖默认配置
+if ($args.Count -ge 1) { $remoteUser = $args[0] }
+if ($args.Count -ge 2) { $remoteHost = $args[1] }
+if ($args.Count -ge 3) { $remotePort = $args[2] }
+
+Write-Host "连接到: $remoteUser@$remoteHost`:$remotePort" -ForegroundColor Yellow
+Write-Host ""
+
+# ============================================================
+# 构建远程执行的 Shell 脚本
+# ============================================================
+
+# 生成远程脚本内容(写入临时文件执行,避免引号嵌套问题)
+$scriptContent = @"
+#!/bin/bash
+cd ~
+if [ -d "$repoDir" ]; then
+    echo "目录已存在,还原本地修改并拉取最新代码..."
+    cd $repoDir
+    git checkout .
+    git pull
+else
+    echo "目录不存在,正在克隆仓库..."
+    git clone $repoUrl
+    cd $repoDir
+fi
+echo "修改 Dockerfile 使用镜像加速..."
+sed -i '1s|^FROM python|FROM \$\{DOCKER_MIRROR\}/library/python|' Dockerfile
+sed -i '1i ARG DOCKER_MIRROR=docker.io' Dockerfile
+echo "正在构建并推送 Docker 镜像..."
+echo "使用镜像加速: $dockerMirror"
+docker buildx build \
+    --platform linux/amd64,linux/arm64 \
+    --build-arg DOCKER_MIRROR=$dockerMirror \
+    $tagParams \
+    --cache-from type=local,src=`$HOME/.buildx-cache \
+    --cache-to type=local,dest=`$HOME/.buildx-cache,mode=max \
+    --push \
+    .
+"@
+
+# 将脚本内容转为 base64,避免特殊字符问题
+$scriptBase64 = [Convert]::ToBase64String([System.Text.Encoding]::UTF8.GetBytes($scriptContent))
+
+# tmux 命令:解码并执行脚本
+$remoteCmd = "tmux has-session -t $tmuxSession 2>/dev/null || tmux new-session -d -s $tmuxSession; " +
+             "tmux send-keys -t $tmuxSession 'echo $scriptBase64 | base64 -d > /tmp/build_cognio.sh && bash /tmp/build_cognio.sh' Enter"
+
+# ============================================================
+# 执行远程命令
+# ============================================================
+
+Write-Host "正在发送命令到 tmux 会话: $tmuxSession" -ForegroundColor Yellow
+Write-Host ""
+
+& ssh -p $remotePort "$remoteUser@$remoteHost" $remoteCmd
+
+if ($LASTEXITCODE -eq 0) {
+    Write-Host ""
+    Write-Host "========================================" -ForegroundColor Green
+    Write-Host "  命令已发送到 tmux 会话: $tmuxSession" -ForegroundColor Green
+    Write-Host "========================================" -ForegroundColor Green
+    Write-Host ""
+    Write-Host "推送目标:" -ForegroundColor Yellow
+    foreach ($tag in $imageTags) {
+        Write-Host "  - ${imageName}:${tag}" -ForegroundColor Cyan
+    }
+    Write-Host ""
+    Write-Host "查看构建进度:" -ForegroundColor Yellow
+    Write-Host "  ssh -t -p $remotePort $remoteUser@$remoteHost `"tmux attach -t $tmuxSession`"" -ForegroundColor Cyan
+} else {
+    Write-Host ""
+    Write-Host "部署失败,请检查错误信息" -ForegroundColor Red
+}
+
+Write-Host ""
+pause