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 { @Autowired private MatService matService; @Autowired private UserService userService; @Autowired private TagService tagService; @Transactional public ReturnT readUser(ExdUser exdUser) { User user = userService.selectOne(new EntityWrapper().eq("username", exdUser.getName())); if (user == null) { user = new User(); user.setUsername(exdUser.getName()); user.setNickname(exdUser.getName()); user.setMobile(exdUser.getName()); user.setRoleId(2L); user.setCreateTime(new Date()); user.setStatus(1); userService.insert(user); } return SUCCESS; } @Transactional public ReturnT readMat(ExdMaterial exdMaterial) { Mat mat = matService.selectOne(new EntityWrapper().eq("matnr", exdMaterial.getFnumber())); if (mat == null) { mat = new Mat(); mat.setMatnr(exdMaterial.getFnumber()); mat.setMaktx(exdMaterial.getFname()); mat.setSpecs(exdMaterial.getFmodel()); mat.setTagId(tagService.getTop().getId()); mat.setStatus(1); mat.setCreateTime(new Date()); matService.insert(mat); } return SUCCESS; } }