4dkankan-agent/src/main/java/com/fdkankan/agent/service/impl/AgentAuthorizeModelingServiceImpl.java

package com.fdkankan.agent.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fdkankan.agent.common.PageInfo;
import com.fdkankan.agent.entity.AgentAuthorizeModeling;
import com.fdkankan.agent.entity.AgentNew;
import com.fdkankan.agent.entity.IncrementUseType;
import com.fdkankan.agent.entity.User;
import com.fdkankan.agent.mapper.IAgentAuthorizeModelingMapper;
import com.fdkankan.agent.request.AuthorizeParam;
import com.fdkankan.agent.service.IAgentAuthorizeModelingService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.fdkankan.agent.service.IAgentNewService;
import com.fdkankan.agent.service.IIncrementUseTypeService;
import com.fdkankan.agent.service.IUserService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author 
 * @since 2025-03-17
 */
@Service
public class AgentAuthorizeModelingServiceImpl extends ServiceImpl<IAgentAuthorizeModelingMapper, AgentAuthorizeModeling> implements IAgentAuthorizeModelingService {

    @Autowired
    IUserService userService;
    @Autowired
    IAgentNewService agentNewService;
    @Autowired
    IIncrementUseTypeService iIncrementUseTypeService;


    @Override
    public Object pageList(AuthorizeParam param) {
        LambdaQueryWrapper<AgentAuthorizeModeling> wrapper = new LambdaQueryWrapper<>();
        if(StringUtils.isNotBlank(param.getCustomerName())){
            wrapper.like(AgentAuthorizeModeling::getCustomerName,param.getCustomerName());
        }
        if(param.getCustomerType() != null){
            wrapper.eq(AgentAuthorizeModeling::getCustomerType,param.getCustomerType());
        }
        if(param.getUseType() != null){
            wrapper.eq(AgentAuthorizeModeling::getUseType,param.getUseType());
        }
        if(StringUtils.isNotBlank(param.getAuthorizeKey())){
            wrapper.like(AgentAuthorizeModeling::getAuthorizeKey,param.getAuthorizeKey());
        }
        wrapper.and(e -> e.eq(AgentAuthorizeModeling::getCreateAgentId,param.getAgentId())
                .or()
                .eq(AgentAuthorizeModeling::getAgentId,param.getAgentId()));

        wrapper.orderByDesc(AgentAuthorizeModeling::getId);
        Page<AgentAuthorizeModeling> page = this.page(new Page<>(param.getPageNum(), param.getPageSize()), wrapper);
        List<Integer> sysIds = page.getRecords().stream().map(AgentAuthorizeModeling::getAgentId).collect(Collectors.toList());

        HashMap<Integer, AgentNew> userMap = agentNewService.getMapByIds(sysIds);
        HashMap<Integer, IncrementUseType> typeMap = iIncrementUseTypeService.getTypeMap();

        for (AgentAuthorizeModeling record : page.getRecords()) {
            if(userMap.get(record.getAgentId())!=null){
                record.setAgentName(userMap.get(record.getAgentId()).getNickName());
            }
            if(typeMap.get(record.getUseType()) !=null){
                record.setUseTypeStr(typeMap.get(record.getUseType()).getName());
            }
        }
        return PageInfo.PageInfo(page);
    }

    @Override
    public Object addOrUpdate(AgentAuthorizeModeling param) {
        this.saveOrUpdate(param);
        return  param;
    }
}