|
@@ -1,6 +1,7 @@
|
|
|
package cn.kdan.compdf.service.impl;
|
|
|
|
|
|
import cn.kdan.compdf.config.RedisConstantServer;
|
|
|
+import cn.kdan.compdf.constant.MyConstant;
|
|
|
import cn.kdan.compdf.constant.UserBalanceConstant;
|
|
|
import cn.kdan.compdf.dto.TenantAssetBalanceDTO;
|
|
|
import cn.kdan.compdf.entity.BackgroundUserRemain;
|
|
@@ -31,25 +32,36 @@ public class BackgroundUserRemainServiceImpl extends ServiceImpl<BackgroundUserR
|
|
|
|
|
|
@Cacheable(value = RedisConstantServer.USER_REMAIN_TENANT, key = "#tenantId", unless = "#result == null")
|
|
|
@Override
|
|
|
- public BackgroundUserRemain getUserRemain(Long tenantId) {
|
|
|
- return this.baseMapper.selectOne(new LambdaQueryWrapper<BackgroundUserRemain>()
|
|
|
+ public List<BackgroundUserRemain> getUserRemain(Long tenantId) {
|
|
|
+ return this.baseMapper.selectList(new LambdaQueryWrapper<BackgroundUserRemain>()
|
|
|
.eq(BackgroundUserRemain::getTenantId, tenantId));
|
|
|
}
|
|
|
|
|
|
+ @Override
|
|
|
+ public List<BackgroundUserRemain> getUserRemainByUserId(Long userId) {
|
|
|
+ return this.baseMapper.selectList(new LambdaQueryWrapper<BackgroundUserRemain>()
|
|
|
+ .eq(BackgroundUserRemain::getUserId, userId));
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
@Override
|
|
|
public void userRemainSync(List<TenantAssetBalanceDTO> tenantBalance) {
|
|
|
Map<Long, List<TenantAssetBalanceDTO>> collect = tenantBalance.stream()
|
|
|
.collect(Collectors.groupingBy(TenantAssetBalanceDTO::getTenantId));
|
|
|
for (Long tenantId : collect.keySet()) {
|
|
|
- BackgroundUserRemain userRemain = this.getUserRemain(tenantId);
|
|
|
- Integer totalFiles = userRemain.getTotalFiles();
|
|
|
- int fileBalance = 0;
|
|
|
- for (TenantAssetBalanceDTO tenantAssetBalanceDTO : collect.get(tenantId)) {
|
|
|
- fileBalance += tenantAssetBalanceDTO.getAsset();
|
|
|
+// List<BackgroundUserRemain> userRemains = this.getUserRemain(tenantId);
|
|
|
+ for (TenantAssetBalanceDTO assetBalanceDTO : tenantBalance) {
|
|
|
+ this.baseMapper.updateUserRemain(assetBalanceDTO.getTenantId(),assetBalanceDTO.getAssetTypeId(),assetBalanceDTO.getAsset());
|
|
|
}
|
|
|
- userRemain.setUsedFiles(totalFiles - fileBalance);
|
|
|
- this.baseMapper.update(userRemain, new LambdaQueryWrapper<BackgroundUserRemain>()
|
|
|
- .eq(BackgroundUserRemain::getTenantId, tenantId));
|
|
|
+// Integer totalFiles = userRemain.getTotalFiles();
|
|
|
+// int fileBalance = 0;
|
|
|
+// for (TenantAssetBalanceDTO tenantAssetBalanceDTO : collect.get(tenantId)) {
|
|
|
+// fileBalance += tenantAssetBalanceDTO.getAsset();
|
|
|
+// }
|
|
|
+// userRemain.setUsedFiles(totalFiles - fileBalance);
|
|
|
+// this.baseMapper.update(userRemain, new LambdaQueryWrapper<BackgroundUserRemain>()
|
|
|
+// .eq(BackgroundUserRemain::getTenantId, tenantId));
|
|
|
}
|
|
|
}
|
|
|
|
|
@@ -65,7 +77,7 @@ public class BackgroundUserRemainServiceImpl extends ServiceImpl<BackgroundUserR
|
|
|
userRemain.setTenantId(tenantId);
|
|
|
userRemain.setTotalFiles(UserBalanceConstant.BALANCE_CHANGE_FREE);
|
|
|
userRemain.setUsedFiles(0);
|
|
|
-
|
|
|
+ userRemain.setAssetType(MyConstant.FREE);
|
|
|
this.baseMapper.insert(userRemain);
|
|
|
}
|
|
|
|