验收报告查询修改

whyclxw

4 天以前 51726e56a2d4b55f0e37b2b22adc4413113d43d0

 src/main/java/com/whyc/service/AttachLockService.java

@@ -2,6 +2,7 @@

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.whyc.dto.Response;
import com.whyc.mapper.AttachLockMapper;
import com.whyc.pojo.AttachLock;
@@ -30,6 +31,7 @@
                    UpdateWrapper uwrapper=new UpdateWrapper();
                    uwrapper.set("lock_flag",attachLock.getLockFlag());
                    uwrapper.set("local_reason",attachLock.getLocalReason());
                    uwrapper.eq("attach_name",attachLock.getAttachName());
                    uwrapper.eq("material_id",attachLock.getMaterialId());
                    mapper.update(null,uwrapper);
                }else{
@@ -53,7 +55,8 @@
                UpdateWrapper uwrapper=new UpdateWrapper();
                uwrapper.set("lock_flag",attachLock.getLockFlag());
                uwrapper.set("local_reason",attachLock.getLocalReason());
                uwrapper.set("product_id",attachLock.getProductId());
                uwrapper.eq("attach_name",attachLock.getAttachName());
                uwrapper.eq("product_id",attachLock.getProductId());
                mapper.update(null,uwrapper);
            }else{
                mapper.insert(attachLock);
@@ -61,4 +64,18 @@
        });
        return new Response().set(1,true,"");
    }

    public void getAndInsert(Integer materialIdInDB, Integer materialId) {
        QueryWrapper<AttachLock> query = Wrappers.query();
        query.eq("material_id",materialIdInDB).eq("lock_flag",1).eq("product_id",0);
        List<AttachLock> attachLocksInDB = mapper.selectList(query);

        attachLocksInDB.forEach(attachLock -> {
            attachLock.setId(null);
            attachLock.setMaterialId(materialId);
        });
        if(attachLocksInDB.size()!=0) {
            mapper.insertBatchSomeColumn(attachLocksInDB);
        }
    }
}