package org.springblade.modules.riverway.service.impl;

import com.baomidou.dynamic.datasource.annotation.DS;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springblade.modules.riverway.entity.RiverwayR;
import org.springblade.modules.riverway.mapper.RiverwayMapper;
import org.springblade.modules.riverway.service.RiverwayService;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.InputStream;
import java.text.DecimalFormat;
import java.util.List;

@Service
@DS("slaves")
public class RiverwayServiceImpl extends ServiceImpl<RiverwayMapper, RiverwayR> implements RiverwayService {

	@Override
	public void upLoadFile(MultipartFile file) {
		try{
			InputStream in = file.getInputStream();
			Workbook workbook = null;
			try{
				workbook = new HSSFWorkbook(in);
			}catch(Exception e){
				workbook = new XSSFWorkbook(in);
			}


			int numberOfSheets = workbook.getNumberOfSheets();
			//第一个sheet
			for (int j=0;j<numberOfSheets;j++) {
				Sheet sheet1 = workbook.getSheetAt(j);

				String sheetName = sheet1.getSheetName();
				String temp[]=sheetName.split("-",1);

				/*//获取当前时间戳作为水面线ID
				Long waterID = new Date().getTime();
				//获取sheet名称作为水面线数据录入
				waterLine.setId(waterID);
				waterLine.setWaterline(temp[0]);
				if(temp.length >1){
					waterLine.setCjtime(temp[1]);
				}*/

				//有效行数
				int rowNubms = sheet1.getPhysicalNumberOfRows();
				for (int i = 1; i < rowNubms; i++) {
					Row rowi = sheet1.getRow(i);
					if (rowi == null) {
						continue;
					}

					RiverwayR riverwayR = new RiverwayR();

					Cell cell1 = rowi.getCell(0);//河段编码
					if(cell1.getCellType() == Cell.CELL_TYPE_BLANK) {
						continue;
					}
					Cell cell2 = rowi.getCell(1);//河段名称
					Cell cell3 = rowi.getCell(2);//河名
					Cell cell4 = rowi.getCell(3);//河段类型
					Cell cell5 = rowi.getCell(4);//河段长
					Cell cell6 = rowi.getCell(5);//起点经度
					Cell cell7 = rowi.getCell(6);//起点纬度
					Cell cell8 = rowi.getCell(7);//终点经度
					Cell cell9 = rowi.getCell(8);//终点纬度
					Cell cell10 = rowi.getCell(9);//关联文站编码
					Cell cell11 = rowi.getCell(10);//关联水文站起点距
					Cell cell12 = rowi.getCell(11);//默认水面线

					//此处判断为null，以防get值的时候 报空指针
					DecimalFormat df = new DecimalFormat("0");

					String hdbm = (cell1 != null) ? getCellValue(cell1) : null;
					String hdmc = (cell2 != null) ? getCellValue(cell2) : null;
					String hm = (cell3!= null) ? getCellValue(cell3) : null;
					String hdlx = (cell4 != null) ? getCellValue(cell4) : null;
					String hdc = df.format(cell5.getNumericCellValue());
					String qdjd = (cell6 != null) ? getCellValue(cell6) : null;
					String qdwd = (cell7 != null) ? getCellValue(cell7) : null;
					String zdjd = (cell8 != null) ? getCellValue(cell8) : null;
					String zdwd = (cell9 != null) ? getCellValue(cell9) : null;
					String glbm = (cell10 != null) ? getCellValue2(cell10) : null;
					String glqdj = (cell11 != null) ? getCellValue(cell11) : null;
					String mrsmx = (cell12 != null) ? getCellValue(cell12) : null;

					riverwayR.setRiverCode(hdbm);
					riverwayR.setRiverway(hdmc);
					riverwayR.setRiverName(hm);
					riverwayR.setRiverType(hdlx);
					riverwayR.setRiverLength(Integer.parseInt(hdc));
					riverwayR.setRiverOrigin(qdjd);
					riverwayR.setQdLat(qdwd);
					riverwayR.setDestination(zdjd);
					riverwayR.setZdLat(zdwd);
					riverwayR.setGlCode(glbm);
					riverwayR.setGlQdj(glqdj);
					riverwayR.setDefaultSmx(mrsmx);

					//河段新增
					RiverwayServiceImpl.super.save(riverwayR);

					in.close();  //关闭流
				}
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	@Override
	public List selectName() {
		return baseMapper.selectName();
	}

	private String getCellValue2(Cell cell){
		String value ="";
		switch (cell.getCellType()){
			case Cell.CELL_TYPE_STRING:
				value = cell.getStringCellValue();
				break;
			case  Cell.CELL_TYPE_NUMERIC:
				DecimalFormat df = new DecimalFormat("0");
				value = df.format(cell.getNumericCellValue());
				break;
			case  Cell.CELL_TYPE_BLANK:
				value = "";
				break;
			case Cell.CELL_TYPE_BOOLEAN:
				value = String.valueOf(cell.getBooleanCellValue());
				break;
			case Cell.CELL_TYPE_ERROR:
				value = String.valueOf(cell.getErrorCellValue());
				break;
			case Cell.CELL_TYPE_FORMULA:
				value = String.valueOf(cell.getCellFormula());
				break;
		}
		return value;
	}

	private String getCellValue(Cell cell){
		String value ="";
		switch (cell.getCellType()){
			case Cell.CELL_TYPE_STRING:
				value = cell.getStringCellValue();
				break;
			case  Cell.CELL_TYPE_NUMERIC:
				value = String.valueOf(cell.getNumericCellValue());
				break;
			case  Cell.CELL_TYPE_BLANK:
				value = "";
				break;
			case Cell.CELL_TYPE_BOOLEAN:
				value = String.valueOf(cell.getBooleanCellValue());
				break;
			case Cell.CELL_TYPE_ERROR:
				value = String.valueOf(cell.getErrorCellValue());
				break;
			case Cell.CELL_TYPE_FORMULA:
				value = String.valueOf(cell.getCellFormula());
				break;
		}
		return value;
	}

}