kimi
2020-03-13 8033eea1502d4cb0a91aa4b86c848ccb00ba8dc0
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
package com.highdatas.mdm.util;
 
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.highdatas.mdm.entity.TUser;
import com.highdatas.mdm.pojo.CodeMsg;
import com.highdatas.mdm.pojo.Result;
import com.highdatas.mdm.service.ITUserService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.StringUtils;
import org.springframework.web.servlet.HandlerInterceptor;
 
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.io.IOException;
import java.io.PrintWriter;
 
/**
 * @author kimi
 * @description
 * @date 2019-12-13 13:29
 */
 
@Slf4j
@Service(value = "testInterceptor")
public class CommonInterceptor implements HandlerInterceptor{
    @Autowired
    ITUserService userService;
 
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws IOException {
        String url = request.getRequestURI();
        log.info(url);
 
        String token = request.getHeader("token");
        String userId = request.getHeader("userId");
        if ("/".equalsIgnoreCase(url)) {
            return true;
        }
        if (StringUtils.isEmpty(userId)) {
            log.error("无登陆信息");
            return false;
        }
 
        token= "U001";
        if (StringUtils.isEmpty(token)) {
            log.error("token is null");
            token= "U001";
           /* response.setStatus(HttpServletResponse.SC_UNAUTHORIZED);
//            String url = "/industryiot/Login.html";
//            response.sendRedirect(url);
 
            response.setCharacterEncoding("UTF-8");
            response.setContentType("application/json; charset=utf-8");
            Result<Object> error = Result.error(CodeMsg.ERROR_TOKEN);
            PrintWriter out = null ;
            out = response.getWriter();
            out.write(JSON.toJSONString(error));
            out.flush();
            out.close();
            return false;*/
        }
        HttpSession session = request.getSession();
 
        String baseUrl = "";
        LinkedMultiValueMap<String, String> body=new LinkedMultiValueMap();
 
        body.add("token", token);
        //String userId = token;
//        ObjectMapper objectMapper = new ObjectMapper();
//        objectMapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
//        TUser user = objectMapper.readValue(login, TUser.class);
//        String userId = HttpUtils.HttpRestClient(baseUrl, HttpMethod.GET, body);
        TUser user = userService.selectOne(new EntityWrapper<TUser>().eq("user_id", userId));
        if (user == null) {
            log.error("user not found by : " + userId);
            response.setStatus(HttpServletResponse.SC_UNAUTHORIZED);
//            String url = "/industryiot/Login.html";
//            response.sendRedirect(url);
 
            response.setCharacterEncoding("UTF-8");
            response.setContentType("application/json; charset=utf-8");
            Result<Object> error = Result.error(CodeMsg.ERROR_TOKEN);
            PrintWriter out = null ;
            out = response.getWriter();
            out.write(JSON.toJSONString(error));
            out.flush();
            out.close();
 
            return false;
        }
 
        session.setAttribute("user", user);
        return true;
    }
}