From 98891d1d1bcaa414f336b9c05ad94d39fe9cb1bc Mon Sep 17 00:00:00 2001 From: Mila <Mila@pollex.com.tw> Date: 星期一, 29 十一月 2021 15:34:57 +0800 Subject: [PATCH] Merge branch 'master' of https://192.168.0.10:8443/r/pcalife/PAM --- pamapi/src/main/java/com/pollex/pam/security/SecurityUtils.java | 23 +++++++++++++++++++++++ 1 files changed, 23 insertions(+), 0 deletions(-) diff --git a/pamapi/src/main/java/com/pollex/pam/security/SecurityUtils.java b/pamapi/src/main/java/com/pollex/pam/security/SecurityUtils.java index 2e170c5..787c1f3 100644 --- a/pamapi/src/main/java/com/pollex/pam/security/SecurityUtils.java +++ b/pamapi/src/main/java/com/pollex/pam/security/SecurityUtils.java @@ -1,8 +1,12 @@ package com.pollex.pam.security; import java.util.Arrays; +import java.util.Map; import java.util.Optional; import java.util.stream.Stream; + +import com.pollex.pam.enums.ConsultantDetailEnum; +import com.pollex.pam.enums.CustomerDetailEnum; import org.springframework.security.core.Authentication; import org.springframework.security.core.GrantedAuthority; import org.springframework.security.core.context.SecurityContext; @@ -97,4 +101,23 @@ private static Stream<String> getAuthorities(Authentication authentication) { return authentication.getAuthorities().stream().map(GrantedAuthority::getAuthority); } + + public static String getAgentNo() { + Map<String, String> userDetails = getCurrentUserDetails(); + return userDetails.get(ConsultantDetailEnum.AGENT_NO.getValue()); + } + + public static Long getCustomerDBId() { + Map<String, String> userDetails = getCurrentUserDetails(); + return Long.parseLong(userDetails.get(CustomerDetailEnum.DB_ID.getValue())); + } + + public static Map<String, String> getCurrentUserDetails() { + SecurityContext securityContext = SecurityContextHolder.getContext(); + if(securityContext.getAuthentication()==null || + securityContext.getAuthentication().getDetails() ==null) { + return null; + } + return (Map<String, String>) securityContext.getAuthentication().getDetails(); + } } -- Gitblit v1.8.0