HTTP header missing in some Rest calls #17118

Bastians-Bits · 2023-02-15T11:42:50Z

Bastians-Bits
Feb 15, 2023

I am trying to extend aznamiers keycloak event listener for rabbitmq to forward some HTTP headers from the rest call as rabbitmq message properties.

This works like a charm for users, but with roles and client-roles I am missing the http headers in the session context.

My changes to the RabbitMqEventListerProvider.java:

package com.github.aznamier.keycloak.event.provider;

import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;

import org.jboss.logging.Logger;
import org.keycloak.events.Event;
import org.keycloak.events.EventListenerProvider;
import org.keycloak.events.EventListenerTransaction;
import org.keycloak.events.admin.AdminEvent;
import org.keycloak.models.KeycloakSession;

import com.rabbitmq.client.AMQP;
import com.rabbitmq.client.AMQP.BasicProperties;
import com.rabbitmq.client.AMQP.BasicProperties.Builder;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;

public class RabbitMqEventListenerProvider implements EventListenerProvider {

	private static final Logger log = Logger.getLogger(RabbitMqEventListenerProvider.class);

	private final RabbitMqConfig cfg;
	private final ConnectionFactory factory;
	private final KeycloakSession session;

	private final EventListenerTransaction tx = new EventListenerTransaction(this::publishAdminEvent, this::publishEvent);

	public RabbitMqEventListenerProvider(RabbitMqConfig cfg, KeycloakSession session) {
		this.cfg = cfg;
		
		this.factory = new ConnectionFactory();

		this.factory.setUsername(cfg.getUsername());
		this.factory.setPassword(cfg.getPassword());
		this.factory.setVirtualHost(cfg.getVhost());
		this.factory.setHost(cfg.getHostUrl());
		this.factory.setPort(cfg.getPort());

		if(cfg.getUseTls()) {
			try {
				this.factory.useSslProtocol();
			} catch (Exception e) {
				log.error("Could not use SSL protocol", e);
			}
		}

		this.session = session;
		this.session.getTransactionManager().enlistAfterCompletion(tx);
	}

	@Override
	public void close() {

	}

	@Override
	public void onEvent(Event event) {
		tx.addEvent(event);
	}

	@Override
	public void onEvent(AdminEvent adminEvent, boolean includeRepresentation) {
		tx.addAdminEvent(adminEvent, includeRepresentation);
	}
	
	private void publishEvent(Event event) {
		EventClientNotificationMqMsg msg = EventClientNotificationMqMsg.create(event);
		String routingKey = RabbitMqConfig.calculateRoutingKey(event);
		String messageString = RabbitMqConfig.writeAsJson(msg, true);
		
		BasicProperties msgProps = getMessageProps(EventClientNotificationMqMsg.class.getName());
		this.publishNotification(messageString, msgProps, routingKey);
	}
	
	private void publishAdminEvent(AdminEvent adminEvent, boolean includeRepresentation) {
		EventAdminNotificationMqMsg msg = EventAdminNotificationMqMsg.create(adminEvent);
		String routingKey = RabbitMqConfig.calculateRoutingKey(adminEvent);
		String messageString = RabbitMqConfig.writeAsJson(msg, true);
		BasicProperties msgProps = getMessageProps(EventAdminNotificationMqMsg.class.getName());
		this.publishNotification(messageString,msgProps, routingKey);
	}
	
	private BasicProperties getMessageProps(String className) {
		
		Map<String,Object> headers = new HashMap<>();
		headers.put("__TypeId__", className);

		if (session != null
			&& session.getContext() != null
			&& session.getContext().getRequestHeaders() != null) {
			String origin = session.getContext().getRequestHeaders().getHeaderString("X-Origin");
			
			if (origin != null && !origin.isEmpty()) {
				log.infof("keycloak-to-rabbitmq ORIGIN: %s %n", origin);
				headers.put("X-Origin", origin);
			}
		}

		Builder propsBuilder = new AMQP.BasicProperties.Builder()
				.appId("Keycloak")
				.headers(headers)
				.contentType("application/json")
				.contentEncoding("UTF-8");
		return propsBuilder.build();
	}
	

	private void publishNotification(String messageString, BasicProperties props, String routingKey) {

		

		try {
			Connection conn = factory.newConnection();
			Channel channel = conn.createChannel();
			
			channel.basicPublish(cfg.getExchange(), routingKey, props, messageString.getBytes(StandardCharsets.UTF_8));
			log.infof("keycloak-to-rabbitmq SUCCESS sending message: %s%n", routingKey);
			channel.close();
			conn.close();

		} catch (Exception ex) {
			log.errorf(ex, "keycloak-to-rabbitmq ERROR sending message: %s%n", routingKey);
		}
	}

}

Has anyone an idea why they are missing or how to access them?

Navigation Menu

Repository navigation

HTTP header missing in some Rest calls #17118

Uh oh!

{{title}}

Uh oh!

Replies: 0 comments

Select a reply

Uh oh!

HTTP header missing in some Rest calls #17118

Uh oh!

Bastians-Bits Feb 15, 2023

Replies: 0 comments

Bastians-Bits
Feb 15, 2023