1000988

src/main/java/com/ylcm/sys/controller/AdMediaController.java

@@ -142,7 +142,12 @@ public class AdMediaController extends BaseController{
 		JSONObject result = new JSONObject();
         result.put("code", Constants.CODE_SUCCESS);
         result.put("msg", "服务器异常");
-        adMediaService.removeById(id);
+
+        String error = adMediaService.deleteById(id);
+        if (error != null) {
+			result.put("code", Constants.CODE_FAIL);
+			result.put("msg", error);
+		}
         
         returnResult(request, response, result);
     }
@@ -155,7 +160,14 @@ public class AdMediaController extends BaseController{
         result.put("msg", "服务器异常");
         String[] idArray = ids.split(",");
         for (String id : idArray) {
-            adMediaService.removeById(id);
+			String error = adMediaService.deleteById(Integer.parseInt(id));
+			if (error != null) {
+				result.put("code", Constants.CODE_FAIL);
+				result.put("msg", error);
+
+				returnResult(request, response, result);
+				return;
+			}
 		}
         
         returnResult(request, response, result);

src/main/java/com/ylcm/sys/service/AdMediaService.java

@@ -21,4 +21,6 @@ public interface AdMediaService extends IService<AdMedia> {
 	String importBatch(List<AdMediaImportDTO> importDTOs);
 	
 	Integer getTotalByName(String name);
+
+	String deleteById(Integer id);
 }

src/main/java/com/ylcm/sys/service/impl/AdMediaServiceImpl.java

@@ -1,9 +1,13 @@
 package com.ylcm.sys.service.impl;
 
 import com.ylcm.sys.domain.AdMedia;
+import com.ylcm.sys.domain.AdSpecial;
+import com.ylcm.sys.domain.AdTheatre;
 import com.ylcm.sys.excel.model.AdMediaImportDTO;
 import com.ylcm.sys.exception.BaseException;
 import com.ylcm.sys.mapper.AdMediaMapper;
+import com.ylcm.sys.mapper.AdSpecialMapper;
+import com.ylcm.sys.mapper.AdTheatreMapper;
 import com.ylcm.sys.service.AdMediaService;
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
@@ -15,6 +19,8 @@ import org.apache.commons.lang.StringUtils;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import javax.annotation.Resource;
+
 /**
  * <p>
  * 媒体管理表 服务实现类
@@ -26,6 +32,12 @@ import org.springframework.transaction.annotation.Transactional;
 @Service
 public class AdMediaServiceImpl extends ServiceImpl<AdMediaMapper, AdMedia> implements AdMediaService {
 
+	@Resource
+	private AdSpecialMapper adSpecialMapper;
+
+	@Resource
+	private AdTheatreMapper adTheatreMapper;
+
 	@Override
 	public AdMedia getByName(String name) {
 		LambdaQueryWrapper<AdMedia> queryWrapper = new LambdaQueryWrapper<AdMedia>().eq(AdMedia::getName, name);
@@ -55,4 +67,22 @@ public class AdMediaServiceImpl extends ServiceImpl<AdMediaMapper, AdMedia> impl
 		}
 		return baseMapper.selectCount(queryWrapper);
 	}
+
+	@Override
+	public String deleteById(Integer id) {
+		AdMedia adMedia = baseMapper.selectById(id);
+
+		AdSpecial adSpecial = adSpecialMapper.selectOne(
+				new LambdaQueryWrapper<AdSpecial>().eq(AdSpecial::getMediaName, adMedia.getName()).last("limit 1"));
+		if (adSpecial == null) {
+			return "该媒体在专题中使用不能删除";
+		}
+		AdTheatre adTheatre = adTheatreMapper.selectOne(
+				new LambdaQueryWrapper<AdTheatre>().eq(AdTheatre::getMediaName, adMedia.getName()).last("limit 1"));
+		if (adTheatre != null) {
+			return "该媒体在剧场中使用不能删除";
+		}
+		baseMapper.deleteById(id);
+		return null;
+	}
 }