From 98205369380710cb8f69ca4e35dd7b3d7f1f4b63 Mon Sep 17 00:00:00 2001
From: haungxing <1203316822@qq.com>
Date: Mon, 10 Apr 2023 09:47:22 +0800
Subject: [PATCH] =?UTF-8?q?#=20=E8=AE=BE=E5=A4=87=E7=BC=96=E7=A0=81?=
 =?UTF-8?q?=E7=94=9F=E6=88=90=E6=96=B9=E5=BC=8F=E6=94=B9=E4=B8=BA=E9=9B=AA?=
 =?UTF-8?q?=E8=8A=B1=E7=BC=96=E7=A0=81=20#=20=E6=B6=88=E6=81=AF=E8=AF=BB?=
 =?UTF-8?q?=E5=8F=96=E6=8E=A5=E5=8F=A3=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../equipment/service/impl/EmInfoServiceImpl.java  |  3 ++-
 .../controller/MessagePushRecordController.java    | 22 +++++++++++++++++++++-
 2 files changed, 23 insertions(+), 2 deletions(-)

diff --git a/hzims-service/equipment/src/main/java/com/hnac/hzims/equipment/service/impl/EmInfoServiceImpl.java b/hzims-service/equipment/src/main/java/com/hnac/hzims/equipment/service/impl/EmInfoServiceImpl.java
index 5430d51..7551fd2 100644
--- a/hzims-service/equipment/src/main/java/com/hnac/hzims/equipment/service/impl/EmInfoServiceImpl.java
+++ b/hzims-service/equipment/src/main/java/com/hnac/hzims/equipment/service/impl/EmInfoServiceImpl.java
@@ -4,6 +4,7 @@ import com.alibaba.fastjson.JSONObject;
 import com.alibaba.fastjson.TypeReference;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.toolkit.IdWorker;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.google.common.collect.Lists;
 import com.hnac.hzims.equipment.EquipmentContants;
@@ -107,7 +108,7 @@ public class EmInfoServiceImpl extends BaseServiceImpl<EmInfoMapper, EmInfoEntit
 		// 验证设备信息
 		this.validEmInfo(em_info);
 		EmInfoEntity infoEntity = em_info.getInfo();
-		//infoEntity.setNumber(Func.);
+		infoEntity.setNumber(Long.toString(IdWorker.getId()));
 		if (this.save(infoEntity)) {
 			this.getStationCodeStoreRedis();
 			Long infoId = em_info.getInfo().getId();
diff --git a/hzims-service/message/src/main/java/com/hnac/hzims/message/controller/MessagePushRecordController.java b/hzims-service/message/src/main/java/com/hnac/hzims/message/controller/MessagePushRecordController.java
index a1f3fa8..6733667 100644
--- a/hzims-service/message/src/main/java/com/hnac/hzims/message/controller/MessagePushRecordController.java
+++ b/hzims-service/message/src/main/java/com/hnac/hzims/message/controller/MessagePushRecordController.java
@@ -1,7 +1,10 @@
 package com.hnac.hzims.message.controller;
 
+import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
+import com.hnac.hzims.message.MessageConstants;
 import com.hnac.hzims.message.entity.MessagePushRecordEntity;
 import com.hnac.hzims.message.service.IMessagePushRecordService;
 import io.swagger.annotations.Api;
@@ -10,9 +13,11 @@ import io.swagger.annotations.ApiParam;
 import lombok.AllArgsConstructor;
 import org.springblade.core.boot.ctrl.BladeController;
 import org.springblade.core.mp.support.Query;
+import org.springblade.core.secure.utils.AuthUtil;
 import org.springblade.core.tool.api.R;
 import org.springframework.web.bind.annotation.*;
 
+import java.time.LocalDateTime;
 import java.util.List;
 
 @RestController
@@ -53,8 +58,23 @@ public class MessagePushRecordController extends BladeController {
 
 	@GetMapping("/send/{id}")
 	@ApiOperation("推送消息记录")
-	@ApiOperationSupport(order = 4)
+	@ApiOperationSupport(order = 5)
 	public R<Boolean> send(@PathVariable @ApiParam("消息记录ID") Long id) {
 		return R.data(messagePushRecordService.send(id));
 	}
+
+	@GetMapping("/readMessage/{id}")
+	@ApiOperation("读取消息")
+	@ApiOperationSupport(order = 6)
+	public R<Boolean> readMessage(@PathVariable @ApiParam("消息记录ID") Long id) {
+		MessagePushRecordEntity record = messagePushRecordService.getById(id);
+		if(!record.getPusher().equals(AuthUtil.getUserId())) {
+			return R.fail("当前登录人不是消息推送人，修改状态失败");
+		}
+		LambdaUpdateWrapper<MessagePushRecordEntity> eq = Wrappers.<MessagePushRecordEntity>lambdaUpdate()
+				.set(MessagePushRecordEntity::getStatus, MessageConstants.CONFIRM)
+				.set(MessagePushRecordEntity::getRespondTime, LocalDateTime.now())
+				.eq(MessagePushRecordEntity::getId, id);
+		return R.status(messagePushRecordService.update(eq));
+	}
 }