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);
  }
}