GH-711 Add initial support for source/target types determination

Resoves #711
This commit is contained in:
Oleg Zhurakousky
2021-07-01 17:06:15 +02:00
parent c86890806e
commit 3b244ea0a5
2 changed files with 74 additions and 0 deletions

View File

@@ -44,6 +44,11 @@ public abstract class MessageUtils {
*/
public static String TARGET_PROTOCOL = "target-protocol";
/**
* Value for 'target-protocol' typically use as header key.
*/
public static String SOURCE_TYPE = "source-type";
/**
* Create a message for the handler. If the handler is a wrapper for a function in an
* isolated class loader, then the message will be created with the target class

View File

@@ -0,0 +1,69 @@
/*
* Copyright 2021-2021 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (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.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springframework.cloud.function.utils;
import org.springframework.cloud.function.context.message.MessageUtils;
import org.springframework.messaging.Message;
import org.springframework.messaging.MessageHeaders;
/**
*
* !!! INTERNAL ONLY !!!
*
* @author Oleg Zhurakousky
*
*/
public final class FunctionMessageUtils {
private FunctionMessageUtils() {
}
public static String getSourceType(String functionDefinition, Message<?> message) {
return determineSourceFromHeaders(message.getHeaders());
}
public static String getTargetType(String functionDefinition, Message<?> message) {
return message.getHeaders().containsKey(MessageUtils.TARGET_PROTOCOL) ? (String) message.getHeaders().get(MessageUtils.TARGET_PROTOCOL) : "unknown";
}
private static String determineSourceFromHeaders(MessageHeaders headers) {
for (String key : headers.keySet()) {
if (key.equals(MessageUtils.SOURCE_TYPE)) {
return (String) headers.get(MessageUtils.SOURCE_TYPE);
}
else if (key.startsWith("amqp_")) {
return "amqp";
}
else if (key.startsWith("kafka_")) {
return "kafka";
}
else if (key.startsWith("aws_")) {
return "aws";
}
else if (key.startsWith("solace_")) {
return "solace";
}
else if (key.toLowerCase().equals("user-agent") || key.toLowerCase().equals("accept-encoding") || key.toLowerCase().equals("host")) {
return "http";
}
// add rsocket
}
return "origin";
}
}