增加用户容量管理

720yun_fd_manage/gis_domain/src/main/java/com/gis/domain/entity/SysUserEntity.java

@@ -33,7 +33,7 @@ public class SysUserEntity extends BaseEntity implements Serializable {
     @ApiModelProperty(value = "性别 0:男  1：女")
     private Integer sex;
 
-    @ApiModelProperty(value = "邮箱")
+    @ApiModelProperty(value = "手机号")
     private String phone;
 
     @ApiModelProperty(value = "状态 0:启用  1：停用 2：注销")
@@ -47,6 +47,8 @@ public class SysUserEntity extends BaseEntity implements Serializable {
 
     @ApiModelProperty(value = "管理者，0：是， 1：否")
     private Integer manager;
+    @ApiModelProperty(value = "容量3G")
+    private Integer size=1024 * 1024 * 3;
 
 //    @JSONField(serialize = false)
 //    @ApiModelProperty(value = "登录次数")

720yun_fd_manage/gis_service/src/main/java/com/gis/service/SysUserService.java

@@ -15,5 +15,6 @@ public interface SysUserService extends IBaseService<SysUserEntity, Long> {
     SysUserEntity findByUserName(String userName);
 
     SysUserEntity findByPhone(String phone);
+    SysUserEntity findByPhoneIfNotBeSave(String phone);
 
 }

720yun_fd_manage/gis_service/src/main/java/com/gis/service/impl/FodderServiceImpl.java

@@ -26,6 +26,7 @@ import com.gis.oss.util.FileAndOssUtil;
 import com.gis.oss.util.FileUtils;
 import com.gis.service.DirService;
 import com.gis.service.FodderService;
+import com.gis.service.SysUserService;
 import com.gis.service.WorkService;
 import com.github.pagehelper.PageInfo;
 import lombok.extern.slf4j.Slf4j;
@@ -75,7 +76,8 @@ public class FodderServiceImpl extends IBaseServiceImpl<FodderEntity, Long> impl
 
     @Autowired
     WorkHotsFodderServiceImpl workHotsFodderService;
-
+    @Autowired
+    SysUserServiceImpl sysUserService;
     @Autowired
     FileAndOssUtil fileAndOssUtil;
 
@@ -618,7 +620,8 @@ public class FodderServiceImpl extends IBaseServiceImpl<FodderEntity, Long> impl
         } else {
             log.info("当前用户空间： {} kb", size);
         }
-        BaseRuntimeException.isTrue(size >= MAX_SIZE, ErrorEnum.FAILURE_CODE_3024.code(), "该用户空间已满");
+
+        BaseRuntimeException.isTrue(size >= sysUserService.findByPhoneIfNotBeSave(phone).getSize(), ErrorEnum.FAILURE_CODE_3024.code(), "该用户空间已满");
     }
 
     /**

720yun_fd_manage/gis_service/src/main/java/com/gis/service/impl/SysUserServiceImpl.java

@@ -1,5 +1,6 @@
 package com.gis.service.impl;
 
+import cn.hutool.core.util.ObjectUtil;
 import com.gis.domain.entity.SysUserEntity;
 import com.gis.mapper.IBaseMapper;
 import com.gis.mapper.SysUserMapper;
@@ -35,5 +36,14 @@ public class SysUserServiceImpl extends IBaseServiceImpl<SysUserEntity, Long> im
         return entityMapper.findByPhone(phone);
     }
 
-
+    @Override
+    public SysUserEntity findByPhoneIfNotBeSave(String phone) {
+        SysUserEntity sysUserEntity = findByPhone(phone);
+        if (ObjectUtil.isNull(sysUserEntity)){
+            sysUserEntity=new SysUserEntity();
+            sysUserEntity.setPhone(phone);
+            save(sysUserEntity);
+        }
+        return sysUserEntity;
+    }
 }