Cache的类型转换的更改优化
This commit is contained in:
@ -1,12 +1,16 @@
|
|||||||
package com.bipt.intelligentapplicationorchestrationservice.controller;
|
package com.bipt.intelligentapplicationorchestrationservice.controller;
|
||||||
|
|
||||||
import com.bipt.intelligentapplicationorchestrationservice.config.IpConfig;
|
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.pojo.*;
|
||||||
|
import com.bipt.intelligentapplicationorchestrationservice.service.ModelDeployer;
|
||||||
import com.bipt.intelligentapplicationorchestrationservice.service.PublishService;
|
import com.bipt.intelligentapplicationorchestrationservice.service.PublishService;
|
||||||
import com.bipt.intelligentapplicationorchestrationservice.util.NacosServiceUtil;
|
import com.bipt.intelligentapplicationorchestrationservice.util.NacosServiceUtil;
|
||||||
import io.swagger.v3.oas.annotations.Operation;
|
import io.swagger.v3.oas.annotations.Operation;
|
||||||
import io.swagger.v3.oas.annotations.tags.Tag;
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
||||||
import lombok.extern.slf4j.Slf4j;
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.beans.BeanUtils;
|
||||||
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
import org.springframework.data.redis.core.RedisTemplate;
|
import org.springframework.data.redis.core.RedisTemplate;
|
||||||
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.transaction.annotation.Transactional;
|
||||||
@ -27,17 +31,29 @@ public class PublishController {
|
|||||||
@Autowired
|
@Autowired
|
||||||
private NacosServiceUtil nacosServiceUtil;
|
private NacosServiceUtil nacosServiceUtil;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private ModelMapper modelMapper;
|
||||||
@Autowired
|
@Autowired
|
||||||
private IpConfig ipConfig;
|
private IpConfig ipConfig;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private ModelDeployer modelDeployer;
|
||||||
|
|
||||||
@PostMapping
|
@PostMapping
|
||||||
@Operation(summary ="新增发布请求")
|
@Operation(summary ="新增发布请求")
|
||||||
@Transactional
|
@Transactional
|
||||||
public OptResult<List<ServicePublishVO>> save(@RequestBody ServicePublishDTO servicePublishDTO) {
|
public OptResult<List<ServicePublishVO>> save(@RequestBody ServicePublishDTO servicePublishDTO) {
|
||||||
log.info("模型发布请求:{}", servicePublishDTO);
|
log.info("模型发布请求:{}", servicePublishDTO);
|
||||||
publishService.save(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字符串
|
// 获取前端传来的IP字符串
|
||||||
String ipListStr = servicePublishDTO.getIp();
|
String ipListStr = servicePublishDTO.getIp();
|
||||||
if (ipListStr == null || ipListStr.trim().isEmpty()) {
|
if (ipListStr == null || ipListStr.trim().isEmpty()) {
|
||||||
|
Reference in New Issue
Block a user