feats

src/main/java/com/shigongli/controller/LoginController.java

@@ -1,18 +1,21 @@
 package com.shigongli.controller;
 
+import cn.binarywang.wx.miniapp.api.WxMaService;
+import cn.binarywang.wx.miniapp.bean.WxMaJscode2SessionResult;
 import com.shigongli.common.*;
+import com.shigongli.constants.StatusConstant;
 import com.shigongli.entity.SysUser;
+import com.shigongli.entity.TaPerson;
 import com.shigongli.service.ISysUserService;
+import com.shigongli.service.ITaPersonService;
 import me.chanjar.weixin.mp.api.WxMpService;
 import me.chanjar.weixin.mp.bean.result.WxMpOAuth2AccessToken;
 import me.chanjar.weixin.mp.bean.result.WxMpUser;
 import org.springframework.beans.factory.annotation.Autowired;
-import com.shigongli.vo.LoginParam;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import io.swagger.annotations.ApiParam;
 import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
 
@@ -27,6 +30,9 @@ public class LoginController extends BaseController {
     @Autowired
     ISysUserService iSysUserService;
 
+    @Autowired
+    ITaPersonService iTaPersonService;
+
     @Autowired
     WxUtils wxUtils;
 
@@ -38,7 +44,7 @@ public class LoginController extends BaseController {
      */
     @PostMapping("/mp/login")
     @ApiOperation(value="登录", notes = "登录", httpMethod = "POST", response = ResponseBean.class)
-    public ResponseBean login(@ApiParam("登录参数") @RequestParam("code") String code) throws Exception {
+    public ResponseBean mpLogin(@ApiParam("登录参数") @RequestParam("code") String code) throws Exception {
         WxMpService mpService = wxUtils.getMpService();
         WxMpOAuth2AccessToken wxMpOAuth2AccessToken = mpService.oauth2getAccessToken(code);
         WxMpUser wxMpUser = mpService.oauth2getUserInfo(wxMpOAuth2AccessToken, null);
@@ -63,4 +69,32 @@ public class LoginController extends BaseController {
 
         return ResponseBean.success(sysUser);
     }
+
+    /**
+     * 小程序登录
+     * @param code
+     * @return
+     * @throws Exception
+     */
+    @PostMapping("/ma/login")
+    @ApiOperation(value="登录", notes = "登录", httpMethod = "POST", response = ResponseBean.class)
+    public ResponseBean maLogin(@ApiParam("登录参数") @RequestParam("code") String code) throws Exception {
+        WxMaService maService = wxUtils.getMaService();
+        WxMaJscode2SessionResult sessionInfo = maService.getUserService().getSessionInfo(code);
+
+        TaPerson taPerson = iTaPersonService.getByOpenId(sessionInfo.getOpenid());
+        if (null == taPerson || StatusConstant.DELETE.equals(taPerson.getStatus())) {
+            taPerson = new TaPerson();
+            taPerson.setOpenid(sessionInfo.getOpenid());
+            iTaPersonService.save(taPerson);
+        }
+
+        // 生成 token
+        Map<String, Object> claims = new HashMap<>();
+        claims.put("personId", taPerson.getPersonId());
+        String jws = JWTUtils.encode(claims);
+        taPerson.setToken(jws);
+
+        return ResponseBean.success(taPerson);
+    }
 }