czt
2026-01-12 dde5ece4eef131d7b02907d4aa6c311a08ef78f5
fzzy-igdss-core/src/main/java/com/fzzy/igds/service/SnapRecordService.java
@@ -1,11 +1,13 @@
package com.fzzy.igds.service;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fzzy.igds.data.BaseResp;
import com.fzzy.igds.data.IgdsBaseParam;
import com.fzzy.igds.domain.SnapRecord;
import com.fzzy.igds.mapper.SnapRecordMapper;
import com.fzzy.igds.utils.ContextUtil;
import com.fzzy.igds.utils.DateUtil;
import com.ruoyi.common.utils.StringUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
@@ -21,34 +23,118 @@
    @Resource
    private SnapRecordMapper snapRecordMapper;
    public List<SnapRecord> listAll(IgdsBaseParam param) {
    /**
     * 分页查询数据
     *
     * @param page
     * @param param
     */
    public void listPage(Page<SnapRecord> page, IgdsBaseParam param) {
        QueryWrapper<SnapRecord> queryWrapper = getQueryWrapper(param);
        snapRecordMapper.selectPage(page, queryWrapper);
    }
        if (null == param)
            return snapRecordMapper.selectList(null);
    /**
     * 封装查询条件
     *
     * @param param
     */
    public QueryWrapper<SnapRecord> getQueryWrapper(IgdsBaseParam param) {
        QueryWrapper<SnapRecord> queryWrapper = new QueryWrapper<>();
        param.setCompanyId(ContextUtil.getCompanyId());
        queryWrapper.eq("company_id", param.getCompanyId());
        //库区检索
        if (StringUtils.isNotBlank(param.getDeptId())) {
            queryWrapper.eq("dept_id", param.getDeptId());
        }
        // 收储公司检索
        if (StringUtils.isNotBlank(param.getKey())) {
            queryWrapper.likeRight("dept_id", param.getKey());
        }
        //抓拍时间检索
        if (null != param.getStart()) {
            queryWrapper.ge("snap_time",param.getStart());
        }
        if (null != param.getEnd()) {
            queryWrapper.le("snap_time", param.getEnd());
        }
        //标签检索
        if (StringUtils.isNotBlank(param.getName())) {
            queryWrapper.like("tags", param.getName());
        }
        queryWrapper.orderByDesc("create_time");
        return queryWrapper;
    }
    /**
     * 查询所有数据
     *
     * @param param
     * @author sgj
     * @date 2025/12/10
     */
    public List<SnapRecord> getListByParam(IgdsBaseParam param) {
        if (null == param){
            param = new IgdsBaseParam();
        }
        QueryWrapper<SnapRecord> queryWrapper = new QueryWrapper<>();
        if (StringUtils.isNotEmpty(param.getName())) {
            queryWrapper.like("name", param.getName());
        if (StringUtils.isNotBlank(param.getDeptId())) {
            queryWrapper.likeRight("dept_id", param.getDeptId());
        }
        if (null != param.getStart()) {
            queryWrapper.ge("snap_time", DateUtil.getCurZero(param.getStart()));
        }
        if (null != param.getEnd()) {
            queryWrapper.le("snap_time", DateUtil.getNextZero(param.getEnd()));
        }
        queryWrapper.orderByDesc("snap_time");
        return snapRecordMapper.selectList(queryWrapper);
    }
    public BaseResp addData(SnapRecord snapRecord) {
        snapRecord.setId(ContextUtil.generateId());
        snapRecord.setCompanyId(ContextUtil.getCompanyId());
    /**
     * 持久化保存
     * @param snapRecord
     */
    public void addData(SnapRecord snapRecord) {
        if(StringUtils.isBlank(snapRecord.getId())){
            snapRecord.setId(ContextUtil.generateId());
        }
        if(StringUtils.isBlank(snapRecord.getCompanyId())){
            snapRecord.setCompanyId(ContextUtil.getCompanyId());
        }
        snapRecord.setUpdateBy(ContextUtil.getLoginUserName());
        snapRecord.setUpdateTime(new Date());
        snapRecord.setCreateBy(ContextUtil.getLoginUserName());
        snapRecord.setCreateTime(new Date());
        return snapRecordMapper.insert(snapRecord) > 0 ? BaseResp.success() : BaseResp.error("添加失败");
        snapRecordMapper.insert(snapRecord);
    }
    /**
     * 更新数据
     *
     * @param snapRecord
     * @author sgj
     * @date 2025/12/10
     */
    public BaseResp updateData(SnapRecord snapRecord) {
        snapRecord.setUpdateBy(ContextUtil.getLoginUserName());
        snapRecord.setUpdateTime(new Date());
        return snapRecordMapper.updateById(snapRecord) > 0 ? BaseResp.success() : BaseResp.error("更新失败");
    }
    /**
     * 删除数据
     *
     * @param snapRecord
     * @author sgj
     * @date 2025/12/10
     */
    public BaseResp deleteData(SnapRecord snapRecord) {
        return snapRecordMapper.deleteById(snapRecord) > 0 ? BaseResp.success() : BaseResp.error("删除失败");
    }