package org.jeecg.modules.doc.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import org.jeecg.modules.doc.api.JeecgSystemClient;
import org.jeecg.modules.doc.entity.UserStorageFlow;
import org.jeecg.modules.doc.mapper.UserStorageFlowMapper;
import org.jeecg.modules.doc.service.IUserStorageFlowService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;

import java.util.Map;

/**
 * @Description: user_storage_flow
 * @Author: jeecg-boot
 * @Date:   2022-08-04
 * @Version: V1.0
 */
@Service
public class UserStorageFlowServiceImpl extends ServiceImpl<UserStorageFlowMapper, UserStorageFlow> implements IUserStorageFlowService {



    @Autowired
    private JeecgSystemClient jeecgSystemClient;
    /**
     * 文件下载时更新限额使用量
     */
    public void updateSurplusFlowByUserId(String token, long size) {
        Map<String, String> userInfo = jeecgSystemClient.getUserSectionInfoByToken(token).getResult();
        UserStorageFlow userStorageFlow = baseMapper.selectOne(new LambdaQueryWrapper<UserStorageFlow>().eq(UserStorageFlow::getUserId, userInfo.get("sysUserId")));
        if (userStorageFlow != null) {
            userStorageFlow.setSurplusFlow((userStorageFlow.getSurplusFlow()==null?0L:userStorageFlow.getSurplusFlow())+size);
            baseMapper.updateById(userStorageFlow);
        } else {
            userStorageFlow = new UserStorageFlow(userInfo.get("sysUserId"),size);
            baseMapper.insert(userStorageFlow);
        }

    }

    @Override
    public void updateSurplusFlowAll() {
        LambdaUpdateWrapper<UserStorageFlow> updateWrapper = new LambdaUpdateWrapper<>();
        updateWrapper.set(UserStorageFlow::getSurplusFlow, 0);
        baseMapper.update(null, updateWrapper);
    }
}