101 lines
3.1 KiB
Java
101 lines
3.1 KiB
Java
|
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);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|