package com.gk.hotwork.Service.ServiceImpl; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.metadata.OrderItem; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.gk.hotwork.Domain.CompanyInfo; import com.gk.hotwork.Domain.Utils.PageInfo; import com.gk.hotwork.Domain.Utils.StringUtils; import com.gk.hotwork.Mapper.CompanyInfoMapper; import com.gk.hotwork.Service.CompanyService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import java.util.ArrayList; import java.util.List; @Service("CompanyService") public class CompanyServiceImpl extends ServiceImpl implements CompanyService { @Autowired CompanyInfoMapper companyInfoMapper; @Override public void selectDataGrid(PageInfo pageInfo) { Page page = new Page<>(pageInfo.getPageIndex(), pageInfo.getPageSize()); List orderItems = new ArrayList<>(); OrderItem orderItem = new OrderItem(); if (StringUtils.isNotBlank(pageInfo.getSort()) && StringUtils.isNotBlank(pageInfo.getOrder())) { orderItem.setAsc(pageInfo.getOrder().equalsIgnoreCase("ascending")); orderItem.setColumn(pageInfo.getSort()); }else { orderItem.setAsc(false); orderItem.setColumn("createddate"); } orderItems.add(orderItem); page.setOrders(orderItems); List list = companyInfoMapper.selectDataGrid(page,pageInfo.getCondition()); pageInfo.setResult(list); pageInfo.setTotalCount(page.getTotal()); } @Override public CompanyInfo selectExistByName(Long id, String company) { return companyInfoMapper.selectExistByName(id,company); } @Override public CompanyInfo selectByName(String company) { LambdaQueryWrapper wrapper = new LambdaQueryWrapper<>(); wrapper.eq(CompanyInfo::getCompany,company); return companyInfoMapper.selectOne(wrapper); } @Override public List selectByAll() { return companyInfoMapper.selectByAll(); } @Override public CompanyInfo selectExistMain(Long id) { return companyInfoMapper.selectExistMain(id); } @Override public boolean isMain(String company) { if (StringUtils.isBlank(company)){return true; } CompanyInfo companyInfo = selectByName(company); if (companyInfo == null){return false;} return companyInfo.getIsmain() == 1; } }