集成redis & task

src/main/java/com/quinn/service/impl/SourceServiceImpl.java

@@ -5,6 +5,7 @@ import com.quinn.mapper.SourceMapper;
 import com.quinn.service.SourceService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.quinn.utils.OSSClientUtil;
+import com.quinn.utils.RedisUtils;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
@@ -26,10 +27,13 @@ public class SourceServiceImpl extends ServiceImpl<SourceMapper, Source> impleme
 
     @Resource
     OSSClientUtil ossClientUtil;
+    @Resource
+    RedisUtils redisUtils;
 
     @Override
     public void downloadSource(ServletOutputStream outputStream, Source source) throws IOException {
         String sourceLink = source.getSourceLink();
+        addDownLoadRecord(source);
         // 读取文件内容。
         BufferedInputStream in = new BufferedInputStream(ossClientUtil.downloadFile(sourceLink));
         BufferedOutputStream out = new BufferedOutputStream(outputStream);
@@ -46,4 +50,21 @@ public class SourceServiceImpl extends ServiceImpl<SourceMapper, Source> impleme
             in.close();
         }
     }
+
+    /**
+     * 更新页码
+     * @param source
+     */
+    private void addDownLoadRecord(Source source) {
+//        Integer downLoadTime = (Integer) redisUtils.get(QuinnConstant.SOURCE_KEY + source.getId());
+//        int downTimes = 0;
+//        if (StringUtils.isEmpty(downLoadTime)){
+//            downTimes = source.getDownRecord() + 1;
+//        }else {
+//            downTimes = downLoadTime + 1;
+//        }
+//        redisUtils.set(QuinnConstant.SOURCE_KEY + source.getId(),downTimes);
+        source.setDownRecord(source.getDownRecord() + 1);
+        updateById(source);
+    }
 }