优化代码

java110

2022-01-17 28e7d3937f819a3486b22bfa1c84aee172aaccb4

 service-store/src/main/java/com/java110/store/smo/impl/StoreInnerServiceSMOImpl.java

old mode 100644
new mode 100755
@@ -1,16 +1,22 @@
package com.java110.store.smo.impl;

import com.java110.dto.store.StoreAttrDto;
import com.java110.utils.util.BeanConvertUtil;
import com.java110.core.base.smo.BaseServiceSMO;
import com.java110.intf.store.IStoreInnerServiceSMO;
import com.java110.dto.PageDto;
import com.java110.dto.owner.OwnerCarDto;
import com.java110.dto.store.StoreAttrDto;
import com.java110.dto.store.StoreDto;
import com.java110.dto.store.StoreUserDto;
import com.java110.dto.user.UserDto;
import com.java110.intf.store.IStoreInnerServiceSMO;
import com.java110.intf.user.IUserInnerServiceSMO;
import com.java110.po.store.StorePo;
import com.java110.store.dao.IStoreServiceDao;
import com.java110.utils.util.BeanConvertUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;

import java.util.ArrayList;
import java.util.List;

/**
@@ -22,10 +28,13 @@
 * add by wuxw 2019/9/20
 **/
@RestController
public class StoreInnerServiceSMOImpl  extends BaseServiceSMO implements IStoreInnerServiceSMO {
public class StoreInnerServiceSMOImpl extends BaseServiceSMO implements IStoreInnerServiceSMO {

    @Autowired
    private IStoreServiceDao storeServiceDaoImpl;

    @Autowired
    private IUserInnerServiceSMO userInnerServiceSMOImpl;

    @Override
    public List<StoreDto> getStores(@RequestBody StoreDto storeDto) {
@@ -53,16 +62,82 @@
        return storeDtos;
    }

    public List<StoreAttrDto> getStoreAttrs(@RequestBody StoreAttrDto storeAttrDto){
    public List<StoreAttrDto> getStoreAttrs(@RequestBody StoreAttrDto storeAttrDto) {
        List<StoreAttrDto> storeAttrDtos = BeanConvertUtil.covertBeanList(storeServiceDaoImpl.getStoreAttrs(BeanConvertUtil.beanCovertMap(storeAttrDto)), StoreAttrDto.class);
        return storeAttrDtos;
    }


    public int getStoreCount(@RequestBody StoreDto storeDto){
    public int getStoreCount(@RequestBody StoreDto storeDto) {
        return storeServiceDaoImpl.getStoreCount(BeanConvertUtil.beanCovertMap(storeDto));
    }

    @Override
    public int updateStore(@RequestBody StorePo storePo) {
        return storeServiceDaoImpl.updateStore(BeanConvertUtil.beanCovertMap(storePo));
    }

    /**
     * 查询员工和员工所属商户信息
     *
     * @param storeUserDto
     * @return
     */
    @Override
    public List<StoreUserDto> getStoreUserInfo(@RequestBody StoreUserDto storeUserDto) {
        List<StoreUserDto> storeUserInfos = BeanConvertUtil.covertBeanList(storeServiceDaoImpl.getStoreUserInfo(BeanConvertUtil.beanCovertMap(storeUserDto)), StoreUserDto.class);
        return storeUserInfos;
    }

    public int getStoreStaffCount(@RequestBody StoreUserDto storeUserDto) {
        return storeServiceDaoImpl.getStoreStaffCount(BeanConvertUtil.beanCovertMap(storeUserDto));
    }
    /**
     * 查询商户员工信息
     *
     * @param storeUserDto
     * @return
     */
    @Override
    public List<StoreUserDto> getStoreStaffs(StoreUserDto storeUserDto) {
        int page = storeUserDto.getPage();

        if (page != PageDto.DEFAULT_PAGE) {
            storeUserDto.setPage((page - 1) * storeUserDto.getRow());
        }

        List<StoreUserDto> storeUserDtos = BeanConvertUtil.covertBeanList(
                storeServiceDaoImpl.getStoreStaffs(BeanConvertUtil.beanCovertMap(storeUserDto)), StoreUserDto.class);

        if (storeUserDtos == null || storeUserDtos.size() == 0) {
            return storeUserDtos;
        }

        String[] userIds = getUserIds(storeUserDtos);
        //根据 userId 查询用户信息
        List<UserDto> users = userInnerServiceSMOImpl.getUserInfo(userIds);

        for (StoreUserDto tmpStoreUserDto : storeUserDtos) {
            for(UserDto userDto : users){
                if(tmpStoreUserDto.getStaffId().equals(userDto.getUserId())){
                    tmpStoreUserDto.setStaffName(userDto.getName());
                    tmpStoreUserDto.setStaffTel(userDto.getTel());
                }
            }
        }
        return storeUserDtos;
    }

    private String[] getUserIds(List<StoreUserDto> storeUserDtos) {

        List<String> userIds = new ArrayList<>();
        for (StoreUserDto storeUserDto : storeUserDtos) {
            userIds.add(storeUserDto.getStaffId());
        }

        return userIds.toArray(new String[userIds.size()]);
    }


    public IStoreServiceDao getStoreServiceDaoImpl() {
        return storeServiceDaoImpl;