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.core.domain.entity.SysUser;
import com.railway.common.enums.PostTypeEnum;
import com.railway.common.utils.SecurityUtils;

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

  private static JcebQueryParamVo getJcebQuery(){
    SysUser user = SecurityUtils.getLoginUser().getUser();
    JcebQueryParamVo jcebQuery = new JcebQueryParamVo();
    if(user.isAdmin()){
      jcebQuery.setSubmitState(JcebStatusEnum.REPORTED.getCode());
      jcebQuery.setCreateBy(null);
      return jcebQuery;
    }
    String post = user.getPost();
    PostTypeEnum postType = PostTypeEnum.ofValue(post);
    jcebQuery.setDeptId(user.getDeptId());
    if(postType == PostTypeEnum.LEADER){
      jcebQuery.setSubmitState(JcebStatusEnum.REPORTED.getCode());
      jcebQuery.setCreateBy(null);
    }else {
      jcebQuery.setCreateBy(user.getUserName());
    }
    return jcebQuery;
  }

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

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

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

}