fix: 修复开源版显示图片的问题
This commit is contained in:
parent
e071f33dbd
commit
723d5aa462
|
@ -5,14 +5,17 @@ import io.metersphere.commons.constants.UserSource;
|
||||||
import io.metersphere.commons.user.SessionUser;
|
import io.metersphere.commons.user.SessionUser;
|
||||||
import io.metersphere.commons.utils.SessionUtils;
|
import io.metersphere.commons.utils.SessionUtils;
|
||||||
import io.metersphere.controller.request.LoginRequest;
|
import io.metersphere.controller.request.LoginRequest;
|
||||||
|
import io.metersphere.service.BaseDisplayService;
|
||||||
import io.metersphere.service.UserService;
|
import io.metersphere.service.UserService;
|
||||||
import org.apache.commons.lang3.StringUtils;
|
import org.apache.commons.lang3.StringUtils;
|
||||||
import org.apache.shiro.SecurityUtils;
|
import org.apache.shiro.SecurityUtils;
|
||||||
import org.springframework.context.i18n.LocaleContextHolder;
|
import org.springframework.context.i18n.LocaleContextHolder;
|
||||||
import org.springframework.core.env.Environment;
|
import org.springframework.core.env.Environment;
|
||||||
|
import org.springframework.http.ResponseEntity;
|
||||||
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.bind.annotation.*;
|
||||||
|
|
||||||
import javax.annotation.Resource;
|
import javax.annotation.Resource;
|
||||||
|
import java.io.IOException;
|
||||||
|
|
||||||
@RestController
|
@RestController
|
||||||
@RequestMapping
|
@RequestMapping
|
||||||
|
@ -22,6 +25,8 @@ public class LoginController {
|
||||||
private UserService userService;
|
private UserService userService;
|
||||||
@Resource
|
@Resource
|
||||||
private Environment env;
|
private Environment env;
|
||||||
|
@Resource
|
||||||
|
private BaseDisplayService baseDisplayService;
|
||||||
|
|
||||||
@GetMapping(value = "/isLogin")
|
@GetMapping(value = "/isLogin")
|
||||||
public ResultHolder isLogin() {
|
public ResultHolder isLogin() {
|
||||||
|
@ -66,4 +71,9 @@ public class LoginController {
|
||||||
public String getDefaultLanguage() {
|
public String getDefaultLanguage() {
|
||||||
return userService.getDefaultLanguage();
|
return userService.getDefaultLanguage();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@GetMapping("display/file/{imageName}")
|
||||||
|
public ResponseEntity<byte[]> image(@PathVariable("imageName") String imageName) throws IOException {
|
||||||
|
return baseDisplayService.getImage(imageName);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,72 @@
|
||||||
|
package io.metersphere.service;
|
||||||
|
|
||||||
|
import io.metersphere.base.domain.SystemParameter;
|
||||||
|
import io.metersphere.base.domain.SystemParameterExample;
|
||||||
|
import io.metersphere.base.mapper.SystemParameterMapper;
|
||||||
|
import org.apache.commons.collections.CollectionUtils;
|
||||||
|
import org.apache.commons.io.IOUtils;
|
||||||
|
import org.apache.commons.lang3.StringUtils;
|
||||||
|
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
|
||||||
|
import org.springframework.http.HttpHeaders;
|
||||||
|
import org.springframework.http.MediaType;
|
||||||
|
import org.springframework.http.ResponseEntity;
|
||||||
|
import org.springframework.stereotype.Service;
|
||||||
|
|
||||||
|
import javax.annotation.Resource;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
@Service
|
||||||
|
public class BaseDisplayService {
|
||||||
|
@Resource
|
||||||
|
private SystemParameterMapper systemParameterMapper;
|
||||||
|
@Resource
|
||||||
|
private FileService fileService;
|
||||||
|
|
||||||
|
public List<SystemParameter> getParamList(String type) {
|
||||||
|
SystemParameterExample example = new SystemParameterExample();
|
||||||
|
example.createCriteria().andParamKeyLike(type + "%");
|
||||||
|
return systemParameterMapper.selectByExample(example);
|
||||||
|
}
|
||||||
|
|
||||||
|
public byte[] loadFileAsBytes(String fileId) {
|
||||||
|
return fileService.loadFileAsBytes(fileId);
|
||||||
|
}
|
||||||
|
|
||||||
|
public ResponseEntity<byte[]> getImage(String imageName) throws IOException {
|
||||||
|
byte[] bytes = null;
|
||||||
|
List<SystemParameter> paramList = getParamList("ui." + imageName);
|
||||||
|
if (!CollectionUtils.isEmpty(paramList)) {
|
||||||
|
SystemParameter sp = paramList.get(0);
|
||||||
|
String paramValue = sp.getParamValue();
|
||||||
|
if (StringUtils.isNotBlank(paramValue)) {
|
||||||
|
bytes = loadFileAsBytes(paramValue);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
MediaType contentType = MediaType.parseMediaType("application/octet-stream");
|
||||||
|
if (bytes == null) {
|
||||||
|
PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver(getClass().getClassLoader());
|
||||||
|
switch (imageName) {
|
||||||
|
case "logo":
|
||||||
|
bytes = IOUtils.toByteArray(resolver.getResources("/static/img/logo-light-MeterSphere.*.svg")[0].getInputStream());
|
||||||
|
contentType = MediaType.valueOf("image/svg+xml");
|
||||||
|
break;
|
||||||
|
case "loginImage":
|
||||||
|
bytes = IOUtils.toByteArray(resolver.getResources("/static/img/info.*.png")[0].getInputStream());
|
||||||
|
break;
|
||||||
|
case "loginLogo":
|
||||||
|
bytes = IOUtils.toByteArray(resolver.getResources("/static/img/logo-dark-MeterSphere.*.svg")[0].getInputStream());
|
||||||
|
contentType = MediaType.valueOf("image/svg+xml");
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return ResponseEntity.ok()
|
||||||
|
.contentType(contentType)
|
||||||
|
.header(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"" + imageName + "\"")
|
||||||
|
.body(bytes);
|
||||||
|
}
|
||||||
|
}
|
|
@ -1 +1 @@
|
||||||
Subproject commit 2ba1351aa0135cdf3e5de740fa2d9c185b60ce9d
|
Subproject commit f514243111c3acb317e80da23857143857a53566
|
Loading…
Reference in New Issue