321 lines
12 KiB
Java
321 lines
12 KiB
Java
package com.guwan.backend.controller;
|
||
|
||
import com.github.dockerjava.api.command.CreateContainerCmd;
|
||
import com.github.dockerjava.api.command.CreateContainerResponse;
|
||
import com.github.dockerjava.api.command.ExecCreateCmdResponse;
|
||
import com.github.dockerjava.api.model.*;
|
||
import com.github.dockerjava.core.DockerClientBuilder;
|
||
import com.github.dockerjava.core.command.ExecStartResultCallback;
|
||
import com.github.dockerjava.httpclient5.ApacheDockerHttpClient;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.http.ResponseEntity;
|
||
import org.springframework.web.bind.annotation.GetMapping;
|
||
import org.springframework.web.bind.annotation.RequestMapping;
|
||
import org.springframework.web.bind.annotation.RestController;
|
||
import com.github.dockerjava.api.DockerClient;
|
||
|
||
import java.io.ByteArrayOutputStream;
|
||
import java.io.File;
|
||
import java.io.IOException;
|
||
import java.nio.file.Files;
|
||
import java.nio.file.Path;
|
||
import java.util.*;
|
||
import java.util.concurrent.TimeUnit;
|
||
|
||
@RestController
|
||
@RequestMapping("/api/common")
|
||
public class DockerController {
|
||
|
||
private final DockerClient dockerClient;
|
||
|
||
@Autowired
|
||
public DockerController(DockerClient dockerClient) {
|
||
this.dockerClient = dockerClient;
|
||
}
|
||
|
||
@GetMapping("/docker-info")
|
||
public String getDockerInfo() {
|
||
Info info = dockerClient.infoCmd().exec(); // 获取 Docker 守护进程的信息
|
||
|
||
|
||
System.out.println("info = " + info);
|
||
|
||
return info.toString(); // 将 Docker 信息返回到 HTTP 响应
|
||
}
|
||
|
||
|
||
/**
|
||
* 获取镜像列表
|
||
*
|
||
* @param
|
||
* @return
|
||
*/
|
||
@GetMapping("/imageList")
|
||
public List<Image> imageList() {
|
||
List<Image> imageList = dockerClient.listImagesCmd().withShowAll(true).exec();
|
||
|
||
System.out.println("imageList = " + imageList);
|
||
return imageList;
|
||
}
|
||
|
||
|
||
@GetMapping("/listContainers")
|
||
public List<Container> listContainers() {
|
||
List<Container> exec = dockerClient.listContainersCmd().exec();
|
||
|
||
exec.forEach(container ->
|
||
System.out.println("name = " + Arrays.toString(container.getNames())));
|
||
return exec;
|
||
}
|
||
|
||
|
||
@GetMapping("/create")
|
||
public void create() throws IOException {
|
||
|
||
|
||
|
||
String javaFilePath = "D:\\00_桌面\\demo.java"; // 这里填入 Java 文件路径
|
||
File javaFile = new File(javaFilePath);
|
||
|
||
Path localDir = Files.createTempDirectory("docker-java");
|
||
File localJavaFile = new File(localDir.toFile(), "Main.java");
|
||
Files.copy(javaFile.toPath(), localJavaFile.toPath());
|
||
|
||
|
||
|
||
String imageName = "openjdk:8"; // 使用官方 OpenJDK 11 镜像
|
||
|
||
|
||
HostConfig hostConfig = HostConfig.newHostConfig()
|
||
.withBinds(new Bind(localDir.toString(), new com.github.dockerjava.api.model.Volume("/java"))); // 绑定挂载
|
||
|
||
|
||
// 启动容器
|
||
String containerId = dockerClient.createContainerCmd(imageName)
|
||
.withCmd("bash", "-c", "javac /java/Main.java && java -cp /java Main && tail -f /dev/null")
|
||
.withHostConfig(hostConfig)
|
||
.withName("java-app-container")
|
||
.exec().getId();
|
||
|
||
dockerClient.startContainerCmd(containerId).exec();
|
||
|
||
System.out.println("Container started. Check the output in Docker.");
|
||
|
||
}
|
||
|
||
private static final String IMAGE_NAME = "gcc:latest";
|
||
private static final int TIMEOUT_SECONDS = 10;
|
||
|
||
/* @GetMapping("/testCpp")
|
||
public void testCpp() throws IOException {
|
||
|
||
System.out.println("dockerClient = " + dockerClient);
|
||
|
||
|
||
try {
|
||
// 创建容器
|
||
CreateContainerResponse container = dockerClient.createContainerCmd(IMAGE_NAME)
|
||
.withTty(true)
|
||
.withWorkingDir("/app")
|
||
.exec();
|
||
|
||
String containerId = container.getId();
|
||
dockerClient.startContainerCmd(containerId).exec();
|
||
|
||
// 写入代码（使用cat和here-doc避免转义问题）
|
||
String cppCode = "#include <iostream>\n"
|
||
+ "int main() { std::cout << \"Hello Docker!\\n\"; return 0; }";
|
||
|
||
execCommand(dockerClient, containerId,
|
||
"sh -c 'cat > /app/main.cpp <<EOF\n" + cppCode + "\nEOF'");
|
||
|
||
// 编译代码（使用相对路径）
|
||
execCommand(dockerClient, containerId, "g++ main.cpp -o main");
|
||
|
||
// 执行程序（使用相对路径）
|
||
String output = execCommand(dockerClient, containerId, "./main");
|
||
System.out.println("程序输出: " + output);
|
||
|
||
} catch (Exception e) {
|
||
throw new RuntimeException(e);
|
||
} finally {
|
||
// 清理容器
|
||
*//*dockerClient.listContainersCmd().withShowAll(true).exec().forEach(c ->
|
||
dockerClient.removeContainerCmd(c.getId()).withForce(true).exec());*//*
|
||
//dockerClient.close();
|
||
}
|
||
}
|
||
|
||
|
||
private static String execCommand(DockerClient dockerClient, String containerId, String command)
|
||
throws Exception {
|
||
ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
|
||
ExecCreateCmdResponse exec = dockerClient.execCreateCmd(containerId)
|
||
.withCmd("sh", "-c", command)
|
||
.withAttachStdout(true)
|
||
.withAttachStderr(true)
|
||
.exec();
|
||
|
||
dockerClient.execStartCmd(exec.getId())
|
||
.exec(new ExecStartResultCallback(outputStream, System.err))
|
||
.awaitCompletion(TIMEOUT_SECONDS, TimeUnit.SECONDS);
|
||
|
||
return outputStream.toString().trim();
|
||
}*/
|
||
|
||
|
||
@GetMapping("/judge")
|
||
public ResponseEntity<Map<String, Object>> judgeCode() {
|
||
Map<String, Object> result = new HashMap<>();
|
||
List<TestCase> testCases = new ArrayList<>();
|
||
|
||
// 真实用户提交的C++代码（示例：A+B问题）
|
||
String cppCode =
|
||
"#include <iostream>\n" +
|
||
"using namespace std;\n" +
|
||
"int main() {\n" +
|
||
" int a, b;\n" +
|
||
" cin >> a >> b;\n" +
|
||
" cout << a + b << endl;\n" +
|
||
" return 0;\n" +
|
||
"}";
|
||
|
||
// 定义测试用例
|
||
testCases.add(new TestCase("1 2", "3")); // 基础测试
|
||
testCases.add(new TestCase("-5 8", "3")); // 负数测试
|
||
testCases.add(new TestCase("1000000 2000000", "3000000")); // 大数测试
|
||
testCases.add(new TestCase("", "")); // 错误输入测试
|
||
testCases.add(new TestCase("3", "")); // 不完整输入测试
|
||
|
||
String containerId = null;
|
||
try {
|
||
// 创建容器（使用官方GCC镜像）
|
||
CreateContainerResponse container = dockerClient.createContainerCmd("gcc:latest")
|
||
.withTty(true)
|
||
.withWorkingDir("/app")
|
||
.withHostConfig(HostConfig.newHostConfig()
|
||
.withMemory(100 * 1024 * 1024L) // 限制内存100MB
|
||
.withCpuCount(1L)) // 限制1核CPU
|
||
.exec();
|
||
|
||
containerId = container.getId();
|
||
dockerClient.startContainerCmd(containerId).exec();
|
||
|
||
// 1. 写入代码文件
|
||
execCommand(containerId, "mkdir -p /app", 10);
|
||
execCommand(containerId,
|
||
"bash -c 'cat > /app/main.cpp <<EOF\n" +
|
||
cppCode.replace("'", "'\"'\"'") + // 处理单引号
|
||
"\nEOF'", 10);
|
||
|
||
// 2. 编译代码（带编译错误检查）
|
||
String compileOutput = execCommand(containerId, "g++ main.cpp -o main -O2 -Wall", 10);
|
||
if (!compileOutput.isEmpty()) {
|
||
result.put("status", "compile_error");
|
||
result.put("message", compileOutput);
|
||
return ResponseEntity.ok(result);
|
||
}
|
||
|
||
// 3. 执行测试用例
|
||
List<Map<String, Object>> caseResults = new ArrayList<>();
|
||
for (int i = 0; i < testCases.size(); i++) {
|
||
TestCase testCase = testCases.get(i);
|
||
Map<String, Object> caseResult = new HashMap<>();
|
||
caseResult.put("case_id", i + 1);
|
||
caseResult.put("input", testCase.input);
|
||
caseResult.put("expected", testCase.expectedOutput);
|
||
|
||
try {
|
||
// 写入输入文件
|
||
execCommand(containerId,
|
||
"bash -c 'cat > /app/input.txt <<EOF\n" +
|
||
testCase.input.replace("'", "'\"'\"'") +
|
||
"\nEOF'", 10);
|
||
|
||
// 执行程序（带超时控制）
|
||
String actualOutput = execCommand(containerId,
|
||
"timeout 2s ./main < /app/input.txt 2>&1", // 捕获标准错误
|
||
3 // 超时3秒
|
||
);
|
||
|
||
// 标准化输出处理
|
||
actualOutput = actualOutput.trim()
|
||
.replaceAll("\r\n", "\n")
|
||
.replaceAll("[ \\t]+", " ");
|
||
|
||
// 结果对比
|
||
boolean isPassed = actualOutput.equals(testCase.expectedOutput);
|
||
caseResult.put("status", isPassed ? "passed" : "failed");
|
||
caseResult.put("actual", actualOutput);
|
||
|
||
// 特殊错误类型检测
|
||
if (actualOutput.contains("Timeout")) {
|
||
caseResult.put("status", "time_limit_exceeded");
|
||
} else if (actualOutput.contains("runtime error")) {
|
||
caseResult.put("status", "runtime_error");
|
||
} else if (actualOutput.isEmpty()) {
|
||
caseResult.put("status", "no_output");
|
||
}
|
||
|
||
} catch (Exception e) {
|
||
caseResult.put("status", "system_error");
|
||
caseResult.put("message", e.getMessage());
|
||
}
|
||
|
||
caseResults.add(caseResult);
|
||
}
|
||
|
||
result.put("status", "completed");
|
||
result.put("cases", caseResults);
|
||
result.put("pass_count", caseResults.stream()
|
||
.filter(c -> "passed".equals(c.get("status")))
|
||
.count());
|
||
|
||
} catch (Exception e) {
|
||
result.put("status", "system_error");
|
||
result.put("message", e.getMessage());
|
||
} finally {
|
||
if (containerId != null) {
|
||
dockerClient.removeContainerCmd(containerId)
|
||
.withForce(true)
|
||
.exec();
|
||
}
|
||
}
|
||
|
||
return ResponseEntity.ok(result);
|
||
}
|
||
|
||
// 辅助方法：执行容器命令
|
||
private String execCommand(String containerId, String command, int timeoutSeconds)
|
||
throws Exception {
|
||
|
||
ByteArrayOutputStream output = new ByteArrayOutputStream();
|
||
ExecCreateCmdResponse exec = dockerClient.execCreateCmd(containerId)
|
||
.withCmd("bash", "-c", command)
|
||
.withAttachStdout(true)
|
||
.withAttachStderr(true)
|
||
.exec();
|
||
|
||
dockerClient.execStartCmd(exec.getId())
|
||
.exec(new ExecStartResultCallback(output, output))
|
||
.awaitCompletion(timeoutSeconds, TimeUnit.SECONDS);
|
||
|
||
return output.toString().trim();
|
||
}
|
||
|
||
// 测试用例类
|
||
static class TestCase {
|
||
String input;
|
||
String expectedOutput;
|
||
|
||
public TestCase(String input, String expectedOutput) {
|
||
this.input = input;
|
||
this.expectedOutput = expectedOutput;
|
||
}
|
||
}
|
||
|
||
|
||
|
||
|
||
}
|