package com.whyc.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.whyc.dto.Response;
|
import com.whyc.mapper.AlarmUserMapper;
|
import com.whyc.pojo.AlarmUser;
|
import org.springframework.stereotype.Service;
|
|
import javax.annotation.Resource;
|
import java.util.Date;
|
import java.util.List;
|
|
@Service
|
public class AlarmUserSendService {
|
@Resource
|
private AlarmUserMapper mapper;
|
|
public Response addList(List<AlarmUser> list){
|
Date now = new Date();
|
for (AlarmUser alarmUser:list) {
|
alarmUser.setRecordTime(now);
|
}
|
int flag = mapper.insertBatchSomeColumn(list);
|
return new Response().set(1,flag>0?true:false);
|
}
|
|
public Response addOrUpdateList(List<AlarmUser> list){
|
Date now = new Date();
|
QueryWrapper<AlarmUser> queryWrapper = new QueryWrapper<>();
|
for (AlarmUser alarmUser:list) {
|
queryWrapper.eq("alarm_id",alarmUser.getAlarmId())
|
.eq("user_id",alarmUser.getUserId())
|
.eq("alarm_source",alarmUser.getAlarmSource());
|
AlarmUser sqlAlm = mapper.selectOne(queryWrapper);
|
alarmUser.setRecordTime(now);
|
if (sqlAlm==null){
|
mapper.insert(alarmUser);
|
}else {
|
sqlAlm.setRecordTime(now);
|
mapper.updateById(sqlAlm);
|
}
|
}
|
return new Response().set(1,true);
|
}
|
|
|
|
public int add(AlarmUser entity){
|
entity.setRecordTime(new Date());
|
return mapper.insert(entity);
|
}
|
|
public int update(AlarmUser entity){
|
entity.setRecordTime(new Date());
|
return mapper.deleteById(entity.getNum());
|
}
|
|
|
|
}
|