railway-prod2/railway-business/src/main/java/com/railway/business/catenary/util/JcebQueryUtil.java

package com.railway.business.catenary.util;

import com.railway.business.catenary.domain.BusJcab;
import com.railway.business.catenary.domain.BusJceb;
import com.railway.business.catenary.domain.vo.JcebQueryParamVo;
import com.railway.business.catenary.domain.vo.JcebQueryVo;
import com.railway.business.catenary.enums.JcebStatusEnum;
import com.railway.common.constant.Constants;
import com.railway.common.core.domain.entity.SysRole;
import com.railway.common.core.domain.entity.SysUser;
import com.railway.common.core.redis.RedisCache;
import com.railway.common.enums.PostTypeEnum;
import com.railway.common.utils.SecurityUtils;
import com.railway.common.utils.StringUtils;
import com.railway.common.utils.spring.SpringUtils;
import com.railway.framework.aspectj.DataScopeAspect;
import java.util.List;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;

/**
 * @author ZhaoMn
 */
@Slf4j
public class JcebQueryUtil {

  private static JcebQueryParamVo getJcebQuery(String submitState){
    JcebQueryParamVo jcebQuery = new JcebQueryParamVo();
    jcebQuery.setSubmitState(submitState);
    SysUser user = SecurityUtils.getLoginUser().getUser();
    if(JcebStatusEnum.UNREPORTED.getCode().equals(submitState)){
      jcebQuery.setCreateBy(String.valueOf(user.getUserId()));
    }else {
      jcebQuery.setCreateBy(null);
    }
    // 管理员全部数据
    if(user.isAdmin()){
      log.debug("^_^ 管理员设置为查询所有 ^_^");
      jcebQuery.setPostType(PostTypeEnum.CENTER.getCode());
      return jcebQuery;
    }
    // 角色数据权限
    String scope = DataScopeAspect.DATA_SCOPE_SELF;
    PostTypeEnum postType = PostTypeEnum.STAFF;
    List<SysRole> roles = user.getRoles();
    if(CollectionUtils.isNotEmpty(roles)){
      for(SysRole role : roles){
        PostTypeEnum tmpPostType = PostTypeEnum.ofRoleName(role.getRoleName());
        if(PostTypeEnum.isCateNary(tmpPostType) && role.getDataScope().compareTo(scope) < 0){
          scope = role.getDataScope();
          postType = tmpPostType;
        }
      }
    }
    // 职位
    log.debug("^_^ postType = {} ^_^", postType);
    if(PostTypeEnum.CENTER == postType) {
      log.debug("^_^ 中心领导根据选择的车间查询 ^_^");
      // jcebQuery.setDeptId(getDeptIdCache(user.getUserId()))
      jcebQuery.setPostType(PostTypeEnum.CENTER.getCode());
    }else if(PostTypeEnum.LEADER == postType || scope.equals(DataScopeAspect.DATA_SCOPE_DEPT)
        || scope.equals(DataScopeAspect.DATA_SCOPE_DEPT_AND_CHILD)){
      log.debug("^_^ 车间领导设置为查询已提交表 ^_^");
      jcebQuery.setDeptId(user.getDeptId());
      jcebQuery.setPostType(PostTypeEnum.LEADER.getCode());
    }else {
      log.debug("^_^ 车间职工只能查看自己的 ^_^");
      jcebQuery.setCreateBy(String.valueOf(user.getUserId()));
      jcebQuery.setPostType(PostTypeEnum.STAFF.getCode());
    }
    return jcebQuery;
  }

  public static void setBusJceb(BusJceb jceb){
    JcebQueryParamVo jcebQuery = getJcebQuery(jceb.getSubmitState());
    jceb.setSubmitState(jcebQuery.getSubmitState());
    jceb.setDeptId(jcebQuery.getDeptId());
    jceb.setCreateBy(jcebQuery.getCreateBy());
  }

  public static void setBusJcab(BusJcab jcab){
    JcebQueryParamVo jcebQuery = getJcebQuery(jcab.getSubmitState());
    jcab.setSubmitState(jcebQuery.getSubmitState());
    jcab.setDeptId(jcebQuery.getDeptId());
    jcab.setCreateBy(jcebQuery.getCreateBy());
  }

  public static void setBusJcebQuery(JcebQueryVo query){
    JcebQueryParamVo jcebQuery = getJcebQuery(query.getSubmitState());
    query.setSubmitState(jcebQuery.getSubmitState());
    query.setDeptId(jcebQuery.getDeptId());
    query.setCreateBy(jcebQuery.getCreateBy());
  }

  /**
   * 获取字典缓存
   *
   * @param userId 参数键
   * @return dictDatas 字典数据列表
   */
  public static Long getDeptIdCache(Long userId) {
    RedisCache redisCache = SpringUtils.getBean(RedisCache.class);
    // 从缓存中取本个所选车间
    Long deptId = redisCache.getCacheObject(getCacheKey(userId));
    if(StringUtils.isEmpty(deptId)) {
      // 取默认选择的车间
      // deptId = redisCache.getCacheObject(Constants.LEADER_VIEW_DEPT_INIT_KEY)
    }
    return deptId;
  }

  /**
   * 设置cache key
   *
   * @param configKey 参数键
   * @return 缓存键key
   */
  public static String getCacheKey(Long configKey) {
    return Constants.LEADER_VIEW_DEPT_KEY + configKey;
  }

}