modules/mainPart/src/main/java/com/zt/life/modules/mainPart/taskReliability/controller/ModelLineController.java
@@ -12,6 +12,7 @@
import com.zt.common.validator.group.AddGroup;
import com.zt.common.validator.group.DefaultGroup;
import com.zt.common.validator.group.UpdateGroup;
import com.zt.life.modules.mainPart.async.PythonLib;
import com.zt.life.modules.mainPart.taskReliability.model.ModelLine;
import com.zt.life.modules.mainPart.taskReliability.model.ModelRbd;
import com.zt.life.modules.mainPart.taskReliability.service.ModelLineService;
@@ -19,9 +20,17 @@
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.scheduling.annotation.Async;
import org.springframework.web.bind.annotation.*;
import springfox.documentation.annotations.ApiIgnore;
import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
/**
@@ -34,8 +43,16 @@
@RequestMapping("/taskReliability/ModelLine/")
@Api(tags="model_line")
public class ModelLineController {
    private static final String URL_REPLACE_STR = "URL_PREFIX_URL";
    private static final String TOKEN_REPLACE_STR = "MY_TOKEN_MY";
    private static final String TOKEN_REPLACE_REGEX = "token=[0-9a-zA-Z]+&";
    @Autowired
    private ModelLineService modelLineService;
    @Autowired
//    private PythonLib pythonLib;
    @GetMapping("page")
    @ApiOperation("分页")
@@ -60,11 +77,32 @@
    @GetMapping("getDiagram")
    @ApiOperation("信息")
    public Result<ModelRbd> getDiagram(Long modelId){
    public Result<ModelRbd> getDiagram(Long modelId, String urlPref, String token) {
        ModelRbd data = modelLineService.getDiagram(modelId);
        if (data.getContent() != null) {
            data.setContent(data.getContent().replace(URL_REPLACE_STR, urlPref));
            data.setContent(data.getContent().replace(TOKEN_REPLACE_STR, "token="+token+"&"));
        }
        return Result.ok(data);
    }
/*
    @GetMapping("callPythonCalc")
    @ApiOperation("信息")
    public Result callPythonCalc() {
        pythonLib.callPython();
        return Result.ok();
    }
    @GetMapping("getPythonCalcResult")
    @ApiOperation("信息")
    public Result<String> getPythonCalcResult() {
        String result = pythonLib.getPythonCalcResult();
        return Result.ok(result);
    }
*/
    @PostMapping
    @ApiOperation("新增")
@@ -72,6 +110,8 @@
    public Result insert(@RequestBody ModelRbd modelRbd){
        //效验数据
        ValidatorUtils.validateEntity(modelRbd, AddGroup.class, DefaultGroup.class);
        modelRbd.setContent(modelRbd.getContent().replace(modelRbd.getUrlPref(), URL_REPLACE_STR));
        modelRbd.setContent(modelRbd.getContent().replaceAll(TOKEN_REPLACE_REGEX, TOKEN_REPLACE_STR));
        modelLineService.insert(modelRbd);
        return Result.ok();
@@ -83,6 +123,8 @@
    public Result update(@RequestBody ModelRbd modelRbd){
        //效验数据
        ValidatorUtils.validateEntity(modelRbd, UpdateGroup.class, DefaultGroup.class);
        modelRbd.setContent(modelRbd.getContent().replace(modelRbd.getUrlPref(), URL_REPLACE_STR));
        modelRbd.setContent(modelRbd.getContent().replaceAll(TOKEN_REPLACE_REGEX, TOKEN_REPLACE_STR));
        modelLineService.update(modelRbd);
        return Result.ok();