From 3c807a6efad5b9a898f38b60e9352bba3844275b Mon Sep 17 00:00:00 2001 From: Kane Wang Date: Thu, 12 Oct 2023 18:31:54 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=9D=E5=AD=98=E8=BF=9B=E5=BA=A6=EF=BC=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../test/BatchInsertTest.java | 140 ++++++++++++++++++ 1 file changed, 140 insertions(+) create mode 100644 code/后端/desktop_archievement_backend/src/test/java/com/cpic/xim/DesktopArchievement/test/BatchInsertTest.java diff --git a/code/后端/desktop_archievement_backend/src/test/java/com/cpic/xim/DesktopArchievement/test/BatchInsertTest.java b/code/后端/desktop_archievement_backend/src/test/java/com/cpic/xim/DesktopArchievement/test/BatchInsertTest.java new file mode 100644 index 0000000..d70e421 --- /dev/null +++ b/code/后端/desktop_archievement_backend/src/test/java/com/cpic/xim/DesktopArchievement/test/BatchInsertTest.java @@ -0,0 +1,140 @@ +/* + * @Author: Kane + * @Date: 2023-10-12 10:43:31 + * @LastEditors: Kane + * @FilePath: /desktop_archievement_backend/src/test/java/com/cpic/xim/DesktopArchievement/test/BatchInsertTest.java + * @Description: + * + * Copyright (c) ${2023} by Kane, All Rights Reserved. + */ +package com.cpic.xim.DesktopArchievement.test; + +import org.apache.ibatis.exceptions.PersistenceException; +import org.apache.ibatis.session.SqlSession; +import org.junit.Test; +import com.cpic.xim.mybatis.mapper.ImportBIArchievementDataMapper; +import com.cpic.xim.mybatis.pojo.*; +import com.cpic.xim.mybatis.utils.MybatisUtils; +import com.cpic.xim.utils.data.ImportBIData; +import java.util.ArrayList; + +public class BatchInsertTest +{ + @Test + public void testBatchInsertIntoBITelsalerAttachingRate() + { + String filePath = "D:/develop/cpicxim/deskop_task_schedule/数据/测试用/坐席车非渗透.xlsx"; + ArrayList records = null; + SqlSession session = null; + ImportBIArchievementDataMapper mapper = null; + + try + { + records = ImportBIData.importBITelsalerAttachingRateRecordFromXlsx( filePath, null, + true ); + session = MybatisUtils.getSqlSessionBatch(); + mapper = session.getMapper( ImportBIArchievementDataMapper.class ); + + mapper.cleanTelsalerAttachingRateData(); + + for ( BITelsalerAttachingRateRecord record : records ) + { + mapper.insertTelsalerAttachingRateDataToDB( record ); + } + + session.commit(); + + assert (true); + } + catch ( PersistenceException error) + { + System.out.println(error); + assert( false ); + } + catch ( Exception error ) + { + if ( session != null ) + { + session.rollback(); + + assert (false); + } + } + } + + @Test + public void testBatchInsertBITeslsalerRenewalRate() + { + String filePath = "D:/develop/cpicxim/deskop_task_schedule/数据/测试用/坐席续保率 .xlsx"; + String sheetName = "当月个车续保率跟踪报表【机构】"; + ArrayList records = null; + SqlSession session = null; + ImportBIArchievementDataMapper mapper = null; + + try + { + records = ImportBIData.importBITelsalerRenewalRateFromXlsx( filePath, sheetName, true, + null ); + + session = MybatisUtils.getSqlSessionBatch(); + mapper = session.getMapper( ImportBIArchievementDataMapper.class ); + + // 清理数据 + mapper.cleanTelsalerRenewalRateData(); + + for ( BITelsalerRenewalRateRecord record : records ) + { + mapper.insertTelsalerRenewalRateDataToDB( record ); + } + + session.commit(); + + System.out.println( records ); + assert (!records.isEmpty()); + } + catch ( Exception error ) + { + assert (false); + + if ( session != null ) + { + session.rollback(); + } + } + } + + @Test + public void testBatchInsertBIDepartmentArchievementToDB() + { + String filePath = "D:/develop/cpicxim/deskop_task_schedule/数据/测试用/BI部门渗透率续保率.xlsx"; + String sheetName = "部门"; + + ArrayList records = null; + SqlSession session = null; + ImportBIArchievementDataMapper mapper = null; + + try + { + records = + ImportBIData.importBIDepartmentArchievementRecords( filePath, sheetName, true ); + session = MybatisUtils.getSqlSessionBatch(); + mapper = session.getMapper( ImportBIArchievementDataMapper.class ); + + mapper.cleanDepartmentAttachingRenewalRateData(); + + for ( BIDepartmentArchievementRecord record : records ) + { + mapper.insertDepartmentAttachingRenewalRateDataToDB( record ); + } + + session.commit(); + } + catch ( Exception error ) + { + if ( session != null ) + { + session.rollback(); + } + } + } +}