diff --git a/ruoyi-modules/management-platform/src/test/java/com/ruoyi/platform/utils/ExcelReaderTest.java b/ruoyi-modules/management-platform/src/test/java/com/ruoyi/platform/utils/ExcelReaderTest.java new file mode 100644 index 00000000..d1639af0 --- /dev/null +++ b/ruoyi-modules/management-platform/src/test/java/com/ruoyi/platform/utils/ExcelReaderTest.java @@ -0,0 +1,52 @@ +package com.ruoyi.platform.utils; + +import com.ruoyi.platform.domain.kg.*; +import com.ruoyi.platform.repository.MatKGRepository; +import com.ruoyi.platform.service.impl.MatKGServiceImpl; +import com.ruoyi.platform.vo.MatKGVo; +import org.junit.jupiter.api.Test; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.boot.test.context.SpringBootTest; + +import java.util.Optional; +import java.util.concurrent.ExecutorService; +import java.util.concurrent.Executors; + +@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT) +public class ExcelReaderTest { + @Autowired + private MatKGServiceImpl service; + + @Autowired + private MatKGRepository repository; + @Test + void read() { + String excelFilePath = "C:\\haha\\matKG\\ENTDOI.csv"; + + ExecutorService executorService = Executors.newFixedThreadPool(10); + // 创建ExcelReader实例 + ExcelReader excelReader = new ExcelReader<>(excelFilePath, MatKGVo.class, data -> { +// Optional existingEntity = service.findById(data.getEntity()); +// executorService.execute(() -> { + MaterialEntity entity = new MaterialEntity(); + entity.setId(data.getUnnamed()); + entity.setEntity(data.getEntity()); + entity.setPreferredEntity(new PreferredEntity(data.getPreferred_Entity())); + entity.setDoi(new DOI(data.getDOI(),data.getYear())); + entity.setNerTag(new NERTag(data.getNER_Tag())); + entity.setPartOfText(new PartOfText(data.getPart_of_text())); + service.save(entity); +// }); +// System.out.println(data); + }); + + // 读取Excel文件 + excelReader.read(); + } + + @Test + public void xx(){ + Optional byId = repository.findById("27358"); + System.out.println(byId.get()); + } +} \ No newline at end of file