hrm-attendance/src/com/engine/attendance/component/persongroup/commonutil/PersongroupCommonUtil.java

package com.engine.attendance.component.persongroup.commonutil;

import com.engine.common.util.DateUtil;
import com.engine.common.util.DbTools;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.collect.Sets;
import lombok.extern.slf4j.Slf4j;
import weaver.conn.RecordSet;
import weaver.general.Util;

import java.util.*;
import java.util.stream.Collectors;

@Slf4j
public class PersongroupCommonUtil {
    /**
     *
     * @param calendarMap  key:年份，value：明细数据
     * @return
     */
    public static Map<String, List<Map<String,Object>>> getDayByCalendarMap(Map<String,String> calendarMap){
        Map<String, List<Map<String,Object>>> dataList = Maps.newHashMap();
        String sql = "select mainid,rq,mc,rqlx,xq,sc from uf_qyrl_dt1 where mainid in ";
        String mainids = "";
        for(Map.Entry<String,String> entry : calendarMap.entrySet()){
            mainids = mainids + entry.getValue() +",";
        }
        mainids = mainids.substring(0,mainids.length()-1);
        sql = sql +" ("+mainids+") ";
        log.info("sql : {}",sql);
        List<Map<String,Object>> list = DbTools.getSqlToList(sql);
        Map<String,List<Map<String,Object>>> collect = list.stream().collect(Collectors.groupingBy(e -> e.get("mainid").toString()));


        calendarMap.entrySet().forEach(e -> {
            dataList.put(e.getKey(),collect.get(e.getValue()));
        });

        return dataList;
    }
    public static int getBetweenDays(Map<String, Object> data){
        //主表数据id
        String id = Util.null2String(data.get("id"));
        //规律排班模块Id
        String modeId = Util.null2String(data.get("modeId"));
        //排班结果模块Id
        String formmodeid = Util.null2String(data.get("formmodeid"));
        //生成天数
        int day = Integer.valueOf(Util.null2String(data.get("day")));
        //企业日历
        String qyrl = Util.null2String(data.get("qyrl"));
        //节假日处理 0:改班，1：跳过，2：不处理
        String jjrcl = Util.null2String(data.get("jjrcl"));
        //公休日处理 0：改班，1：跳过，2：不处理
        String gxrcl = Util.null2String(data.get("gxrcl"));
        //生效日期
        String bdate = Util.null2String(data.get("bdate"));
        //当前日期
        String nowDate = DateUtil.getCurrentDate();
        //排班方式
        String pbfs = Util.null2String(data.get("pbfs"));
        //改班班次
        String gbbc = Util.null2String(data.get("gbbc"));


        //规律排班明细表数据
        String sql = "select ksts,jsts,hxbz,bcmc from uf_jcl_kq_glpb_dt1 where mainid=?";
        List<Map<String,Object>> detailDataList = DbTools.getSqlToList(sql,id);
        //规律排班适用组织数据
        sql = "select dx,dxlx,aqjb from uf_syzz where modeid=? and dataid=?";
       // List<Map<String,Object>> organizationList = DbTools.getSqlToList(sql,modeId,id);
        //规律排班所用企业日历数据
        sql = "select rlmc,nd,rq,rqlx from uf_jcl_kq_rlxx where rlmc=?";
       // List<Map<String,Object>> calendarList = DbTools.getSqlToList(sql,qyrl);
//        Map<String,String> calendarMap = calendarList.stream().collect(Collectors.toMap(e->Util.null2String(e.get("nd")),e->Util.null2String(e.get("id"))));
//        log.info("calendarMap : {}",calendarMap);
//        Map<String, List<Map<String,Object>>> detailDataGroupingByYear = calendarList.stream().collect(Collectors.groupingBy(e ->e.get("nd").toString()));
//
        //生效日期与当前日期相差年份
        int betweenYears =  DateUtil.getBetWeenYears(bdate,nowDate);
        //循环总天数
       // int circulateDays = Integer.valueOf(detailDataList.get(detailDataList.size()-1).get("jsts").toString());
        int betweenDays = DateUtil.getBetWeenDays(bdate,nowDate);
        /**计算生效日期到当前日期的总天数*/
        if (betweenYears == 0){
            String dateCondition = "rq >='"+bdate+"' and rq <'"+nowDate+"'";
            int needSkipDay = getNeedSkipDay(jjrcl,gxrcl,qyrl,dateCondition);
            log.info("needSkipDay : [{}]",needSkipDay);
            betweenDays = betweenDays-needSkipDay;

        }else if(betweenYears > 0){
            int needSkipDay = 0;
            int bgYear = Integer.valueOf(bdate.split("-")[0]);
            for(int i=0;i<=betweenYears;i++){
                int currentYear = bgYear +i;
                String endTime = currentYear+"-12-31";
                if(i == 0){
                    String dateCondition = "rq >='"+bdate+"' and rq <='"+endTime+"'";
                    needSkipDay += getNeedSkipDay(jjrcl,gxrcl,qyrl,dateCondition);
                }else if (i == betweenYears){
                    String beginTime = currentYear + "-01-01";
                    String dateCondition = "rq >='"+beginTime+"' and rq <'"+nowDate+"'";
                    needSkipDay += getNeedSkipDay(jjrcl,gxrcl,qyrl,dateCondition);
                }else {
                    String beginTime = currentYear + "-01-01";
                    String dateCondition = "rq >='"+beginTime+"' and rq <='"+endTime+"'";
                    needSkipDay += getNeedSkipDay(jjrcl,gxrcl,qyrl,dateCondition);
                }
            }
            log.info("needSkipDay : [{}]",needSkipDay);
            betweenDays = betweenDays-needSkipDay;
        }
        return betweenDays;
    }


    public static int getNeedSkipDay(String jjrcl,String gxrcl,String rlmc,String dateCondition){
        String sql = "select count(0) num from uf_jcl_kq_rlxx where rlmc=? and "+dateCondition;
        String conditionSql = "";
        if ("1".equals(jjrcl)){
            //节假日为跳过
            conditionSql = conditionSql + " rqlx=1";
        }
        if ("1".equals(gxrcl)){
            //公休日为跳过
            if ("".equals(conditionSql)){
                conditionSql = conditionSql + " rqlx=2 or rqlx=3";
            }else {
                conditionSql = conditionSql + " or rqlx=2 or rqlx=3";
            }
        }
        if (!"".equals(conditionSql)){
            sql = sql + " and ( "+conditionSql+")";
        }
        log.info("getNeedSkipDay sql : {},rlmc : {}",sql,rlmc);
        Map<String,Object> objectMap = DbTools.getSqlToMap(sql,rlmc);
        int needSkipDay = Integer.valueOf(objectMap.get("num").toString());
        return needSkipDay;
    }


    public static boolean insertShiftSchedulingResults(List<Map<String,Object>> organizationList,List<Map<String,Object>> insertDataList,String formmodeid){
        RecordSet rs = new RecordSet();
        List<List> addlist = Lists.newArrayList();
        String sql = "delete from uf_pbjg where bcrq>= ? and bcrq<=? ";
        String startDate = Util.null2String(insertDataList.get(0).get("bcrq"));
        String endDate = Util.null2String(insertDataList.get(insertDataList.size()-1).get("bcrq"));
        Set<String> psersons = Sets.newHashSet();
        Set<String> psersonsGroups = Sets.newHashSet();
        Set<String> departments = Sets.newHashSet();
        Set<String> subCompanys = Sets.newHashSet();
        for (Map<String,Object> organization: organizationList){

            for (Map<String,Object> insertData : insertDataList){
                ArrayList<Object> list = new ArrayList<>();
                list.add(insertData.get("pbtj"));
                list.add(insertData.get("pbgl"));
                list.add(organization.get("dxlx"));
                if ("0".equals(organization.get("dxlx"))){
                    //人员
                    list.add(Util.null2String(organization.get("dx")).split("-")[0]);
                    list.add(null);
                    list.add(null);
                    list.add(null);
                    psersons.add(Util.null2String(organization.get("dx")).split("-")[0]);

                }else if ("1".equals(organization.get("dxlx"))){
                    //人员分组
                    list.add(null);
                    list.add(Util.null2String(organization.get("dx")).split("-")[0]);
                    list.add(null);
                    list.add(null);
                    psersonsGroups.add(Util.null2String(organization.get("dx")).split("-")[0]);
                }else if ("2".equals(organization.get("dxlx"))){
                    //部门
                    list.add(null);
                    list.add(null);
                    list.add(Util.null2String(organization.get("dx")).split("-")[0]);
                    list.add(null);
                    departments.add(Util.null2String(organization.get("dx")).split("-")[0]);
                }else if ("3".equals(organization.get("dxlx"))){
                    //分部
                    list.add(null);
                    list.add(null);
                    list.add(null);
                    list.add(Util.null2String(organization.get("dx")).split("-")[0]);
                    subCompanys.add(Util.null2String(organization.get("dx")).split("-")[0]);
                }
                list.add(insertData.get("bcrq"));
                list.add(insertData.get("rqlx"));
                list.add(insertData.get("sfxx"));
                list.add(insertData.get("sfdkpp"));
                list.add(insertData.get("bcxx"));
                list.add(insertData.get("hxbz"));
                list.add(insertData.get("pbrq"));
                list.add(insertData.get("pbsj"));
                list.add(formmodeid);
                list.add(1);
                list.add(0);
                list.add(DateUtil.getCurrentDate());
                list.add(DateUtil.getCurrentTime().split(" ")[1]);
                list.add(UUID.randomUUID().toString());
                addlist.add(list);
            }

        }

        boolean result = false;
        if (psersons.size() >0){
            String deleteSql = sql + " and  pbdxry in ("+String.join(",",psersons)+")";
            log.info("deleteSql : {}",deleteSql);
            DbTools.update(rs,deleteSql,startDate,endDate);
        }
        if (psersonsGroups.size() >0){
            String deleteSql = sql + " and  pbdxryfz in ("+String.join(",",psersonsGroups)+")";
            log.info("deleteSql : {}",deleteSql,startDate,endDate);
            DbTools.update(rs,deleteSql,startDate,endDate);
        }
        if (departments.size() >0){
            String deleteSql = sql + " and  pbdxbm in ("+String.join(",",departments)+")";
            log.info("deleteSql : {}",deleteSql,startDate,endDate);
            DbTools.update(rs,deleteSql,startDate,endDate);
        }
        if (subCompanys.size() >0){
            String deleteSql = sql + " and  pbdxfb in ("+String.join(",",subCompanys)+")";
            log.info("deleteSql : {}",deleteSql,startDate,endDate);
            DbTools.update(rs,deleteSql,startDate,endDate);
        }

        if (addlist.size() >0){
            log.info("addlist :[{}]",addlist);
            String insertSql = "insert into uf_pbjg (pbtj,pbgl,dxlx,pbdxry,pbdxryfz,pbdxbm,pbdxfb,bcrq,rqlx,sfxx,sfdkpp,bcxx,hxbz,pbrq,pbsj,formmodeid,modedatacreater,modedatacreatertype,modedatacreatedate,modedatacreatetime,MODEUUID) values (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)";
             result =rs.executeBatchSql(insertSql,addlist);
            log.info("result :[{}]",result);
        }
        return result;
    }


    /***
     * 获得人员是否在人员分组里
     * @param personnelGrouping 人员分组集合
     * @param userId 人员id
     * @return 人员分组主键ID集合
     */
    public static Set<String> getPersonnelGroupingByPerson(List<Map<String,Object>> personnelGrouping,String userId,String startDate,String endDate){
        Map<String,List<Map<String,Object>>> collect = personnelGrouping.stream().collect(Collectors.groupingBy(e -> Util.null2String(e.get("mainid"))));
        Set<String> personnelGroupIds = Sets.newHashSet();
        log.info("getPersonnelGroupingByPerson collect : {}",collect);
        if (userId == null || "".equals(userId)){
            return personnelGroupIds;
        }
        for (Map.Entry<String,List<Map<String,Object>>> e : collect.entrySet()){
            List<Map<String,Object>> value = e.getValue();
            for (Map<String,Object> data :value ){
                String edate = Util.null2String(data.get("edate"));
                String bdate = Util.null2String(data.get("bdate"));
                if (!"".equals(startDate) && startDate != null){
                    if ((!"".equals(edate) && DateUtil.getBetWeenDays(edate,startDate) >0) ||  DateUtil.getBetWeenDays(bdate,endDate) < 0){
                        continue;
                    }
                }
                if (!"".equals(edate)  && DateUtil.getBetWeenDays(edate) >0){
                    continue;
                }
                String empid = Util.null2String(data.get("empid"));
                String filters = Util.null2String(data.get("filters"));
                log.info(" empid : {},userId : {}",empid,userId);
                if (empid.equals(userId)){
                    personnelGroupIds.add(e.getKey());
                }else if (!"".equals(filters)){
                    filters = filters.replace("ａｎｄ","and");
                    filters = filters.replace("ｏｒ","or");
                    String sql = "select id from hrmresource where 1=1 and " +filters;
                    log.info("getPersonnelGroupingByPerson filter sql : {}",sql);
                    List<Map<String,Object>> dataList = DbTools.getSqlToList(sql);
                    for (Map<String,Object> dataMap :dataList){
                        String hrmId = Util.null2String(dataMap.get("id"));
                        if (hrmId.equals(userId)){
                            personnelGroupIds.add(e.getKey());
                        }
                    }
                }
            }
        }


        return personnelGroupIds;
    }

    /**
     * 获得人员分组集合
     * @return
     */
    public static Map<String,Object> getPersonGroup(){
        String sql = "select id,name from uf_ryqz";
        return DbTools.getSqlToMapList(sql);
    }

    /**
     * 获得部门集合
     * @return
     */
    public static Map<String,Object> getDepartMent(){
        String sql = "select id,departmentname name from hrmdepartment";
        return DbTools.getSqlToMapList(sql);
    }
    /**
     * 获得分部集合
     * @return
     */
    public static Map<String,Object> getSubCompany(){
        String sql = "select id,subcompanyname name from hrmsubcompany";
        return DbTools.getSqlToMapList(sql);
    }
    /**
     * 获得班次信息集合
     * @return
     */
    public static Map<String,Object> getClassesInformation(){
        String sql = "select id,mc name from uf_jcl_kq_bcxx";
        return DbTools.getSqlToMapList(sql);
    }
    /**
     * 获得候选班次集合
     * @return
     */
    public static Map<String,Object> getCandidateTeam(){
        String sql = "select id,mc name from uf_jcl_kq_hxbz";
        return DbTools.getSqlToMapList(sql);
    }


}