gdc-finance-api/src/main/java/com/example/service/house/HouseWzLandManageDetailsService.java

package com.example.service.house;

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.example.common.PageVo;
import com.example.common.Rsp;
import com.example.dao.WzOtnAreaDao;
import com.example.entity.house.HouseSitePo;
import com.example.pojo.bo.ListHouseSiteBo;
import com.example.pojo.dto.ListSiteNameDto;
import com.example.pojo.dto.ListSiteNumDto;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.List;

@Slf4j
@Service
public class HouseWzLandManageDetailsService {
    private final WzOtnAreaDao wzOtnAreaDao;

    public HouseWzLandManageDetailsService(WzOtnAreaDao wzOtnAreaDao) {
        this.wzOtnAreaDao = wzOtnAreaDao;
    }

    public Rsp<PageVo<String>> listSiteName(ListSiteNameDto dto) {
        Integer yearMonth = dto.getMap().getStatisticalMonth();
        if (yearMonth == null) {
            yearMonth = Integer.valueOf(LocalDate.now().minusDays(40)
                    .format(DateTimeFormatter.ofPattern("yyyyMM")));
        }
        if ("全省".equals(dto.getMap().getCity())) {
            dto.getMap().setCity(null);
            dto.getMap().setCounty(null);
        }
        ListHouseSiteBo bo = new ListHouseSiteBo();
        bo.setYearMonth(yearMonth);
        bo.setAreaNo(dto.getMap().getCity());
        bo.setCityNo(dto.getMap().getCounty());
        bo.setSiteName(dto.getMap().getAlias());
        Page<HouseSitePo> page = new Page<>(dto.getPage().getPageNum(), dto.getPage().getPageSize());
        List<HouseSitePo> list = wzOtnAreaDao.listHouseSite(page, bo);
        List<String> vos = list.stream().map(HouseSitePo::getSiteName).toList();
        PageVo<String> pageVo = new PageVo<>(vos, page);
        return Rsp.ok(pageVo);
    }

    public Rsp<PageVo<String>> listSiteNum(ListSiteNumDto dto) {
        Integer yearMonth = dto.getMap().getStatisticalMonth();
        if (yearMonth == null) {
            yearMonth = Integer.valueOf(LocalDate.now().minusDays(40)
                    .format(DateTimeFormatter.ofPattern("yyyyMM")));
        }
        if ("全省".equals(dto.getMap().getCity())) {
            dto.getMap().setCity(null);
            dto.getMap().setCounty(null);
        }
        ListHouseSiteBo bo = new ListHouseSiteBo();
        bo.setYearMonth(yearMonth);
        bo.setAreaNo(dto.getMap().getCity());
        bo.setCityNo(dto.getMap().getCounty());
        bo.setSiteNum(dto.getMap().getCode());
        Page<HouseSitePo> page = new Page<>(dto.getPage().getPageNum(), dto.getPage().getPageSize());
        List<HouseSitePo> list = wzOtnAreaDao.listHouseSite(page, bo);
        List<String> vos = list.stream().map(HouseSitePo::getSiteNum).toList();
        PageVo<String> pageVo = new PageVo<>(vos, page);
        return Rsp.ok(pageVo);
    }
}