DownloadPdfService.java 2.33 KB
Newer Older
刘基明's avatar
刘基明 committed
1 2 3 4 5 6 7 8 9
package com.tanpu.fund.service;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.tanpu.fund.entity.generator.ReportDownloadSummary;
import com.tanpu.fund.mapper.generator.ReportDownloadSummaryMapper;
import org.apache.commons.collections4.CollectionUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

刘基明's avatar
刘基明 committed
10
import javax.annotation.Resource;
刘基明's avatar
刘基明 committed
11 12 13 14 15
import java.util.List;

@Service
public class DownloadPdfService {

刘基明's avatar
刘基明 committed
16
    @Resource
刘基明's avatar
刘基明 committed
17 18 19 20 21 22
    private ReportDownloadSummaryMapper reportDownloadSummaryMapper;

    private static final Integer onceAddTimes = 200;

    @Transactional(rollbackFor = Exception.class)
    public void emptyStandardVipCount(String userId) {
刘基明's avatar
刘基明 committed
23 24
        LambdaQueryWrapper<ReportDownloadSummary> qw = new LambdaQueryWrapper<>();
        List<ReportDownloadSummary> reportDownloadSummaries = reportDownloadSummaryMapper.selectList(qw.eq(ReportDownloadSummary::getUserId, userId));
刘基明's avatar
刘基明 committed
25 26 27 28 29 30
        if (CollectionUtils.isEmpty(reportDownloadSummaries)){
            ReportDownloadSummary entity = ReportDownloadSummary.builder().userId(userId).stdVipLimit(0).build();
            reportDownloadSummaryMapper.insert(entity);
        }else {
            ReportDownloadSummary entity = reportDownloadSummaries.get(0);
            entity.setStdVipLimit(0);
刘基明's avatar
刘基明 committed
31 32 33
            entity.setDeepReportStd(0);
            entity.setHoldFundReportStd(0);
            entity.setCumulativeProfitReportStd(0);
刘基明's avatar
刘基明 committed
34 35 36 37 38 39
            reportDownloadSummaryMapper.updateById(entity);
        }
    }

    @Transactional(rollbackFor = Exception.class)
    public void addStandardVipCount(String userId) {
刘基明's avatar
刘基明 committed
40 41
        LambdaQueryWrapper<ReportDownloadSummary> qw = new LambdaQueryWrapper<>();
        List<ReportDownloadSummary> reportDownloadSummaries = reportDownloadSummaryMapper.selectList(qw.eq(ReportDownloadSummary::getUserId, userId));
刘基明's avatar
刘基明 committed
42 43 44 45 46 47 48 49 50 51 52 53 54
        if (CollectionUtils.isEmpty(reportDownloadSummaries)){
            ReportDownloadSummary entity = ReportDownloadSummary.builder().userId(userId).stdVipLimit(onceAddTimes).build();
            reportDownloadSummaryMapper.insert(entity);
        }else {
            ReportDownloadSummary entity = reportDownloadSummaries.get(0);
            entity.setStdVipLimit(entity.getStdVipLimit() + onceAddTimes);
            reportDownloadSummaryMapper.updateById(entity);
        }



    }
}