From a51147871c8e862d4882f4087171b8d25e743ebd Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期五, 07 四月 2023 15:07:11 +0800
Subject: [PATCH] 1
---
src/main/java/com/terra/coal/mapper/BasicMapper.java | 28 ++++++++++++++++++++++++++++
src/main/java/service/MainService.java | 22 ++++++++++++++++++----
src/main/java/com/terra/coal/mapper/Coal2000Mapper.java | 3 +--
src/main/java/com/terra/coal/mapper/Coal54Mapper.java | 3 +--
4 files changed, 48 insertions(+), 8 deletions(-)
diff --git a/src/main/java/com/terra/coal/mapper/BasicMapper.java b/src/main/java/com/terra/coal/mapper/BasicMapper.java
new file mode 100644
index 0000000..7cc6313
--- /dev/null
+++ b/src/main/java/com/terra/coal/mapper/BasicMapper.java
@@ -0,0 +1,28 @@
+package com.terra.coal.mapper;
+
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import org.apache.ibatis.annotations.Param;
+
+import java.util.List;
+
+/**
+ * BasicMapper
+ * @author WWW
+ */
+public interface BasicMapper<T> extends BaseMapper<T> {
+ /**
+ * insertBatch
+ *
+ * @param list
+ * @return
+ */
+ public int insertBatch(@Param("list") List<T> list);
+
+ /**
+ * updateBatch
+ *
+ * @param list
+ * @return
+ */
+ public int updateBatch(@Param("list") List<T> list);
+}
diff --git a/src/main/java/com/terra/coal/mapper/Coal2000Mapper.java b/src/main/java/com/terra/coal/mapper/Coal2000Mapper.java
index 699926a..5dd1a55 100644
--- a/src/main/java/com/terra/coal/mapper/Coal2000Mapper.java
+++ b/src/main/java/com/terra/coal/mapper/Coal2000Mapper.java
@@ -1,6 +1,5 @@
package com.terra.coal.mapper;
-import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.terra.coal.entity.Coal2000Entity;
import org.apache.ibatis.annotations.Mapper;
import org.springframework.stereotype.Repository;
@@ -11,6 +10,6 @@
*/
@Mapper
@Repository
-public interface Coal2000Mapper extends BaseMapper<Coal2000Entity> {
+public interface Coal2000Mapper extends BasicMapper<Coal2000Entity> {
//
}
diff --git a/src/main/java/com/terra/coal/mapper/Coal54Mapper.java b/src/main/java/com/terra/coal/mapper/Coal54Mapper.java
index 97272f6..6826419 100644
--- a/src/main/java/com/terra/coal/mapper/Coal54Mapper.java
+++ b/src/main/java/com/terra/coal/mapper/Coal54Mapper.java
@@ -1,6 +1,5 @@
package com.terra.coal.mapper;
-import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.terra.coal.entity.Coal54Entity;
import org.apache.ibatis.annotations.Mapper;
import org.springframework.stereotype.Repository;
@@ -11,6 +10,6 @@
*/
@Mapper
@Repository
-public interface Coal54Mapper extends BaseMapper<Coal54Entity> {
+public interface Coal54Mapper extends BasicMapper<Coal54Entity> {
//
}
diff --git a/src/main/java/service/MainService.java b/src/main/java/service/MainService.java
index 8dbf393..2e0848f 100644
--- a/src/main/java/service/MainService.java
+++ b/src/main/java/service/MainService.java
@@ -4,6 +4,8 @@
import com.terra.coal.entity.Coal54Entity;
import com.terra.coal.entity.StaticData;
import com.terra.coal.helper.StringHelper;
+import com.terra.coal.mapper.Coal54Mapper;
+import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.io.BufferedReader;
@@ -20,6 +22,9 @@
*/
@Service
public class MainService {
+ @Autowired
+ Coal54Mapper coal54Mapper;
+
/**
* 鍔犺浇54鏁版嵁(鍏ュ簱)
*/
@@ -27,23 +32,32 @@
FileReader fr = new FileReader(f);
BufferedReader reader = new BufferedReader(fr);
- int rows = 0;
+ int rows = 0, cc = 0;
String line = reader.readLine();
List<Coal54Entity> list = new ArrayList<>();
while (null != line) {
- rows++;
- if (rows > 2) {
+ cc++;
+ if (cc > 2) {
Coal54Entity entity = getCoal54(line);
if (null != entity) {
list.add(entity);
}
+ }
+ if (list.size() > 99) {
+ rows += coal54Mapper.insertBatch(list);
+ list.clear();
}
line = reader.readLine();
}
reader.close();
- return 0;
+ if (list.size() > 0) {
+ rows += coal54Mapper.insertBatch(list);
+ list.clear();
+ }
+
+ return rows;
}
private Coal54Entity getCoal54(String str) {
--
Gitblit v1.9.3