devbox
空白诗007
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

VUE+elementui照片上传和zip压缩包上传后端目录解压_vue elementui 文件上传

作者：空白诗007 | 2024-07-10 09:51:03

赞

踩

vue elementui 文件上传

 

1.Vue代码

  1. <template>
  2.   <div class="app-container">
  3.     <el-form :model="queryParams" ref="queryForm" :inline="true" label-width="85px">
  4.       <el-form-item label="工号" prop="userCode">
  5.         <el-input v-model="queryParams.userCode" placeholder="请输入工号" clearable size="small"
  6.           @keyup.enter.native="handleQuery"/>
  7.       </el-form-item>
  8.       <el-form-item label="姓名" prop="nickName">
  9.         <el-input v-model="queryParams.nickName" placeholder="请输入姓名" clearable
  10.           size="small" @keyup.enter.native="handleQuery"/>
  11.       </el-form-item>
  12.       <el-form-item label="部门" prop="deptName">
  13.         <el-input v-model="queryParams.deptName" placeholder="请输入部门" clearable
  14.           size="small" @keyup.enter.native="handleQuery"/>
  15.       </el-form-item>
  16.       <el-form-item>
  17.         <el-button type="primary" icon="el-icon-search" size="mini" @click="handleQuery">查询</el-button>
  18.         <el-button icon="el-icon-refresh" size="mini" @click="resetQuery">重置</el-button>
  19.         <el-button type="info" icon="el-icon-upload" size="mini" @click="handleImport" v-hasPermi="['riskPredictionData:roster:batchUploadPhoto']">批量上传</el-button>
  20.       </el-form-item>
  21.     </el-form>
  22.  
  23.     <el-row :gutter="10" class="mb8">
  24.       <el-col :span="1.5" v-if="false">
  25.         <el-button
  26.           type="success"
  27.           icon="el-icon-edit"
  28.           size="mini"
  29.           :disabled="single"
  30.           @click="handleUpdate"
  31.           v-hasPermi="['riskPredictionData:roster:uploadPhoto']"
  32.         >上传照片</el-button>
  33.       </el-col>
  34.     </el-row>
  35.  
  36.     <el-table ref="reportSupplier" :key="tamptime" :max-height="tableHeight" v-loading="loading" :data="rosterList" >
  37.       <el-table-column type="index" width="55" label="序号" align="center" />
  38.       <el-table-column label="工号" align="center" prop="userCode" />
  39.       <el-table-column label="姓名" align="center" prop="nickName" />
  40.       <el-table-column label="部门" align="center" prop="deptName" show-overflow-tooltip />
  41.       <el-table-column label="职位" align="center" prop="position" show-overflow-tooltip v-if="false"/>
  42.       <el-table-column label="职务" align="center" prop="post"  show-overflow-tooltip/>
  43.       <el-table-column label="头像" align="center" prop="photo" >
  44.         <template slot-scope="scope">
  45.           <el-image :src="baseUrl + '/profile/upload/cadrePhoto/' + scope.row.userCode +'.jpg/'" style="height:50px;width:50px" />
  46.         </template>
  47.       </el-table-column>
  48.       <el-table-column label="操作" align="center" >
  49.         <template slot-scope="scope">
  50.           <el-button
  51.             size="mini"
  52.             type="text"
  53.             icon="el-icon-edit"
  54.             @click="handleUpdate(scope.row)"
  55.             v-hasPermi="['riskPredictionData:roster:uploadPhoto']"
  56.           >上传照片</el-button>
  57.         </template>
  58.       </el-table-column>
  59.     </el-table>
  60.  
  61.     <pagination
  62.       v-show="total>0"
  63.       :total="total"
  64.       :page.sync="queryParams.pageNum"
  65.       :limit.sync="queryParams.pageSize"
  66.       @pagination="getList"
  67.     />
  68.  
  69.      <!-- 花名册入对话框 -->
  70.     <el-dialog :title="upload.title" :visible.sync="upload.open"  width="400px">
  71.       <el-upload ref="upload" :limit="1"  accept=".zip" :headers="upload.headers"
  72.         :action="upload.url + '?updateSupport=' + upload.updateSupport"
  73.         :disabled="upload.isUploading" :on-progress="handleFileUploadProgress"
  74.         :on-success="handleFileSuccess" :auto-upload="false" drag >
  75.         <i class="el-icon-upload"></i>
  76.         <div class="el-upload__text">
  77.           将文件拖到此处，或
  78.           <em>点击上传</em>
  79.         </div>
  80.         <div
  81.           class="el-upload__tip"
  82.           slot="tip"
  83.         >
  84.           <!-- <el-checkbox v-model="upload.updateSupport" /> 是否更新已经存在的用户数据 -->
  85. <!--          <el-link
  86.             type="info"
  87.             style="font-size:12px"
  88.             @click="importTemplate"
  89.           >下载模板</el-link>-->
  90.         </div>
  91.         <div
  92.           class="el-upload__tip"
  93.           style="color:red"
  94.           slot="tip"
  95.         >提示：仅允许导入“zip”格式文件！</div>
  96.       </el-upload>
  97.       <div
  98.         slot="footer"
  99.         class="dialog-footer"
  100.       >
  101.         <el-button
  102.           type="primary"
  103.           @click="submitFileForm"
  104.         >确 定</el-button>
  105.         <el-button @click="upload.open = false">取 消</el-button>
  106.       </div>
  107.     </el-dialog>
  108.     <!-- 添加或修改成员照对话框 -->
  109.     <el-dialog :title="title" :visible.sync="open" width="500px" v-dialog-drag :close-on-click-modal="false">
  110.       <el-form ref="form" :model="form" :rules="rules" label-width="95px">
  111.         <el-form-item label="工号" prop="userCode">
  112.           {{form.userCode}}
  113.         </el-form-item>
  114. <!--        <el-form-item label="上传证件照" prop="photo">
  115.           <el-upload class="upload-demo"
  116.                      ref="upload"
  117.                      accept=".jpg"
  118.                      :action="uploadPhoto.url+form.userCode"
  119.                      :headers="uploadPhoto.headers"
  120.                      :before-upload="handleBefore"
  121.                      :on-remove="handleRemove"
  122.                      :on-success="(res, file)=>{handleSuccess( res, file)}"
  123.                      :show-file-list="false"
  124.           >
  125.             <el-button slot="trigger" type="primary">选取文件</el-button>
  126.             <div slot="tip" class="el-upload__tip">上传文件不能超过10M</div>
  127.           </el-upload>
  128.           <el-image style="width: 100px; height: 100px" :src="uploadPhoto.imgUrl" fit="fill"></el-image>
  129.         </el-form-item>-->
  130.         <el-form-item label="上传照片" prop="photo">
  131.           <el-upload
  132.             ref="uploadPhoto"
  133.             class="avatar-uploader"
  134.             :key="tamptimeUploadPhoto"
  135.             :action="uploadPhoto.url+form.userCode"
  136.             :headers="uploadPhoto.headers"
  137.             :show-file-list="false"
  138.             :on-success="(res, file)=>{handleSuccess( res, file)}"
  139.             :before-upload="handleBefore"
  140.             :auto-upload="true">
  141.             <img v-if="uploadPhoto.isShowPhoto" :src="uploadPhoto.imgUrl" class="avatar">
  142.             <i v-else class="el-icon-plus avatar-uploader-icon"></i>
  143.           </el-upload>
  144.         </el-form-item>
  145.       </el-form>
  146.       <div slot="footer" class="dialog-footer">
  147.         <el-button type="primary" @click="submitForm">确 定</el-button>
  148.         <el-button @click="cancel">取 消</el-button>
  149.       </div>
  150.     </el-dialog>
  151.  
  152.   </div>
  153. </template>
  154.  
  155. <script>
  156. import {listRoster, getRoster, exportRoster, importTemplate}
  157.   from "@/api/hr/riskPrediction/data/roster";
  158. import { getToken } from "@/utils/auth";
  159. import store from "@/store";
  160. import axios from "axios";
  161. export default {
  162.   name: "CadrePhoto",
  163.   data() {
  164.     return {
  165.       // 遮罩层
  166.       loading: true,
  167.       // 选中数组
  168.       ids: [],
  169.       // 非单个禁用
  170.       single: true,
  171.       // 非多个禁用
  172.       multiple: true,
  173.       // 总条数
  174.       total: 0,
  175.       tableHeight:0,
  176.       // 员工风险预判-花名册导入
  177.       rosterList: [],
  178.       // 弹出层标题
  179.       title: "",
  180.       // 是否显示弹出层
  181.       open: false,
  182.       // 查询参数
  183.       queryParams: {
  184.         pageNum: 1,
  185.         pageSize: 10,
  186.         userCode: undefined,
  187.         nickName: undefined,
  188.         deptName: undefined,
  189.         post: '部门负责人',
  190.       },
  191.       // 表单参数
  192.       form: {
  193.         userCode: undefined,
  194.         photo: undefined,
  195.       },
  196.       // 表单校验
  197.       rules: {
  198.         userCode: [
  199.           { required: true, message: "工号不能为空", trigger: "blur" }
  200.         ],
  201.         nickName: [
  202.           { required: true, message: "姓名不能为空", trigger: "blur" }
  203.         ],
  204.       },
  205.       upload: {
  206.         // 是否显示弹出层
  207.         open: false,
  208.         // 弹出层标题
  209.         title: "批量上传",
  210.         // 是否禁用上传
  211.         isUploading: false,
  212.         // 是否更新已经存在的数据
  213.         updateSupport: 0,
  214.         // 设置上传的请求头部
  215.         headers: { Authorization: "Bearer " + getToken() },
  216.         // 上传的地址
  217.         url: process.env.VUE_APP_BASE_API + "/riskPrediction/data/roster/batchUpload"
  218.       },
  219.       options: {
  220.         img: store.getters.avatar, //裁剪图片的地址
  221.         autoCrop: true, // 是否默认生成截图框
  222.         autoCropWidth: 200, // 默认生成截图框宽度
  223.         autoCropHeight: 200, // 默认生成截图框高度
  224.         fixedBox: true // 固定截图框大小 不允许改变
  225.       },
  226.       uploadPhoto: {
  227.         // 设置上传的请求头部
  228.         headers: { Authorization: "Bearer " + getToken() },
  229.         // 上传的地址
  230.         url: process.env.VUE_APP_BASE_API + "/riskPrediction/data/roster/upload/",
  231.         imgUrl: '',
  232.         isShowPhoto: false,
  233.       },
  234.       baseUrl: process.env.VUE_APP_BASE_API,
  235.       tamptime:new Date().valueOf(),
  236.       tamptimeUploadPhoto:new Date().valueOf(),
  237.     };
  238.   },
  239.   created() {
  240.     this.getList();
  241.   },
  242.   mounted() {
  243.     // 设置表格最大高度
  244.     this.$nextTick(() => {
  245.       //屏幕可视化高度-表格上方高度-需求空出高度
  246.       this.tableHeight = window.innerHeight - this.$refs.reportSupplier.$el.offsetTop - 140;
  247.     });
  248.   },
  249.   methods: {
  250.     /** 导入按钮操作 */
  251.     handleImport () {
  252.       this.upload.title = "批量导入";
  253.       this.upload.open = true;
  254.     },
  255.     // 文件上传中处理
  256.     handleFileUploadProgress (event, file, fileList) {
  257.       this.upload.isUploading = true;
  258.     },
  259.     // 文件上传成功处理
  260.     handleFileSuccess (response, file, fileList) {
  261.       this.upload.open = false;
  262.       this.upload.isUploading = false;
  263.       this.$refs.upload.clearFiles();
  264.       if(this.strIsEmpty(response.msg)){
  265.         this.$message({
  266.           message: '请使用模板导入',
  267.           type: 'warning'
  268.         });
  269.         return
  270.       }
  271.       this.$alert(response.msg, "导入结果", { dangerouslyUseHTMLString: true });
  272.       this.getList();
  273.     },
  274.     /** 下载模板操作 */
  275.     importTemplate () {
  276.       importTemplate().then(response => {
  277.         this.download(response.msg);
  278.       });
  279.     },
  280.     // 提交上传文件
  281.     submitFileForm () {
  282.       this.$refs.upload.submit();
  283.     },
  284.     /** 查询员工风险预判-花名册导入 */
  285.     getList() {
  286.       this.loading = true;
  287.       this.tamptime = new Date().valueOf()
  288.       listRoster(this.queryParams).then(response => {
  289.         this.rosterList = response.rows;
  290.         this.total = response.total;
  291.         this.loading = false;
  292.         // 数据并不会更新, 随便调用一下页面事件，什么事件都可以，聚焦等都可以，这里使用 element 表格自带的高亮行事件
  293.         this.$refs.reportSupplier.setCurrentRow(this.rosterList[0])
  294.       });
  295.     },
  296.     /** 搜索按钮操作 */
  297.     handleQuery() {
  298.       this.queryParams.pageNum = 1;
  299.       this.getList();
  300.     },
  301.     /** 重置按钮操作 */
  302.     resetQuery() {
  303.       this.resetForm("queryForm");
  304.       this.handleQuery();
  305.     },
  306.     /** 修改按钮操作 */
  307.     handleUpdate(row) {
  308.       this.reset();
  309.       const userCode = row.userCode
  310.       const _that = this
  311.       getRoster(userCode).then(response => {
  312.         this.form = response.data;
  313.         if(this.form.userCode){
  314.           this.uploadPhoto.imgUrl = process.env.VUE_APP_BASE_API + "/profile/upload/cadrePhoto/" + this.form.userCode +'.jpg'
  315.           axios.get(_that.uploadPhoto.imgUrl)
  316.           .then(function (response) {
  317.             if(response.status == 200){
  318.               _that.uploadPhoto.isShowPhoto = true;
  319.             }
  320.           })
  321.           .catch(function (error) {
  322.             // 处理请求失败的结果
  323.           });
  324.           /*getImgUrl(this.uploadPhoto.imgUrl).then(response=>{
  325.             if(response.status == 200){
  326.               uploadPhoto.isShowPhoto = true;
  327.             }
  328.           }).catch(error => {
  329.           })*/
  330.         }
  331.         this.open = true;
  332.         this.title = "修改照片";
  333.       });
  334.     },
  335.     // 取消按钮
  336.     cancel() {
  337.       this.open = false;
  338.       this.getList()
  339.     },
  340.     /** 提交按钮 */
  341.     submitForm: function() {
  342.       this.open = false;
  343.       this.getList()
  344.       //this.$refs.uploadPhoto.submit();
  345.     },
  346.     reset(){
  347.       this.form = {
  348.        userCode : undefined,
  349.        photo: undefined,
  350.       },
  351.       this.uploadPhoto = {
  352.         // 设置上传的请求头部
  353.         headers: { Authorization: "Bearer " + getToken() },
  354.         // 上传的地址
  355.         url: process.env.VUE_APP_BASE_API + "/riskPrediction/data/roster/upload/",
  356.         imgUrl: '',
  357.         isShowPhoto: false,
  358.       }
  359.     },
  360.     handleBefore (file) {
  361.       var fileExtension = file.name.substring(file.name.lastIndexOf('.') + 1);
  362.       if (["jpg"].indexOf(fileExtension) == -1) {
  363.         //文件类型
  364.         this.$message({
  365.           message: "不支持上传." + fileExtension+ "类型的附件",
  366.           type: 'warning'
  367.         })
  368.         return false;
  369.       }
  370.       if ( file.size > 10485760) {
  371.         //文件不能大于10M
  372.         this.$message({
  373.           message: "上传附件不能大于10M",
  374.           type: 'warning'
  375.         })
  376.         return false;
  377.       }
  378.     },
  379.     handleRemove(file, fileList) {
  380.     },
  381.     handleSuccess(res, file){
  382.       if (res.code === 200) {
  383.         this.uploadPhoto.imgUrl = res.url
  384.         this.form.photo =file.fileName
  385.         this.uploadPhoto.isShowPhoto = true;
  386.         this.tamptimeUploadPhoto = new Date().valueOf()
  387.       }else{
  388.         this.uploadPhoto.imgUrl = null
  389.         this.form.photo = null
  390.       }
  391.     },
  392.     getImgUrl(){
  393.       return this.uploadPhoto.imgUrl
  394.     }
  395.  
  396.   }
  397. };
  398. </script>
  399. <style>
  400. .avatar-uploader .el-upload {
  401.   border: 1px dashed #d9d9d9;
  402.   border-radius: 6px;
  403.   cursor: pointer;
  404.   position: relative;
  405.   overflow: hidden;
  406. }
  407. .avatar-uploader .el-upload:hover {
  408.   border-color: #409EFF;
  409. }
  410. .avatar-uploader-icon {
  411.   font-size: 28px;
  412.   color: #8c939d;
  413.   width: 178px;
  414.   height: 178px;
  415.   line-height: 178px;
  416.   text-align: center;
  417. }
  418. .avatar {
  419.   width: 178px;
  420.   height: 178px;
  421.   display: block;
  422. }
  423. </style>

2.roster.js代码

  1. import request from '@/utils/request'
  2.  
  3. // 查询员工风险预判-花名册导入+员工基础信息列表
  4. export function listRoster(query) {
  5.   return request({
  6.     url: '/riskPrediction/data/roster/list',
  7.     method: 'get',
  8.     params: query
  9.   })
  10. }
  11.  
  12. // 查询员工风险预判-花名册导入+员工基础信息详细
  13. export function getRoster(userCode) {
  14.   return request({
  15.     url: '/riskPrediction/data/roster/' + userCode,
  16.     method: 'get'
  17.   })
  18. }
  19.  
  20.  
  21. // 导出员工风险预判-花名册导入+员工基础信息
  22. export function exportRoster(query) {
  23.   return request({
  24.     url: '/riskPrediction/data/roster/export',
  25.     method: 'get',
  26.     params: query
  27.   })
  28. }
  29.  
  30. // 下载模板
  31. export function importTemplate() {
  32.   return request({
  33.     url: '/riskPrediction/data/roster/importTemplate',
  34.     method: 'get'
  35.   })
  36. }

3.auth.js代码

  1. import Cookies from 'js-cookie'
  2. import router from './../router'
  3. import store from '@/store'
  4.  
  5. const TokenKey = 'Admin-Token'
  6.  
  7.  
  8. const ssoToken = 'sso_session';
  9.  
  10.  
  11.  
  12. export function getAuthHeader() {
  13.   let header = {};
  14.   if (getToken()) {
  15.     header['Authorization'] = 'Bearer ' + getToken() // 让每个请求携带自定义token 请根据实际情况自行修改
  16.   }
  17.   let route = router.currentRoute;
  18.   if (route && route.meta && route.meta.roleIds) {
  19.     header['routerRoleIds'] = route.meta.roleIds // 让每个请求携带权限表示
  20.   } 
  21.   return header;
  22. }
  23.  
  24.  
  25. export function getRedirectRoutePath(toRoute) {
  26.   if(!toRoute) {
  27.     return  {path:"/"};
  28.   }
  29.   let toPath = toRoute.path;
  30.   const permissionRoutes = store.getters && store.getters.permission_routes
  31.   if (!permissionRoutes || permissionRoutes.length < 1) {
  32.     return {path:"/"};
  33.   }
  34.   if (toPath == "/") {
  35.     let permissionRootPaths = permissionRoutes.filter(route=>{
  36.       if (route.path  == ""  && route.children 
  37.       && route.children.length > 0 ) {
  38.         let t = route.children[0];
  39.         return t.path == "/bgBoard"
  40.       }
  41.       return false;
  42.     })
  43.     if (permissionRootPaths.length > 0) {
  44.       let t = permissionRootPaths[0].children[0];
  45.       return {path:t.path};
  46.     }
  47.     return {path:"/index"};
  48.   }
  49.   return toRoute;
  50. }
  51.  
  52. export function getToken() {
  53.   return Cookies.get(TokenKey)
  54. }
  55.  
  56. export function setToken(token) {
  57.   return Cookies.set(TokenKey, token)
  58. }
  59.  
  60. export function removeToken() {
  61.   return Cookies.remove(TokenKey)
  62. }
  63.  
  64. export function getSSOToken() {
  65.   return Cookies.get(ssoToken)
  66. }
  67.  
  68. export function setSSOToken(token) {
  69.   return Cookies.set(ssoToken, token)
  70. }
  71.  
  72. export function removeSSOToken() {
  73.   return Cookies.remove(ssoToken)
  74. }
  75.

4.后端代码

  1. package com.goertek.riskPrediction.data.controller;
  2.  
  3. import com.goertek.framework.aspectj.lang.enums.BusinessType;
  4. import com.goertek.framework.config.GoerTekConfig;
  5. import com.goertek.framework.config.ServerConfig;
  6. import com.goertek.framework.utils.poi.ExcelUtil;
  7. import com.goertek.framework.web.controller.BaseController;
  8. import com.goertek.framework.web.domain.AjaxResult;
  9. import com.goertek.framework.web.page.TableDataInfo;
  10. import com.goertek.hrLog.HrLog;
  11. import com.goertek.hrUtils.CadreFileUploadUtils;
  12. import com.goertek.riskPrediction.data.domain.HrRiskPredictionImportRoster;
  13. import com.goertek.riskPrediction.data.service.IHrRiskPredictionImportRosterService;
  14. import org.springframework.beans.factory.annotation.Autowired;
  15. import org.springframework.security.access.prepost.PreAuthorize;
  16. import org.springframework.web.bind.annotation.*;
  17. import org.springframework.web.multipart.MultipartFile;
  18.  
  19. import java.io.File;
  20. import java.util.List;
  21.  
  22. /**
  23.  * 员工风险预判-花名册导入Controller
  24.  * 
  25.  * @author xx
  26.  * @date 2021-08-30
  27.  */
  28. @RestController
  29. @RequestMapping("/riskPrediction/data/roster")
  30. public class HrRiskPredictionImportRosterControllerTest extends BaseController
  31. {
  32.     @Autowired
  33.     private ServerConfig serverConfig;
  34.  
  35.     /**
  36.      * 照片上传请求
  37.      */
  38.     @HrLog(title = "照片上传", businessType = BusinessType.IMPORT)
  39.     @PostMapping("/upload/{userCode}")
  40.     public AjaxResult uploadFile(MultipartFile file,@PathVariable("userCode") String userCode) throws Exception
  41.     {
  42.         try
  43.         {
  44.             String originalFilename = userCode + ".jpg";
  45.             // 上传文件路径
  46.             String filePath = GoerTekConfig.getUploadPath()+ "/cadrePhoto";;
  47.             // 上传并返回新文件名称
  48.             String fileName = CadreFileUploadUtils.upload(filePath, file, originalFilename);
  49.             String url = serverConfig.getUrl() + fileName;
  50.             AjaxResult ajax = AjaxResult.success();
  51.             ajax.put("fileName", fileName);
  52.             ajax.put("url", url);
  53.             ajax.put("realName", file.getOriginalFilename());
  54.             return ajax;
  55.         }
  56.         catch (Exception e)
  57.         {
  58.             return AjaxResult.error(e.getMessage());
  59.         }
  60.     }
  61.  
  62.     @HrLog(title = "批量导入", businessType = BusinessType.IMPORT)
  63.     @PostMapping("/batchUpload")
  64.     public AjaxResult batchUpload(MultipartFile file, boolean updateSupport) throws Exception
  65.     {
  66.         try
  67.         {
  68.             String originalFilename = file.getOriginalFilename();
  69.             // 上传文件路径
  70.             String filePath = GoerTekConfig.getUploadPath()+ "/cadrePhoto";;
  71.             // 上传并返回新文件名称
  72.             String fileName = CadreFileUploadUtils.upload(filePath, file, originalFilename);
  73.             String url = serverConfig.getUrl() + fileName;
  74.             AjaxResult ajax = AjaxResult.success();
  75.             ajax.put("fileName", fileName);
  76.             ajax.put("url", url);
  77.             ajax.put("realName", file.getOriginalFilename());
  78.             //解压
  79.             File zipFile = new File(filePath+"/"+originalFilename);
  80.             File destDir = new File(filePath);
  81.             CadreFileUploadUtils.zipDecompress(zipFile,destDir);
  82.             return ajax;
  83.         }
  84.         catch (Exception e)
  85.         {
  86.             return AjaxResult.error(e.getMessage());
  87.         }
  88.     }
  89.  
  90.  
  91.  
  92.  
  93. }
5.CadreFileUploadUtils.java
  1. package com.goertek.hrUtils;
  2.  
  3. import com.goertek.framework.config.GoerTekConfig;
  4. import com.goertek.framework.constant.Constants;
  5. import com.goertek.framework.exception.file.FileNameLengthLimitExceededException;
  6. import com.goertek.framework.exception.file.FileSizeLimitExceededException;
  7. import com.goertek.framework.exception.file.InvalidExtensionException;
  8. import com.goertek.framework.utils.DateUtils;
  9. import com.goertek.framework.utils.StringUtils;
  10. import com.goertek.framework.utils.file.MimeTypeUtils;
  11. import com.goertek.framework.utils.security.Md5Utils;
  12. import org.apache.commons.io.FilenameUtils;
  13. import org.springframework.web.multipart.MultipartFile;
  14.  
  15. import java.io.File;
  16. import java.io.FileInputStream;
  17. import java.io.FileOutputStream;
  18. import java.io.IOException;
  19. import java.util.zip.ZipEntry;
  20. import java.util.zip.ZipInputStream;
  21.  
  22. /**
  23.  * 文件上传工具类
  24.  *
  25.  * @author admin
  26.  */
  27. public class CadreFileUploadUtils {
  28.  
  29.     /**
  30.      * 默认的文件名最大长度 100
  31.      */
  32.     public static final int DEFAULT_FILE_NAME_LENGTH = 100;
  33.  
  34.     /**
  35.      * 默认上传的地址
  36.      */
  37.     private static String defaultBaseDir = GoerTekConfig.getProfile();
  38.  
  39.     private static int counter = 0;
  40.  
  41.     public static void setDefaultBaseDir(String defaultBaseDir) {
  42.         CadreFileUploadUtils.defaultBaseDir = defaultBaseDir;
  43.     }
  44.  
  45.     public static String getDefaultBaseDir() {
  46.         return defaultBaseDir;
  47.     }
  48.  
  49.     /**
  50.      * 以默认配置进行文件上传
  51.      *
  52.      * @param file 上传的文件
  53.      * @return 文件名称
  54.      * @throws Exception
  55.      */
  56.     public static final String upload(MultipartFile file, String fileName) throws IOException {
  57.         try {
  58.             return upload(getDefaultBaseDir(), file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION, fileName);
  59.         } catch (Exception e) {
  60.             throw new IOException(e.getMessage(), e);
  61.         }
  62.     }
  63.  
  64.     /**
  65.      * 根据文件路径上传
  66.      *
  67.      * @param baseDir 相对应用的基目录
  68.      * @param file    上传的文件
  69.      * @return 文件名称
  70.      * @throws IOException
  71.      */
  72.     public static final String upload(String baseDir, MultipartFile file, String fileName) throws IOException {
  73.         try {
  74.             return upload(baseDir, file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION, fileName);
  75.         } catch (Exception e) {
  76.             throw new IOException(e.getMessage(), e);
  77.         }
  78.     }
  79.  
  80.  
  81.  
  82.     /**
  83.      * 根据文件路径上传
  84.      *
  85.      * @param baseDir 相对应用的基目录
  86.      * @param file    上传的文件
  87.      * @return 文件名称
  88.      * @throws IOException
  89.      */
  90.     public static final String uploadWithoutDate(String baseDir, MultipartFile file) throws IOException {
  91.         try {
  92.             int fileNamelength = file.getOriginalFilename().length();
  93.             if (fileNamelength > com.goertek.framework.utils.file.FileUploadUtils.DEFAULT_FILE_NAME_LENGTH) {
  94.                 throw new FileNameLengthLimitExceededException(com.goertek.framework.utils.file.FileUploadUtils.DEFAULT_FILE_NAME_LENGTH);
  95.             }
  96.  
  97.             // assertAllowed(file, allowedExtension);
  98.  
  99.             String fileName =  file.getOriginalFilename();
  100.             File desc = getAbsoluteFile(baseDir, fileName);
  101.             file.transferTo(desc);
  102.             String pathFileName = getPathFileName(baseDir, fileName);
  103.             return pathFileName;
  104.         } catch (Exception e) {
  105.             throw new IOException(e.getMessage(), e);
  106.         }
  107.     }
  108.  
  109.  
  110.     /**
  111.      * 文件上传
  112.      *
  113.      * @param baseDir 相对应用的基目录
  114.      * @param file    上传的文件
  115.      * @return 返回上传成功的文件名
  116.      * @throws FileSizeLimitExceededException       如果超出最大大小
  117.      * @throws FileNameLengthLimitExceededException 文件名太长
  118.      * @throws IOException                          比如读写文件出错时
  119.      * @throws InvalidExtensionException            文件校验异常
  120.      */
  121.     public static final String upload(String baseDir, MultipartFile file, String[] allowedExtension, String fileName)
  122.             throws FileSizeLimitExceededException, IOException, FileNameLengthLimitExceededException,
  123.             InvalidExtensionException {
  124.         int fileNamelength = file.getOriginalFilename().length();
  125.         if (fileNamelength > CadreFileUploadUtils.DEFAULT_FILE_NAME_LENGTH) {
  126.             throw new FileNameLengthLimitExceededException(CadreFileUploadUtils.DEFAULT_FILE_NAME_LENGTH);
  127.         }
  128.  
  129.  
  130.         //String fileName = extractFilename(file);
  131.         //String fileName = file.getOriginalFilename();
  132.  
  133.         File desc = getAbsoluteFile(baseDir, fileName);
  134.         file.transferTo(desc);
  135.         String pathFileName = getPathFileName(baseDir, fileName);
  136.         return pathFileName;
  137.     }
  138.  
  139.     /**
  140.      * 编码文件名
  141.      */
  142.     public static final String extractFilename(MultipartFile file) {
  143.         String fileName = file.getOriginalFilename();
  144.         String extension = getExtension(file);
  145.         fileName = DateUtils.datePath() + "/" + encodingFilename(fileName) + "." + extension;
  146.         return fileName;
  147.     }
  148.  
  149.  
  150.  
  151.     private static final File getAbsoluteFile(String uploadDir, String fileName) throws IOException {
  152.         File desc = new File(uploadDir + File.separator + fileName);
  153.  
  154.         if (!desc.getParentFile().exists()) {
  155.             desc.getParentFile().mkdirs();
  156.         }
  157.         if (!desc.exists()) {
  158.             desc.createNewFile();
  159.         }else{
  160.             desc.createNewFile();
  161.         }
  162.         return desc;
  163.     }
  164.  
  165.     private static final String getPathFileName(String uploadDir, String fileName) throws IOException {
  166.         int dirLastIndex = GoerTekConfig.getProfile().length() + 1;
  167.         String currentDir = StringUtils.substring(uploadDir, dirLastIndex);
  168.         String pathFileName = Constants.RESOURCE_PREFIX + "/" + currentDir + "/" + fileName;
  169.         return pathFileName;
  170.     }
  171.  
  172.     /**
  173.      * 编码文件名
  174.      */
  175.     private static final String encodingFilename(String fileName) {
  176.         fileName = fileName.replace("_", " ");
  177.         fileName = Md5Utils.hash(fileName + System.nanoTime() + counter++);
  178.         return fileName;
  179.     }
  180.  
  181.     /**
  182.      * 文件大小校验
  183.      *
  184.      * @param file 上传的文件
  185.      * @return
  186.      * @throws FileSizeLimitExceededException 如果超出最大大小
  187.      * @throws InvalidExtensionException
  188.      */
  189.     public static final void assertAllowed(MultipartFile file, String[] allowedExtension)
  190.             throws FileSizeLimitExceededException, InvalidExtensionException {
  191.         long size = file.getSize();
  192.  
  193.         String fileName = file.getOriginalFilename();
  194.         String extension = getExtension(file);
  195.         if (allowedExtension != null && !isAllowedExtension(extension, allowedExtension)) {
  196.             if (allowedExtension == MimeTypeUtils.IMAGE_EXTENSION) {
  197.                 throw new InvalidExtensionException.InvalidImageExtensionException(allowedExtension, extension,
  198.                         fileName);
  199.             } else if (allowedExtension == MimeTypeUtils.FLASH_EXTENSION) {
  200.                 throw new InvalidExtensionException.InvalidFlashExtensionException(allowedExtension, extension,
  201.                         fileName);
  202.             } else if (allowedExtension == MimeTypeUtils.MEDIA_EXTENSION) {
  203.                 throw new InvalidExtensionException.InvalidMediaExtensionException(allowedExtension, extension,
  204.                         fileName);
  205.             } else {
  206.                 throw new InvalidExtensionException(allowedExtension, extension, fileName);
  207.             }
  208.         }
  209.  
  210.     }
  211.  
  212.     /**
  213.      * 判断MIME类型是否是允许的MIME类型
  214.      *
  215.      * @param extension
  216.      * @param allowedExtension
  217.      * @return
  218.      */
  219.     public static final boolean isAllowedExtension(String extension, String[] allowedExtension) {
  220.         for (String str : allowedExtension) {
  221.             if (str.equalsIgnoreCase(extension)) {
  222.                 return true;
  223.             }
  224.         }
  225.         return false;
  226.     }
  227.  
  228.     /**
  229.      * 获取文件名的后缀
  230.      *
  231.      * @param file 表单文件
  232.      * @return 后缀名
  233.      */
  234.     public static final String getExtension(MultipartFile file) {
  235.         String extension = FilenameUtils.getExtension(file.getOriginalFilename());
  236.         if (StringUtils.isEmpty(extension)) {
  237.             extension = MimeTypeUtils.getExtension(file.getContentType());
  238.         }
  239.         return extension;
  240.     }
  241.  
  242.  
  243.     public static String getUploadFile(String url) {
  244.         if (!url.startsWith("/")) {
  245.             url = "/" + url;
  246.         }
  247.         if (url.startsWith(Constants.RESOURCE_PREFIX)) {
  248.             url = url.substring(Constants.RESOURCE_PREFIX.length());
  249.         }
  250.         return GoerTekConfig.getProfile() + "/" + url;
  251.     }
  252.  
  253.     /**
  254.      * 解压缩ZIP文件
  255.      * @param zipFile ZIP文件
  256.      * @param destDir 目标路径
  257.      */
  258.     public static void zipDecompress(File zipFile, File destDir) {
  259.         byte[] buffer = new byte[1024];
  260.         try (ZipInputStream zis = new ZipInputStream(new FileInputStream(zipFile))) {
  261.             ZipEntry entry = zis.getNextEntry();
  262.             while (entry != null) {
  263.                 File file = new File(destDir, entry.getName());
  264.                 if (entry.isDirectory()) {
  265.                     file.mkdirs();
  266.                 } else {
  267.                     File parent = file.getParentFile();
  268.                     if (!parent.exists()) {
  269.                         parent.mkdirs();
  270.                     }
  271.                     try (FileOutputStream fos = new FileOutputStream(file)) {
  272.                         int len;
  273.                         while ((len = zis.read(buffer)) > 0) {
  274.                             fos.write(buffer, 0, len);
  275.                         }
  276.                     }
  277.                 }
  278.                 entry = zis.getNextEntry();
  279.             }
  280.         } catch (IOException e) {
  281.             e.printStackTrace();
  282.         }
  283.     }
  284.  
  285. }

本文内容由网友自发贡献，转载请注明出处：https://www.wpsshop.cn/w/空白诗007/article/detail/805641
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号