Package com.alibaba.cloud.ai.service.code.impl

Class AbstractCodePoolExecutorService

java.lang.Object

com.alibaba.cloud.ai.service.code.impl.AbstractCodePoolExecutorService

All Implemented Interfaces:

CodePoolExecutorService

Direct Known Subclasses:

DockerCodePoolExecutorService

public abstract class AbstractCodePoolExecutorService
extends Object
implements CodePoolExecutorService

运行Python任务的容器池

Since:

2025/7/12

Author:

vlsmb

Nested Class Summary

Nested classes/interfaces inherited from interface com.alibaba.cloud.ai.service.code.CodePoolExecutorService

CodePoolExecutorService.State, CodePoolExecutorService.TaskRequest, CodePoolExecutorService.TaskResponse

Field Summary

Fields

Modifier and Type	Field	Description
protected final ExecutorService	consumerThreadPool
protected final ConcurrentHashMap<String,CodePoolExecutorService.State>	coreContainerState
protected final AtomicInteger	currentCoreContainerSize
protected final AtomicInteger	currentTempContainerSize
protected final CodeExecutorProperties	properties
protected final ArrayBlockingQueue<String>	readyCoreContainer
protected final ArrayBlockingQueue<String>	readyTempContainer
protected final ArrayBlockingQueue<FutureTask<CodePoolExecutorService.TaskResponse>>	taskQueue
protected final ConcurrentHashMap<String,Future<?>>	tempContainerRemoveFuture
protected final ConcurrentHashMap<String,CodePoolExecutorService.State>	tempContainerState

Constructor Summary

Constructors

Constructor	Description
AbstractCodePoolExecutorService(CodeExecutorProperties properties)

Method Summary

Modifier and Type	Method	Description
protected void	clearTempDir(Path tempDir)	删除临时目录
protected abstract String	createNewContainer()
protected void	createWritableFile(Path tempDir, String fileName)	创建可写的临时文件
protected abstract CodePoolExecutorService.TaskResponse	execTaskInContainer(CodePoolExecutorService.TaskRequest request, String containerId)
protected String	generateContainerName()	生成唯一的容器名称
protected abstract void	removeContainer(String containerId)
CodePoolExecutorService.TaskResponse	runTask(CodePoolExecutorService.TaskRequest request)
protected void	shutdownPool()
protected abstract void	stopContainer(String containerId)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

coreContainerState

protected final ConcurrentHashMap<String,CodePoolExecutorService.State> coreContainerState

tempContainerState

protected final ConcurrentHashMap<String,CodePoolExecutorService.State> tempContainerState

tempContainerRemoveFuture

protected final ConcurrentHashMap<String,Future<?>> tempContainerRemoveFuture

taskQueue

protected final ArrayBlockingQueue<FutureTask<CodePoolExecutorService.TaskResponse>> taskQueue

readyCoreContainer

protected final ArrayBlockingQueue<String> readyCoreContainer

readyTempContainer

protected final ArrayBlockingQueue<String> readyTempContainer

currentCoreContainerSize

protected final AtomicInteger currentCoreContainerSize

currentTempContainerSize

protected final AtomicInteger currentTempContainerSize

consumerThreadPool

protected final ExecutorService consumerThreadPool

properties

protected final CodeExecutorProperties properties

Constructor Details

AbstractCodePoolExecutorService

public AbstractCodePoolExecutorService(CodeExecutorProperties properties)

Method Details

createNewContainer

protected abstract String createNewContainer()
                                      throws Exception

Throws:

Exception

execTaskInContainer

protected abstract CodePoolExecutorService.TaskResponse execTaskInContainer(CodePoolExecutorService.TaskRequest request,
 String containerId)
                                                                     throws Exception

Throws:

Exception

stopContainer

protected abstract void stopContainer(String containerId)
                               throws Exception

Throws:

Exception

removeContainer

protected abstract void removeContainer(String containerId)
                                 throws Exception

Throws:

Exception

shutdownPool

protected void shutdownPool()
                     throws Exception

Throws:

Exception

runTask

public CodePoolExecutorService.TaskResponse runTask(CodePoolExecutorService.TaskRequest request)

Specified by:

runTask in interface CodePoolExecutorService

clearTempDir

protected void clearTempDir(Path tempDir)

删除临时目录

createWritableFile

protected void createWritableFile(Path tempDir,
 String fileName)
                           throws IOException

创建可写的临时文件

Parameters:

tempDir - 临时目录

fileName - 文件名

Throws:

IOException - IO异常

generateContainerName

protected String generateContainerName()

生成唯一的容器名称