月球大数据地理空间分析展示平台-【后端】-月球后台服务
1
13693261870
2024-11-17 796b44ea813a1133beae4f3a67f1c0263510c0c7
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
package com.moon.server.service.all;
 
import com.moon.server.entity.ctrl.FileInfoEntity;
import com.moon.server.helper.PathHelper;
import com.moon.server.helper.StringHelper;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.apache.commons.fileupload.servlet.ServletRequestContext;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
 
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
 
@Service
@SuppressWarnings("ALL")
public class CommonsFileuploadService {
    @Autowired
    PathHelper pathHelper;
 
    private final static Log log = LogFactory.getLog(CommonsFileuploadService.class);
 
    public Object fileUpload(HttpServletRequest req, HttpServletResponse res) throws Exception {
        List<FileItem> items = getFileItem(req, res);
 
        return copeFileItems(items, req);
    }
 
    private List<FileItem> getFileItem(HttpServletRequest req, HttpServletResponse res) throws Exception {
        req.setCharacterEncoding("utf-8");
        res.setContentType("application/json;charset=utf-8");
 
        if (!ServletFileUpload.isMultipartContent(req)) {
            throw new RuntimeException("表单的enctype属性不是multipart/form-data类型");
        }
 
        DiskFileItemFactory factory = new DiskFileItemFactory();
        ServletFileUpload sfu = new ServletFileUpload(factory);
        ServletRequestContext ctx = new ServletRequestContext(req);
 
        sfu.setHeaderEncoding("utf-8");
        factory.setSizeThreshold(4096);
 
        List<FileItem> list = sfu.parseRequest(ctx);
 
        return list;
    }
 
    private Object copeFileItems(List<FileItem> items, HttpServletRequest req) throws Exception {
        Map<String, String> map = new HashMap<String, String>(3);
        List<FileInfoEntity> list = new ArrayList<FileInfoEntity>();
 
        String path = pathHelper.getTempPath();
        for (FileItem item : items) {
            if (item.isFormField()) {
                String key = item.getFieldName();
                String value = item.getString("utf-8");
                map.put(key, value);
                continue;
            }
 
            FileInfoEntity fi = copeFile(item, path, req);
            if (fi != null) {
                list.add(fi);
            }
        }
 
        return map;
    }
 
    private FileInfoEntity copeFile(FileItem item, String path, HttpServletRequest req) {
        try {
            FileInfoEntity fi = new FileInfoEntity(item.getName());
            if (StringHelper.isEmpty(fi.getFileName())) {
                return null;
            }
 
            fi.setSize(item.getSize());
            fi.setPath(path + File.separator + fi.getFileName());
            item.write(new File(fi.getPath()));
 
            return fi;
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            return null;
        }
    }
}