潘钊 / bsth_control · Files

package com.bsth.service.impl;

import com.bsth.entity.Refuel;
import com.bsth.entity.realcontrol.ChildTaskPlan;
import com.bsth.entity.realcontrol.ScheduleRealInfo;
import com.bsth.entity.sys.SysUser;
import com.bsth.repository.RefuelRepository;
import com.bsth.security.util.SecurityUtils;
import com.bsth.service.RefuelService;
import com.bsth.service.report.CulateMileageService;

import java.io.File;
import java.io.FileInputStream;
import java.math.BigDecimal;
import java.text.DecimalFormat;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

/**
 * Created in 19/9/3.
 */
@Service
public class RefuelServiceImpl extends BaseServiceImpl<Refuel, Long> implements RefuelService {

	@Autowired
	private RefuelRepository repository;
	
	@Autowired
	private CulateMileageService culateMileageService;

	@Override
	public List<Refuel> query(Map<String, Object> map) {
		// TODO Auto-generated method stub
		String date = "", car = "%%", driver = "%%";
		if(map.get("date") != null){
			date = map.get("date").toString();
		}
		if(map.get("car") != null){
			car = "%" + map.get("car").toString() + "%";
		}
		if(map.get("driver") != null){
			driver = "%" + map.get("driver").toString() + "%";
		}
		List<Refuel> list = repository.findByDate(date, car, driver);
		return list;
	}
	
	@Override
	public List<Map<String, Object>> queryDaily(Map<String, Object> map) {
		// TODO Auto-generated method stub
		DecimalFormat df = new DecimalFormat("0.##");
		
		String date = "", car = "%%", driver = "%%";
		if(map.get("date") != null){
			date = map.get("date").toString();
		}
		if(map.get("car") != null){
			car = "%" + map.get("car").toString() + "%";
		}
		if(map.get("driver") != null){
			driver = "%" + map.get("driver").toString() + "%";
		}
		
		List<Refuel> list = repository.findByDate(date, car, driver);
		
		List<String> keyList = new ArrayList<String>();
		Map<String, Refuel> rMap = new HashMap<String, Refuel>();
		for(Refuel rf : list){
			String key = rf.getCar() + "/" + rf.getDriver();
			if("/".equals(key)){
				continue;
			}
			if(rMap.containsKey(key)){
				Refuel r = rMap.get(key);
				if(rf.getOutOil().length() > 0){
					if(r.getOutOil().length() == 0){
						r.setOutOil(rf.getOutOil());
					} else if(Double.valueOf(rf.getOutOil()) > Double.valueOf(r.getOutOil())){
						r.setOutOil(rf.getOutOil());
					}
				}
				if(rf.getInOil().length() > 0){
					if(r.getInOil().length() == 0){
						r.setInOil(rf.getInOil());
					} else if(Double.valueOf(rf.getInOil()) < Double.valueOf(r.getInOil())){
						r.setInOil(rf.getInOil());
					}
				}
				if(rf.getInStation0().length() > 0){
					if(r.getInStation0().length() == 0){
						r.setInStation0(rf.getInStation0());
					} else {
						r.setInStation0(add(r.getInStation0(), rf.getInStation0()));
					}
				}
				if(rf.getInStation5().length() > 0){
					if(r.getInStation5().length() == 0){
						r.setInStation5(rf.getInStation5());
					} else {
						r.setInStation5(add(r.getInStation5(), rf.getInStation5()));
					}
				}
				if(rf.getOilCard0().length() > 0){
					if(r.getOilCard0().length() == 0){
						r.setOilCard0(rf.getOilCard0());
					} else {
						r.setOilCard0(add(r.getOilCard0(), rf.getOilCard0()));
					}
				}
				if(rf.getOilCard10().length() > 0){
					if(r.getOilCard10().length() == 0){
						r.setOilCard10(rf.getOilCard10());
					} else {
						r.setOilCard10(add(r.getOilCard10(), rf.getOilCard10()));
					}
				}
				if(rf.getEastStation0().length() > 0){
					if(r.getEastStation0().length() == 0){
						r.setEastStation0(rf.getEastStation0());
					} else {
						r.setEastStation0(add(r.getEastStation0(), rf.getEastStation0()));
					}
				}
				if(rf.getEastStation10().length() > 0){
					if(r.getEastStation10().length() == 0){
						r.setEastStation10(rf.getEastStation10());
					} else {
						r.setEastStation10(add(r.getEastStation10(), rf.getEastStation10()));
					}
				}
				if(rf.getOutStation0().length() > 0){
					if(r.getOutStation0().length() == 0){
						r.setOutStation0(rf.getOutStation0());
					} else {
						r.setOutStation0(add(r.getOutStation0(), rf.getOutStation0()));
					}
				}
				if(rf.getOutStation10().length() > 0){
					if(r.getOutStation10().length() == 0){
						r.setOutStation10(rf.getOutStation10());
					} else {
						r.setOutStation10(add(r.getOutStation10(), rf.getOutStation10()));
					}
				}
			} else {
				keyList.add(key);
				rMap.put(key, rf);
			}
		}
		
		List<Map<String, Object>> resList = new ArrayList<Map<String, Object>>();
		for(String key : keyList){
			Refuel r = rMap.get(key);
			Map<String, Object> m = new HashMap<String, Object>();
			
			String realMileage = "", oil = "0", consume = "";
			List<ScheduleRealInfo> temp = repository.scheduleByDate(date, r.getCar(), r.getDriver());
			List<ScheduleRealInfo> temp2 = new ArrayList<ScheduleRealInfo>();
			if(temp.size() > 0){
				for(ScheduleRealInfo s : temp){
					Set<ChildTaskPlan> cts = s.getcTasks();
					if(cts != null && cts.size() > 0){
						temp2.add(s);
		            }else{
			            if(s.getZdsjActual()!=null 
			            		&& s.getFcsjActual()!=null){
			            	temp2.add(s);
			            }
		            }
				}
				Double sjgl = culateMileageService.culateSjgl(temp2);
				Double ksgl = culateMileageService.culateKsgl(temp);
				Double jccgl = culateMileageService.culateJccgl(temp2);
				Double ljgl = culateMileageService.culateLjgl(temp2);
				
				realMileage = add(add(sjgl, ksgl), add(jccgl, ljgl));
			}
			
			if(r.getOutOil() != null && r.getInOil() != null && r.getOutOil().length() > 0 && r.getInOil().length() > 0){
				oil = new BigDecimal(r.getOutOil()).subtract(new BigDecimal(r.getInOil())).toString();
			}
			if(r.getInStation0().length() > 0){
				oil = add(oil, r.getInStation0());
			}
			if(r.getInStation5().length() > 0){
				oil = add(oil, r.getInStation5());
			}
			if(r.getOilCard0().length() > 0){
				oil = add(oil, r.getOilCard0());
			}
			if(r.getOilCard10().length() > 0){
				oil = add(oil, r.getOilCard10());
			}
			if(r.getEastStation0().length() > 0){
				oil = add(oil, r.getEastStation0());
			}
			if(r.getEastStation10().length() > 0){
				oil = add(oil, r.getEastStation10());
			}
			if(r.getOutStation0().length() > 0){
				oil = add(oil, r.getOutStation0());
			}
			if(r.getOutStation10().length() > 0){
				oil = add(oil, r.getOutStation10());
			}
			
			if(realMileage.length() > 0 && oil.length() > 0 && !("0".equals(realMileage))){
				consume = df.format(new BigDecimal(oil).divide(new BigDecimal(realMileage), 4, BigDecimal.ROUND_HALF_UP).multiply(new BigDecimal(100)));
			}
			
			m.put("date", r.getDateStr());
			m.put("line", r.getLineName());
			m.put("car", r.getCar());
			m.put("driver", r.getDriver());
			m.put("realMileage", realMileage);
			m.put("oil", oil);
			m.put("consume", consume);
			
			resList.add(m);
		}
		
		return resList;
	}
	
	@Override
	public String importExcel(File file) {
		SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
		SimpleDateFormat sd = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
		DecimalFormat df = new DecimalFormat("0.###");
		List<Refuel> list = new ArrayList<Refuel>();
		List<String> textList = new ArrayList<String>();
		String msg = "";
		try {
			POIFSFileSystem fs = new POIFSFileSystem(new FileInputStream(file));
			HSSFWorkbook wb = new HSSFWorkbook(fs);
			HSSFSheet sheet = wb.getSheetAt(0);
			// 取得总行数
			int rowNum = sheet.getLastRowNum() + 1;
			// 取得总列数
			int cellNum = sheet.getRow(0).getLastCellNum();
			HSSFRow row = null;
			HSSFCell cell = null;
			
			String userName = "";
			SysUser user = SecurityUtils.getCurrentUser();
			if(user != null && user.getUserName() != null){
				userName = user.getUserName();
			}
			
			//取值
			for(int i = 1; i < rowNum; i++){
				row = sheet.getRow(i);
				if (row == null){
					continue;
				}
				String text = "";
				for(int j = 0; j < cellNum; j++){
					cell = row.getCell(j);
					if(cell == null){
						text += ",";
						continue;
					}
						text += String.valueOf(cell) + ",";
				}
				String[] split = (text+";").split(",");
				String str = "";
				for(int j = 0; j < split.length && j < 14; j++){
					str += split[j];
				}
				if(str.trim().length() == 0)
					continue;
				textList.add(text + ";");
			}
			
			wb.close();
			fs.close();
			
			//验证
			for(int i = 0; i < textList.size(); i++){
				String text = textList.get(i);
				String[] split = text.split(",");
				String rq = split[0].trim();
				String line = split[1].trim();
				String car = split[2].trim();
				String driver = split[3].trim();
				if((car + driver).trim().length() == 0){
					continue;
				}
				String outOil = split[4].trim();
				String inOil = split[5].trim();
				String inStation0 = split[6].trim();
				String inStation5 = split[7].trim();
				String oilCard0 = split[8].trim();
				String oilCard10 = split[9].trim();
				String eastStation0 = split[10].trim();
				String eastStation10 = split[11].trim();
				String outStation0 = split[12].trim();
				String outStation10 = split[13].trim();
				
				driver = driver.replace(".0", "");
				
				String l = "", obj = "";
				try {
					rq = sdf.format(sdf.parse(rq));
					
					if(outOil.length() != 0){
						l = "E";obj = outOil;
						outOil = df.format(new BigDecimal(outOil));
					}
					if(inOil.length() != 0){
						l = "F";obj = inOil;
						inOil = df.format(new BigDecimal(inOil));
					}
					if(inStation0.length() != 0){
						l = "G";obj = inStation0;
						inStation0 = df.format(new BigDecimal(inStation0));
					}
					if(inStation5.length() != 0){
						l = "H";obj = inStation5;
						inStation5 = df.format(new BigDecimal(inStation5));
					}
					if(oilCard0.length() != 0){
						l = "I";obj = oilCard0;
						oilCard0 = df.format(new BigDecimal(oilCard0));
					}
					if(oilCard10.length() != 0){
						l = "J";obj = oilCard10;
						oilCard10 = df.format(new BigDecimal(oilCard10));
					}
					if(eastStation0.length() != 0){
						l = "K";obj = eastStation0;
						eastStation0 = df.format(new BigDecimal(eastStation0));
					}
					if(eastStation10.length() != 0){
						l = "L";obj = eastStation10;
						eastStation10 = df.format(new BigDecimal(eastStation10));
					}
					if(outStation0.length() != 0){
						l = "M";obj = outStation0;
						outStation0 = df.format(new BigDecimal(outStation0));
					}
					if(outStation10.length() != 0){
						l = "N";obj = outStation10;
						outStation10 = df.format(new BigDecimal(outStation10));
					}
				} catch (ParseException e) {
					// TODO: handle exception
					msg += "\\n"+(i+2)+"行日期格式错误";
					continue;
				} catch (NumberFormatException e) {
					// TODO: handle exception
					msg += "\\n"+(i+2)+"行"+l+"列内容：“"+obj+"”数据异常或有误";
					continue;
				}
				
				Refuel r = new Refuel();
				r.setDateStr(rq);
				r.setLineName(line);
				r.setCar(car);
				r.setDriver(driver);
				r.setOutOil(outOil);
				r.setInOil(inOil);
				r.setInStation0(inStation0);
				r.setInStation5(inStation5);
				r.setOilCard0(oilCard0);
				r.setOilCard10(oilCard10);
				r.setEastStation0(eastStation0);
				r.setEastStation10(eastStation10);
				r.setOutStation0(outStation0);
				r.setOutStation10(outStation10);
				list.add(r);
			}
			
			//导入（msg是验证中产生的格式报错信息）
			if(msg.length() == 0){
				for(Refuel r : list){
					if(msg.length() == 0){
						List<Refuel> rs = repository.selectByDateAndCar(r.getDateStr(), r.getLineName(), r.getCar(), r.getDriver());
						if(rs.size() > 0){
							for(Refuel re : rs){
								if(re.getId() != null){
									repository.update(re.getId(), r.getDateStr(), r.getDateStr(), r.getLineName(), r.getCar(), r.getDriver(),
											r.getOutOil(), r.getInOil(), r.getInStation0(), r.getInStation5(), r.getOilCard0(), r.getOilCard10(),
											r.getEastStation0(), r.getEastStation10(), r.getOutStation0(), r.getOutStation10(), userName, sd.format(new Date()));
								}
							}
						} else {
							repository.insertData(r.getDateStr(), r.getDateStr(), r.getLineName(), r.getCar(), r.getDriver(),
									r.getOutOil(), r.getInOil(), r.getInStation0(), r.getInStation5(), r.getOilCard0(), r.getOilCard10(),
									r.getEastStation0(), r.getEastStation10(), r.getOutStation0(), r.getOutStation10(), userName, sd.format(new Date()));
						}
					}
				}
			} else {
				throw new Exception();
			}
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
			return "文件导入失败" + msg;
		} finally {
			file.delete();
		}
		return "文件导入成功";
	}
	
	public String add(Object v1, Object v2)
    {
        BigDecimal b1 = new BigDecimal(v1.toString());
        BigDecimal b2 = new BigDecimal(v2.toString());
        return b1.add(b2).toString();
    }
	
}