|
@@ -1,6 +1,7 @@
|
|
|
package cn.com.qmth.examcloud.tool.controller;
|
|
|
|
|
|
import cn.com.qmth.examcloud.tool.config.Constants;
|
|
|
+import cn.com.qmth.examcloud.tool.vo.user.Role;
|
|
|
import cn.com.qmth.examcloud.tool.vo.user.User;
|
|
|
import org.slf4j.Logger;
|
|
|
import org.slf4j.LoggerFactory;
|
|
@@ -31,6 +32,18 @@ public class BaseController {
|
|
|
return (User) currentSession().getAttribute(Constants.LOGIN_USER);
|
|
|
}
|
|
|
|
|
|
+ protected boolean isSuperAdmin() {
|
|
|
+ User user = currentLoginUser();
|
|
|
+ if (user != null) {
|
|
|
+ for (Role role : user.getRoleList()) {
|
|
|
+ if ("SUPER_ADMIN".equals(role.getRoleCode())) {
|
|
|
+ return true;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+
|
|
|
protected void download(HttpServletResponse response, byte[] bytes, String title) {
|
|
|
try (OutputStream out = response.getOutputStream();) {
|
|
|
response.reset();
|
|
@@ -49,9 +62,7 @@ public class BaseController {
|
|
|
return;
|
|
|
}
|
|
|
|
|
|
- try (FileInputStream fis = new FileInputStream(file);
|
|
|
- OutputStream out = response.getOutputStream();
|
|
|
- ) {
|
|
|
+ try (FileInputStream fis = new FileInputStream(file); OutputStream out = response.getOutputStream();) {
|
|
|
response.reset();
|
|
|
response.setHeader("Accept-Length", String.valueOf(fis.available()));
|
|
|
response.setContentType("application/octet-stream;charset=UTF-8");
|