gunshi-project-ss/src/main/java/com/gunshi/project/hsz/service/MentenceTemplateService.java

package com.gunshi.project.hsz.service;


import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gunshi.project.hsz.entity.so.MentenceTemplatePageSo;
import com.gunshi.project.hsz.mapper.MentenceStMapper;
import com.gunshi.project.hsz.mapper.MentenceTemplateMapper;
import com.gunshi.project.hsz.model.MentencePlan;
import com.gunshi.project.hsz.model.MentenceSt;
import com.gunshi.project.hsz.model.MentenceTemplate;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.system.mapper.SysUserMapper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.Date;
import java.util.Objects;

@Service
@Slf4j
@Transactional(rollbackFor = Exception.class)
public class MentenceTemplateService extends ServiceImpl<MentenceTemplateMapper, MentenceTemplate> {

    @Autowired
    private SysUserMapper sysUserMapper;

    public boolean saveDate(MentenceTemplate dto) {
        dto.setId(IdWorker.getId());
        dto.setCreateTime(new Date());
        if(dto.getCreateId() != null){
            SysUser sysUser = sysUserMapper.selectUserById(Long.valueOf(dto.getCreateId()));
            if(sysUser != null){
                dto.setCreateName(sysUser.getUserName());
            }
        }
        save(dto);
        return true;
    }

    public boolean update(MentenceTemplate dto) {
        LambdaQueryWrapper<MentenceTemplate> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(MentenceTemplate::getId, dto.getId());
        MentenceTemplate entity = getOne(wrapper);
        if(Objects.isNull(entity)){
            throw new IllegalArgumentException("该模板不存在，请检查");
        }
        updateById(dto);
        return true;
    }

    public Page<MentenceTemplate> pageQuery(MentenceTemplatePageSo pageSo) {
        LambdaQueryWrapper<MentenceTemplate> wrapper = new LambdaQueryWrapper<>();
        if(!StringUtils.isBlank(pageSo.getName())){
            wrapper.like(MentenceTemplate::getName,pageSo.getName());
        }
        if(pageSo.getType() != null){
            wrapper.eq(MentenceTemplate::getType,pageSo.getType());
        }
        Page<MentenceTemplate> mentenceTemplatePage = baseMapper.selectPage(pageSo.getPageSo().toPage(), wrapper);
        return mentenceTemplatePage;
    }
}