自动化立体仓库 - WMS系统
zwl
2025-10-13 03b5b8a480fb201e288486299eb385a00d2aac66
src/main/java/com/zy/third/task/handler/BaseDataHandler.java
New file
@@ -0,0 +1,82 @@
package com.zy.third.task.handler;
import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.zy.asrs.entity.Mat;
import com.zy.asrs.service.MatService;
import com.zy.asrs.service.TagService;
import com.zy.asrs.task.AbstractHandler;
import com.zy.asrs.task.core.ReturnT;
import com.zy.system.entity.User;
import com.zy.system.service.UserService;
import com.zy.third.entity.ExdMaterial;
import com.zy.third.entity.ExdUser;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;
import java.util.Date;
@Component
@Slf4j
public class BaseDataHandler extends AbstractHandler<String> {
    @Autowired
    private MatService matService;
    @Autowired
    private UserService userService;
    @Autowired
    private TagService tagService;
    @Transactional
    public ReturnT<String> readUser(ExdUser exdUser) {
        User user = userService.selectOne(new EntityWrapper<User>().eq("username", exdUser.getName()));
        if (user == null) {
            user = new User();
            user.setUsername(exdUser.getName());
            user.setNickname(exdUser.getName());
            user.setMobile(exdUser.getName());
            user.setEmail(exdUser.getId());
            user.setRoleId(2L);
            user.setCreateTime(new Date());
            user.setStatus(1);
            userService.insert(user);
        }else {
            user.setUsername(exdUser.getName());
            user.setNickname(exdUser.getName());
            user.setMobile(exdUser.getName());
            user.setEmail(exdUser.getId());
            userService.updateById(user);
        }
        return SUCCESS;
    }
    @Transactional
    public ReturnT<String> readMat(ExdMaterial exdMaterial) {
        Mat mat = matService.selectOne(new EntityWrapper<Mat>().eq("matnr", exdMaterial.getFnumber()));
        if (mat == null) {
            mat = new Mat();
            mat.setMatnr(exdMaterial.getFnumber());
            mat.setMaktx(exdMaterial.getFname());
            mat.setSpecs(exdMaterial.getFmodel());
            mat.setSku(exdMaterial.getId());
            mat.setTagId(tagService.getTop().getId());
            mat.setStatus(1);
            mat.setCreateTime(new Date());
            matService.insert(mat);
        }else  {
            mat.setMatnr(exdMaterial.getFnumber());
            mat.setMaktx(exdMaterial.getFname());
            mat.setSpecs(exdMaterial.getFmodel());
            mat.setSku(exdMaterial.getId());
            mat.setTagId(tagService.getTop().getId());
            mat.setStatus(1);
            matService.updateById(mat);
        }
        return SUCCESS;
    }
}