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.setManu(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.setManu(exdMaterial.getId());
|
mat.setTagId(tagService.getTop().getId());
|
mat.setStatus(1);
|
matService.updateById(mat);
|
}
|
return SUCCESS;
|
}
|
}
|