diff --git a/src/main/java/com/zhgd/xmgl/modules/project/controller/LargeScreenConfigController.java b/src/main/java/com/zhgd/xmgl/modules/project/controller/LargeScreenConfigController.java index 6a292a5be..3add87afa 100644 --- a/src/main/java/com/zhgd/xmgl/modules/project/controller/LargeScreenConfigController.java +++ b/src/main/java/com/zhgd/xmgl/modules/project/controller/LargeScreenConfigController.java @@ -1,5 +1,6 @@ package com.zhgd.xmgl.modules.project.controller; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; @@ -91,7 +92,16 @@ public class LargeScreenConfigController { @ApiOperation(value = " 添加自定义大屏配置信息", notes = "添加自定义大屏配置信息", httpMethod = "POST") @PostMapping(value = "/add") public Result add(@RequestBody LargeScreenConfig largeScreenConfig) { - largeScreenConfigService.save(largeScreenConfig); + LargeScreenConfig dbConfig = largeScreenConfigService.getOne(new LambdaQueryWrapper() + .eq(LargeScreenConfig::getProjectSn, largeScreenConfig.getProjectSn()) + .eq(LargeScreenConfig::getConfigKey, largeScreenConfig.getConfigKey()) + ); + if (dbConfig != null) { + largeScreenConfig.setId(dbConfig.getId()); + largeScreenConfigService.updateById(largeScreenConfig); + } else { + largeScreenConfigService.save(largeScreenConfig); + } return Result.success("添加成功!"); }