代码优化

This commit is contained in:
thinkgem
2024-07-26 11:04:34 +08:00
parent 75e5e2765c
commit 1c539f3796
5 changed files with 15 additions and 22 deletions

View File

@@ -9,7 +9,6 @@ import com.jeesite.common.collect.MapUtils;
import com.jeesite.common.config.Global; import com.jeesite.common.config.Global;
import com.jeesite.common.idgen.IdGen; import com.jeesite.common.idgen.IdGen;
import com.jeesite.common.lang.StringUtils; import com.jeesite.common.lang.StringUtils;
import com.jeesite.common.lang.TimeUtils;
import com.jeesite.common.web.BaseController; import com.jeesite.common.web.BaseController;
import com.jeesite.modules.cms.entity.Article; import com.jeesite.modules.cms.entity.Article;
import com.jeesite.modules.cms.entity.Category; import com.jeesite.modules.cms.entity.Category;
@@ -19,7 +18,6 @@ import com.jeesite.modules.cms.service.FileTempleteService;
import com.jeesite.modules.cms.utils.CmsUtils; import com.jeesite.modules.cms.utils.CmsUtils;
import com.jeesite.modules.sys.entity.Office; import com.jeesite.modules.sys.entity.Office;
import com.jeesite.modules.sys.utils.DictUtils; import com.jeesite.modules.sys.utils.DictUtils;
import com.jeesite.modules.sys.utils.UserUtils;
import org.apache.shiro.authz.annotation.Logical; import org.apache.shiro.authz.annotation.Logical;
import org.apache.shiro.authz.annotation.RequiresPermissions; import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
@@ -325,7 +323,7 @@ public class CategoryController extends BaseController {
@RequestMapping(value = "fixTreeData") @RequestMapping(value = "fixTreeData")
@ResponseBody @ResponseBody
public String fixTreeData(Category category) { public String fixTreeData(Category category) {
if (!UserUtils.getUser().isAdmin()) { if (!category.currentUser().isAdmin()) {
return renderResult(Global.FALSE, "操作失败,只有管理员才能进行修复!"); return renderResult(Global.FALSE, "操作失败,只有管理员才能进行修复!");
} }
categoryService.fixTreeData(); categoryService.fixTreeData();

View File

@@ -12,7 +12,6 @@ import com.jeesite.common.lang.StringUtils;
import com.jeesite.common.web.BaseController; import com.jeesite.common.web.BaseController;
import com.jeesite.modules.biz.entity.BizCategory; import com.jeesite.modules.biz.entity.BizCategory;
import com.jeesite.modules.biz.service.BizCategoryService; import com.jeesite.modules.biz.service.BizCategoryService;
import com.jeesite.modules.sys.utils.UserUtils;
import io.swagger.annotations.Api; import io.swagger.annotations.Api;
import org.apache.shiro.authz.annotation.RequiresPermissions; import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
@@ -236,7 +235,7 @@ public class BizCategoryController extends BaseController {
@RequestMapping(value = "fixTreeData") @RequestMapping(value = "fixTreeData")
@ResponseBody @ResponseBody
public String fixTreeData(BizCategory bizCategory){ public String fixTreeData(BizCategory bizCategory){
if (!UserUtils.getUser().isAdmin()){ if (!bizCategory.currentUser().isAdmin()){
return renderResult(Global.FALSE, "操作失败,只有管理员才能进行修复!"); return renderResult(Global.FALSE, "操作失败,只有管理员才能进行修复!");
} }
bizCategoryService.fixTreeData(); bizCategoryService.fixTreeData();

View File

@@ -35,7 +35,6 @@ import com.jeesite.common.collect.ListUtils;
import com.jeesite.common.collect.MapUtils; import com.jeesite.common.collect.MapUtils;
import com.jeesite.common.lang.StringUtils; import com.jeesite.common.lang.StringUtils;
import com.jeesite.common.idgen.IdGen; import com.jeesite.common.idgen.IdGen;
import com.jeesite.modules.sys.utils.UserUtils;
<% }else{ %> <% }else{ %>
import com.jeesite.common.entity.Page; import com.jeesite.common.entity.Page;
<% } %> <% } %>
@@ -368,7 +367,7 @@ public class ${ClassName}Controller extends BaseController {
@RequestMapping(value = "fixTreeData") @RequestMapping(value = "fixTreeData")
@ResponseBody @ResponseBody
public String fixTreeData(${ClassName} ${className}){ public String fixTreeData(${ClassName} ${className}){
if (!UserUtils.getUser().isAdmin()){ if (!${className}.currentUser().isAdmin()){
return renderResult(Global.FALSE, "操作失败,只有管理员才能进行修复!"); return renderResult(Global.FALSE, "操作失败,只有管理员才能进行修复!");
} }
${className}Service.fixTreeData(); ${className}Service.fixTreeData();

View File

@@ -35,7 +35,6 @@ import com.jeesite.common.collect.ListUtils;
import com.jeesite.common.collect.MapUtils; import com.jeesite.common.collect.MapUtils;
import com.jeesite.common.lang.StringUtils; import com.jeesite.common.lang.StringUtils;
import com.jeesite.common.idgen.IdGen; import com.jeesite.common.idgen.IdGen;
import com.jeesite.modules.sys.utils.UserUtils;
<% }else{ %> <% }else{ %>
import com.jeesite.common.entity.Page; import com.jeesite.common.entity.Page;
<% } %> <% } %>
@@ -370,7 +369,7 @@ public class ${ClassName}Controller extends BaseController {
@RequestMapping(value = "fixTreeData") @RequestMapping(value = "fixTreeData")
@ResponseBody @ResponseBody
public String fixTreeData(${ClassName} ${className}){ public String fixTreeData(${ClassName} ${className}){
if (!UserUtils.getUser().isAdmin()){ if (!${className}.currentUser().isAdmin()){
return renderResult(Global.FALSE, "操作失败,只有管理员才能进行修复!"); return renderResult(Global.FALSE, "操作失败,只有管理员才能进行修复!");
} }
${className}Service.fixTreeData(); ${className}Service.fixTreeData();

View File

@@ -4,9 +4,14 @@
*/ */
package com.jeesite.modules.test.web; package com.jeesite.modules.test.web;
import java.util.List; import com.jeesite.common.collect.ListUtils;
import java.util.Map; import com.jeesite.common.collect.MapUtils;
import com.jeesite.common.config.Global;
import com.jeesite.common.idgen.IdGen;
import com.jeesite.common.lang.StringUtils;
import com.jeesite.common.web.BaseController;
import com.jeesite.modules.test.entity.TestTree;
import com.jeesite.modules.test.service.TestTreeService;
import org.apache.shiro.authz.annotation.RequiresPermissions; import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller; import org.springframework.stereotype.Controller;
@@ -17,15 +22,8 @@ import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody; import org.springframework.web.bind.annotation.ResponseBody;
import com.jeesite.common.config.Global; import java.util.List;
import com.jeesite.common.collect.ListUtils; import java.util.Map;
import com.jeesite.common.collect.MapUtils;
import com.jeesite.common.lang.StringUtils;
import com.jeesite.common.idgen.IdGen;
import com.jeesite.modules.sys.utils.UserUtils;
import com.jeesite.common.web.BaseController;
import com.jeesite.modules.test.entity.TestTree;
import com.jeesite.modules.test.service.TestTreeService;
/** /**
* 测试树表Controller * 测试树表Controller
@@ -214,7 +212,7 @@ public class TestTreeController extends BaseController {
@RequestMapping(value = "fixTreeData") @RequestMapping(value = "fixTreeData")
@ResponseBody @ResponseBody
public String fixTreeData(TestTree testTree){ public String fixTreeData(TestTree testTree){
if (!UserUtils.getUser().isAdmin()){ if (!testTree.currentUser().isAdmin()){
return renderResult(Global.FALSE, "操作失败,只有管理员才能进行修复!"); return renderResult(Global.FALSE, "操作失败,只有管理员才能进行修复!");
} }
testTreeService.fixTreeData(); testTreeService.fixTreeData();