|
@@ -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());
|
|
|
}
|