文件上传及入库

@@ -7,6 +7,8 @@ readDirectory = /Users/mrz/Documents/java项目/test
 #接收存储报文目录
 #接收存储报文目录
 bakDirectory = kakoRevice
 bakDirectory = kakoRevice
 #是否需要发送报文,默认N不发,Y将发送readDirectory下的XML扩展名的报文
 #是否需要发送报文,默认N不发,Y将发送readDirectory下的XML扩展名的报文
+#上传文件路径，必填
+uploadDirectory = upload
 isNeedSend = Y
 isNeedSend = Y
 
 
 #IMF MEAT报头配置
 #IMF MEAT报头配置

+package com.tianbo.warehouse.controller;
+
+import com.tianbo.warehouse.controller.response.ResultJson;
+import com.tianbo.warehouse.model.Attachment;
+import com.tianbo.warehouse.service.AttachmentService;
+import com.tianbo.warehouse.util.Date.DateUtil;
+import com.tianbo.warehouse.util.Helper;
+import org.apache.commons.io.FileUtils;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.File;
+import java.io.IOException;
+
+import static com.tianbo.warehouse.util.IO.FileTool.readProperties;
+
+@RestController
+public class UploadController {
+
+    @Autowired
+    private AttachmentService attachmentService;
+
+    /**
+     * 单个文件上传
+     * @param file
+     * @return
+     * @throws Exception
+     */
+    @PostMapping("/upload")
+    public ResultJson upload(@RequestParam MultipartFile file) throws Exception {
+       ResultJson resultJson =  new ResultJson();
+                if (null!=file && !file.isEmpty() && file.getSize()>0){
+                    byte[] filebyte = file.getBytes();
+                    String fileLitleName = file.getName();
+                    String fileName = file.getOriginalFilename();
+                    String fileType =  fileName.substring((fileName.lastIndexOf(".")));
+                    String saveDir = readProperties("uploadDirectory");
+                    StringBuffer stringBuffer = new StringBuffer();
+                    String saveFileName =  stringBuffer.append(saveDir).append("/").append(DateUtil.getToday()).append("/").append(Helper.getUUID()).append(fileType).toString();
+                    File saveFile = new File(saveFileName);
+                    if(!saveFile.getParentFile().exists()){
+                        saveFile.mkdirs();
+                    }
+
+                    try{
+                        FileUtils.writeByteArrayToFile(saveFile,filebyte);
+                        Attachment attachment = new Attachment(saveFileName);
+                        int i =attachmentService.insertSelective(attachment);
+                        resultJson = (i==1 ? new ResultJson("200","附件入库成功",saveFileName) :new ResultJson("500","附件入库失败"));
+                    }catch (IOException e){
+                        resultJson = new ResultJson("400","附件存储失败",e);
+                        e.printStackTrace();
+                    }catch (IllegalStateException e){
+                        resultJson = new ResultJson("400","附件存储失败",e);
+                        e.printStackTrace();
+                    }
+                }
+                return resultJson;
+    }
+
+    /**
+     * 批量上传
+     * @param files
+     * @return
+     * @throws Exception
+     */
+    @PostMapping("/uploads")
+    public ResultJson upload(MultipartFile[] files) throws Exception {
+        if (files!=null && files.length>0){
+            // ... 处理文件储存逻辑
+            for (MultipartFile file : files) {
+                if (!file.isEmpty() && file.getSize()>0){
+                    String fileName = file.getName();
+                    String fileType = file.getContentType();
+                    String saveDir = readProperties("uploadDirectory");
+                    StringBuffer stringBuffer = new StringBuffer();
+                    stringBuffer.append(saveDir).append("/").append(DateUtil.getToday()).append("/").append(fileType).append(Helper.getUUID()).append(fileName);
+                    File saveFile = new File(stringBuffer.toString());
+                    try{
+                        file.transferTo(saveFile);
+                    }catch (IOException e){
+                        e.printStackTrace();
+                    }catch (IllegalStateException e){
+                        e.printStackTrace();
+                    }
+
+                }
+
+            }
+        }
+        return new ResultJson();
+    }
+}

+package com.tianbo.warehouse.dao;
+
+import com.tianbo.warehouse.model.Attachment;
+
+public interface AttachmentMapper {
+    int deleteByPrimaryKey(Integer aid);
+
+    int insert(Attachment record);
+
+    int insertSelective(Attachment record);
+
+    Attachment selectByPrimaryKey(Integer aid);
+
+    int updateByPrimaryKeySelective(Attachment record);
+
+    int updateByPrimaryKey(Attachment record);
+}

+package com.tianbo.warehouse.model;
+
+import lombok.Data;
+
+@Data
+public class Attachment {
+    private Integer aid;
+
+    private String path;
+
+    public Attachment(String path){
+        this.path = path;
+    }
+}

+package com.tianbo.warehouse.service;
+
+import com.tianbo.warehouse.model.Attachment;
+
+public interface AttachmentService {
+
+    int insertSelective(Attachment record);
+}

+package com.tianbo.warehouse.service.imp;
+
+import com.tianbo.warehouse.dao.AttachmentMapper;
+import com.tianbo.warehouse.model.Attachment;
+import com.tianbo.warehouse.service.AttachmentService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+@Service
+public class AttachmentServiceImp implements AttachmentService {
+
+    @Autowired
+    private AttachmentMapper attachmentMapper;
+
+    @Override
+    public int insertSelective(Attachment record){
+        return attachmentMapper.insertSelective(record);
+    }
+}

@@ -45,6 +45,6 @@
             <property name="enableSubPackages" value="true"/>
             <property name="enableSubPackages" value="true"/>
         </javaClientGenerator>
         </javaClientGenerator>
         <!-- 要生成的表 tableName是数据库中的表名或视图名 domainObjectName是实体类名-->
         <!-- 要生成的表 tableName是数据库中的表名或视图名 domainObjectName是实体类名-->
-        <table tableName="log" domainObjectName="LOG" enableCountByExample="false" enableUpdateByExample="false" enableDeleteByExample="false" enableSelectByExample="false" selectByExampleQueryId="false"></table>
+        <table tableName="attachment" domainObjectName="Attachment" enableCountByExample="false" enableUpdateByExample="false" enableDeleteByExample="false" enableSelectByExample="false" selectByExampleQueryId="false"></table>
     </context>
     </context>
 </generatorConfiguration>
 </generatorConfiguration>

+<?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.tianbo.warehouse.dao.AttachmentMapper" >
+  <resultMap id="BaseResultMap" type="com.tianbo.warehouse.model.Attachment" >
+    <id column="aid" property="aid" jdbcType="INTEGER" />
+    <result column="path" property="path" jdbcType="VARCHAR" />
+  </resultMap>
+  <sql id="Base_Column_List" >
+    aid, path
+  </sql>
+  <select id="selectByPrimaryKey" resultMap="BaseResultMap" parameterType="java.lang.Integer" >
+    select 
+    <include refid="Base_Column_List" />
+    from attachment
+    where aid = #{aid,jdbcType=INTEGER}
+  </select>
+  <delete id="deleteByPrimaryKey" parameterType="java.lang.Integer" >
+    delete from attachment
+    where aid = #{aid,jdbcType=INTEGER}
+  </delete>
+  <insert id="insert" parameterType="com.tianbo.warehouse.model.Attachment" >
+    insert into attachment (aid, path)
+    values (#{aid,jdbcType=INTEGER}, #{path,jdbcType=VARCHAR})
+  </insert>
+  <insert id="insertSelective" parameterType="com.tianbo.warehouse.model.Attachment" >
+    insert into attachment
+    <trim prefix="(" suffix=")" suffixOverrides="," >
+      <if test="aid != null" >
+        aid,
+      </if>
+      <if test="path != null" >
+        path,
+      </if>
+    </trim>
+    <trim prefix="values (" suffix=")" suffixOverrides="," >
+      <if test="aid != null" >
+        #{aid,jdbcType=INTEGER},
+      </if>
+      <if test="path != null" >
+        #{path,jdbcType=VARCHAR},
+      </if>
+    </trim>
+  </insert>
+  <update id="updateByPrimaryKeySelective" parameterType="com.tianbo.warehouse.model.Attachment" >
+    update attachment
+    <set >
+      <if test="path != null" >
+        path = #{path,jdbcType=VARCHAR},
+      </if>
+    </set>
+    where aid = #{aid,jdbcType=INTEGER}
+  </update>
+  <update id="updateByPrimaryKey" parameterType="com.tianbo.warehouse.model.Attachment" >
+    update attachment
+    set path = #{path,jdbcType=VARCHAR}
+    where aid = #{aid,jdbcType=INTEGER}
+  </update>
+</mapper>