BaseController.java 3.7 KB
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;
    }
}