|
@@ -0,0 +1,84 @@
|
|
|
+package cn.keking.service;
|
|
|
+
|
|
|
+import cn.keking.model.FileAttribute;
|
|
|
+import cn.keking.model.FileType;
|
|
|
+import cn.keking.utils.FileUtils;
|
|
|
+import org.redisson.api.RBlockingQueue;
|
|
|
+import org.redisson.api.RedissonClient;
|
|
|
+import org.slf4j.Logger;
|
|
|
+import org.slf4j.LoggerFactory;
|
|
|
+import org.springframework.beans.factory.annotation.Autowired;
|
|
|
+import org.springframework.stereotype.Service;
|
|
|
+import org.springframework.ui.ExtendedModelMap;
|
|
|
+import javax.annotation.PostConstruct;
|
|
|
+import java.util.concurrent.ExecutorService;
|
|
|
+import java.util.concurrent.Executors;
|
|
|
+
|
|
|
+/**
|
|
|
+ * Created by kl on 2018/1/19.
|
|
|
+ * Content :消费队列中的转换文件
|
|
|
+ */
|
|
|
+@Service
|
|
|
+public class FileConverQueueTask {
|
|
|
+
|
|
|
+ Logger logger= LoggerFactory.getLogger(getClass());
|
|
|
+ public static final String queueTaskName="FileConverQueueTask";
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ FilePreviewFactory previewFactory;
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ RedissonClient redissonClient;
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ FileUtils fileUtils;
|
|
|
+
|
|
|
+ @PostConstruct
|
|
|
+ public void startTask(){
|
|
|
+ ExecutorService executorService = Executors.newFixedThreadPool(3);
|
|
|
+ executorService.submit(new ConverTask(previewFactory,redissonClient,fileUtils));
|
|
|
+ logger.info("队列处理文件转换任务启动完成 ");
|
|
|
+ }
|
|
|
+
|
|
|
+ class ConverTask implements Runnable{
|
|
|
+
|
|
|
+ FilePreviewFactory previewFactory;
|
|
|
+
|
|
|
+ RedissonClient redissonClient;
|
|
|
+
|
|
|
+ FileUtils fileUtils;
|
|
|
+
|
|
|
+ public ConverTask(FilePreviewFactory previewFactory, RedissonClient redissonClient,FileUtils fileUtils) {
|
|
|
+ this.previewFactory = previewFactory;
|
|
|
+ this.redissonClient = redissonClient;
|
|
|
+ this.fileUtils=fileUtils;
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public void run() {
|
|
|
+ while (true) {
|
|
|
+ try {
|
|
|
+ final RBlockingQueue<String> queue = redissonClient.getBlockingQueue(FileConverQueueTask.queueTaskName);
|
|
|
+ String url = queue.take();
|
|
|
+ if(url!=null){
|
|
|
+ FileAttribute fileAttribute=fileUtils.getFileAttribute(url);
|
|
|
+ logger.info("正在处理转换任务,文件名称【{}】",fileAttribute.getName());
|
|
|
+ FileType fileType=fileAttribute.getType();
|
|
|
+ if(fileType.equals(FileType.compress) || fileType.equals(FileType.office)){
|
|
|
+ FilePreview filePreview=previewFactory.get(url);
|
|
|
+ filePreview.filePreviewHandle(url,new ExtendedModelMap());
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } catch (Exception e) {
|
|
|
+ try {
|
|
|
+ Thread.sleep(1000*10);
|
|
|
+ }catch (Exception ex){
|
|
|
+ ex.printStackTrace();
|
|
|
+ }
|
|
|
+ e.printStackTrace();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+}
|