/* * Copyright (c) 2018-2028, DreamLu All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * Redistributions of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * Neither the name of the dreamlu.net developer nor the names of its * contributors may be used to endorse or promote products derived from * this software without specific prior written permission. * Author: DreamLu 卢春梦 (596392912@qq.com) */ package org.springblade.core.cloud.http; import lombok.AllArgsConstructor; import org.springblade.core.tool.constant.BladeConstant; import org.springblade.core.tool.utils.ThreadLocalUtil; import org.springframework.http.HttpHeaders; import org.springframework.http.HttpRequest; import org.springframework.http.client.ClientHttpRequestExecution; import org.springframework.http.client.ClientHttpRequestInterceptor; import org.springframework.http.client.ClientHttpResponse; import org.springframework.lang.NonNull; import java.io.IOException; /** * RestTemplateHeaderInterceptor 传递Request header * * @author L.cm */ @AllArgsConstructor public class RestTemplateHeaderInterceptor implements ClientHttpRequestInterceptor { @NonNull @Override public ClientHttpResponse intercept(@NonNull HttpRequest request, @NonNull byte[] bytes, @NonNull ClientHttpRequestExecution execution) throws IOException { HttpHeaders headers = ThreadLocalUtil.get(BladeConstant.CONTEXT_KEY); if (headers != null && !headers.isEmpty()) { HttpHeaders httpHeaders = request.getHeaders(); headers.forEach((key, values) -> values.forEach(value -> httpHeaders.add(key, value))); } return execution.execute(request, bytes); } }