|  | @@ -190,41 +190,66 @@ public class UmsMemberServiceImpl implements UmsMemberService {
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      @Autowired
 | 
	
		
			
				|  |  |      private UmsCompanyShareMapper umsCompanyShareMapper;
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |      @Override
 | 
	
		
			
				|  |  | -    public Map<String,Object> getCurrentWxUser() {
 | 
	
		
			
				|  |  | +    public Map<String, Object> getCurrentWxUser() {
 | 
	
		
			
				|  |  |          SecurityContext ctx = SecurityContextHolder.getContext();
 | 
	
		
			
				|  |  |          Authentication auth = ctx.getAuthentication();
 | 
	
		
			
				|  |  |          MemberDetails memberDetails = (MemberDetails) auth.getPrincipal();
 | 
	
		
			
				|  |  |          UmsMember umsMember = memberDetails.getUmsMember();
 | 
	
		
			
				|  |  | -        Map<String,Object> map = new HashMap<>();
 | 
	
		
			
				|  |  | +        Map<String, Object> map = new HashMap<>();
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -        if (umsMember!=null){
 | 
	
		
			
				|  |  | -            map.put("user",umsMember);
 | 
	
		
			
				|  |  | +        if (umsMember != null) {
 | 
	
		
			
				|  |  | +            map.put("user", umsMember);
 | 
	
		
			
				|  |  |              UmsMemberPlatformExample example = new UmsMemberPlatformExample();
 | 
	
		
			
				|  |  |              UmsMemberPlatformExample.Criteria criteria = example.createCriteria();
 | 
	
		
			
				|  |  |              criteria.andMemberIdEqualTo(umsMember.getId());
 | 
	
		
			
				|  |  |              List<UmsMemberPlatform> platforms = umsMemberPlatformMapper.selectByExample(example);
 | 
	
		
			
				|  |  | -            if (platforms==null||platforms.size()<1){
 | 
	
		
			
				|  |  | +            if (platforms == null || platforms.size() < 1) {
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -                throw  new RuntimeException("登录失败");
 | 
	
		
			
				|  |  | +                throw new RuntimeException("登录失败");
 | 
	
		
			
				|  |  |              }
 | 
	
		
			
				|  |  |              UmsMemberPlatform umsMemberPlatform = platforms.get(0);
 | 
	
		
			
				|  |  | -            map.put("wxUsers",umsMemberPlatform);
 | 
	
		
			
				|  |  | -            Map<String, Object> ownData = umsMemberDao.getOwnData(umsMember.getId(),umsMemberPlatform.getId());
 | 
	
		
			
				|  |  | -            map.put("pre",ownData);
 | 
	
		
			
				|  |  | +            map.put("wxUsers", umsMemberPlatform);
 | 
	
		
			
				|  |  | +            Map<String, Object> ownData = umsMemberDao.getOwnData(umsMember.getId(), umsMemberPlatform.getId());
 | 
	
		
			
				|  |  | +            map.put("pre", ownData);
 | 
	
		
			
				|  |  |              getCompany(umsMember, map);
 | 
	
		
			
				|  |  |          }
 | 
	
		
			
				|  |  |          return map;
 | 
	
		
			
				|  |  |      }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | +    @Override
 | 
	
		
			
				|  |  | +    public UmsMemberPlatform getWxUser() {
 | 
	
		
			
				|  |  | +        SecurityContext ctx = SecurityContextHolder.getContext();
 | 
	
		
			
				|  |  | +        Authentication auth = ctx.getAuthentication();
 | 
	
		
			
				|  |  | +        MemberDetails memberDetails = (MemberDetails) auth.getPrincipal();
 | 
	
		
			
				|  |  | +        UmsMember umsMember = memberDetails.getUmsMember();
 | 
	
		
			
				|  |  | +        Map<String, Object> map = new HashMap<>();
 | 
	
		
			
				|  |  | +        UmsMemberPlatform umsMemberPlatform = null;
 | 
	
		
			
				|  |  | +        if (umsMember != null) {
 | 
	
		
			
				|  |  | +            map.put("user", umsMember);
 | 
	
		
			
				|  |  | +            UmsMemberPlatformExample example = new UmsMemberPlatformExample();
 | 
	
		
			
				|  |  | +            UmsMemberPlatformExample.Criteria criteria = example.createCriteria();
 | 
	
		
			
				|  |  | +            criteria.andMemberIdEqualTo(umsMember.getId());
 | 
	
		
			
				|  |  | +            List<UmsMemberPlatform> platforms = umsMemberPlatformMapper.selectByExample(example);
 | 
	
		
			
				|  |  | +            if (platforms == null || platforms.size() < 1) {
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +                throw new RuntimeException("登录失败");
 | 
	
		
			
				|  |  | +            }
 | 
	
		
			
				|  |  | +            umsMemberPlatform = platforms.get(0);
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +        }
 | 
	
		
			
				|  |  | +        return umsMemberPlatform;
 | 
	
		
			
				|  |  | +    }
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  |      private void getCompany(UmsMember umsMember, Map<String, Object> map) {
 | 
	
		
			
				|  |  | -        UmsCompanyShareExample example=new UmsCompanyShareExample();
 | 
	
		
			
				|  |  | +        UmsCompanyShareExample example = new UmsCompanyShareExample();
 | 
	
		
			
				|  |  |          UmsCompanyShareExample.Criteria criteria = example.createCriteria();
 | 
	
		
			
				|  |  |          criteria.andMemberIdEqualTo(umsMember.getId());
 | 
	
		
			
				|  |  |          criteria.andStatusEqualTo(0);
 | 
	
		
			
				|  |  |          List<UmsCompanyShare> umsCompanyShares = umsCompanyShareMapper.selectByExample(example);
 | 
	
		
			
				|  |  | -        if(umsCompanyShares.size()>0){
 | 
	
		
			
				|  |  | -            map.put("company",umsCompanyShares.get(0));
 | 
	
		
			
				|  |  | +        if (umsCompanyShares.size() > 0) {
 | 
	
		
			
				|  |  | +            map.put("company", umsCompanyShares.get(0));
 | 
	
		
			
				|  |  |          }
 | 
	
		
			
				|  |  |      }
 | 
	
		
			
				|  |  |  
 | 
	
	
		
			
				|  | @@ -247,7 +272,6 @@ public class UmsMemberServiceImpl implements UmsMemberService {
 | 
	
		
			
				|  |  |      }
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  |      @Override
 | 
	
		
			
				|  |  |      public String login(String username, String password) {
 | 
	
		
			
				|  |  |          String token = null;
 | 
	
	
		
			
				|  | @@ -260,7 +284,7 @@ public class UmsMemberServiceImpl implements UmsMemberService {
 | 
	
		
			
				|  |  |              UsernamePasswordAuthenticationToken authentication = new UsernamePasswordAuthenticationToken(userDetails, null, userDetails.getAuthorities());
 | 
	
		
			
				|  |  |              SecurityContextHolder.getContext().setAuthentication(authentication);
 | 
	
		
			
				|  |  |              token = jwtTokenUtil.generateToken(userDetails);
 | 
	
		
			
				|  |  | -            memberCacheService.setToken(token,username);
 | 
	
		
			
				|  |  | +            memberCacheService.setToken(token, username);
 | 
	
		
			
				|  |  |          } catch (AuthenticationException e) {
 | 
	
		
			
				|  |  |              LOGGER.warn("登录异常:{}", e.getMessage());
 | 
	
		
			
				|  |  |          }
 | 
	
	
		
			
				|  | @@ -269,21 +293,21 @@ public class UmsMemberServiceImpl implements UmsMemberService {
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |      @Override
 | 
	
		
			
				|  |  |      public String loginPhone(String phone, String code) {
 | 
	
		
			
				|  |  | -        if (StringUtils.isEmpty(code)){
 | 
	
		
			
				|  |  | +        if (StringUtils.isEmpty(code)) {
 | 
	
		
			
				|  |  |              throw new RuntimeException("验证码错误");
 | 
	
		
			
				|  |  |          }
 | 
	
		
			
				|  |  |          String token = null;
 | 
	
		
			
				|  |  |          //密码需要客户端加密后传递
 | 
	
		
			
				|  |  |          try {
 | 
	
		
			
				|  |  | -            UserDetails userDetails =new MemberDetails(getByPhone(phone)) ;
 | 
	
		
			
				|  |  | +            UserDetails userDetails = new MemberDetails(getByPhone(phone));
 | 
	
		
			
				|  |  |              String loginCode = memberCacheService.getLoginCode(phone);
 | 
	
		
			
				|  |  | -            if (!code.equals(loginCode)){
 | 
	
		
			
				|  |  | +            if (!code.equals(loginCode)) {
 | 
	
		
			
				|  |  |                  throw new BadCredentialsException("验证码不正确");
 | 
	
		
			
				|  |  |              }
 | 
	
		
			
				|  |  |              UsernamePasswordAuthenticationToken authentication = new UsernamePasswordAuthenticationToken(userDetails, null, userDetails.getAuthorities());
 | 
	
		
			
				|  |  |              SecurityContextHolder.getContext().setAuthentication(authentication);
 | 
	
		
			
				|  |  |              token = jwtTokenUtil.generateToken(userDetails);
 | 
	
		
			
				|  |  | -            memberCacheService.setToken(token,userDetails.getUsername());
 | 
	
		
			
				|  |  | +            memberCacheService.setToken(token, userDetails.getUsername());
 | 
	
		
			
				|  |  |          } catch (AuthenticationException e) {
 | 
	
		
			
				|  |  |              LOGGER.warn("登录异常:{}", e.getMessage());
 | 
	
		
			
				|  |  |          }
 |