浏览代码

Merge branch '2022DoubleEleven' of http://git.dayaedu.com/yonge/mec into master_saas

zouxuan 2 年之前
父节点
当前提交
3651e1b275

+ 10 - 0
mec-biz/src/main/java/com/ym/mec/biz/dal/entity/LuckDrawCount.java

@@ -16,8 +16,18 @@ public class LuckDrawCount extends BaseEntity {
 
 	private Date modifyOn;
 
+	private Boolean HZFlag = false;
+
 	private Integer groupId;
 
+	public Boolean getHZFlag() {
+		return HZFlag;
+	}
+
+	public void setHZFlag(Boolean HZFlag) {
+		this.HZFlag = HZFlag;
+	}
+
 	public Integer getGroupId() {
 		return groupId;
 	}

+ 11 - 1
mec-student/src/main/java/com/ym/mec/student/controller/LuckDrawController.java

@@ -1,6 +1,7 @@
 package com.ym.mec.student.controller;
 
 import com.ym.mec.auth.api.entity.SysUser;
+import com.ym.mec.biz.dal.entity.LuckDrawCount;
 import com.ym.mec.biz.dal.entity.LuckDrawLog;
 import com.ym.mec.biz.dal.page.LuckDrawQueryInfo;
 import com.ym.mec.biz.service.LuckDrawCountService;
@@ -18,6 +19,8 @@ import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
+import java.util.Objects;
+
 @Api(tags = "抽奖")
 @RestController
 @RequestMapping(value = "luckDraw")
@@ -80,7 +83,14 @@ public class LuckDrawController extends BaseController {
 		if(groupId.equals(3) && user.getOrganId().equals(4)){
 			groupId = 4;
 		}
-		return succeed(luckDrawCountService.findByIdAndGroupId(user.getId(),groupId));
+		LuckDrawCount luckDrawCount = luckDrawCountService.findByIdAndGroupId(user.getId(), groupId);
+		if(Objects.isNull(luckDrawCount)){
+			luckDrawCount = new LuckDrawCount();
+			luckDrawCount.setUserId(user.getId().longValue());
+			luckDrawCount.setAvailableCount(0);
+		}
+		luckDrawCount.setHZFlag(user.getOrganId().equals(4));
+		return succeed(luckDrawCount);
 	}
 
 }