package com.lxw.test3d.Service;
|
|
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
|
import com.lxw.test3d.mapper.NjHomeConfigMapper;
|
import com.lxw.test3d.pojo.NjHomeConfig;
|
import org.apache.poi.ss.usermodel.Sheet;
|
import org.apache.poi.ss.usermodel.Workbook;
|
import org.apache.poi.ss.usermodel.WorkbookFactory;
|
import org.springframework.stereotype.Service;
|
|
import javax.annotation.Resource;
|
import java.io.File;
|
import java.io.FileInputStream;
|
import java.io.IOException;
|
import java.util.ArrayList;
|
import java.util.List;
|
|
@Service
|
public class NjHomeConfigService {
|
@Resource
|
private NjHomeConfigMapper mapper;
|
|
|
public void addToSql(String filepath){
|
//ExcelReader ec = new ExcelReader("E:\\ffmpeg\\4.xls", "ac");
|
//清空
|
UpdateWrapper wrapper=new UpdateWrapper();
|
wrapper.ge("num",0);
|
mapper.delete(wrapper);
|
List<String> listName=new ArrayList();
|
FileInputStream inStream = null;
|
try {
|
inStream = new FileInputStream(new File(filepath));
|
Workbook workBook = WorkbookFactory.create(inStream);
|
int numberOfSheets = workBook.getNumberOfSheets();
|
|
for (int i=0; i<numberOfSheets; i++) {
|
Sheet sheet = workBook.getSheetAt(i);
|
String sheetName=sheet.getSheetName();
|
listName.add(sheetName);
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
} finally {
|
try {
|
if (inStream != null) {
|
inStream.close();
|
}
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
for (String sheetName:listName){
|
ExcelReader ec = new ExcelReader(filepath, sheetName);
|
List<NjHomeConfig> list=ec.getSheet();
|
mapper.insertBatchSomeColumn(list);
|
}
|
}
|
}
|
