java-insagent-pc/src/main/java/com/ydtech/modules/admin/service/impl/SysGeographyPositionServiceImpl.java

package com.ydtech.modules.admin.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollectionUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ydtech.modules.admin.dao.SysDeptMapper;
import com.ydtech.modules.admin.dao.SysGeographyPositionMapper;
import com.ydtech.modules.admin.dao.SysGeographyPositionRecordMapper;
import com.ydtech.modules.admin.model.SysDept;
import com.ydtech.modules.admin.model.SysGeographyPosition;
import com.ydtech.modules.admin.model.SysGeographyPositionRecord;
import com.ydtech.modules.admin.model.SysUser;
import com.ydtech.modules.admin.model.dto.*;
import com.ydtech.modules.admin.model.vo.*;
import com.ydtech.modules.admin.service.GeoService;
import com.ydtech.modules.admin.service.SysGeographyPositionRecordService;
import com.ydtech.modules.admin.service.SysGeographyPositionService;
import com.ydtech.modules.admin.service.SysUserService;
import com.ydtech.modules.order.entity.BasePageResult;
import com.ydtech.utils.StringUtils;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.*;
import java.util.stream.Collectors;

@Service("SysGeographyPositionService")
public class SysGeographyPositionServiceImpl extends ServiceImpl<SysGeographyPositionMapper, SysGeographyPosition> implements SysGeographyPositionService {
    @Resource
    private SysGeographyPositionMapper sysGeographyPositionMapper;

    @Resource
    private SysGeographyPositionRecordMapper sysGeographyPositionRecordMapper;

    @Resource
    private SysGeographyPositionRecordService sysGeographyPositionRecordService;
    @Resource
    private SysUserService sysUserService;

    @Resource
    private SysDeptMapper sysDeptMapper;
    @Resource
    private GeoService geoService;

    private static final String APP_GEO_KEY = "APPLocations";


    @Override
    public SysGeographyPositionVo queryPosition(SysGeographyPositionDto sysGeographyPositionDto) {
        SysGeographyPositionVo sysGeographyPositionVo = new SysGeographyPositionVo();

        List<SysGeographyPosition> list = sysGeographyPositionMapper.selectList(new LambdaQueryWrapper<SysGeographyPosition>()
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getProvince()), SysGeographyPosition::getProvince, sysGeographyPositionDto.getProvince())
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getCity()), SysGeographyPosition::getCity, sysGeographyPositionDto.getCity())
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getDistrict()), SysGeographyPosition::getDistrict, sysGeographyPositionDto.getDistrict())
                .in(SysGeographyPosition::getSource, Arrays.asList("1", "2"))
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getSource()), SysGeographyPosition::getSource, sysGeographyPositionDto.getSource())
                .eq(sysGeographyPositionDto.getStatus() != null, SysGeographyPosition::getStatus, sysGeographyPositionDto.getStatus())
                .orderByDesc(SysGeographyPosition::getCreateTime));
        if (list.size() == 0) return new SysGeographyPositionVo();

        List<SysGeographyPositionInfoVo> sysGeographyPositionList = BeanUtil.copyToList(list, SysGeographyPositionInfoVo.class);
        //填充机构名
        for (SysGeographyPositionInfoVo sysGeographyPositionInfoVo : sysGeographyPositionList) {
            SysUser sysUser = sysUserService.getById(sysGeographyPositionInfoVo.getId());
            if (sysUser == null) continue;
            SysDept sysDept = sysDeptMapper.selectOne(new LambdaQueryWrapper<SysDept>()
                    .eq(SysDept::getId, sysUser.getDeptId()).eq(SysDept::getDelFlag, 0));
            if (sysDept == null) continue;
            sysGeographyPositionInfoVo.setDeptName(sysDept.getName());

        }

        sysGeographyPositionVo.setList(sysGeographyPositionList);
        Long aLong = sysGeographyPositionMapper.selectCount(new LambdaQueryWrapper<SysGeographyPosition>().eq(SysGeographyPosition::getStatus, 1).in(SysGeographyPosition::getSource, Arrays.asList("1", "2")));
        sysGeographyPositionVo.setManNum(aLong);

        return sysGeographyPositionVo;
    }

    @Override
    public BasePageResult<SysGeographyPositionRecord> queryPositionList(SysGeographyPositionPageDto sysGeographyPositionPageDto) {
        Page<SysGeographyPositionRecord> page = new Page<>(sysGeographyPositionPageDto.getPageNum(), sysGeographyPositionPageDto.getPageSize());

        LambdaQueryWrapper<SysGeographyPositionRecord> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(StringUtils.isNotEmpty(sysGeographyPositionPageDto.getId()), SysGeographyPositionRecord::getUserId, sysGeographyPositionPageDto.getId());
        wrapper.orderByDesc(SysGeographyPositionRecord::getCreateTime);

        Page<SysGeographyPositionRecord> insOrdersPage = sysGeographyPositionRecordService.page(page, wrapper);

        return new BasePageResult<>(sysGeographyPositionPageDto.getPageNum(), page.getSize(), insOrdersPage.getTotal(), page.getPages(),
                insOrdersPage.getRecords());

    }


    @Override
    public SysPcAddressNumVo queryPcNum(SysGeographyPositionDto sysGeographyPositionDto) {
        SysPcAddressNumVo sysPcAddressNumVo = new SysPcAddressNumVo();

        Long all = sysGeographyPositionMapper.selectCount(new LambdaQueryWrapper<SysGeographyPosition>()
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getProvince()), SysGeographyPosition::getProvince, sysGeographyPositionDto.getProvince())
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getCity()), SysGeographyPosition::getCity, sysGeographyPositionDto.getCity())
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getDistrict()), SysGeographyPosition::getDistrict, sysGeographyPositionDto.getDistrict())
                .in(SysGeographyPosition::getSource, Arrays.asList("1", "2"))
                .eq(sysGeographyPositionDto.getStatus() != null, SysGeographyPosition::getStatus, sysGeographyPositionDto.getStatus()));
        Long channelNum = sysGeographyPositionMapper.selectCount(new LambdaQueryWrapper<SysGeographyPosition>()
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getProvince()), SysGeographyPosition::getProvince, sysGeographyPositionDto.getProvince())
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getCity()), SysGeographyPosition::getCity, sysGeographyPositionDto.getCity())
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getDistrict()), SysGeographyPosition::getDistrict, sysGeographyPositionDto.getDistrict())
                .eq(SysGeographyPosition::getSource, 1)
                .eq(sysGeographyPositionDto.getStatus() != null, SysGeographyPosition::getStatus, sysGeographyPositionDto.getStatus()));
        Long proxyNum = sysGeographyPositionMapper.selectCount(new LambdaQueryWrapper<SysGeographyPosition>()
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getProvince()), SysGeographyPosition::getProvince, sysGeographyPositionDto.getProvince())
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getCity()), SysGeographyPosition::getCity, sysGeographyPositionDto.getCity())
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getDistrict()), SysGeographyPosition::getDistrict, sysGeographyPositionDto.getDistrict())
                .eq(SysGeographyPosition::getSource, 2)
                .eq(sysGeographyPositionDto.getStatus() != null, SysGeographyPosition::getStatus, sysGeographyPositionDto.getStatus()));
        Long onlineNum = sysGeographyPositionMapper.selectCount(new LambdaQueryWrapper<SysGeographyPosition>()
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getProvince()), SysGeographyPosition::getProvince, sysGeographyPositionDto.getProvince())
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getCity()), SysGeographyPosition::getCity, sysGeographyPositionDto.getCity())
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getDistrict()), SysGeographyPosition::getDistrict, sysGeographyPositionDto.getDistrict())
                .in(SysGeographyPosition::getSource, Arrays.asList("1", "2"))
                .eq(sysGeographyPositionDto.getStatus() == null, SysGeographyPosition::getStatus, 1)
                .eq(sysGeographyPositionDto.getStatus() != null, SysGeographyPosition::getStatus, sysGeographyPositionDto.getStatus()));
        sysPcAddressNumVo.setAllNum(all);
        sysPcAddressNumVo.setProxyNum(proxyNum);
        sysPcAddressNumVo.setChannelNum(channelNum);
        sysPcAddressNumVo.setOnlineNum(onlineNum);


        return sysPcAddressNumVo;
    }

    @Override
    public List<RegionInfoVo> queryRegionInfo(SysGeographyPositionDto sysGeographyPositionDto) {
        List<RegionInfoVo> regionInfoVos = new ArrayList<>();
        //全部：null  渠道：1 代理：2
        List<SysGeographyPosition> sysGeographyPositionList = sysGeographyPositionMapper.selectList(new LambdaQueryWrapper<SysGeographyPosition>()
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getProvince()), SysGeographyPosition::getProvince, sysGeographyPositionDto.getProvince())
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getCity()), SysGeographyPosition::getCity, sysGeographyPositionDto.getCity())
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getDistrict()), SysGeographyPosition::getDistrict, sysGeographyPositionDto.getDistrict())
                .in(SysGeographyPosition::getSource, Arrays.asList("1", "2"))
                .eq(StringUtils.isNotEmpty(sysGeographyPositionDto.getSource()), SysGeographyPosition::getSource, sysGeographyPositionDto.getSource())
                .eq(sysGeographyPositionDto.getStatus() != null, SysGeographyPosition::getStatus, sysGeographyPositionDto.getStatus()));

        if (CollectionUtil.isEmpty(sysGeographyPositionList)) return regionInfoVos;

        Map<String, List<SysGeographyPosition>> map;
        if (StringUtils.isEmpty(sysGeographyPositionDto.getProvince())) {
            map = sysGeographyPositionList.stream().collect(Collectors.groupingBy(SysGeographyPosition::getProvince));
        } else {
            if (StringUtils.isNotEmpty(sysGeographyPositionDto.getCity())) {
                map = sysGeographyPositionList.stream().collect(Collectors.groupingBy(SysGeographyPosition::getDistrict));

            } else {
                map = sysGeographyPositionList.stream().collect(Collectors.groupingBy(SysGeographyPosition::getCity));

            }
        }

        for (Map.Entry<String, List<SysGeographyPosition>> entry : map.entrySet()) {
            RegionInfoVo regionInfoVo = new RegionInfoVo();
            regionInfoVo.setDistrictName(entry.getKey());
            regionInfoVo.setNumber(entry.getValue().size());
            regionInfoVos.add(regionInfoVo);
        }

        return regionInfoVos;

    }


    @Override
    public List<SysGeographyPositionInfoVo> findLocationsWithinRadius(AdressRangeDto adressRangeDto) {

        geoService.addLocationGeography(list(), APP_GEO_KEY);
        List<String> nearbyIds = geoService.findNearbyIds(APP_GEO_KEY, adressRangeDto.getLongitude(), adressRangeDto.getLatitude(), adressRangeDto.getRadius());
        if (nearbyIds.size() == 0) return new ArrayList<>();
        geoService.move(nearbyIds, APP_GEO_KEY);
        List<SysGeographyPosition> sysGeographyPositions = listByIds(nearbyIds);
        List<SysGeographyPositionInfoVo> sysGeographyPositionInfoVoList = BeanUtil.copyToList(sysGeographyPositions, SysGeographyPositionInfoVo.class);
        if (sysGeographyPositionInfoVoList.size() == 0) return sysGeographyPositionInfoVoList;

        for (SysGeographyPositionInfoVo sysGeographyPositionInfoVo : sysGeographyPositionInfoVoList) {
            SysUser sysUser = sysUserService.getById(sysGeographyPositionInfoVo.getId());
            if (sysUser == null) continue;
            SysDept sysDept = sysDeptMapper.selectOne(new LambdaQueryWrapper<SysDept>()
                    .eq(SysDept::getId, sysUser.getDeptId()).eq(SysDept::getDelFlag, 0));
            if (sysDept == null) continue;
            sysGeographyPositionInfoVo.setDeptName(sysDept.getName());
        }

        return sysGeographyPositionInfoVoList;

    }

    @Override
    public BasePageResult<SysGeographyPositionInfoVo> queryPcAddressInfo(PcAddressInfoDto addressInfoDto) {
        Page<SysGeographyPosition> page = new Page<>(addressInfoDto.getPageNum(), addressInfoDto.getPageSize());

        LambdaQueryWrapper<SysGeographyPosition> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(StringUtils.isNotEmpty(addressInfoDto.getLng()), SysGeographyPosition::getLongitude, addressInfoDto.getLng());
        wrapper.eq(StringUtils.isNotEmpty(addressInfoDto.getLat()), SysGeographyPosition::getLatitude, addressInfoDto.getLat());
        wrapper.in(SysGeographyPosition::getSource, Arrays.asList("1", "2"));
        wrapper.eq(StringUtils.isNotEmpty(addressInfoDto.getSource()), SysGeographyPosition::getSource, addressInfoDto.getSource());
        wrapper.orderByDesc(SysGeographyPosition::getCreateTime);

        Page<SysGeographyPosition> insOrdersPage = page(page, wrapper);
        List<SysGeographyPosition> records = insOrdersPage.getRecords();

        List<SysGeographyPositionInfoVo> sysGeographyPositionList = BeanUtil.copyToList(records, SysGeographyPositionInfoVo.class);
        //填充机构名
        for (SysGeographyPositionInfoVo sysGeographyPositionInfoVo : sysGeographyPositionList) {
            SysUser sysUser = sysUserService.getById(sysGeographyPositionInfoVo.getId());
            if (sysUser == null) continue;
            SysDept sysDept = sysDeptMapper.selectOne(new LambdaQueryWrapper<SysDept>()
                    .eq(SysDept::getId, sysUser.getDeptId()).eq(SysDept::getDelFlag, 0));
            if (sysDept == null) continue;
            sysGeographyPositionInfoVo.setDeptName(sysDept.getName());

        }

        return new BasePageResult<>(addressInfoDto.getPageNum(), page.getSize(), insOrdersPage.getTotal(), page.getPages(),
                sysGeographyPositionList);
    }

    @Override
    public BasePageResult<SysGeographyPositionInfoVo> queryGeographyRecord(PcAddressInfoDto addressInfoDto) {
        Page page = new Page(addressInfoDto.getPageNum(), addressInfoDto.getPageSize());
        IPage<SysGeographyPositionInfoVo> sysGeographyPositionList = sysGeographyPositionRecordMapper.queryGeographyHistory(page,addressInfoDto);
        List<SysGeographyPositionInfoVo> records = sysGeographyPositionList.getRecords();
        //填充机构名
        for (SysGeographyPositionInfoVo sysGeographyPositionInfoVo : records) {
            SysUser sysUser = sysUserService.getById(sysGeographyPositionInfoVo.getUserId());
            if (sysUser == null) continue;
            SysDept sysDept = sysDeptMapper.selectOne(new LambdaQueryWrapper<SysDept>()
                    .eq(SysDept::getId, sysUser.getDeptId()).eq(SysDept::getDelFlag, 0));
            if (sysDept == null) continue;
            sysGeographyPositionInfoVo.setDeptName(sysDept.getName());
        }

        return new BasePageResult<>(addressInfoDto.getPageNum(), addressInfoDto.getPageSize(), sysGeographyPositionList.getTotal(), sysGeographyPositionList.getPages(),
                records);
    }

    @Override
    public AppGeographyHistoryInfoVo getGeographyPosition(String startTime, String endTime, String source) {
        AppGeographyHistoryInfoVo appGeographyHistoryInfoVo = new AppGeographyHistoryInfoVo();
        if (StringUtils.isEmpty(startTime) || startTime.equals(endTime)) {

            if (StringUtils.isNullOrEmpty(startTime)) {
                Date date = new Date();
                SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd");
                startTime = simpleDateFormat.format(date);
            }
            QueryWrapper<SysGeographyPositionRecord> queryWrapper = new QueryWrapper<>();
            // 指定 COUNT(DISTINCT user_id) 查询
            queryWrapper.select("COUNT(DISTINCT user_id) AS total")
                    .likeRight(StringUtils.isNotEmpty(startTime), "create_time", startTime)
                    .in("source", Arrays.asList(1, 2))
                    .eq(StringUtils.isNotEmpty(source),"source",source);;
            Map<String, Object> result = sysGeographyPositionRecordService.getMap(queryWrapper);
            Long total = (Long) result.get("total");
            appGeographyHistoryInfoVo.setOnlineNum(total);
            List<PcAddressHistoryTimeVo> geographyPosition = sysGeographyPositionRecordMapper.getGeographyPosition(startTime,source);
            appGeographyHistoryInfoVo.setList(geographyPosition);

            List<SysGeographyPositionRecord> sysGeographyPositionRecords =sysGeographyPositionRecordMapper.getGeographyPositionRecord(startTime,source);
            System.out.println("********************" + sysGeographyPositionRecords.size());
            appGeographyHistoryInfoVo.setPositionRecordList(sysGeographyPositionRecords);
        }else {
            SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
            startTime = startTime + " 00:00:01";
            endTime = endTime + " 23:59:59";
            try {
                Date parse = simpleDateFormat.parse(startTime);
                Date parse1 = simpleDateFormat.parse(endTime);
                startTime = simpleDateFormat.format(parse);
                endTime = simpleDateFormat.format(parse1);

            } catch (ParseException e) {
                throw new RuntimeException(e);
            }
            QueryWrapper<SysGeographyPositionRecord> queryWrapper = new QueryWrapper<>();
            queryWrapper.select("COUNT(DISTINCT user_id) AS total")
                    .between("create_time", startTime, endTime)
                    .in("source", Arrays.asList(1, 2))
                    .eq(StringUtils.isNotEmpty(source),"source",source);;
            Map<String, Object> result = sysGeographyPositionRecordService.getMap(queryWrapper);
            Long total = (Long) result.get("total");
            appGeographyHistoryInfoVo.setOnlineNum(total);

            List<PcAddressHistoryTimeVo> geographyPosition = sysGeographyPositionRecordMapper.getGeographyPositionSection(startTime,endTime,source);
            appGeographyHistoryInfoVo.setList(geographyPosition);

            List<SysGeographyPositionRecord> sysGeographyPositionRecords =sysGeographyPositionRecordMapper.getGeographyPositionHistory(startTime,endTime,source);
            for (SysGeographyPositionRecord sysGeographyPositionRecord : sysGeographyPositionRecords) {

                SysUser sysUser = sysUserService.getById(sysGeographyPositionRecord.getUserId());
                if (sysUser == null) continue;
                SysDept sysDept = sysDeptMapper.selectOne(new LambdaQueryWrapper<SysDept>()
                        .eq(SysDept::getId, sysUser.getDeptId()).eq(SysDept::getDelFlag, 0));
                if (sysDept == null) continue;
                sysGeographyPositionRecord.setDeptName(sysDept.getName());

                SysGeographyPosition sysGeographyPosition = sysGeographyPositionMapper.selectOne(new LambdaQueryWrapper<SysGeographyPosition>()
                        .eq(SysGeographyPosition::getId,sysGeographyPositionRecord.getUserId()));
                sysGeographyPositionRecord.setLeaderName(sysGeographyPosition.getLeaderName());
            }
            System.out.println("********************" + sysGeographyPositionRecords.size());
            appGeographyHistoryInfoVo.setPositionRecordList(sysGeographyPositionRecords);

        }

        return appGeographyHistoryInfoVo;
    }

    @Override
    public List<SysGeographyPosition> getOnlinePeople() {

        return sysGeographyPositionMapper.selectList(new LambdaQueryWrapper<SysGeographyPosition>().eq(SysGeographyPosition::getStatus,1));
    }


}