zqy
5 天以前 b02beccf4567068cb47a3f1181a00039456c872d
zhang-content/src/main/java/com/ruoyi/service/impl/ZCertificateServiceImpl.java
@@ -1,11 +1,22 @@
package com.ruoyi.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.domain.ZCertificate;
import com.ruoyi.domain.ZExperience;
import com.ruoyi.mapper.ZCertificateMapper;
import com.ruoyi.service.ZCertificateService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
/**
 * <p>
@@ -15,7 +26,45 @@
 * @author ojq
 * @since 2023-03-14
 */
@Slf4j
@Service
public class ZCertificateServiceImpl extends ServiceImpl<ZCertificateMapper, ZCertificate> implements ZCertificateService {
    @Override
    public List<ZCertificate> selectExperienceList(Long userId) {
        LambdaQueryWrapper<ZCertificate> lqw = new LambdaQueryWrapper<>();
        lqw.eq(ZCertificate::getUserId,userId);
        return list(lqw);
    }
    @Override
    public boolean importExcel(MultipartFile file, Long userId) {
        ExcelUtil<ZCertificate> util = new ExcelUtil<>(ZCertificate.class);
        List<ZCertificate> certificateList = null;
        try {
            certificateList = util.importExcel(file.getInputStream());
        } catch (Exception e) {
            e.printStackTrace();
        }
        assert certificateList != null;
        List<ZCertificate> list = certificateList.stream().peek(
                (certificate) -> certificate.setUserId(userId)
        ).collect(Collectors.toList());
        log.info("资产列表为:{}",list);
        return saveBatch(list);
    }
    @Override
    public AjaxResult listType() {
        List<ZCertificate> typeList = list(new QueryWrapper<ZCertificate>().select("distinct type"));
        ArrayList<String> types = new ArrayList<>();
        for (ZCertificate zCertificate : typeList) {
           types.add(zCertificate.getType());
        }
        return AjaxResult.success(types);
    }
}