管道基础大数据平台系统开发-【后端】-Server
1
13693261870
2022-10-21 edacde3b8be28eda7455a5612dea1ba1dd87d581
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
package com.lf.server.service.data;
 
import com.lf.server.entity.data.MetaFileEntity;
import com.lf.server.entity.sys.UserEntity;
import com.lf.server.helper.FileHelper;
import com.lf.server.helper.PathHelper;
import com.lf.server.helper.StringHelper;
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 org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.support.StandardMultipartHttpServletRequest;
 
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.lang.reflect.Field;
import java.sql.Timestamp;
import java.util.*;
 
/**
 * 数据上传服务类
 * @author WWW
 */
@Service
public class UploaderService {
    @Autowired
    PathHelper pathHelper;
 
    private final static Log log = LogFactory.getLog(UploaderService.class);
 
    /**
     * 上传文件
     */
    public <T> List<MetaFileEntity> uploadData(T t, UserEntity ue, HttpServletRequest request, HttpServletResponse res) throws Exception {
        StandardMultipartHttpServletRequest req = (StandardMultipartHttpServletRequest) request;
        req.setCharacterEncoding("utf-8");
        res.setContentType("application/json;charset=utf-8");
 
        if (t != null) {
            setEntity(t, req);
        }
        List<MetaFileEntity> list = getFiles(ue, req);
 
        return list;
    }
 
    /**
     * 设置实体类
     * class<T> clazz
     * T t = clazz.newInstance();
     */
    private <T> void setEntity(T t, StandardMultipartHttpServletRequest req) {
        Enumeration<String> enumeration = req.getParameterNames();
        while (enumeration.hasMoreElements()) {
            String key = enumeration.nextElement();
 
            try {
                Field field = t.getClass().getDeclaredField(key);
                if (field != null) {
                    field.setAccessible(true);
                    String value = req.getParameter(key);
 
                    switch (field.getType().toString()) {
                        case "long":
                            field.set(t, Long.valueOf(value));
                            break;
                        case "int":
                            field.set(t, Integer.valueOf(value));
                            break;
                        case "class java.sql.Timestamp":
                            field.set(t, Timestamp.valueOf(value));
                            break;
                        //case "class java.lang.String":
                        default:
                            field.set(t, value);
                            break;
                    }
                }
            } catch (Exception ex) {
                log.error(ex.getStackTrace());
            }
        }
    }
 
    /**
     * 获取文件
     */
    private List<MetaFileEntity> getFiles(UserEntity ue, StandardMultipartHttpServletRequest req) throws Exception {
        List<MetaFileEntity> list = new ArrayList<MetaFileEntity>();
 
        String path = pathHelper.getTempPath(ue.getId());
        Iterator<String> iterator = req.getFileNames();
        while (iterator.hasNext()) {
            MultipartFile file = req.getFile(iterator.next());
            if (StringHelper.isEmpty(file.getOriginalFilename())) {
                continue;
            }
 
            double sizes = FileHelper.sizeToMb(file.getSize());
            MetaFileEntity mf = new MetaFileEntity();
            mf.setName(file.getOriginalFilename());
            mf.setSizes(sizes);
            mf.setPath(path + File.separator + mf.getName());
 
            file.transferTo(new File(mf.getPath()));
            mf.setGuid(FileHelper.getFileMd5(mf.getPath()));
 
            list.add(mf);
        }
 
        return list;
    }
 
    /**
     * 获取参数
     * Enumeration<String> headers = req.getHeaderNames();
     * Enumeration<String> attributes = req.getAttributeNames();
     */
    private Map<String, String> getParams(StandardMultipartHttpServletRequest req) {
        Map<String, String> map = new HashMap<String, String>(3);
 
        Enumeration<String> enumeration = req.getParameterNames();
        while (enumeration.hasMoreElements()) {
            String key = enumeration.nextElement();
            String value = req.getParameter(key);
            map.put(key, value);
        }
 
        return map;
    }
}