From 64c9ab17a187d6087acbda902c01d14f5707922a Mon Sep 17 00:00:00 2001
From: houruijun <411269194@kidgrow.com>
Date: Sun, 28 Jun 2020 16:34:36 +0800
Subject: [PATCH] 1.调整扣费业务支持feign调用 2.更新运营平台支持统一格式化时间 3.修改zuu网关,增加请求头doctorId字段,修改departmentId字段的取值

---
 kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java |   54 +++++++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 45 insertions(+), 9 deletions(-)

diff --git a/kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java b/kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java
index 79a5510..20b0e16 100644
--- a/kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java
+++ b/kidgrow-springcloud/kidgrow-springcloud-zuul/src/main/java/com/kidgrow/zuul/filter/UserInfoHeaderFilter.java
@@ -1,18 +1,20 @@
 package com.kidgrow.zuul.filter;
 
-import cn.hutool.core.collection.CollectionUtil;
 import com.alibaba.fastjson.JSON;
 import com.kidgrow.common.constant.CommonConstant;
 import com.kidgrow.common.constant.SecurityConstants;
 import com.kidgrow.common.model.ResultBody;
+import com.kidgrow.common.model.SysDoctor;
 import com.kidgrow.common.model.SysOrganization;
 import com.kidgrow.common.model.SysUser;
 import com.kidgrow.common.utils.AddrUtil;
+import com.kidgrow.zuul.feign.SysDoctorService;
 import com.netflix.zuul.ZuulFilter;
 import com.netflix.zuul.context.RequestContext;
 import eu.bitwalker.useragentutils.UserAgent;
 import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.cloud.netflix.zuul.filters.support.FilterConstants;
 import org.springframework.security.authentication.AnonymousAuthenticationToken;
 import org.springframework.security.core.Authentication;
@@ -21,12 +23,12 @@
 import org.springframework.stereotype.Component;
 
 import javax.servlet.http.HttpServletRequest;
-import javax.servlet.http.HttpServletResponse;
 import java.net.URLEncoder;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 import static org.springframework.cloud.netflix.zuul.filters.support.FilterConstants.FORM_BODY_WRAPPER_FILTER_ORDER;
-import static org.springframework.cloud.netflix.zuul.filters.support.FilterConstants.PRE_DECORATION_FILTER_ORDER;
 
 /**
  * 石家庄喜高科技有限责任公司 版权所有 © Copyright 2020<br>
@@ -39,6 +41,8 @@
 @Slf4j
 @Component
 public class UserInfoHeaderFilter extends ZuulFilter {
+    @Autowired
+    private SysDoctorService sysDoctorService;
     @Override
     public String filterType() {
         return FilterConstants.PRE_TYPE;
@@ -54,6 +58,8 @@
         return true;
     }
 
+
+
     @SneakyThrows
     @Override
     public Object run() {
@@ -66,15 +72,45 @@
             String tenantId = "";
             //客户端模式只返回一个clientId
             if (principal instanceof SysUser) {
+
                 SysUser user = (SysUser) authentication.getPrincipal();
+                Map<String,Object> map=new HashMap<>();
+                map.put("user_id",user.getId());
+                ResultBody listByMap = sysDoctorService.getListByMap(map);
+
+                List<SysDoctor> sysDoctors = JSON.parseArray(JSON.toJSONString(listByMap.getData()), SysDoctor.class);
+                String str="";
+                Long hospital=-1L;
+                Long doctorId=-1L;
+                Long departmentId=-1L;
+                String departmentName="";
+                if(sysDoctors!=null&&sysDoctors.size()>0){
+                    str= sysDoctors.get(0).getDoctorName();
+                    hospital=sysDoctors.get(0).getHospitalId();
+                    doctorId=sysDoctors.get(0).getId();
+                    departmentId=sysDoctors.get(0).getDepartmentId();
+                    departmentName=sysDoctors.get(0).getDepartmentName();
+                }
                 ctx.addZuulRequestHeader(SecurityConstants.USER_ID_HEADER, String.valueOf(user.getId()));
+                ctx.addZuulRequestHeader(SecurityConstants.DOCTOR_ID_HEADER, String.valueOf(doctorId));
                 ctx.addZuulRequestHeader(SecurityConstants.USER_HEADER, user.getUsername());
-                ctx.addZuulRequestHeader(SecurityConstants.USER_REAL_NAME, "");//待完善
+
+                ctx.addZuulRequestHeader(SecurityConstants.USER_REAL_NAME,URLEncoder.encode(str,"UTF-8"));//待完善
+                ctx.addZuulRequestHeader(SecurityConstants.USER_HOSPITAL_ID_HEADER,hospital.toString());
+                tenantId=hospital.toString();
                 List<SysOrganization> organizations = (List<SysOrganization>) user.getOrganizations();
-                //如果没有有组织架构
                 //将组织数据 存到header 里面
-                ctx.addZuulRequestHeader(SecurityConstants.USER_ORGS_HEADER, JSON.toJSONString(organizations));
-//                log.error("用户ID写入Header成功,用户ID为:"+String.valueOf(user.getId()));
+                organizations.sort((e1,e2)->e1.getOrgLevel().compareTo(e2.getOrgLevel()));
+                if (organizations != null&&organizations.size()>0) {
+                    ctx.addZuulRequestHeader(SecurityConstants.USER_ORG_ID_HEADER,String.valueOf(organizations.get(0).getId()));
+                    ctx.addZuulRequestHeader(SecurityConstants.USER_ORG_NAME_HEADER, URLEncoder.encode(organizations.get(0).getOrgName(),"UTF-8"));
+                    ctx.addZuulRequestHeader(SecurityConstants.USER_DEP_ID_HEADER,String.valueOf(departmentId));
+                    ctx.addZuulRequestHeader(SecurityConstants.USER_DEP_NAME_HEADER,(departmentName.equals("")?URLEncoder.encode(organizations.get(organizations.size()-1).getOrgName(),"UTF-8"):departmentName));
+                    ctx.addZuulRequestHeader(SecurityConstants.USER_ORGS_HEADER,JSON.toJSONString(organizations));
+                }
+                //将角色放到header
+                ctx.addZuulRequestHeader(SecurityConstants.ROLE_HEADER, JSON.toJSONString(user.getRoles()));
+
             }
 
 
@@ -84,7 +120,7 @@
             //保存租户id,租户id根据业务尽进行替换
             switch (clientId) {
                 case "hospital":
-                    tenantId = tenantId;
+                    tenantId =  tenantId;
                     break;
                 case "webApp":
                     tenantId = "webApp";
@@ -101,7 +137,7 @@
 
             ctx.addZuulRequestHeader(SecurityConstants.TENANT_HEADER, tenantId);
             ctx.addZuulRequestHeader(SecurityConstants.CLIENT_HEADER, clientId);
-            ctx.addZuulRequestHeader(SecurityConstants.ROLE_HEADER, CollectionUtil.join(authentication.getAuthorities(), ","));
+
         }
         return null;
     }

--
Gitblit v1.8.0