KeyInfoServiceImpl.java 7.33 KB
package com.ruoyi.service.impl.keyinfo;

import com.ruoyi.domain.OrderEntity;
import com.ruoyi.domain.keyInfo.KeyInfo;
import com.ruoyi.service.keyinfo.KeyInfoService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;

import com.github.pagehelper.PageHelper;

import javax.servlet.http.HttpServletResponse;

import java.util.Collections;
import java.util.List;


@Service
public class KeyInfoServiceImpl extends ServiceImpl<com.ruoyi.mapper.keyinfo.KeyInfoMapper, KeyInfo> implements KeyInfoService {
    @Autowired
    private com.ruoyi.mapper.keyinfo.KeyInfoMapper KeyInfoMapper;


    /**
     * 分页查询
     */
    @Override
    public IPage<KeyInfo> pageList(Page<KeyInfo> page, KeyInfo entity, OrderEntity orderEntity) {
        LambdaQueryWrapper<KeyInfo> countWrapper = new LambdaQueryWrapper<>(entity);
        countWrapper.select(KeyInfo::getId);
        int count = count(countWrapper);

        List<KeyInfo> lists = Collections.emptyList();
        if (count > 0) {
            PageHelper.startPage((int) page.getCurrent(), (int) page.getSize(), false);
            LambdaQueryWrapper<KeyInfo> selectWrapper = new LambdaQueryWrapper<>(entity);
            orderColumn(selectWrapper, orderEntity);
            lists = list(selectWrapper);
        }

        IPage<KeyInfo> returnPage = new Page<KeyInfo>();
        returnPage.setRecords(lists);
        returnPage.setPages(count % page.getSize() == 0 ? count / page.getSize() : count / page.getSize() + 1);
        returnPage.setCurrent(page.getCurrent());
        returnPage.setSize(page.getSize());
        returnPage.setTotal(count);

        return returnPage;
    }


    @Override
    public List<KeyInfo> list(KeyInfo entity) {
        return list(new LambdaQueryWrapper<>(entity));
    }

    @Override
    public List<KeyInfo> listOfSelect(KeyInfo entity) {
        LambdaQueryWrapper<KeyInfo> wrapper = new LambdaQueryWrapper<>(entity);
        wrapper.select(KeyInfo::getId, KeyInfo::getName);
        return list(wrapper);
    }

    @Override
    public KeyInfo getOne(KeyInfo entity) {
        return getOne(new LambdaQueryWrapper<>(entity));
    }

    @Override
    public Integer countId(KeyInfo entity) {
        LambdaQueryWrapper<KeyInfo> wrapper = new LambdaQueryWrapper<>(entity);
        wrapper.select(KeyInfo::getId);
        return count(wrapper);
    }


    /**
     * 插入有值的列
     */
    @Override
    public int insertSelective(KeyInfo entity) {
        return KeyInfoMapper.insertSelective(entity);
    }

    /**
     * 插入数据
     */
    @Override
    public boolean insert(KeyInfo entity) {
        return save(entity);
    }

    /**
     * 根据主键修改数据
     */
    @Override
    public boolean updateByPrimaryKey(KeyInfo entity) {
        return updateById(entity);
    }


    public static void orderColumn(LambdaQueryWrapper<KeyInfo> wrapper, com.ruoyi.domain.OrderEntity orderEntity) {
        if (StringUtils.equals("ascending", orderEntity.getOrder())) {
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "id")) {
                wrapper.orderByAsc(KeyInfo::getId);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "name")) {
                wrapper.orderByAsc(KeyInfo::getName);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "status")) {
                wrapper.orderByAsc(KeyInfo::getStatus);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "delFlag")) {
                wrapper.orderByAsc(KeyInfo::getDelFlag);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "createBy")) {
                wrapper.orderByAsc(KeyInfo::getCreateBy);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "createTime")) {
                wrapper.orderByAsc(KeyInfo::getCreateTime);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "updateby")) {
                wrapper.orderByAsc(KeyInfo::getUpdateby);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "updateTime")) {
                wrapper.orderByAsc(KeyInfo::getUpdateTime);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "yardId")) {
                wrapper.orderByAsc(KeyInfo::getYardId);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "deviceId")) {
                wrapper.orderByAsc(KeyInfo::getDeviceId);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "cabinetno")) {
                wrapper.orderByAsc(KeyInfo::getCabinetno);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "plateNum")) {
                wrapper.orderByAsc(KeyInfo::getPlateNum);
            }
        } else if (StringUtils.equals("descending", orderEntity.getOrder())) {
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "id")) {
                wrapper.orderByDesc(KeyInfo::getId);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "name")) {
                wrapper.orderByDesc(KeyInfo::getName);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "status")) {
                wrapper.orderByDesc(KeyInfo::getStatus);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "delFlag")) {
                wrapper.orderByDesc(KeyInfo::getDelFlag);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "createBy")) {
                wrapper.orderByDesc(KeyInfo::getCreateBy);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "createTime")) {
                wrapper.orderByDesc(KeyInfo::getCreateTime);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "updateby")) {
                wrapper.orderByDesc(KeyInfo::getUpdateby);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "updateTime")) {
                wrapper.orderByDesc(KeyInfo::getUpdateTime);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "yardId")) {
                wrapper.orderByDesc(KeyInfo::getYardId);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "deviceId")) {
                wrapper.orderByDesc(KeyInfo::getDeviceId);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "cabinetno")) {
                wrapper.orderByDesc(KeyInfo::getCabinetno);
            }
            if (org.apache.commons.lang3.StringUtils.equals(orderEntity.getProp(), "plateNum")) {
                wrapper.orderByDesc(KeyInfo::getPlateNum);
            }
        }

    }
}