diff --git a/src/main/java/com/baiyee/adcallback/api/controller/AdPlatformController.java b/src/main/java/com/baiyee/adcallback/api/controller/AdPlatformController.java index deff3a9..3b86736 100644 --- a/src/main/java/com/baiyee/adcallback/api/controller/AdPlatformController.java +++ b/src/main/java/com/baiyee/adcallback/api/controller/AdPlatformController.java @@ -1,6 +1,7 @@ package com.baiyee.adcallback.api.controller; import cn.hutool.json.JSONUtil; +import com.baiyee.adcallback.api.dto.DeleteTagDTO; import com.baiyee.adcallback.repository.entity.TbPlatformCompanyEntity; import com.baiyee.adcallback.service.AdPlatformService; import lombok.RequiredArgsConstructor; @@ -35,4 +36,16 @@ public class AdPlatformController { adPlatformService.getTag(tbPlatformCompanyEntity); return new ResponseEntity<>(HttpStatus.OK); } + + + + @PostMapping(value = "/deleteTag") + public ResponseEntity deleteTag(@RequestBody String json) { + DeleteTagDTO deleteTagDTO = JSONUtil.toBean(json, DeleteTagDTO.class); + if (!token.equals(deleteTagDTO.getToken())) { + return new ResponseEntity<>(HttpStatus.BAD_REQUEST); + } + adPlatformService.deleteTagByTaskId(deleteTagDTO.getTaskId()); + return new ResponseEntity<>(HttpStatus.OK); + } } diff --git a/src/main/java/com/baiyee/adcallback/api/dto/DeleteTagDTO.java b/src/main/java/com/baiyee/adcallback/api/dto/DeleteTagDTO.java new file mode 100644 index 0000000..70b8a4c --- /dev/null +++ b/src/main/java/com/baiyee/adcallback/api/dto/DeleteTagDTO.java @@ -0,0 +1,15 @@ +package com.baiyee.adcallback.api.dto; + +import lombok.Data; + +/** + * @author Enzo + * @date : 2023/4/18 + * + */ +@Data +public class DeleteTagDTO { + private String token; + + private Long taskId; +} diff --git a/src/main/java/com/baiyee/adcallback/repository/TbPlatformCompanyEntityRepository.java b/src/main/java/com/baiyee/adcallback/repository/TbPlatformCompanyEntityRepository.java index a414fd0..4c0f047 100644 --- a/src/main/java/com/baiyee/adcallback/repository/TbPlatformCompanyEntityRepository.java +++ b/src/main/java/com/baiyee/adcallback/repository/TbPlatformCompanyEntityRepository.java @@ -2,6 +2,8 @@ package com.baiyee.adcallback.repository; import com.baiyee.adcallback.repository.entity.TbPlatformCompanyEntity; import org.springframework.data.jpa.repository.JpaRepository; +import org.springframework.data.jpa.repository.Modifying; +import org.springframework.data.jpa.repository.Query; import java.util.List; @@ -25,4 +27,13 @@ public interface TbPlatformCompanyEntityRepository extends JpaRepository findByStatus(Integer number); + + /** + * 任务删除配置 tag + * @param taskId + * @return + */ + @Modifying + @Query(value = "delete from TbPlatformCompanyEntity where taskId =?1") + int deleteByTaskId(Long taskId); } diff --git a/src/main/java/com/baiyee/adcallback/service/AdPlatformService.java b/src/main/java/com/baiyee/adcallback/service/AdPlatformService.java index 640a7a7..da4da1a 100644 --- a/src/main/java/com/baiyee/adcallback/service/AdPlatformService.java +++ b/src/main/java/com/baiyee/adcallback/service/AdPlatformService.java @@ -22,4 +22,10 @@ public interface AdPlatformService { * @return */ List findByPlatTag(Integer number); + + /** + * 删除 + * @param taskId + */ + void deleteTagByTaskId(Long taskId); } diff --git a/src/main/java/com/baiyee/adcallback/service/impl/AdPlatformServiceImpl.java b/src/main/java/com/baiyee/adcallback/service/impl/AdPlatformServiceImpl.java index a9e34f4..2be2afa 100644 --- a/src/main/java/com/baiyee/adcallback/service/impl/AdPlatformServiceImpl.java +++ b/src/main/java/com/baiyee/adcallback/service/impl/AdPlatformServiceImpl.java @@ -7,6 +7,7 @@ import com.baiyee.adcallback.repository.entity.TbPlatformCompanyEntity; import com.baiyee.adcallback.service.AdPlatformService; import lombok.RequiredArgsConstructor; import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; import java.util.List; @@ -40,4 +41,10 @@ public class AdPlatformServiceImpl implements AdPlatformService { public List findByPlatTag(Integer number) { return tbPlatformCompanyEntityRepository.findByStatus(number); } + + @Override + @Transactional(rollbackFor = Exception.class) + public void deleteTagByTaskId(Long taskId) { + tbPlatformCompanyEntityRepository.deleteByTaskId(taskId); + } }