zwhui 1 year ago
parent
commit
651e1853e3

+ 6 - 0
wholesale/src/main/java/com/yhlxj/dao/model/entity/WholesaleUser.java

@@ -1,5 +1,6 @@
 package com.yhlxj.dao.model.entity;
 
+import com.baomidou.mybatisplus.annotation.TableField;
 import com.yhlxj.dao.model.BaseEntity;
 import lombok.Data;
 
@@ -22,5 +23,10 @@ public class WholesaleUser extends BaseEntity {
     private String password;
     private String token;
     private Date expireTime;
+    /**
+     * 总时长
+     */
+    @TableField(exist = false)
+    private Integer totalTime;
     private String exchangeCode;
 }

+ 7 - 0
wholesale/src/main/java/com/yhlxj/web/controller/login/WholesaleUserController.java

@@ -9,8 +9,10 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.cyksj.common.exception.BusinessRuntimeException;
 import com.cyksj.common.util.StringUtil;
 import com.cyksj.enums.GatewayApiCode;
+import com.yhlxj.dao.mapper.WebWholesaleOrderDonCodeRelateMapper;
 import com.yhlxj.dao.model.dto.Result;
 import com.yhlxj.dao.model.entity.WebInfo;
+import com.yhlxj.dao.model.entity.WebWholesaleOrderDonCodeRelate;
 import com.yhlxj.dao.model.entity.WholesaleNotice;
 import com.yhlxj.dao.model.entity.WholesaleUser;
 import com.yhlxj.dao.model.enums.GatewayResponse;
@@ -49,6 +51,8 @@ public class WholesaleUserController {
 
     private final WholesaleNoticeService wholesaleNoticeService;
 
+    private final WebWholesaleOrderDonCodeRelateMapper webWholesaleOrderDonCodeRelateMapper;
+
     /**
      * 批发网站登录
      */
@@ -125,6 +129,9 @@ public class WholesaleUserController {
         long loginIdAsLong = StpUserUtil.getLoginIdAsLong();
         WholesaleUser wholesaleUser = wholesaleUserService.getById(loginIdAsLong);
         wholesaleUser.setPassword(null);
+        Integer count = webWholesaleOrderDonCodeRelateMapper.selectCount(Wrappers.lambdaQuery(WebWholesaleOrderDonCodeRelate.class)
+                .eq(WebWholesaleOrderDonCodeRelate::getDhUserId, loginIdAsLong));
+        wholesaleUser.setTotalTime(count*30);
         return wholesaleUser;
     }