com.github.xuchen93.web.common.RequestContextBean Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of module-web Show documentation
There is a newer version: 1.1.3
package com.github.xuchen93.web.common;


import cn.hutool.core.util.StrUtil;
import cn.hutool.extra.servlet.ServletUtil;
import cn.hutool.extra.spring.SpringUtil;
import cn.hutool.http.useragent.UserAgentUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.github.xuchen93.core.config.XuchenProperties;
import lombok.Data;
import org.springframework.context.annotation.Scope;
import org.springframework.context.annotation.ScopedProxyMode;
import org.springframework.stereotype.Component;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;

@Data
@Component
@Scope(value = WebApplicationContext.SCOPE_REQUEST, proxyMode = ScopedProxyMode.TARGET_CLASS)
public class RequestContextBean {
	private JSONObject tokenObj;
	private String userId;
	private String userName;
	private String nickName;


	private String clientIp;
	private Map header;
	private String url;
	private String browser;
	private String hostIp;
	private String requestMethod;
	private String requestArgs;


	public RequestContextBean() {
		XuchenProperties properties = SpringUtil.getBean(XuchenProperties.class);
		if (properties.getRequest().isDetail() && RequestContextHolder.getRequestAttributes() != null) {
			ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
			HttpServletRequest request = attributes.getRequest();
			String userAgent = request.getHeader("User-Agent");
			String hostIp = null;
			try {
				hostIp = InetAddress.getLocalHost().getHostAddress();
			} catch (UnknownHostException e) {

			}
			setClientIp(ServletUtil.getClientIP(request));
			setBrowser(UserAgentUtil.parse(userAgent).getBrowser().toString());
			setHeader(getHeaderMap(request));
			setUrl(request.getRequestURI());
			setRequestArgs(JSONUtil.toJsonStr(request.getParameterMap()));
			setHostIp(hostIp);
			setRequestMethod(StrUtil.isBlank(request.getMethod()) ? "未知请求" : request.getMethod());
		}
	}

	private static Map getHeaderMap(HttpServletRequest request) {
		Enumeration names = request.getHeaderNames();
		Map headerMap = new HashMap<>();
		while (names.hasMoreElements()) {
			String name = names.nextElement();
			headerMap.put(name, request.getHeader(name));
		}
		return headerMap;
	}
}