From e6e1bee8df392a26451fd816b7800ca9dfba5ae9 Mon Sep 17 00:00:00 2001 From: xiaohucoding <2307520758@qq.com> Date: Tue, 1 Jul 2025 12:07:00 +0800 Subject: [PATCH] =?UTF-8?q?Cache=E7=9A=84=E7=B1=BB=E5=9E=8B=E8=BD=AC?= =?UTF-8?q?=E6=8D=A2=E7=9A=84=E6=9B=B4=E6=94=B9=E4=BC=98=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../controller/PublishController.java | 20 +++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) diff --git a/src/main/java/com/bipt/intelligentapplicationorchestrationservice/controller/PublishController.java b/src/main/java/com/bipt/intelligentapplicationorchestrationservice/controller/PublishController.java index 1917af5..482dc41 100644 --- a/src/main/java/com/bipt/intelligentapplicationorchestrationservice/controller/PublishController.java +++ b/src/main/java/com/bipt/intelligentapplicationorchestrationservice/controller/PublishController.java @@ -1,12 +1,16 @@ package com.bipt.intelligentapplicationorchestrationservice.controller; import com.bipt.intelligentapplicationorchestrationservice.config.IpConfig; +import com.bipt.intelligentapplicationorchestrationservice.entity.DeployRequest; +import com.bipt.intelligentapplicationorchestrationservice.mapper.ModelMapper; import com.bipt.intelligentapplicationorchestrationservice.pojo.*; +import com.bipt.intelligentapplicationorchestrationservice.service.ModelDeployer; import com.bipt.intelligentapplicationorchestrationservice.service.PublishService; import com.bipt.intelligentapplicationorchestrationservice.util.NacosServiceUtil; import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.tags.Tag; import lombok.extern.slf4j.Slf4j; +import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.data.redis.core.RedisTemplate; import org.springframework.transaction.annotation.Transactional; @@ -27,17 +31,29 @@ public class PublishController { @Autowired private NacosServiceUtil nacosServiceUtil; + @Autowired + private ModelMapper modelMapper; @Autowired private IpConfig ipConfig; + @Autowired + private ModelDeployer modelDeployer; + @PostMapping @Operation(summary ="新增发布请求") @Transactional public OptResult> save(@RequestBody ServicePublishDTO servicePublishDTO) { log.info("模型发布请求:{}", servicePublishDTO); publishService.save(servicePublishDTO); - //todo 调用模型部署 - + //调用模型部署 + DeployRequest request = new DeployRequest(); + Long modelId = servicePublishDTO.getModelId(); + ModelVersion modelVersion = modelMapper.selectById(modelId); + String modelConfig = modelVersion.getModelConfig(); + //假设modelConfig只存GPU数据 + request.setModelId(String.valueOf(modelId)); + request.setRequiredMemory(Integer.parseInt(modelConfig)); + modelDeployer.deploy(request); // 获取前端传来的IP字符串 String ipListStr = servicePublishDTO.getIp(); if (ipListStr == null || ipListStr.trim().isEmpty()) {