|
|
@@ -5,8 +5,11 @@ import com.ym.mec.auth.api.entity.SysUser;
|
|
|
import com.ym.mec.auth.service.SysUserService;
|
|
|
import com.ym.mec.common.controller.BaseController;
|
|
|
import com.ym.mec.common.entity.HttpResponseResult;
|
|
|
+import com.ym.mec.common.entity.ImResult;
|
|
|
+import com.ym.mec.common.entity.ImUserModel;
|
|
|
import com.ym.mec.common.security.AuthUser;
|
|
|
import com.ym.mec.common.security.SecurityUtils;
|
|
|
+import com.ym.mec.im.ImFeignService;
|
|
|
import io.swagger.annotations.Api;
|
|
|
import io.swagger.annotations.ApiImplicitParam;
|
|
|
import io.swagger.annotations.ApiImplicitParams;
|
|
|
@@ -25,6 +28,7 @@ import org.springframework.web.bind.annotation.RestController;
|
|
|
import org.springframework.web.client.RestTemplate;
|
|
|
import java.io.IOException;
|
|
|
import java.util.Base64;
|
|
|
+import java.util.Date;
|
|
|
import java.util.Map;
|
|
|
|
|
|
@RestController
|
|
|
@@ -42,6 +46,9 @@ public class TokenController extends BaseController {
|
|
|
|
|
|
@Autowired
|
|
|
private SysUserService userService;
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ private ImFeignService imFeignService;
|
|
|
|
|
|
@PostMapping(value = "/smsLogin",consumes = MediaType.APPLICATION_JSON_VALUE)
|
|
|
@ApiOperation(value = "短信验证码的方式登录")
|
|
|
@@ -57,6 +64,13 @@ public class TokenController extends BaseController {
|
|
|
public SysUser queryUserInfo() {
|
|
|
AuthUser authUser = SecurityUtils.getUser();
|
|
|
if(authUser != null){
|
|
|
+ SysUser sysUser = userService.get(authUser.getUserId());
|
|
|
+ if(StringUtils.isEmpty(sysUser.getImToken())){
|
|
|
+ ImResult register = imFeignService.register(new ImUserModel(sysUser.getId().toString(), sysUser.getUsername(), null));
|
|
|
+ sysUser.setImToken(register.getToken());
|
|
|
+ sysUser.setUpdateTime(new Date());
|
|
|
+ userService.update(sysUser);
|
|
|
+ }
|
|
|
return userService.get(authUser.getUserId());
|
|
|
}
|
|
|
return null;
|
zouxuan