1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
package com.vci.web.service.impl;
 
import com.vci.starter.web.pagemodel.Tree;
import com.vci.starter.web.util.LocalFileUtil;
import com.vci.web.service.VciMdServiceI;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;
import org.springframework.util.ResourceUtils;
 
import java.io.*;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
 
/**
 * markdown的服务
 * @author weidy
 * @date 2021/3/16
 */
@Service
public class VciMdServiceImpl implements VciMdServiceI {
 
    /**
     * 日志
     */
    private Logger logger = LoggerFactory.getLogger(getClass());
 
    /**
     * 文档文件路径和标题的映射
     */
    private static Map<String,String> markDownFileMap = new ConcurrentHashMap<>();
    /**
     * 扫描所有的md文件
     */
    @Override
    public void scanAllMdFile() {
        String jarFilePath = null;
        //这个是在运行环境
        jarFilePath = LocalFileUtil.getProjectFolder(getClass()) + File.separator + "md";
        if(StringUtils.isNotBlank(jarFilePath)){
            File folder = new File(jarFilePath);
            if(folder.exists()){
                File[] files = folder.listFiles();
                for(File file : files){
                    readFromFile(file);
                }
            }
        }
        if(CollectionUtils.isEmpty(markDownFileMap)){
            URL resource = null;
            try {//这个是在开发环境
                resource = ResourceUtils.getURL("classpath:md");
            } catch (FileNotFoundException e) {
            }
            if(resource!=null){
                File folder = new File(resource.getPath() );
                if(folder.exists()){
                    File[] files = folder.listFiles();
                    for(File file : files){
                        readFromFile(file);
                    }
                }
            }
        }
//        //看看外置的文件夹
//        File folder = new File(LocalFileUtil.getProjectFolder() +File.separator + "md");
//        if(folder.exists()){
//            File[] files = folder.listFiles();
//            for(File file : files){
//                readFromFile(file);
//            }
//        }
    }
 
    /**
     * 从文件中读取
     * @param file 文件
     */
    private void readFromFile(File file){
        String fileName = file.getName();
        RandomAccessFile  raf = null;
        String title = "";
        try{
            raf = new RandomAccessFile(file,"r");
            title = raf.readLine();
        } catch (IOException e  ) {
            if(logger.isErrorEnabled()){
                logger.error("读取文件",e);
            }
        } finally {
            if(raf!=null){
                try {
                    raf.close();
                } catch (IOException e) {
                    if(logger.isErrorEnabled()){
                        logger.error("关闭文件",e);
                    }
                }
            }
        }
        try {
            title = title==null?"":new String(title.getBytes("ISO-8859-1"),"utf-8").replace("#","");
        } catch (UnsupportedEncodingException e) {
            if(logger.isErrorEnabled()){
                logger.error("转换标题",e);
            }
        }
        markDownFileMap.put(fileName,title);
    }
 
    /**
     * 获取md文件的名字
     *
     * @return 名称的树形展示
     */
    @Override
    public List<Tree> treeAllMD() {
        List<Tree> treeList = new ArrayList<>();
        markDownFileMap.forEach((fileName,title)->{
            Tree tree = new Tree();
            tree.setOid(fileName);
            if(StringUtils.isNotBlank(title)){
                tree.setText(title);
            }else{
                tree.setText(fileName);
            }
            treeList.add(tree);
        });
        return treeList;
    }
}