快捷導(dǎo)航

解決子線程中獲取不到HttpServletRequest對(duì)象的問(wèn)題

更新時(shí)間：2024年07月08日 11:56:14 作者：沐雨橙風(fēng)ιε

這篇文章主要介紹了解決子線程中獲取不到HttpServletRequest對(duì)象的問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

子線程中獲取不到HttpServletRequest對(duì)象

本文主要分享一下項(xiàng)目里遇到的獲取request對(duì)象為null的問(wèn)題，具體是在登錄的時(shí)候觸發(fā)的郵箱提醒，獲取客戶端ip地址，然后通過(guò)ip地址定位獲取定位信息，從而提示賬號(hào)在哪里登錄。

但是登錄卻發(fā)現(xiàn)獲取request對(duì)象的時(shí)候報(bào)錯(cuò)了。

具體的代碼

如下：這個(gè)異常是自己手動(dòng)拋出的。

package cn.edu.sgu.www.mhxysy.util;
 
import cn.edu.sgu.www.mhxysy.consts.MimeType;
import cn.edu.sgu.www.mhxysy.exception.GlobalException;
import cn.edu.sgu.www.mhxysy.restful.JsonResult;
import cn.edu.sgu.www.mhxysy.restful.ResponseCode;
import com.alibaba.fastjson.JSON;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
 
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
 
/**
 * http工具類(lèi)
 * @author heyunlin
 * @version 1.0
 */
public class HttpUtils {
 
    /**
     * 獲取HttpServletRequest對(duì)象
     * @return HttpServletRequest
     */
    public static HttpServletRequest getRequest() {
        RequestAttributes attributes = RequestContextHolder.getRequestAttributes();
 
        if (attributes != null ) {
            return ((ServletRequestAttributes) attributes).getRequest();
        }
 
        throw new GlobalException(ResponseCode.ERROR, "獲取request對(duì)象失敗");
    }
 
}

在項(xiàng)目其他地方也有用這個(gè)工具了獲取HttpServletRequest對(duì)象，都能獲取到，覺(jué)得很是奇怪。

點(diǎn)進(jìn)去RequestContextHolder這個(gè)類(lèi)的代碼里看了一下，好像找到問(wèn)題了~

這是基于ThreadLocal實(shí)現(xiàn)的，可能與子線程無(wú)法訪問(wèn)父線程中設(shè)置的數(shù)據(jù)的問(wèn)題有關(guān)。

不會(huì)ThreadLocal的童鞋，通過(guò)下面文章簡(jiǎn)單了解一下ThreadLocal
子線程無(wú)法訪問(wèn)父線程中通過(guò)ThreadLocal設(shè)置的變量

為了驗(yàn)證自己的猜測(cè)，點(diǎn)開(kāi)RequestContextHolder的源代碼~

public abstract class RequestContextHolder  {    
    private static final ThreadLocal<RequestAttributes> requestAttributesHolder =
			new NamedThreadLocal<>("Request attributes");
 
	private static final ThreadLocal<RequestAttributes> inheritableRequestAttributesHolder =
			new NamedInheritableThreadLocal<>("Request context");
 
	/**
	 * 根據(jù)inheritable的值決定通過(guò)ThreadLocal或InhertitableThreadLocal保存RequestAttributes對(duì)象 
	 */
	public static void setRequestAttributes(@Nullable RequestAttributes attributes, boolean inheritable) {
		if (attributes == null) {
			resetRequestAttributes();
		} else {
			if (inheritable) {
				inheritableRequestAttributesHolder.set(attributes);
 
				requestAttributesHolder.remove();
			} else {
				requestAttributesHolder.set(attributes);
 
				inheritableRequestAttributesHolder.remove();
			}
		}
	}
 
    /**
	 * 通過(guò)Ctrl+鼠標(biāo)點(diǎn)擊，發(fā)現(xiàn)實(shí)際調(diào)用的是這個(gè)方法
     * 所以默認(rèn)是通過(guò)ThreadLocal保存的變量
	 */
    public static void setRequestAttributes(@Nullable RequestAttributes attributes) {
		setRequestAttributes(attributes, false);
	}
 
    /**
	 * 獲取ThreadLocal中設(shè)置的RequestAttributes對(duì)象 
	 */
	@Nullable
	public static RequestAttributes getRequestAttributes() {
        // 因?yàn)槟J(rèn)是通過(guò)ThreadLocal而不是InheritableThreadLocal保存，
        // 在子線程訪問(wèn)不到在父線程中通過(guò)set()方法設(shè)置的變量
		RequestAttributes attributes = requestAttributesHolder.get();
 
        // 所以在子線程中，會(huì)走這個(gè)分支
		if (attributes == null) {
            // 然后從InheritableThreadLocal中獲取RequestAttributes對(duì)象
            // 因?yàn)檎{(diào)用上面第一個(gè)setRequestAttributes(RequestAttributes, boolean)方法時(shí)傳的參數(shù)是false，所以InheritableThreadLocal中沒(méi)有設(shè)置RequestAttributes對(duì)象，因此，這里get()還是null，最后attributes的值為null
			attributes = inheritableRequestAttributesHolder.get();
		}
 
		return attributes;
	}
 
}

于是，把涉及獲取request對(duì)象ip地址獲取的代碼放在線程外面，這樣就避免了空指針問(wèn)題了~

package cn.edu.sgu.www.mhxysy.chain.login.impl;
 
import cn.edu.sgu.www.mhxysy.chain.login.UserLoginHandler;
import cn.edu.sgu.www.mhxysy.config.property.EmailProperties;
import cn.edu.sgu.www.mhxysy.config.property.SystemSettingsProperties;
import cn.edu.sgu.www.mhxysy.entity.location.Location;
import cn.edu.sgu.www.mhxysy.util.EmailUtils;
import cn.edu.sgu.www.mhxysy.util.IpUtils;
import cn.edu.sgu.www.mhxysy.util.LocationUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
 
import java.time.LocalDateTime;
import java.time.ZoneId;
import java.time.format.DateTimeFormatter;
 
/**
 * @author heyunlin
 * @version 1.0
 */
@Component
public class EmailSendHandler implements UserLoginHandler {
 
    private Object params;
    private UserLoginHandler next;
 
    private final EmailUtils emailUtils;
    private final EmailProperties emailProperties;
    private final SystemSettingsProperties systemSettingsProperties;
 
    @Autowired
    public EmailSendHandler(
            EmailUtils emailUtils,
            EmailProperties emailProperties,
            SystemSettingsProperties systemSettingsProperties) {
        this.emailUtils = emailUtils;
        this.emailProperties = emailProperties;
        this.systemSettingsProperties = systemSettingsProperties;
    }
 
    @Override
    public void handle() {
        if (emailProperties.isEnable()) {
            String ip = IpUtils.getIp();
            String username = (String) params;
            String zoneId = systemSettingsProperties.getZoneId();
            // 定義日期格式
            DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy年MM月dd日 HH:mm:ss");
 
            new Thread(() -> {
                try {
                    String address = "廣東廣州";
                    Location location = LocationUtils.getLocation(ip);
 
                    if (systemSettingsProperties.isUseRealLocation()) {
                        String locationAddress = location.getAddress();
 
                        if (locationAddress != null) {
                            address = locationAddress;
                        }
                    }
 
                    String text = "您的賬號(hào)" + username + "在" + address + "登錄了。" +
                            "[" + LocalDateTime.now(ZoneId.of(zoneId)).format(formatter) + "]";
 
                    emailUtils.sendMessage(text);
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }).start();
        }
 
        if (next != null) {
            next.handle();
        }
    }
 
    @Override
    public void setNext(UserLoginHandler next) {
        this.next = next;
    }
 
    @Override
    public void setParams(Object params) {
        this.params = params;
    }
 
}