SpringCloud/spring-cloud-function
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

GH-434 Added generic FunctionInvoker for AWS

Browse Source 
- Added generic FunctionInvoker capable of handling the request generically without requiring user to implemen specific AWS request handler

Resolves #434
This commit is contained in:
Oleg Zhurakousky
2019-12-05 19:28:54 +01:00
parent 0f38ea47b8
commit 52b0fdea50
15 changed files with 641 additions and 234 deletions
Download Patch File Download Diff File Expand all files Collapse all files

154
spring-cloud-function-adapters/spring-cloud-function-adapter-aws/src/test/java/org/springframework/cloud/function/adapter/aws/FunctionInvokerTests.java Normal file
View File

@@ -0,0 +1,154 @@
/*
* Copyright 2012-2019 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.adapter.aws;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.nio.charset.StandardCharsets;
import java.util.function.Function;
import com.amazonaws.services.lambda.runtime.events.KinesisEvent;
import org.junit.jupiter.api.Test;
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.messaging.Message;
import static org.assertj.core.api.Assertions.assertThat;
/**
*
* @author Oleg Zhurakousky
*
*/
public class FunctionInvokerTests {
String sampleEvent = "{" +
" \"Records\": [" +
" {" +
" \"kinesis\": {" +
" \"kinesisSchemaVersion\": \"1.0\"," +
" \"partitionKey\": \"1\"," +
" \"sequenceNumber\": \"49590338271490256608559692538361571095921575989136588898\"," +
" \"data\": \"SGVsbG8sIHRoaXMgaXMgYSB0ZXN0Lg==\"," +
" \"approximateArrivalTimestamp\": 1545084650.987" +
" }," +
" \"eventSource\": \"aws:kinesis\"," +
" \"eventVersion\": \"1.0\"," +
" \"eventID\": \"shardId-000000000006:49590338271490256608559692538361571095921575989136588898\"," +
" \"eventName\": \"aws:kinesis:record\"," +
" \"invokeIdentityArn\": \"arn:aws:iam::123456789012:role/lambda-role\"," +
" \"awsRegion\": \"us-east-2\"," +
" \"eventSourceARN\": \"arn:aws:kinesis:us-east-2:123456789012:stream/lambda-stream\"" +
" }," +
" {" +
" \"kinesis\": {" +
" \"kinesisSchemaVersion\": \"1.0\"," +
" \"partitionKey\": \"1\"," +
" \"sequenceNumber\": \"49590338271490256608559692540925702759324208523137515618\"," +
" \"data\": \"VGhpcyBpcyBvbmx5IGEgdGVzdC4=\"," +
" \"approximateArrivalTimestamp\": 1545084711.166" +
" }," +
" \"eventSource\": \"aws:kinesis\"," +
" \"eventVersion\": \"1.0\"," +
" \"eventID\": \"shardId-000000000006:49590338271490256608559692540925702759324208523137515618\"," +
" \"eventName\": \"aws:kinesis:record\"," +
" \"invokeIdentityArn\": \"arn:aws:iam::123456789012:role/lambda-role\"," +
" \"awsRegion\": \"us-east-2\"," +
" \"eventSourceARN\": \"arn:aws:kinesis:us-east-2:123456789012:stream/lambda-stream\"" +
" }" +
" ]" +
"}";
@Test
public void testKinesisStringMessageEvent() throws Exception {
System.setProperty("MAIN_CLASS", KinesisConfiguration.class.getName());
System.setProperty("spring.cloud.function.definition", "echoStringMessage");
FunctionInvoker invoker = new FunctionInvoker();
InputStream targetStream = new ByteArrayInputStream(this.sampleEvent.getBytes());
ByteArrayOutputStream output = new ByteArrayOutputStream();
invoker.handleRequest(targetStream, output, null);
String result = new String(output.toByteArray(), StandardCharsets.UTF_8);
assertThat(result).isEqualTo(this.sampleEvent);
}
@Test
public void testKinesisStringEvent() throws Exception {
System.setProperty("MAIN_CLASS", KinesisConfiguration.class.getName());
System.setProperty("spring.cloud.function.definition", "echoStringMessage");
FunctionInvoker invoker = new FunctionInvoker();
InputStream targetStream = new ByteArrayInputStream(this.sampleEvent.getBytes());
ByteArrayOutputStream output = new ByteArrayOutputStream();
invoker.handleRequest(targetStream, output, null);
String result = new String(output.toByteArray(), StandardCharsets.UTF_8);
System.out.println(result);
assertThat(result).isEqualTo(this.sampleEvent);
}
@Test
public void testKinesisEvent() throws Exception {
System.setProperty("MAIN_CLASS", KinesisConfiguration.class.getName());
System.setProperty("spring.cloud.function.definition", "echoKinesisEvent");
FunctionInvoker invoker = new FunctionInvoker();
InputStream targetStream = new ByteArrayInputStream(this.sampleEvent.getBytes());
ByteArrayOutputStream output = new ByteArrayOutputStream();
invoker.handleRequest(targetStream, output, null);
String result = new String(output.toByteArray(), StandardCharsets.UTF_8);
System.out.println(result);
assertThat(result).contains("\"sequenceNumber\":\"49590338271490256608559692538361571095921575989136588898\"");
}
@EnableAutoConfiguration
@Configuration
public static class KinesisConfiguration {
@Bean
public Function<Message<String>, Message<String>> echoStringMessage() {
return v -> {
System.out.println("Received: " + v);
return v;
};
}
@Bean
public Function<String, String> echoString() {
return v -> {
System.out.println("Received: " + v);
return v;
};
}
@Bean
public Function<KinesisEvent, KinesisEvent> echoKinesisEvent() {
return v -> {
System.out.println("Received: " + v);
return v;
};
}
}
}