edit

Fancy

2025-01-07 8000dd179e5a1eef3113cbd75dddb32bbb805d1c

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package com.dy.pmsWechat.login;
 
import com.alibaba.fastjson2.JSONObject;
import com.dy.pmsGlobal.daoMp.MpOpenIdMapper;
import com.dy.pmsGlobal.pojoMp.MpOpenId;
import com.dy.pmsWechat.util.JwtUtil;
import com.dy.pmsWechat.util.RestTemplateUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
 
import java.io.IOException;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
 
@Service
public class WeChatService {
 
    @Value("${wechat.appId}")
    private String appId;
 
    @Value("${wechat.appSecret}")
    private String appSecret;
 
    @Value("${wechat.loginUrl}")
    private String loginUrl;
 
    @Value("${wechat.jwt.secret-key}")
    private String secretKey;
 
    @Value("${wechat.jwt.ttl}")
    private  long  ttlMillis;
 
    private final RestTemplateUtil restTemplateUtil;
    private MpOpenIdMapper mpOpenIdDao;
 
    public WeChatService(RestTemplateUtil restTemplateUtil) {
        this.restTemplateUtil = restTemplateUtil;
    }
 
    @Autowired
    private void setMpOpenIdDao(MpOpenIdMapper mpOpenIdDao) {
        this.mpOpenIdDao = mpOpenIdDao;
    }
 
    public JSONObject login(String code) throws IOException {
        Map<String, Object> queryParams = new HashMap<>();
        queryParams.put("appid", appId);
        queryParams.put("secret", appSecret);
        queryParams.put("js_code", code);
        queryParams.put("grant_type", "authorization_code");
        Map<String, String> headerParams = new HashMap<>();
        JSONObject  wxMpUser = restTemplateUtil.get(loginUrl, queryParams, headerParams);
        String openId = wxMpUser.get("openid").toString();
        //获取微信信息添加到数据库  //先根据openid查询用户信息
        MpOpenId userInfo = mpOpenIdDao.getInfoByOpenId(openId);
        if(userInfo == null){ //数据库没有时添加到数据库中
            userInfo = new MpOpenId();
            userInfo.setOpenId(openId);
            userInfo.setCreateTime(new Date());
            mpOpenIdDao.insert(userInfo);
        }
        //授权完成之后，跳转到具体的功能页面
        //生成token,按照一定规则生成字符串，可以包含用户信息
        //登陆controller中生成，返回给客户端
        Map<String, Object> claims = new HashMap<>();
        claims.put("openId",openId);
        String token = JwtUtil.createJwt(secretKey,ttlMillis,claims);
        return new JSONObject().fluentPut("token",token);
    }
}