From 203e2664c13883f8bb64f638cb1794c0b4480ec2 Mon Sep 17 00:00:00 2001 From: pang.jiabao <pang_jiabao@163.com> Date: 星期日, 27 四月 2025 08:24:20 +0800 Subject: [PATCH] 完成米多泉州wms开发 --- src/main/java/com/zy/common/web/BaseController.java | 54 +++++++++++++++++++++++++++++++++++++++++++++++++++++- 1 files changed, 53 insertions(+), 1 deletions(-) diff --git a/src/main/java/com/zy/common/web/BaseController.java b/src/main/java/com/zy/common/web/BaseController.java index 8c880d1..995c3d9 100644 --- a/src/main/java/com/zy/common/web/BaseController.java +++ b/src/main/java/com/zy/common/web/BaseController.java @@ -7,7 +7,13 @@ import com.core.common.Cools; import com.core.controller.AbstractBaseController; import com.core.exception.CoolException; +import com.zy.asrs.entity.Node; +import com.zy.asrs.entity.Tag; +import com.zy.asrs.service.NodeService; +import com.zy.asrs.service.TagService; import com.zy.system.entity.User; +import com.zy.system.entity.UserLogin; +import com.zy.system.service.UserLoginService; import com.zy.system.service.UserService; import io.swagger.annotations.ApiModelProperty; import org.springframework.beans.factory.annotation.Autowired; @@ -22,10 +28,45 @@ */ public class BaseController extends AbstractBaseController { + protected static final String RANGE_TIME_LINK = " - "; + @Autowired protected HttpServletRequest request; @Autowired private UserService userService; + @Autowired + private TagService tagService; + @Autowired + private NodeService nodeService; + @Autowired + private UserLoginService userLoginService; + + protected Node getOriginNode(){ + Node node = nodeService.getTop(); + if (node == null) { + throw new CoolException("搴撳尯鏁版嵁閿欒"); + } + return node; + } + + protected Long getHostId(){ + if (getUserId() == 9527) { + return null; + } + User user = getUser(); + if (user.getRoleId() == 1) { + String hostId = String.valueOf(request.getAttribute("hostId")); + if (Cools.isEmpty(hostId)) { + UserLogin userLogin = userLoginService.selectOne(new EntityWrapper<UserLogin>().eq("user_id", user.getId())); + if (userLogin != null) { + return userLogin.getHostId(); + } + } + return Long.parseLong(hostId); + } else { + return user.getHostId(); + } + } protected Long getUserId(){ return Long.parseLong(String.valueOf(request.getAttribute("userId"))); @@ -49,6 +90,17 @@ return ""; } + protected Tag getOriginTag(){ + return tagService.getTop(); + } + + protected <T> void hostEq(EntityWrapper<T> wrapper){ + Long hostId = getHostId(); + if (hostId != null) { + wrapper.eq("host_id", hostId); + } + } + /** * 鍒嗛〉缁勮 * @param pageNumber @@ -69,7 +121,7 @@ // 鍏ㄥ瓧娈垫ā绯婃悳绱� todo if (!Cools.isEmpty(map.get("condition"))) { Set<String> columns = new HashSet<>(); - for (Field field :Cools.getAllFields(cls)){ + for (Field field : Cools.getAllFields(cls)){ if (Modifier.isFinal(field.getModifiers()) || Modifier.isStatic(field.getModifiers()) || Modifier.isTransient(field.getModifiers())){ -- Gitblit v1.9.1