SpringBoot 整合docker,执行容器服务

我使用以下文章的镜像作为演示镜像,读者有自己的镜像可以使用自己的
TencentARC/GFPGAN人脸恢复Ubuntu-22.04搭建(附带Docker镜像)_tencentarc gfpgan-CSDN博客

1. 封装springboot 启动docker容器的方法

java 复制代码
   public String runDockerCommand(String[] command) {
        StringBuilder result = new StringBuilder();
        try {
            // 使用 ProcessBuilder 执行命令
            ProcessBuilder processBuilder = new ProcessBuilder(command);
            processBuilder.redirectErrorStream(true);  // 将标准错误重定向到标准输出
            Process process = processBuilder.start();

            // 读取容器内的执行日志并打印到控制台
            BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
            String line;
            log.info("====== Docker logs start ======");
            while ((line = reader.readLine()) != null) {
               log.info(line);  // 打印日志到控制台
            }
            log.info("====== Docker logs end  ======");
            // 等待进程执行完毕
            int exitCode = process.waitFor();
            if (exitCode != 0) {
                throw new RuntimeException("Docker 命令执行失败,退出代码: " + exitCode);
            }

        } catch (Exception e) {
            e.printStackTrace();
            return "Docker 命令执行失败: " + e.getMessage();
        }

        return result.toString();  // 返回执行日志
    }

2. 构造启动docker容器的启动命令,并调用启动docker容器的方法

此方法参数可以根据自己镜像所需修改参数

  • inputPath为输入目录,如果用我的镜像的话就需要这个目录下有图片
  • outputPath为图片清晰生成目录
java 复制代码
    public void runGFPGAN(String inputPath,String outputPath) {
        // 定义 Docker 命令,使用传入的参数
        String[] command = {
                "docker", "run", "--rm",
                "-v", outputPath + ":/workspace/GFPGAN/results",
                "-v", inputPath + ":/workspace/GFPGAN/inputs/whole_imgs",
                "-e", "INPUT_PATH=/workspace/GFPGAN/inputs/whole_imgs",
                "-e", "OUTPUT_PATH=/workspace/GFPGAN/results",
                "registry.cn-hangzhou.aliyuncs.com/zr-dev/tencentarc:v1.0"
        };

        // 执行 Docker 命令
        this.runDockerCommand(command);
    }

3. 启动测试

相关推荐
皮皮林5516 小时前
IDEA 源码阅读利器,你居然还不会?
java·intellij idea
卡尔特斯10 小时前
Android Kotlin 项目代理配置【详细步骤(可选)】
android·java·kotlin
白鲸开源10 小时前
Ubuntu 22 下 DolphinScheduler 3.x 伪集群部署实录
java·ubuntu·开源
ytadpole10 小时前
Java 25 新特性 更简洁、更高效、更现代
java·后端
纪莫11 小时前
A公司一面:类加载的过程是怎么样的? 双亲委派的优点和缺点? 产生fullGC的情况有哪些? spring的动态代理有哪些?区别是什么? 如何排查CPU使用率过高?
java·java面试⑧股
JavaGuide11 小时前
JDK 25(长期支持版) 发布,新特性解读!
java·后端
用户37215742613511 小时前
Java 轻松批量替换 Word 文档文字内容
java
白鲸开源12 小时前
教你数分钟内创建并运行一个 DolphinScheduler Workflow!
java
Java中文社群12 小时前
有点意思!Java8后最有用新特性排行榜!
java·后端·面试
代码匠心12 小时前
从零开始学Flink:数据源
java·大数据·后端·flink