tx-lims-after.git

			@@ -2,6 +2,7 @@

			import cn.hutool.core.date.DateUtil;
			import com.baomidou.mybatisplus.core.metadata.IPage;
			import com.baomidou.mybatisplus.core.toolkit.Wrappers;
			import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
			import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
			import com.yuanchu.mom.dto.PerformanceShiftAddDto;
			@@ -12,10 +13,13 @@
			import com.yuanchu.mom.service.PerformanceShiftService;
			import org.springframework.beans.factory.annotation.Autowired;
			import org.springframework.stereotype.Service;
			import org.springframework.transaction.annotation.Transactional;

			import java.text.SimpleDateFormat;
			import java.time.LocalDate;
			import java.time.LocalDateTime;
			import java.time.format.DateTimeFormatter;
			import java.time.temporal.TemporalAdjusters;
			import java.util.*;

			/**
			@@ -32,51 +36,88 @@
			@Autowired
			private EnumService enumService;

			public List<PerformanceShift> list = new ArrayList<>();

			@Transactional(rollbackFor = Exception.class)
			@Override
			public void performanceShiftAdd(PerformanceShiftAddDto performanceShiftAddDto) {
			DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd");
			String formattedDateTime = performanceShiftAddDto.getStartWeek().format(formatter);
			boolean exists = baseMapper.exists(Wrappers.<PerformanceShift>lambdaQuery()
			.eq(PerformanceShift::getWorkTime, formattedDateTime));
			String[] splitUserId = performanceShiftAddDto.getUserId().split(",");
			// 如果不存在添加数据
			if (!exists) {
			LocalDate firstDayOfMonth = performanceShiftAddDto.getEndWeek().toLocalDate().withDayOfMonth(1);
			LocalDate lastDayOfMonth = performanceShiftAddDto.getEndWeek().toLocalDate().with(TemporalAdjusters.lastDayOfMonth());
			List<LocalDateTime> localDateTimesBetween = getLocalDateTimesBetween(firstDayOfMonth.atStartOfDay(), lastDayOfMonth.atStartOfDay());
			localDateTimesBetween.forEach(i -> {
			for (String s : splitUserId) {
			PerformanceShift performanceShift = new PerformanceShift();
			performanceShift.setUserId(Integer.valueOf(s));
			performanceShift.setWorkTime(i);
			performanceShift.setShift("");
			list.add(performanceShift);
			}
			if (list.size() >= 1000) {
			baseMapper.insertBatchSomeColumn(list);
			list.clear();
			}
			});
			if (!list.isEmpty()) {
			baseMapper.insertBatchSomeColumn(list);
			}
			}
			// 再次更新
			List<LocalDateTime> datesBetween = getLocalDateTimesBetween(performanceShiftAddDto.getStartWeek(), performanceShiftAddDto.getEndWeek());
			for (LocalDateTime date : datesBetween) {
			String[] splitUserId = performanceShiftAddDto.getUserId().split(",");
			for (String s : splitUserId) {
			PerformanceShift performanceShift = new PerformanceShift();
			performanceShift.setShift(performanceShiftAddDto.getShift());
			performanceShift.setUserId(Integer.valueOf(s));
			performanceShift.setWorkTime(date);
			baseMapper.insert(performanceShift);
			String formatterDateTime = date.format(formatter);
			baseMapper.update(new PerformanceShift(), Wrappers.<PerformanceShift>lambdaUpdate()
			.set(PerformanceShift::getShift, performanceShiftAddDto.getShift())
			.eq(PerformanceShift::getUserId, s)
			.eq(PerformanceShift::getWorkTime, formatterDateTime));
			}
			}
			}

			@Override
			public IPage<Map<String, Object>> performanceShiftPage(Page<Object> page, LocalDateTime time, String userName, String laboratory) {
			public IPage<Map<String, Object>> performanceShiftPage(Page<Object> page, String time, String userName, String laboratory) {
			IPage<Map<String, Object>> mapIPage = baseMapper.performanceShiftPage(page, time, userName, laboratory);
			List<Enums> shiftType = enumService.selectEnumByCategory("班次类型");
			for (Map<String, Object> i : mapIPage.getRecords()) {
			String[] shiftTimes = i.get("shiftTime").toString().split(";");
			int totalAttendance = 0;
			Map<Object, Object> map = new HashMap<>();
			for (Enums enums : shiftType) {
			int num = 0;
			// 分割日期
			for (String shiftTime : shiftTimes) {
			Map<Object, Object> hashMap = new HashMap<>();
			String[] shiftTimeAndShift = shiftTime.split("：");
			if (enums.getValue().equals(shiftTimeAndShift[1])) {
			num++;
			List<Object> map = new ArrayList<>();
			// 分割日期
			for (String shiftTime : shiftTimes) {
			Map<Object, Object> hashMap = new HashMap<>();
			String[] shiftTimeAndShift = shiftTime.split("：");
			for (Enums enums : shiftType) {
			if (!i.containsKey(enums.getLabel())) {
			i.put(enums.getLabel(), 0);
			}
			hashMap.put("id", shiftTimeAndShift[2]);
			hashMap.put("shift", shiftTimeAndShift[1]);
			hashMap.put("weekly", DateUtil.weekOfYear(DateUtil.parse(shiftTimeAndShift[0])));
			hashMap.put("time", shiftTimeAndShift[0]);
			map.put(getWeek(shiftTimeAndShift[0]), hashMap);
			if (enums.getValue().equals(shiftTimeAndShift[1])) {
			Integer num = (Integer) i.get(enums.getLabel());
			i.put(enums.getLabel(), num += 1);
			}
			}
			i.put(enums.getLabel(), num);
			i.put("list", map);
			if(!enums.getValue().equals("3") && !enums.getValue().equals("4")) {
			i.put("totalAttendance", totalAttendance += num);
			if (shiftTimeAndShift[1].equals("1") \|\| shiftTimeAndShift[1].equals("2") \|\| shiftTimeAndShift[1].equals("0")) {
			i.put("totalAttendance", totalAttendance += 1);
			}
			i.remove("shiftTime");
			hashMap.put("id", shiftTimeAndShift[2]);
			hashMap.put("shift", shiftTimeAndShift[1]);
			hashMap.put("weekly", DateUtil.weekOfYear(DateUtil.parse(shiftTimeAndShift[0])));
			hashMap.put("time", shiftTimeAndShift[0]);
			hashMap.put("headerTime", getWeek(shiftTimeAndShift[0]));
			map.add(hashMap);
			}
			i.put("list", map);
			i.remove("shiftTime");
			}
			return mapIPage;
			}