net.guerlab.cloud.web.webmvc.interceptor.TransferHeaderInterceptor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of guerlab-cloud-web-webmvc Show documentation

The newest version!

/*
 * Copyright 2018-2025 guerlab.net and other contributors.
 *
 * Licensed under the GNU LESSER GENERAL PUBLIC LICENSE, Version 3 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.gnu.org/licenses/lgpl-3.0.html
 *
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package net.guerlab.cloud.web.webmvc.interceptor;

import java.util.Enumeration;

import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;

import org.springframework.web.servlet.HandlerInterceptor;

import net.guerlab.cloud.commons.Constants;
import net.guerlab.cloud.context.core.ContextAttributesHolder;

/**
 * 透传header处理.
 *
 * @author guer
 */
@Slf4j
public class TransferHeaderInterceptor implements HandlerInterceptor {

	@Override
	public final boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
		Enumeration headers = request.getHeaderNames();
		String header;
		while (headers.hasMoreElements()) {
			header = headers.nextElement();
			if (!header.startsWith(Constants.ALLOW_TRANSFER_HEADER_PREFIX)) {
				continue;
			}
			String value = request.getHeader(header);
			log.debug("transferHeader: {} = {}", header, value);

			ContextAttributesHolder.get().put(header, value);
		}
		return true;
	}
}