From d0e236993a0d7624748ec22a4f3e4acef38c5313 Mon Sep 17 00:00:00 2001
From: wjt <wujingtao_t@126.com>
Date: Mon, 24 Apr 2023 14:54:38 +0800
Subject: [PATCH] =?UTF-8?q?=E5=88=A4=E6=96=AD=E5=88=A0=E9=99=A4=E5=88=86?=
 =?UTF-8?q?=E6=9C=BA=E5=8F=B7=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../modules/telemarkting/dao/ExtensionNumberRepository.java   | 4 ++++
 .../telemarkting/service/impl/ExtensionNumberServiceImpl.java | 4 +++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/ad-platform-manage/ad-platform-management/src/main/java/com/baiye/modules/telemarkting/dao/ExtensionNumberRepository.java b/ad-platform-manage/ad-platform-management/src/main/java/com/baiye/modules/telemarkting/dao/ExtensionNumberRepository.java
index b017fca6..80ce8a25 100644
--- a/ad-platform-manage/ad-platform-management/src/main/java/com/baiye/modules/telemarkting/dao/ExtensionNumberRepository.java
+++ b/ad-platform-manage/ad-platform-management/src/main/java/com/baiye/modules/telemarkting/dao/ExtensionNumberRepository.java
@@ -17,6 +17,7 @@ import java.util.List;
 public interface ExtensionNumberRepository extends JpaRepository<ExtensionNumber, Long>, JpaSpecificationExecutor<ExtensionNumber> {
 
     ExtensionNumber findByNumber(Integer number);
+
     /**
      * 根据状态和公司id查询分机号
      *
@@ -65,4 +66,7 @@ public interface ExtensionNumberRepository extends JpaRepository<ExtensionNumber
     @Modifying
     @Query(value = "update tb_extension_number c set c.status =?1 where c.number in (select teu.number from tb_extension_user teu where teu.member_id =?2)", nativeQuery = true)
     int updateExtensionNumber(Integer status, Long userId);
+
+    @Query(value = "SELECT number FROM tb_extension_number WHERE  company_id = ?1 ", nativeQuery = true)
+    List<Integer> findByCompanyId(Long companyId);
 }
diff --git a/ad-platform-manage/ad-platform-management/src/main/java/com/baiye/modules/telemarkting/service/impl/ExtensionNumberServiceImpl.java b/ad-platform-manage/ad-platform-management/src/main/java/com/baiye/modules/telemarkting/service/impl/ExtensionNumberServiceImpl.java
index fb1b8daf..eaedfd1c 100644
--- a/ad-platform-manage/ad-platform-management/src/main/java/com/baiye/modules/telemarkting/service/impl/ExtensionNumberServiceImpl.java
+++ b/ad-platform-manage/ad-platform-management/src/main/java/com/baiye/modules/telemarkting/service/impl/ExtensionNumberServiceImpl.java
@@ -227,7 +227,9 @@ public class ExtensionNumberServiceImpl implements ExtensionNumberService {
     @Override
     @Transactional(rollbackFor = Exception.class)
     public void deleteExtensionNumberByCompany(Long companyId) {
-        extensionUserRepository.deleteByCompanyId(companyId);
+        //查出公司的所有分机号
+        List<Integer> byCompanyId = extensionNumberRepository.findByCompanyId(companyId);
+        extensionUserRepository.deleteByNumbers(byCompanyId);
         extensionNumberRepository.deleteByCompanyId(companyId);
         extensionDisplayRepository.deleteByCompanyId(companyId);
     }