package com.dobbinsoft.fw.launcher.controller;
|
|
import cn.hutool.core.io.file.FileNameUtil;
|
import com.alibaba.fastjson.JSONObject;
|
import com.dobbinsoft.fw.core.Const;
|
import com.dobbinsoft.fw.core.entiy.inter.IdentityOwner;
|
import com.dobbinsoft.fw.core.entiy.inter.PermissionOwner;
|
import com.dobbinsoft.fw.core.util.GeneratorUtil;
|
import com.dobbinsoft.fw.launcher.inter.AfterFileUpload;
|
import com.dobbinsoft.fw.launcher.inter.BeforeFileUpload;
|
import com.dobbinsoft.fw.launcher.inter.BeforeFileUploadPath;
|
import com.dobbinsoft.fw.launcher.permission.IAdminAuthenticator;
|
import com.dobbinsoft.fw.launcher.permission.IUserAuthenticator;
|
import com.dobbinsoft.fw.support.storage.StorageClient;
|
import com.dobbinsoft.fw.support.storage.StoragePrivateResult;
|
import com.dobbinsoft.fw.support.storage.StorageRequest;
|
import com.dobbinsoft.fw.support.storage.StorageResult;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.data.redis.core.StringRedisTemplate;
|
import org.springframework.stereotype.Controller;
|
import org.springframework.util.FileCopyUtils;
|
import org.springframework.util.ObjectUtils;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.servlet.ServletException;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
/**
|
* Created by rize on 2019/2/13.
|
*/
|
@Controller
|
@RequestMapping("/upload")
|
public class FileUploadController {
|
|
@Autowired
|
private IUserAuthenticator userAuthenticator;
|
|
@Autowired
|
private StorageClient storageClient;
|
|
@Autowired
|
private IAdminAuthenticator adminAuthenticator;
|
|
@Autowired(required = false)
|
private AfterFileUpload afterFileUpload;
|
|
@Autowired(required = false)
|
private BeforeFileUpload beforeFileUpload;
|
|
@Autowired(required = false)
|
private BeforeFileUploadPath beforeFileUploadPath;
|
|
private static final Logger logger = LoggerFactory.getLogger(FileUploadController.class);
|
|
|
/**
|
* 后台通过服务器间接传文件
|
*
|
* @param file
|
* @return
|
* @throws IOException
|
*/
|
@PostMapping("/admin")
|
@ResponseBody
|
public Object createAdmin(@RequestParam("file") MultipartFile file, HttpServletRequest request, HttpServletResponse response) throws Exception {
|
if (this.beforeFileUpload != null) {
|
this.beforeFileUpload.before(request);
|
}
|
String accessToken = request.getHeader(Const.ADMIN_ACCESS_TOKEN);
|
PermissionOwner admin = adminAuthenticator.getAdmin(accessToken);
|
if (admin != null) {
|
return commonsUpload(file);
|
}
|
throw new RuntimeException("权限不足");
|
}
|
|
/**
|
* 后台通过服务器代理间接上传 私密文件（非公开URL）
|
*
|
* @param file
|
* @param request
|
* @param response
|
* @return
|
* @throws IOException
|
*/
|
@PostMapping("/admin/private")
|
@ResponseBody
|
public Object createAdminPrivate(@RequestParam("file") MultipartFile file, HttpServletRequest request, HttpServletResponse response) throws Exception {
|
if (this.beforeFileUpload != null) {
|
this.beforeFileUpload.before(request);
|
}
|
String accessToken = request.getHeader(Const.ADMIN_ACCESS_TOKEN);
|
PermissionOwner admin = adminAuthenticator.getAdmin(accessToken);
|
if (admin != null) {
|
InputStream inputStream = null;
|
Map<String, Object> data = new HashMap<>();
|
try {
|
String ext = FileNameUtil.getSuffix(file.getOriginalFilename());
|
String uuid = GeneratorUtil.genFileName();
|
StorageRequest storageRequest = new StorageRequest();
|
storageRequest.setContentType(file.getContentType());
|
storageRequest.setFilename(uuid + "." + ext);
|
storageRequest.setSize(file.getSize());
|
inputStream = file.getInputStream();
|
storageRequest.setIs(inputStream);
|
storageRequest.setPath("private");
|
StoragePrivateResult result = storageClient.savePrivate(storageRequest);
|
if (this.afterFileUpload != null) {
|
this.afterFileUpload.afterPrivate(storageRequest.getFilename(), result.getUrl(), result.getKey(), file.getSize());
|
}
|
if (result.isSuc()) {
|
data.put("key", result.getKey());
|
data.put("url", result.getUrl());
|
data.put("errno", 200);
|
data.put("errmsg", "成功");
|
return data;
|
}
|
} catch (IOException e) {
|
throw new RuntimeException("网络错误");
|
} finally {
|
if (inputStream != null) {
|
inputStream.close();
|
}
|
}
|
}
|
throw new RuntimeException("权限不足");
|
}
|
|
@PostMapping("/user")
|
@ResponseBody
|
public Object createUser(@RequestParam("file") MultipartFile file, HttpServletRequest request) throws Exception {
|
if (this.beforeFileUpload != null) {
|
this.beforeFileUpload.before(request);
|
}
|
String accessToken = request.getHeader(Const.USER_ACCESS_TOKEN);
|
IdentityOwner user = userAuthenticator.getUser(accessToken);
|
if (user != null) {
|
return commonsUpload(file);
|
}
|
throw new RuntimeException("权限不足");
|
}
|
|
private Object commonsUpload(MultipartFile file) throws IOException {
|
InputStream inputStream = null;
|
Map<String, Object> data = new HashMap<>();
|
try {
|
String ext = FileNameUtil.getSuffix(file.getOriginalFilename());
|
String uuid = GeneratorUtil.genFileName();
|
StorageRequest storageRequest = new StorageRequest();
|
storageRequest.setContentType(file.getContentType());
|
storageRequest.setFilename(uuid + "." + ext);
|
storageRequest.setSize(file.getSize());
|
inputStream = file.getInputStream();
|
storageRequest.setIs(inputStream);
|
if (this.beforeFileUploadPath != null) {
|
storageRequest.setPath(this.beforeFileUploadPath.setPath("commons"));
|
} else {
|
storageRequest.setPath("commons");
|
}
|
StorageResult result = storageClient.save(storageRequest);
|
if (afterFileUpload != null) {
|
afterFileUpload.afterPublic(storageRequest.getFilename(), result.getUrl(), file.getSize());
|
}
|
if (result.isSuc()) {
|
data.put("url", result.getUrl());
|
data.put("errno", 200);
|
data.put("errmsg", "成功");
|
}
|
} catch (IOException e) {
|
throw new RuntimeException("网络错误");
|
} finally {
|
if (inputStream != null) {
|
inputStream.close();
|
}
|
}
|
return data;
|
}
|
|
/**
|
* 上传文件到文件系统
|
*
|
* @param file
|
* @param fsf FileSystem File 上传覆盖具体某个文件
|
* @return
|
*/
|
@PostMapping("/local")
|
@ResponseBody
|
public Object local(@RequestParam("file") MultipartFile file, String fsf, HttpServletRequest request) throws IOException {
|
String accessToken = request.getHeader(Const.ADMIN_ACCESS_TOKEN);
|
IdentityOwner user = userAuthenticator.getUser(accessToken);
|
if (user != null) {
|
int i = fsf.lastIndexOf("/");
|
if (i > 0) {
|
String substring = fsf.substring(0, i);
|
File dir = new File(substring);
|
if (!dir.exists()) {
|
dir.mkdir();
|
}
|
File fsFile = new File(fsf);
|
if (fsFile.exists()) {
|
fsFile.delete();
|
}
|
InputStream fis = file.getInputStream();
|
OutputStream os = new FileOutputStream(fsf);
|
FileCopyUtils.copy(fis, os);
|
fis.close();
|
os.close();
|
Map<String, Object> data = new HashMap<>();
|
data.put("errno", 200);
|
data.put("errmsg", "成功");
|
return data;
|
} else {
|
throw new RuntimeException("文件系统路径不正确");
|
}
|
}
|
throw new RuntimeException("权限不足");
|
}
|
|
|
/**
|
* Post请求
|
*/
|
@RequestMapping(method = RequestMethod.POST)
|
public void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
|
Map<String, String[]> parameterMap = request.getParameterMap();
|
JSONObject responseJson = new JSONObject();
|
responseJson.put("code", 200);
|
for (String key : parameterMap.keySet()) {
|
responseJson.put(key, parameterMap.get(key)[0]);
|
}
|
response(request, response, responseJson.toJSONString(), HttpServletResponse.SC_OK);
|
}
|
|
/**
|
* 服务器响应结果
|
*
|
* @param request
|
* @param response
|
* @param results
|
* @param status
|
* @throws IOException
|
*/
|
private void response(HttpServletRequest request, HttpServletResponse response, String results, int status)
|
throws IOException {
|
String callbackFunName = request.getParameter("callback");
|
response.addHeader("Content-Length", String.valueOf(results.length()));
|
if (callbackFunName == null || callbackFunName.equalsIgnoreCase(""))
|
response.getWriter().println(results);
|
else
|
response.getWriter().println(callbackFunName + "( " + results + " )");
|
response.setStatus(status);
|
response.flushBuffer();
|
}
|
|
/**
|
* 服务器响应结果
|
*/
|
private void response(HttpServletRequest request, HttpServletResponse response, String results) throws IOException {
|
String callbackFunName = request.getParameter("callback");
|
if (callbackFunName == null || callbackFunName.equalsIgnoreCase(""))
|
response.getWriter().println(results);
|
else
|
response.getWriter().println(callbackFunName + "( " + results + " )");
|
response.setStatus(HttpServletResponse.SC_OK);
|
response.flushBuffer();
|
}
|
|
}
|
