管道基础大数据平台系统开发-【后端】-Server
1
13693261870
2022-11-18 3bd470aabe93e6922dd0c66bfacf48bea8cae8fc
src/main/java/com/lf/server/service/data/UploaderService.java
@@ -1,7 +1,7 @@
package com.lf.server.service.data;
import com.lf.server.entity.ctrl.KeyValueEntity;
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;
@@ -33,23 +33,21 @@
    /**
     * 上传文件
     */
    public <T> List<MetaFileEntity> uploadData(T t, UserEntity ue, HttpServletRequest request, HttpServletResponse res) throws Exception {
        StandardMultipartHttpServletRequest req = (StandardMultipartHttpServletRequest) request;
    public <T> List<MetaFileEntity> uploadData(T t, String path, HttpServletRequest req, HttpServletResponse res) throws Exception {
        StandardMultipartHttpServletRequest request = (StandardMultipartHttpServletRequest) req;
        req.setCharacterEncoding("utf-8");
        res.setContentType("application/json;charset=utf-8");
        if (t != null) {
            setEntity(t, req);
            setEntity(t, request);
        }
        List<MetaFileEntity> list = getFiles(ue, req);
        List<MetaFileEntity> list = getFiles(path, request);
        return list;
    }
    /**
     * 设置实体类
     * class<T> clazz
     * T t = clazz.newInstance();
     */
    private <T> void setEntity(T t, StandardMultipartHttpServletRequest req) {
        Enumeration<String> enumeration = req.getParameterNames();
@@ -58,25 +56,26 @@
            try {
                Field field = t.getClass().getDeclaredField(key);
                if (field != null) {
                    field.setAccessible(true);
                    String value = req.getParameter(key);
                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;
                    }
                switch (field.getType().toString()) {
                    case "double":
                        field.set(t, Double.valueOf(value));
                        break;
                    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.getMessage(), ex);
@@ -87,10 +86,10 @@
    /**
     * 获取文件
     */
    private List<MetaFileEntity> getFiles(UserEntity ue, StandardMultipartHttpServletRequest req) throws Exception {
    private List<MetaFileEntity> getFiles(String subPath, StandardMultipartHttpServletRequest req) throws Exception {
        List<MetaFileEntity> list = new ArrayList<MetaFileEntity>();
        String path = pathHelper.getTempPath(ue.getId());
        String path = pathHelper.getTempPath(subPath);
        Iterator<String> iterator = req.getFileNames();
        while (iterator.hasNext()) {
            MultipartFile file = req.getFile(iterator.next());
@@ -114,6 +113,75 @@
    }
    /**
     * 自理实体集合
     */
    public void dealEntities(List<MetaFileEntity> list) {
        for (MetaFileEntity entity : list) {
            entity.setPath(FileHelper.getRelativePath(entity.getPath()));
        }
    }
    /**
     * 删除文件
     */
    public Integer deleteFiles(List<MetaFileEntity> list) {
        String root = pathHelper.getConfig().getTempPath();
        int count = 0;
        for (MetaFileEntity entity : list) {
            if (!StringHelper.isEmpty(entity.getPath())) {
                String file = root + File.separator + entity.getPath();
                File f = new File(file);
                if (f.exists()) {
                    f.delete();
                    count++;
                }
            }
        }
        return count;
    }
    /**
     * 插入元数据
     */
    public List<KeyValueEntity> insertMetas(List<KeyValueEntity> kvs) {
        String root = pathHelper.getConfig().getTempPath();
        List<KeyValueEntity> rs = new ArrayList<KeyValueEntity>();
        for (KeyValueEntity kv : kvs) {
            if (kv.getKey().toLowerCase().contains(".xls")) {
                KeyValueEntity entity = readExcel(root + File.separator + kv.getKey(), kv.getValue());
                if (null != entity) {
                    rs.add(entity);
                }
                continue;
            }
        }
        return rs;
    }
    /**
     * 读取Excel
     */
    private KeyValueEntity readExcel(String fileName, String entityName) {
        File f = new File(fileName);
        if (!f.exists() || f.isDirectory()) {
            return null;
        }
        KeyValueEntity kv = new KeyValueEntity();
        kv.setKey(FileHelper.getFileName(fileName));
        return kv;
    }
    //
    /**
     * 获取参数
     * Enumeration<String> headers = req.getHeaderNames();
     * Enumeration<String> attributes = req.getAttributeNames();