package com.whyc.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.github.pagehelper.PageHelper;
|
import com.github.pagehelper.PageInfo;
|
import com.whyc.dto.Response;
|
import com.whyc.mapper.*;
|
import com.whyc.pojo.*;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.TransactionDefinition;
|
import org.springframework.transaction.TransactionStatus;
|
|
import javax.annotation.Resource;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
|
@Service
|
public class WindingExperimentService {
|
@Autowired
|
DataSourceTransactionManager dataSourceTransactionManager;
|
@Autowired
|
TransactionDefinition transactionDefinition;
|
|
@Resource
|
private ExperimentMapper mapper;
|
|
@Resource
|
private ExperimentWindingStep1Mapper mapper1;
|
|
@Resource
|
private ExperimentWindingStep2Mapper mapper2;
|
|
@Resource
|
private ProjectMapper projectMapper;
|
|
|
@Resource
|
private UserMapper userMapper;
|
|
/**
|
* 绕组实验 开始实验
|
* @param experiment
|
* @return
|
*/
|
public Response<Object> addWindingExperiment(Experiment experiment) {
|
experiment.setCreateTime(new Date());
|
experiment.setStatus(1);
|
try {
|
if (mapper.insert(experiment) > 0) {
|
return new Response().setMsg(1, "添加成功");
|
} else {
|
return new Response().setMsg(0, "添加失败");
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
return new Response().setMsg(0, "添加失败");
|
}
|
}
|
|
/**
|
* 绕组实验步骤 step参数 1 2 对应实验步骤1、2
|
* @param list
|
* @return
|
*/
|
public Response<Object> addExperimentWindingStep1(List<ExperimentWindingStep1> list) {
|
Boolean flag = true;
|
if (list != null) {
|
for (int i = 0; i < list.size(); i++) {
|
ExperimentWindingStep1 experimentWindingStep1 = list.get(i);
|
experimentWindingStep1.setSaveTime(new Date());
|
list.set(i, experimentWindingStep1);
|
}
|
}
|
TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(transactionDefinition);
|
try {
|
for (int i = 0; i < list.size(); i++) {
|
ExperimentWindingStep1 windingStep1 = list.get(i);
|
if (mapper1.insert(windingStep1) > 0) {
|
flag = true;
|
} else {
|
flag = false;
|
}
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
dataSourceTransactionManager.rollback(transactionStatus); //事务回滚
|
return new Response().setMsg(0, "添加失败");
|
}
|
if (flag) {
|
dataSourceTransactionManager.commit(transactionStatus);
|
return new Response().setMsg(1, "添加成功");
|
} else {
|
dataSourceTransactionManager.rollback(transactionStatus);
|
return new Response().setMsg(0, "添加失败");
|
}
|
|
}
|
|
/**
|
* 绕组实验步骤2
|
* @param list
|
* @return
|
*/
|
public Response<Object> addExperimentWindingStep2(List<ExperimentWindingStep2> list) {
|
Boolean flag = true;
|
if (list != null) {
|
for (int i = 0; i < list.size(); i++) {
|
ExperimentWindingStep2 experimentWindingStep2 = list.get(i);
|
experimentWindingStep2.setSaveTime(new Date());
|
list.set(i, experimentWindingStep2);
|
}
|
}
|
TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(transactionDefinition);
|
try {
|
for (int i = 0; i < list.size(); i++) {
|
ExperimentWindingStep2 windingStep2 = list.get(i);
|
if (mapper2.insert(windingStep2) > 0) {
|
flag = true;
|
} else {
|
flag = false;
|
}
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
dataSourceTransactionManager.rollback(transactionStatus); //事务回滚
|
return new Response().setMsg(0, "添加失败");
|
}
|
if (flag) {
|
dataSourceTransactionManager.commit(transactionStatus);
|
return new Response().setMsg(1, "添加成功");
|
} else {
|
dataSourceTransactionManager.rollback(transactionStatus);
|
return new Response().setMsg(0, "添加失败");
|
}
|
|
}
|
|
public Response<List<Project>> getProjectName() {
|
QueryWrapper<Project> wrapper = new QueryWrapper<>();
|
ArrayList<Object> arr = new ArrayList<>();
|
arr.add(1); //项目状态:0-已取消,1-进行中,2-完成,3-延期
|
arr.add(3);
|
wrapper.in("status", arr);
|
List<Project> list = projectMapper.selectList(wrapper);
|
Response<List<Project>> response = new Response<>();
|
response.set(1, list);
|
return response;
|
}
|
|
|
public Response delExperimentWindingbyId(String experimentId) {
|
|
try {
|
mapper1.delExperimentWindingbyId(experimentId);
|
mapper2.delExperimentWindingbyId(experimentId);
|
} catch (Exception e) {
|
e.printStackTrace();
|
return new Response().setMsg(0, "删除失败");
|
}
|
|
return new Response().setMsg(1, "删除成功");
|
|
}
|
|
/**
|
* 根据 experimentId 编辑参数
|
* step参数 1 2 对应实验步骤1、2
|
* @param windingStep1s
|
* @return
|
*/
|
public Response updateExperimentWindingStep1(List<ExperimentWindingStep1> windingStep1s) {
|
boolean bl = true;
|
TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(transactionDefinition);
|
if (windingStep1s != null && windingStep1s.size() > 0) {
|
for (int i = 0; i < windingStep1s.size(); i++) {
|
try {
|
ExperimentWindingStep1 windingStep1 = windingStep1s.get(i);
|
if (mapper1.updateById(windingStep1) > 0) {
|
bl = true;
|
} else {
|
bl = false;
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
dataSourceTransactionManager.rollback(transactionStatus); //事务回滚
|
return new Response().setMsg(0, "修改失败");
|
}
|
}
|
}
|
if (bl) {
|
dataSourceTransactionManager.commit(transactionStatus);
|
return new Response().setMsg(1, "修改成功");
|
} else {
|
dataSourceTransactionManager.rollback(transactionStatus); //事务回滚
|
return new Response().setMsg(0, "修改失败");
|
}
|
}
|
|
/**
|
* 根据 experiment_id 编辑试验数据
|
* @param windingStep2s
|
* @return
|
*/
|
public Response updateExperimentWindingStep2(List<ExperimentWindingStep2> windingStep2s) {
|
boolean bl = true;
|
TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(transactionDefinition);
|
if (windingStep2s != null && windingStep2s.size() > 0) {
|
for (int i = 0; i < windingStep2s.size(); i++) {
|
try {
|
ExperimentWindingStep2 windingStep2 = windingStep2s.get(i);
|
if (mapper2.updateById(windingStep2) > 0) {
|
bl = true;
|
} else {
|
bl = false;
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
dataSourceTransactionManager.rollback(transactionStatus); //事务回滚
|
return new Response().setMsg(0, "修改失败");
|
}
|
}
|
}
|
if (bl) {
|
dataSourceTransactionManager.commit(transactionStatus);
|
return new Response().setMsg(1, "修改成功");
|
} else {
|
dataSourceTransactionManager.rollback(transactionStatus); //事务回滚
|
return new Response().setMsg(0, "修改失败");
|
}
|
}
|
|
/**
|
* 根据唯一ID编辑实验数据1
|
* @param windingStep1s
|
* @return
|
*/
|
public Response updateExperimentWindingStep1ById(List<ExperimentWindingStep1> windingStep1s) {
|
boolean bl = true;
|
TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(transactionDefinition);
|
if (windingStep1s != null && windingStep1s.size() > 0) {
|
for (int i = 0; i < windingStep1s.size(); i++) {
|
try {
|
ExperimentWindingStep1 windingStep1 = windingStep1s.get(i);
|
if (mapper1.updateByPrimaryKey(windingStep1) > 0) {
|
bl = true;
|
} else {
|
bl = false;
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
dataSourceTransactionManager.rollback(transactionStatus); //事务回滚
|
return new Response().setMsg(0, "修改失败");
|
}
|
}
|
}
|
if (bl) {
|
dataSourceTransactionManager.commit(transactionStatus);
|
return new Response().setMsg(1, "修改成功");
|
} else {
|
dataSourceTransactionManager.rollback(transactionStatus); //事务回滚
|
return new Response().setMsg(0, "修改失败");
|
}
|
}
|
|
public Response updateExperimentWindingStep2ById(List<ExperimentWindingStep2> windingStep2s) {
|
boolean bl = true;
|
TransactionStatus transactionStatus = dataSourceTransactionManager.getTransaction(transactionDefinition);
|
if (windingStep2s != null && windingStep2s.size() > 0) {
|
for (int i = 0; i < windingStep2s.size(); i++) {
|
try {
|
ExperimentWindingStep2 windingStep2 = windingStep2s.get(i);
|
if (mapper2.updateByPrimaryKey(windingStep2) > 0) {
|
bl = true;
|
} else {
|
bl = false;
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
dataSourceTransactionManager.rollback(transactionStatus); //事务回滚
|
return new Response().setMsg(0, "修改失败");
|
}
|
}
|
}
|
if (bl) {
|
dataSourceTransactionManager.commit(transactionStatus);
|
return new Response().setMsg(1, "修改成功");
|
} else {
|
dataSourceTransactionManager.rollback(transactionStatus); //事务回滚
|
return new Response().setMsg(0, "修改失败");
|
}
|
}
|
|
/**
|
* 实验数据 step参数 1 2 对应实验步骤1、2
|
*
|
* @param step
|
* @param pageNum
|
* @param pageSize
|
* @return
|
*/
|
public Response getExperimentWindingStep1(int step, int pageNum, int pageSize) {
|
|
PageHelper.startPage(pageNum, pageSize);
|
QueryWrapper<ExperimentWindingStep1> wrapper = new QueryWrapper<>();
|
wrapper.eq("step",step);
|
List<ExperimentWindingStep1> list = mapper1.selectList(wrapper);
|
PageInfo<ExperimentWindingStep1> pageInfo = new PageInfo<>(list);
|
return new Response<PageInfo<ExperimentWindingStep1>>().set(1, pageInfo);
|
}
|
|
|
/**
|
* 实验数据2
|
* @param pageNum
|
* @param pageSize
|
* @return
|
*/
|
public Response getExperimentWindingStep2(int pageNum, int pageSize) {
|
|
PageHelper.startPage(pageNum, pageSize);
|
List<ExperimentWindingStep2> list = mapper2.selectList(null);
|
|
PageInfo<ExperimentWindingStep2> pageInfo = new PageInfo<>(list);
|
return new Response<PageInfo<ExperimentWindingStep2>>().set(1, pageInfo);
|
}
|
|
|
/**
|
* 获取测试名
|
* @param name
|
* @return
|
*/
|
public Response<Object> getRealName(String name) {
|
Response<Object> response = new Response<>();
|
QueryWrapper<User> wrapper = new QueryWrapper<>();
|
wrapper.eq("name", name);
|
User user = userMapper.selectOne(wrapper);
|
if (null != user) {
|
response.set(1, user.getRealName());
|
}else{
|
response.setMsg(0,"查询失败");
|
}
|
return response;
|
}
|
|
|
/**
|
* 状态:-1-取消,0-未开始,1-进行中,2-完成
|
* * 不对controller接口提供取消功能
|
* @param id
|
* @return
|
*/
|
public Response updateFlag(String id) {
|
Response<Object> response = new Response<>();
|
QueryWrapper<Experiment> queryWrapper = new QueryWrapper<>();
|
queryWrapper.eq("id",id);
|
Experiment experiment1 = mapper.selectOne(queryWrapper);
|
if (experiment1 ==null){
|
response.setMsg(0,"试验编号有误");
|
}
|
try {
|
Experiment experiment = new Experiment<>();
|
experiment.setId(id);
|
experiment.setEndTime(new Date());
|
mapper.updateFlag(experiment);
|
} catch (Exception e) {
|
e.printStackTrace();
|
response.setMsg(0,"更新失败");
|
}
|
return response.setMsg(1,"更新成功");
|
}
|
}
|
