fxkh-txl-service/src/main/java/com/whdc/service/impl/OrganizationServiceImpl.java

package com.whdc.service.impl;

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.whdc.mapper.OrganizationMapper;
import com.whdc.model.entity.Organization;
import com.whdc.model.entity.UnitDict;
import com.whdc.model.vo.AdcdTree;
import com.whdc.service.IAdinfoService;
import com.whdc.service.IOrganizationService;
import com.whdc.service.IUnitDictService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * <p>
 * 服务实现类
 * </p>
 * @author xusan
 * @date 2024-05-11
 */
@Service
@Slf4j
public class OrganizationServiceImpl extends ServiceImpl<OrganizationMapper, Organization> implements IOrganizationService {

    @Autowired
    private IUnitDictService unitDictService;

    @Autowired
    private IAdinfoService adinfoService;

    @Override
    public IPage<Organization> page(Organization dto) {
        return baseMapper.page(new Page<>(), dto);
    }

    @Override
    public List<Organization> find(Organization dto) {
        return null;
    }


    @Override
    public List<Organization> findByUserId(String userId) {
        List<Organization> organizations = baseMapper.findByUserId(userId);

        List<UnitDict> list = unitDictService.findByUserId(userId);

        Map<Integer, List<UnitDict>> collect = list.stream().collect(Collectors.groupingBy(UnitDict::getOId));
        organizations.forEach(organization -> {
            List<UnitDict> unitDicts = collect.get(organization.getId());
            organization.setUnitDicts(unitDicts);
            String name = organization.getName();
            if (name.contains("市") && name.contains("县")){
                organization.setAdinfoList(adinfoService.tree(null, null));
            }
        });

        return organizations;
    }



}