package org.ruoyi.system.controller.system;
|
|
import cn.dev33.satoken.annotation.SaIgnore;
|
import lombok.RequiredArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.ruoyi.common.core.exception.ServiceException;
|
import org.ruoyi.common.wechat.itchat4j.api.WechatTools;
|
import org.ruoyi.common.wechat.itchat4j.controller.LoginController;
|
import org.ruoyi.common.wechat.itchat4j.core.MsgCenter;
|
import org.ruoyi.common.wechat.itchat4j.face.IMsgHandlerFace;
|
import org.ruoyi.common.wechat.web.base.BaseException;
|
import org.ruoyi.system.domain.bo.WxRobConfigBo;
|
import org.ruoyi.system.domain.vo.WxRobConfigVo;
|
import org.ruoyi.system.handler.MyMsgHandler;
|
import org.ruoyi.system.service.ISseService;
|
import org.ruoyi.system.service.IWxRobConfigService;
|
import org.springframework.validation.annotation.Validated;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestParam;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import java.util.List;
|
|
/**
|
* 个人微信扩展控制器
|
*
|
* @author WangLe
|
*/
|
@SaIgnore
|
@Slf4j
|
@Validated
|
@RequiredArgsConstructor
|
@RestController
|
public class WeChatController {
|
|
private final ISseService sseService;
|
|
private final IWxRobConfigService wxRobConfigService;
|
|
/**
|
* 登录第一步，获取二维码链接
|
* @throws BaseException
|
*/
|
@PostMapping("/getQr")
|
public String getQr(@RequestParam String uniqueKey) {
|
LoginController login = new LoginController(uniqueKey);
|
try {
|
return login.login_1();
|
} catch (BaseException e) {
|
throw new ServiceException("获取二维码失败："+ e.getMessage());
|
}
|
}
|
|
@PostMapping("/wxLogin")
|
public Boolean wxLogin(@RequestParam String uniqueKey) {
|
LoginController login = new LoginController(uniqueKey);
|
return login.login_2();
|
}
|
|
@PostMapping("/wxInit")
|
public Boolean wxInit(@RequestParam String uniqueKey) {
|
LoginController login = new LoginController(uniqueKey);
|
// 开启消息处理线程
|
WxRobConfigBo wxRobConfigBo = new WxRobConfigBo();
|
wxRobConfigBo.setUniqueKey(uniqueKey);
|
List<WxRobConfigVo> wxRobConfigVos = wxRobConfigService.queryList(wxRobConfigBo);
|
//查询机器人对应的用户
|
start(uniqueKey,new MyMsgHandler(uniqueKey,sseService,wxRobConfigVos.get(0)));
|
return login.login_3();
|
}
|
|
@PostMapping("/wxLogout")
|
public void wxLogout(@RequestParam String uniqueKey) {
|
WechatTools.logout(uniqueKey);
|
}
|
|
public void start(String uniqueKey,IMsgHandlerFace msgHandler) {
|
log.info("7.+++开启消息处理线程["+uniqueKey+"]+++");
|
new Thread(() -> MsgCenter.handleMsg(uniqueKey,msgHandler)).start();
|
}
|
}
|
