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

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springblade.common.node.TreeStringNode;
import org.springblade.common.utils.NodeTreeUtil;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.modules.police.entity.PoliceStationEntity;
import org.springblade.modules.police.excel.PoliceStationExcel;
import org.springblade.modules.police.vo.PoliceStationVO;
import org.springblade.modules.police.mapper.PoliceStationMapper;
import org.springblade.modules.police.service.IPoliceStationService;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.metadata.IPage;

import java.util.Date;
import java.util.List;

/**
 * 派出所信息表 服务实现类
 *
 * @author BladeX
 * @since 2024-02-01
 */
@Service
public class PoliceStationServiceImpl extends ServiceImpl<PoliceStationMapper, PoliceStationEntity> implements IPoliceStationService {

	@Override
	public IPage<PoliceStationVO> selectPoliceStationPage(IPage<PoliceStationVO> page, PoliceStationVO policeStation) {
		return page.setRecords(baseMapper.selectPoliceStationPage(page, policeStation));
	}

	/**
	 * 导入派出所数据
	 * @param data
	 * @param isCovered
	 */
	@Override
	public void importPoliceStation(List<PoliceStationExcel> data, Boolean isCovered) {
		for (PoliceStationExcel policeStationExcel : data) {
			PoliceStationEntity policeStationEntity = new PoliceStationEntity();
			policeStationEntity.setCode(policeStationExcel.getCode());
			policeStationEntity.setName(policeStationExcel.getName());
			// 数据异常，无法导入
//			policeStationEntity.setGeom(policeStationExcel.getGeom());
			// 判断是否已录入
			QueryWrapper<PoliceStationEntity> queryWrapper = new QueryWrapper<>();
			queryWrapper.eq("is_deleted",0).eq("code",policeStationExcel.getCode());
			PoliceStationEntity one = getOne(queryWrapper);
			if (null!=one){
				// 更新
				policeStationEntity.setId(one.getId());
				updateById(policeStationEntity);
			}else {
				// 新增
				policeStationEntity.setCreateUser(AuthUtil.getUserId());
				policeStationEntity.setCreateTime(new Date());
				policeStationEntity.setUpdateUser(AuthUtil.getUserId());
				policeStationEntity.setUpdateTime(new Date());
				save(policeStationEntity);
			}
		}
	}

	/**
	 * 派出所树查询
	 * @param policeStation
	 * @return
	 */
	@Override
	public List<TreeStringNode> tree(PoliceStationEntity policeStation) {
		return NodeTreeUtil.getStringNodeTree(baseMapper.tree(policeStation));
	}
}