孙永军 1 jaar geleden
bovenliggende
commit
180b6c9627
1 gewijzigde bestanden met toevoegingen van 15 en 7 verwijderingen
  1. 15 7
      src/main/java/com/smcic/api/epg/service/impl/EpgDisableConfigService.java

+ 15 - 7
src/main/java/com/smcic/api/epg/service/impl/EpgDisableConfigService.java

@@ -42,14 +42,22 @@ public class EpgDisableConfigService extends ServiceImpl<EpgDisableConfigMapper,
         if(null == program){
             throw new APIException("不存在的节目");
         }
+        EpgDisableConfig epgDisableConfig;
 
-        EpgDisableConfig epgDisableConfig = new EpgDisableConfig();
-        epgDisableConfig.setChannelId(disableDTO.getChannelId());
-        epgDisableConfig.setChannelName(channel.getTitle());
-        epgDisableConfig.setStartTime(disableDTO.getStartTime());
-        epgDisableConfig.setProgram(program.getTitle());
-        epgDisableConfig.setStatus(disableDTO.getStatus());
-        save(epgDisableConfig);
+        epgDisableConfig = lambdaQuery().eq(EpgDisableConfig::getChannelId, disableDTO.getChannelId()).eq(EpgDisableConfig::getStartTime, disableDTO.getStartTime()).one();
+
+        if(epgDisableConfig == null){
+            epgDisableConfig = new EpgDisableConfig();
+            epgDisableConfig.setChannelId(disableDTO.getChannelId());
+            epgDisableConfig.setChannelName(channel.getTitle());
+            epgDisableConfig.setStartTime(disableDTO.getStartTime());
+            epgDisableConfig.setProgram(program.getTitle());
+            epgDisableConfig.setStatus(disableDTO.getStatus());
+            save(epgDisableConfig);
+        }else {
+            epgDisableConfig.setStatus(disableDTO.getStatus());
+            updateById(epgDisableConfig);
+        }
         programService.setCache(disableDTO.getChannelId());
     }