栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

使用ThreadLocal在线程间传递参数

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

使用ThreadLocal在线程间传递参数

使用ThreadLocal在线程间传递参数 使用场景

用户参数透传,如租户ID,tid,userID,Token等。这些参数与具体业务无关,但是又是必须的(如:租户ID,在数据落库时需要传入,但是在其他业务代码中基本用不到)如果作为方法的参数一层一层向下传递,必然造成代码的冗余和扩展性查等影响。

整片文章就以租户ID为例子说明吧。

1.使用ThreadLocal结合AOP

大致思路:在请求接口时,租户ID往往可以从Token或用户信息中取出,这时,在请求拦截器中,将用户TenantID取出,放入线程中。后面在需要TenantID时,直接从线程中取出,这样,既保证了简洁性也增加了易用性。

2.Pom依赖

本文使用阿里线程工具。


    com.alibaba
    transmittable-thread-local
    2.11.4

3.RequestFilter
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import org.springframework.web.filter.GenericFilterBean;

import javax.servlet.FilterChain;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;


@Slf4j
@Component
@Order(Ordered.HIGHEST_PRECEDENCE)
public class TenantContextHolderFilter extends GenericFilterBean {

	@Override
	@SneakyThrows
	public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) {
		HttpServletRequest request = (HttpServletRequest) servletRequest;
		HttpServletResponse response = (HttpServletResponse) servletResponse;
    // 从Token/Header/param中获取租户ID
		String headerTenantId = request.getHeader("TENANT_ID");

		log.debug("TenantContextHolderFilter|header中的TENANT_ID为:{}", headerTenantId);

    // 判断逻辑,自行定义
		if (headerTenantId != null && headerTenantId != "" && !headerTenantId.equals("null")) {
			TenantContextHolder.setTenantId(Integer.parseInt(headerTenantId));
    } else {
      // 默认值
			TenantContextHolder.setTenantId(0);
		}
		filterChain.doFilter(request, response);
    // 请求结束,一定要清除线程中自己设置的参数。
		TenantContextHolder.clear();
	}

}

4.TenantContextHolder(线程参数设置工具)
@UtilityClass
public class TenantContextHolder {

	private final ThreadLocal THREAD_LOCAL_TENANT = new TransmittableThreadLocal<>();

	
	public void setTenantId(Integer tenantId) {
		THREAD_LOCAL_TENANT.set(tenantId);
	}

	
	public Integer getTenantId() {
		return THREAD_LOCAL_TENANT.get();
	}

	
	public void clear() {
		THREAD_LOCAL_TENANT.remove();
	}

}
5.使用

在具体的Controller,Service,Mapper中,只要是当前请求的线程,都是可以直接使用TenantContextHolder.getTenantId()获取租户ID的。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/274968.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号