文件上传

common/src/main/java/com/kym/common/exception/BaseException.java

@@ -32,4 +32,10 @@ public class BaseException extends RuntimeException {
         // TODO: 2023-07-04 args cause 处理 
     }
 
+    BaseException(int code,String message) {
+        this.code = code;
+        this.message = message;
+    }
+
+
 }

common/src/main/java/com/kym/common/exception/BusinessException.java

@@ -22,5 +22,9 @@ public class BusinessException extends BaseException {
     public BusinessException(IResponseCode responseEnum, Object[] args, String message, Throwable cause) {
         super(responseEnum, args, message, cause);
     }
+
+    public BusinessException(int code,String message) {
+        super(code,message);
+    }
 }
 

common/src/main/java/com/kym/common/utils/CommUtil.java

@@ -0,0 +1,234 @@
+package com.kym.common.utils;
+
+import com.kym.common.constant.ResponseEnum;
+import com.kym.common.exception.BusinessException;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.lang.reflect.Array;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.List;
+import java.util.Map;
+
+
+/**
+ * 通用工具类
+ */
+public class CommUtil {
+
+    private static Logger logger = LoggerFactory.getLogger(CommUtil.class);
+
+    private CommUtil() {
+    }
+
+    /**
+     * 断言
+     *
+     * @param expected
+     * @param error
+     * @param code
+     */
+    public static void asserts(boolean expected, String error, int code) {
+        if (!expected) {
+            throw new BusinessException(code, error);
+        }
+    }
+
+    /**
+     * 断言
+     *
+     * @param expected
+     * @param error
+     */
+    public static void asserts(boolean expected, String error) {
+        if (!expected) {
+            throw new BusinessException(ResponseEnum.FAILED.getCode(), error);
+        }
+    }
+
+
+    /**
+     * 断言
+     *
+     * @param expected
+     * @param msgEnum
+     */
+    public static void asserts(boolean expected, ResponseEnum msgEnum) {
+        if (!expected) {
+            throw new BusinessException(msgEnum);
+        }
+    }
+
+    /**
+     * privatmitive
+     * 判断一个对象是否是基础数据类型或String
+     */
+    public static boolean isBasicType(Object obj) {
+        if (obj instanceof String) {
+            return true;
+        } else if (obj instanceof Integer) {
+            return true;
+        } else if (obj instanceof Short) {
+            return true;
+        } else if (obj instanceof Float) {
+            return true;
+        } else if (obj instanceof Double) {
+            return true;
+        } else if (obj instanceof Byte) {
+            return true;
+        } else if (obj instanceof Boolean) {
+            return true;
+        } else if (obj instanceof Long) {
+            return true;
+        } else {
+            return obj instanceof Character;
+        }
+    }
+
+    public static boolean isNotEmptyAndNull(Object value) {
+        if (null == value) {
+            return false;
+        }
+        if (isBasicType(value)) {
+            return !isEmptyOrNull(value.toString());
+        } else {
+            if (value instanceof Array) {
+                return Array.getLength(value) != 0;
+            } else if (value instanceof Collection) {
+                return !((Collection<?>) value).isEmpty();
+            } else if (value instanceof Map) {
+                return !((Map<?, ?>) value).isEmpty();
+            } else {
+                return !isEmptyOrNull(value.toString());
+            }
+        }
+    }
+
+    /**
+     * 判断集合是否为空
+     */
+    public static boolean isEmptyOrNull(Collection<?> collection) {
+        return null == collection || collection.isEmpty();
+    }
+
+
+    /**
+     * 判断集合是否为空
+     */
+    public static boolean isEmptyOrNull(List<Object> args) {
+        if (null == args || args.isEmpty()) {
+            return true;
+        }
+        for (Object arg : args) {
+            if (isEmptyOrNull(arg)) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    /**
+     * 判断Map是否为空
+     */
+    public static boolean isEmptyOrNull(Map<?, ?> map) {
+        return null == map || map.isEmpty();
+    }
+
+    /**
+     * 判断数组是否为空
+     */
+    public static boolean isEmptyOrNull(Object[] array) {
+        return null == array || array.length == 0 || isEmptyOrNull(array[0]);
+    }
+
+    /**
+     * 判断字符串是否为空
+     */
+    public static boolean isEmptyOrNull(String value) {
+        return null == value || value.length() == 0 || value.trim().length() == 0;
+    }
+
+    /**
+     * 判断对象是否为空
+     */
+    public static boolean isEmptyOrNull(Object value) {
+        if (null == value) {
+            return true;
+        }
+        if (isBasicType(value)) {
+            return isEmptyOrNull(value.toString());
+        } else {
+            if (value.getClass().isArray()) {
+                return Array.getLength(value) == 0;
+            } else if (value instanceof Collection) {
+                return ((Collection<?>) value).isEmpty();
+            } else if (value instanceof Map) {
+                return ((Map<?, ?>) value).isEmpty();
+            } else {
+                return isEmptyOrNull(value.toString());
+            }
+        }
+    }
+
+
+    /**
+     * 集合过滤空转List
+     *
+     * @param iterable
+     * @param <T>
+     * @return
+     */
+    public static <T> List<T> null2List(Iterable<T> iterable) {
+        List<T> list = new ArrayList<>();
+        if (isEmptyOrNull(iterable)) {
+            return list;
+        }
+
+        for (T next : iterable) {
+            if (!isEmptyOrNull(next)) {
+                list.add(next);
+            }
+        }
+
+        return list;
+    }
+
+    /**
+     * 判断是否为脚本类可执行文件
+     *
+     * @param suffix
+     * @return
+     */
+    public static boolean isExecutable(String suffix) {
+        if (isEmptyOrNull(suffix)) return false;
+        List<String> forbidSuffixs = List.of("sh", "bat", "exe", "msi", "py");
+        return forbidSuffixs.contains(suffix.toLowerCase());
+    }
+
+
+    /**
+     * 对象转Long
+     *
+     * @param s
+     * @return
+     */
+    public static Long null2Long(Object s) {
+        long v = 0L;
+        if (null != s) {
+            try {
+                v = Long.parseLong(s.toString());
+            } catch (Exception e) {
+                logger.error("null2Long error :{}", s);
+            }
+        }
+        return v;
+    }
+
+    public static String null2String(Object val) {
+        if(isEmptyOrNull(val)){
+            return "";
+        }
+        return val.toString().trim();
+    }
+}

entity/src/main/java/com/kym/entity/common/SimpleVo.java

@@ -0,0 +1,6 @@
+package com.kym.entity.common;
+
+public class SimpleVo {
+    public Long id;
+    public String name;
+}

entity/src/main/java/com/kym/entity/miniapp/Attachment.java

@@ -0,0 +1,69 @@
+package com.kym.entity.miniapp;
+
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.Getter;
+import lombok.Setter;
+
+import java.io.Serializable;
+import java.time.LocalDateTime;
+
+/**
+ * <p>
+ * 附件
+ * </p>
+ *
+ * @author zuy
+ * @since 2023-08-12
+ */
+@Getter
+@Setter
+@TableName("t_attachment")
+public class Attachment implements Serializable {
+
+    private static final long serialVersionUID = 1L;
+
+    /**
+     * ID
+     */
+    private Long id;
+
+    /**
+     * uuid
+     */
+    private String uuid;
+
+    /**
+     * 文件名
+     */
+    private String name;
+
+    /**
+     * 文件大小
+     */
+    private Long size;
+
+    /**
+     * 上传人ID
+     */
+    private Long uploadBy;
+
+    /**
+     * 上传人
+     */
+    private String uploadName;
+
+    /**
+     * 是否删除
+     */
+    private Boolean isDelete;
+
+    /**
+     * 创建时间
+     */
+    private LocalDateTime createTime;
+
+    /**
+     * 更新时间
+     */
+    private LocalDateTime updateTime;
+}

mapper/src/main/java/com/kym/mapper/miniapp/AttachmentMapper.java

@@ -0,0 +1,16 @@
+package com.kym.mapper.miniapp;
+
+import com.kym.entity.miniapp.Attachment;
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+
+/**
+ * <p>
+ * 附件 Mapper 接口
+ * </p>
+ *
+ * @author zuy
+ * @since 2023-08-12
+ */
+public interface AttachmentMapper extends BaseMapper<Attachment> {
+
+}

mapper/src/main/resources/mappers/miniapp/AttachmentMapper.xml

@@ -0,0 +1,23 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
+<mapper namespace="com.kym.mapper.miniapp.AttachmentMapper">
+
+    <!-- 通用查询映射结果 -->
+    <resultMap id="BaseResultMap" type="com.kym.entity.miniapp.Attachment">
+        <result column="id" property="id" />
+        <result column="uuid" property="uuid" />
+        <result column="name" property="name" />
+        <result column="size" property="size" />
+        <result column="upload_by" property="uploadBy" />
+        <result column="upload_name" property="uploadName" />
+        <result column="is_delete" property="isDelete" />
+        <result column="create_time" property="createTime" />
+        <result column="update_time" property="updateTime" />
+    </resultMap>
+
+    <!-- 通用查询结果列 -->
+    <sql id="Base_Column_List">
+        id, uuid, name, size, upload_by, upload_name, is_delete, create_time, update_time
+    </sql>
+
+</mapper>

miniapp/src/main/java/com/kym/miniapp/controller/AttachmentController.java

@@ -0,0 +1,18 @@
+package com.kym.miniapp.controller;
+
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+/**
+ * <p>
+ * 附件 前端控制器
+ * </p>
+ *
+ * @author zuy
+ * @since 2023-08-12
+ */
+@RestController
+@RequestMapping("/attachment")
+public class AttachmentController {
+
+}

miniapp/src/main/java/com/kym/miniapp/controller/FileController.java

@@ -1,14 +1,39 @@
 package com.kym.miniapp.controller;
 
+import cn.hutool.core.io.FileUtil;
+import com.kym.common.R;
+import com.kym.common.annotation.SysLog;
+import com.kym.common.utils.CommUtil;
+import com.kym.entity.miniapp.Attachment;
+import com.kym.service.miniapp.AttachmentService;
+import jakarta.annotation.Resource;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.IOException;
+
 
 /**
  * @author skyline
  * @description 文件上传
  * @date 2023-07-29 10:54
  */
-@RestController("/api/file")
+@RestController
+@RequestMapping("/api/file")
 public class FileController {
 
+    @Resource
+    private AttachmentService attachmentService;
+
+    @SysLog("文件上传")
+    @PostMapping("/upload")
+    public R<?> upload(MultipartFile file) throws IOException {
+        var fileName = file.getOriginalFilename();
+        CommUtil.asserts(!CommUtil.isExecutable(FileUtil.getSuffix(fileName)),"上传文件类型不支持");
+        Attachment attachment = attachmentService.upload(file);
+        return R.success(attachment);
+    }
 
 }

miniapp/src/main/java/com/kym/miniapp/utils/MybatisPlusGeneratorForMiniApp.java

@@ -17,7 +17,7 @@ public class MybatisPlusGeneratorForMiniApp {
         // 指定数据源
         FastAutoGenerator.create("jdbc:mysql://121.40.98.15:3307/charge_app", "root", "123456")
                 .globalConfig(builder -> {
-                    builder.author("skyline")
+                    builder.author("zuy")
                             // 指定输出目录
                             .disableOpenDir();
                 })
@@ -34,12 +34,12 @@ public class MybatisPlusGeneratorForMiniApp {
                             .controller("miniapp.controller")
                             // 设置mapperXml生成路径
                             .pathInfo(Map.of(
-                                    OutputFile.entity, "D://快与慢充电桩/code/charge-java/entity/src/main/java/com/kym/entity/miniapp",
-                                    OutputFile.service, "D://快与慢充电桩/code/charge-java/service/src/main/java/com/kym/service/miniapp",
-                                    OutputFile.serviceImpl, "D://快与慢充电桩/code/charge-java/service/src/main/java/com/kym/service/miniapp/impl",
-                                    OutputFile.mapper, "D://快与慢充电桩/code/charge-java/mapper/src/main/java/com/kym/mapper/miniapp",
-                                    OutputFile.controller, "D://快与慢充电桩/code/charge-java/miniapp/src/main/java/com/kym/miniapp/controller",
-                                    OutputFile.xml, "D://快与慢充电桩/code/charge-java/miniapp/src/main/resources/mappers"
+                                    OutputFile.entity, "/home/zuy/02.code/charge/charge-java/entity/src/main/java/com/kym/entity/miniapp",
+                                    OutputFile.service, "/home/zuy/02.code/charge/charge-java/service/src/main/java/com/kym/service/miniapp",
+                                    OutputFile.serviceImpl, "/home/zuy/02.code/charge/charge-java/service/src/main/java/com/kym/service/miniapp/impl",
+                                    OutputFile.mapper, "/home/zuy/02.code/charge/charge-java/mapper/src/main/java/com/kym/mapper/miniapp",
+                                    OutputFile.controller, "/home/zuy/02.code/charge/charge-java/miniapp/src/main/java/com/kym/miniapp/controller",
+                                    OutputFile.xml, "/home/zuy/02.code/charge/charge-java/mapper/src/main/resources/mappers/miniapp"
 
                             ));
                 })
@@ -50,7 +50,7 @@ public class MybatisPlusGeneratorForMiniApp {
 //                                    "t_user",
 //                                    "t_account"
 //                                    "t_charge_order",
-                                    "t_wallet_detail"
+                                    "t_attachment"
 //                                    "t_cars",
 //                            "t_collect"
                             )

miniapp/src/main/resources/application.yml

@@ -118,3 +118,9 @@ en-plus:
   # 最小充电余额（分）
   charge-min-amount: 200
 
+#文件上传配置
+upload:
+  file:
+   storage: file_storage  #文件存放地址
+   size: 20  #最大上传Mb
+

service/src/main/java/com/kym/service/miniapp/AttachmentService.java

@@ -0,0 +1,21 @@
+package com.kym.service.miniapp;
+
+import com.kym.entity.miniapp.Attachment;
+import com.baomidou.mybatisplus.extension.service.IService;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.IOException;
+
+
+/**
+ * <p>
+ * 附件 服务类
+ * </p>
+ *
+ * @author zuy
+ * @since 2023-08-12
+ */
+public interface AttachmentService extends IService<Attachment> {
+
+    Attachment upload(MultipartFile file) throws IOException;
+}

service/src/main/java/com/kym/service/miniapp/impl/AttachmentServiceImpl.java

@@ -0,0 +1,68 @@
+package com.kym.service.miniapp.impl;
+
+import cn.dev33.satoken.stp.StpUtil;
+import cn.hutool.core.io.FileUtil;
+import cn.hutool.core.lang.UUID;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.kym.common.utils.CommUtil;
+import com.kym.entity.miniapp.Attachment;
+import com.kym.mapper.miniapp.AttachmentMapper;
+import com.kym.service.miniapp.AttachmentService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.File;
+import java.io.IOException;
+
+
+/**
+ * <p>
+ * 附件 服务实现类
+ * </p>
+ *
+ * @author zuy
+ * @since 2023-08-12
+ */
+@Service
+public class AttachmentServiceImpl extends ServiceImpl<AttachmentMapper, Attachment> implements AttachmentService {
+
+    @Value("${upload.file.storage}")
+    private String fileStoragePath;
+
+    @Value("${upload.file.size}")
+    private int fileSize;
+
+    @Autowired
+    private AttachmentMapper attachmentMapper;
+
+    @Override
+    public Attachment upload(MultipartFile file) throws IOException {
+        String fileName = file.getOriginalFilename();
+        StpUtil.checkLogin();
+        long size = file.getSize();
+        CommUtil.asserts((long) fileSize << 20 > size, "文件超出上传大小");
+        Long userId = CommUtil.null2Long(StpUtil.getSession().get("userId"));
+
+        Attachment attachment = new Attachment();
+        attachment.setName(fileName);
+        attachment.setSize(file.getSize());
+        attachment.setUuid(UUID.fastUUID().toString(true) + "." + FileUtil.getSuffix(fileName));
+        attachment.setUploadBy(userId);
+        attachment.setUploadName(CommUtil.null2String(StpUtil.getSession().get("username")));
+        attachment.setIsDelete(false);
+
+        String applicationPath = System.getProperty("user.dir");
+
+        File dir = new File(applicationPath,fileStoragePath);
+        if(!dir.exists()){
+            CommUtil.asserts(dir.mkdirs(),"文件上传异常");
+        }
+        File storeFile = new File(dir,attachment.getUuid());
+        file.transferTo(storeFile);
+
+        attachmentMapper.insert(attachment);
+        return attachment;
+    }
+}