BaseController.java
3.7 KB
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package com.server.web.controller;
import com.server.utils.weixin.WeixinBaseUtil;
import com.server.utils.weixin.bean.AccessTokenBean;
import com.server.web.common.mapper.TKzyUserMapper;
import com.server.web.common.model.TKzyUser;
import com.server.web.common.model.TKzyUserExample;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;
import org.springframework.web.client.RestTemplate;
import javax.servlet.http.HttpServletRequest;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;
/**
 * Created by dell on 2018/8/27.
 */
public class BaseController {
    public final static String WX_NAMESPACE = "/rest";
    @Autowired
    RedisTemplate redisTemplate;
    @Autowired
    TKzyUserMapper userMapper;
    Logger logger = LoggerFactory.getLogger(getClass());
    public final static String WX_USER_SESSION_KEY = "weixin_user_session_key";
    /**
     * 获取appaccessToken
     * @param appId
     * @return
     */
    public String getAccessToken(String appId,String appSecret){
        if(StringUtils.hasText(appSecret)){
            String json ="";
            boolean redisAccessToken = true;
            if(redisTemplate.hasKey("wx_accessToken_"+appId)){
                json = (String)redisTemplate.opsForValue().get("wx_accessToken_"+appId);
                if(StringUtils.hasText(json)){
                    redisAccessToken =false;
                }
            }
            if(redisAccessToken){
                String url = WeixinBaseUtil.ACCESS_TOKEN_URL + "&appid=" + appId + "&secret=" + appSecret;
                RestTemplate restTemplate = new RestTemplate();
                json = restTemplate.getForObject(url, String.class);
                redisTemplate.opsForValue().set("wx_accessToken_"+appId,json,60*60, TimeUnit.SECONDS);
            }
            AccessTokenBean accessTokenBean = new AccessTokenBean(json);
            return accessTokenBean.getAccessToken();
        }else{
            return null;
        }
    }
    public Map<String,Object> returnSuccess(Object obj){
        Map<String,Object> returnResult = new HashMap<String,Object>();
        returnResult.put("status", "1");
        returnResult.put("message","操作成功");
        returnResult.put("data",obj);
        return returnResult;
    }
    public Map<String,Object> returnError(String code, String message ,Object obj){
        Map<String,Object> returnResult = new HashMap<String,Object>();
        returnResult.put("status", code);
        returnResult.put("message",message);
        returnResult.put("data",obj);
        return returnResult;
    }
    public TKzyUser getLoginUser(HttpServletRequest request){
        TKzyUser user;
        if(request.getSession().getAttribute(WX_USER_SESSION_KEY)!=null){
            user = (TKzyUser)request.getSession().getAttribute(WX_USER_SESSION_KEY);
            return user;
        }
        String userId = request.getParameter("userId");
        String wxOpenId = request.getParameter("wxOpenId");
        if(StringUtils.isEmpty(userId) || StringUtils.isEmpty(wxOpenId)){
           return null;
        }
        TKzyUserExample example = new TKzyUserExample();
        example.createCriteria().andWxOpenidEqualTo(wxOpenId).andIdEqualTo(Long.parseLong(userId));
        List<TKzyUser> users = userMapper.selectByExample(example);
        if (CollectionUtils.isEmpty(users)) {
            return null;
        }
        user = users.get(0);
        request.getSession().setAttribute(WX_USER_SESSION_KEY,user);
        return user;
    }
}