Ver Fonte

调整数据库测试逻辑,不请求全部数据,而是查询数据条数。

DESKTOP-6LTVLN7\Liumouren há 2 meses atrás
pai
commit
c408fb7f80

+ 2 - 0
src/main/java/com/skyversation/poiaddr/service/AmapAddressV3Repository.java

@@ -15,4 +15,6 @@ public interface AmapAddressV3Repository extends JpaRepository<AmapAddressV3, St
     @Query(value = "SELECT * FROM amap_address_v3 WHERE name ILIKE %:addr% OR address ILIKE %:addr% LIMIT 20", nativeQuery = true)
     List<AmapAddressV3> getAddressPoisByAddr(@Param("addr") String addr);
 
+    @Query(value = "SELECT COUNT(*) FROM amap_address_v3", nativeQuery = true)
+    long countAllUsersNative();
 }

+ 2 - 2
src/main/java/com/skyversation/poiaddr/service/AreaService.java

@@ -91,9 +91,9 @@ public class AreaService {
         try {
             long startTime = System.currentTimeMillis();
             System.out.println("<<<<<<<<------开始数据库连通性测试");
-            List<AmapAddressV3> amapAddressV3List = amapAddressV3Service.getAllData();
+            Long amapAddressV3List = amapAddressV3Service.countAllUsersNative();
             long endTime = System.currentTimeMillis();
-            System.out.println(">>>>>>>>------数据库连通性测试完成!用时" + (endTime - startTime) / 1000 + "秒!结果:" + amapAddressV3List.size());
+            System.out.println(">>>>>>>>------数据库连通性测试完成!用时" + (endTime - startTime) / 1000 + "秒!当前数据库数据条数:" + amapAddressV3List);
 //            System.out.println("<<<<<<<<------开始执行任务!!!");
 //            yyskAddressStandardizationService.searchLoadFileData("D:\\1.项目\\5. 松江项目\\4. 地名地址\\地名地址库治理数据\\原始数据\\",
 //                    0, 10000, 1);

+ 2 - 2
src/main/java/com/skyversation/poiaddr/service/impl/AmapAddressV3Service.java

@@ -17,8 +17,8 @@ public class AmapAddressV3Service {
         return amapAddressV3Repository.saveAll(amapAddressV3List);
     }
 
-    public List<AmapAddressV3> getAllData() {
-        return amapAddressV3Repository.findAll();
+    public Long countAllUsersNative() {
+        return amapAddressV3Repository.countAllUsersNative();
     }
 
     public List<AmapAddressV3> getAddressPoisByAddr(String addr){