file

2025-12-31 18:10:10 +08:00
parent d8cecbe135
commit e99b717e48
49 changed files with 2626 additions and 11 deletions
--- a/src/main/java/com/senmesh/sfile/service/impl/UserSpaceServiceImpl.java
+++ b/src/main/java/com/senmesh/sfile/service/impl/UserSpaceServiceImpl.java
@@ -0,0 +1,100 @@
+package com.senmesh.sfile.service.impl;
+
+import com.senmesh.common.exception.util.Assert;
+import com.senmesh.common.util.ListUtil;
+import com.senmesh.common.util.UserContextHolder;
+import com.senmesh.frame.redis.dto.LockDTO;
+import com.senmesh.frame.redis.redisson.RedissonUtils;
+import com.senmesh.sfile.convert.UserSpaceConvert;
+import com.senmesh.sfile.dao.UserSpaceMapper;
+import com.senmesh.sfile.dto.UserSpaceCreateDTO;
+import com.senmesh.sfile.dto.UserSpaceDTO;
+import com.senmesh.sfile.dto.UserSpaceModifyDTO;
+import com.senmesh.sfile.dto.UserSpaceQueryDTO;
+import com.senmesh.sfile.entity.UserSpace;
+import com.senmesh.sfile.service.UserSpaceService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.List;
+
+/**
+ * 用户空间ServiceImpl
+ */
+@Service
+public class UserSpaceServiceImpl implements UserSpaceService {
+
+  @Autowired
+  private UserSpaceMapper userSpaceMapper;
+
+  /**
+   * 默认总容量：5GB
+   */
+  private static final Long DEFAULT_TOTAL_SIZE = 5L * 1024 * 1024 * 1024;
+  
+
+  public static LockDTO getLockDTO(Long userId) {
+    Assert.notNull(userId, "用户id不能为空");
+    return LockDTO.builder().methodName("userspace").tryLockWaitTime(0L)
+        .arg(userId.toString())
+        .failMsg("该用户空间正在被操作")
+        .build();
+  }
+
+  @Override
+  @Transactional
+  public Long create(UserSpaceCreateDTO createDTO) {
+    Long userId = createDTO.getUserId();
+    UserSpaceQueryDTO queryDTO = UserSpaceQueryDTO.builder().userId(userId).build();
+
+    return RedissonUtils.lockAndSave(getLockDTO(userId),
+        () -> this.find(queryDTO), () -> {
+          UserSpace userSpace = UserSpaceConvert.INSTANCE.toEntity(createDTO);
+          userSpace.setTotalSize(DEFAULT_TOTAL_SIZE);
+          userSpace.setUsedSize(0L);
+          userSpaceMapper.saveEntity(userSpace);
+          return userSpace.getId();
+        });
+  }
+
+  @Override
+  @Transactional
+  public void modify(UserSpaceModifyDTO modifyDTO) {
+    Long userId = modifyDTO.getUserId();
+    UserSpaceQueryDTO queryDTO = UserSpaceQueryDTO.builder().userId(userId).build();
+    UserSpaceDTO userSpaceDTO = this.find(queryDTO);
+    Assert.notNull(userSpaceDTO, "用户空间不存在");
+
+    UserSpace userSpace = UserSpace.builder()
+        .id(userSpaceDTO.getId())
+        .usedSize(modifyDTO.getUsedSize())
+        .build();
+    userSpaceMapper.updateEntitySelective(userSpace);
+  }
+
+  @Override
+  @Transactional
+  public void remove(Long id) {
+    userSpaceMapper.deleteEntity(UserSpace.builder().id(id).build(),
+        UserContextHolder.getLoginUserId());
+  }
+
+  @Override
+  public List<UserSpaceDTO> list(UserSpaceQueryDTO queryDTO) {
+    queryDTO.disablePage();
+    return userSpaceMapper.list(queryDTO);
+  }
+
+  @Override
+  public void page(UserSpaceQueryDTO queryDTO) {
+    userSpaceMapper.list(queryDTO);
+  }
+
+  @Override
+  public UserSpaceDTO find(UserSpaceQueryDTO queryDTO) {
+    List<UserSpaceDTO> userSpaceDTOs = this.list(queryDTO);
+    return ListUtil.first(userSpaceDTOs);
+  }
+}
+