public MiniappDTO getMiniappToken(String code) { MiniappDTO dto = null; try { //调用微信小程序接口,得到openid和sessionKey WxMaJscode2SessionResult result = getWxSessionInfo(code,"userMini"); String openid = result.getOpenid(); String sessionKey = result.getSessionKey(); log.info("openid:"+openid + ",sessionKey:"+sessionKey); //保存用户信息 User user = null; if(userInfoService.selectCountById(openid)) { user = User.builder().userId(openid).sessionKey(sessionKey).build(); log.info("用户登录并修改用户信息:"+JSONObject.toJSONString(user)); userInfoService.updateUserInfo(user); } else { user = User.builder().userId(openid).sessionKey(sessionKey).role(1).build(); log.info("用户登录并保存用户信息:"+JSONObject.toJSONString(user)); userInfoService.saveUserInfo(user); } // 将 openid 和 sessionKey 存入自定义的登录态 token 中并返回 String token = generateToken(openid, sessionKey); //查询用户信息 User userRole = userInfoService.selectUserByKey(openid); //返回token和用户角色 dto = MiniappDTO.builder() .roleType(2 == userRole.getRole() ? true : false) .token(token).msg("登录成功").build(); } catch (Exception e) { log.error("登录失败:" + e.getMessage(), e); dto = MiniappDTO.builder().msg("登录失败").build(); } return dto; }简化代码
时间: 2024-02-14 16:15:22 浏览: 27
下面是简化后的代码:
```
public MiniappDTO getMiniappToken(String code) {
MiniappDTO dto = null;
try {
//调用微信小程序接口,得到openid和sessionKey
WxMaJscode2SessionResult result = getWxSessionInfo(code,"userMini");
String openid = result.getOpenid();
String sessionKey = result.getSessionKey();
log.info("openid:"+openid + ",sessionKey:"+sessionKey);
//保存或更新用户信息
User user = User.builder().userId(openid).sessionKey(sessionKey).build();
if(userInfoService.selectCountById(openid)) {
log.info("用户登录并修改用户信息:"+JSONObject.toJSONString(user));
userInfoService.updateUserInfo(user);
} else {
user.setRole(1);
log.info("用户登录并保存用户信息:"+JSONObject.toJSONString(user));
userInfoService.saveUserInfo(user);
}
//生成token并返回
String token = generateToken(openid, sessionKey);
//查询用户信息
User userRole = userInfoService.selectUserByKey(openid);
//返回token和用户角色
dto = MiniappDTO.builder().token(token).roleType(userRole.getRole() == 2).msg("登录成功").build();
} catch (Exception e) {
log.error("登录失败:" + e.getMessage(), e);
dto = MiniappDTO.builder().msg("登录失败").build();
}
return dto;
}
```