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

initial commit

Browse Source
This commit is contained in:
markfisher
2016-09-21 22:33:06 -04:00
commit bdd39b74ff
36 changed files with 2715 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

107
spring-cloud-function-core/src/main/java/org/springframework/cloud/function/gateway/FunctionGateway.java Normal file
View File

@@ -0,0 +1,107 @@
/*
* Copyright 2016 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
*
* http://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.gateway;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Supplier;
import org.reactivestreams.Publisher;
import org.reactivestreams.Subscriber;
import org.reactivestreams.Subscription;
import org.springframework.cloud.function.invoker.FunctionInvokingRunnable;
import org.springframework.cloud.function.registry.FunctionRegistry;
import org.springframework.scheduling.TaskScheduler;
import org.springframework.scheduling.Trigger;
import org.springframework.util.Assert;
/**
* @author Mark Fisher
*/
public class FunctionGateway {
private final FunctionRegistry registry;
private final TaskScheduler scheduler;
public FunctionGateway(FunctionRegistry registry, TaskScheduler scheduler) {
Assert.notNull(registry, "FunctionRegistry must not be null");
Assert.notNull(scheduler, "TaskScheduler must not be null");
this.registry = registry;
this.scheduler = scheduler;
}
@SuppressWarnings("unchecked")
public void compose(String name, Function<?, ?>... functions) {
Assert.isTrue(functions != null && functions.length > 1, "more than one Function is required");
@SuppressWarnings("rawtypes")
Function function = functions[0];
for (int i = 1; i < functions.length; i++) {
function = function.andThen(functions[i]);
}
this.registry.register(name, function);
}
@SuppressWarnings("unchecked")
public void compose(String composedFunctionName, String... functionNames) {
Assert.isTrue(functionNames != null && functionNames.length > 1, "more than one Function is required");
@SuppressWarnings("rawtypes")
Function function = this.registry.lookup(functionNames[0]);
for (int i = 1; i < functionNames.length; i++) {
function = function.andThen(this.registry.lookup(functionNames[i]));
}
this.registry.register(composedFunctionName, function);
}
public <T, R> R invoke(String functionName, T request) {
Function<T, R> function = this.registry.lookup(functionName);
return function.apply(request);
}
public <T, R> void schedule(String functionName, Trigger trigger, Supplier<T> supplier, Consumer<R> consumer) {
Function<T, R> function = this.registry.lookup(functionName);
this.scheduler.schedule(new FunctionInvokingRunnable(supplier, function, consumer), trigger);
}
public <T, R> void subscribe(Publisher<T> publisher, String functionName, final Consumer<R> consumer) {
final Function<T, R> function = this.registry.lookup(functionName);
publisher.subscribe(new Subscriber<T>() {
@Override
public void onComplete() {}
@Override
public void onError(Throwable error) {}
@Override
public void onNext(T next) {
if (consumer != null) {
consumer.accept(function.apply(next));
}
else {
function.apply(next);
}
}
@Override
public void onSubscribe(Subscription subscription) {
subscription.request(Long.MAX_VALUE);
}
});
}
}

41
spring-cloud-function-core/src/main/java/org/springframework/cloud/function/invoker/AbstractFunctionInvoker.java Normal file
View File

@@ -0,0 +1,41 @@
/*
* Copyright 2016 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
*
* http://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.invoker;
import java.util.function.Function;
import org.springframework.util.Assert;
/**
* @author Mark Fisher
*
* @param <T> function parameter type
* @param <R> function return type
*/
public abstract class AbstractFunctionInvoker<T, R> {
private final Function<T, R> function;
protected AbstractFunctionInvoker(Function<T, R> function) {
Assert.notNull(function, "Function must not be null");
this.function = function;
}
protected R doInvoke(T input) {
return this.function.apply(input);
}
}

47
spring-cloud-function-core/src/main/java/org/springframework/cloud/function/invoker/FunctionInvokingRunnable.java Normal file
View File

@@ -0,0 +1,47 @@
/*
* Copyright 2016 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
*
* http://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.invoker;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Supplier;
/**
* @author Mark Fisher
*
* @param <T> output of supplier, input to function
* @param <R> output of function, input to consumer
*/
public class FunctionInvokingRunnable<T, R> implements Runnable {
private final Supplier<T> supplier;
private final Function<T, R> function;
private final Consumer<R> consumer;
public FunctionInvokingRunnable(Supplier<T> supplier, Function<T, R> function, Consumer<R> consumer) {
this.supplier = supplier;
this.function = function;
this.consumer = consumer;
}
@Override
public void run() {
this.consumer.accept(this.function.apply(this.supplier.get()));
}
}

32
spring-cloud-function-core/src/main/java/org/springframework/cloud/function/registry/FunctionRegistry.java Normal file
View File

@@ -0,0 +1,32 @@
/*
* Copyright 2016 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
*
* http://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.registry;
import java.util.function.Function;
/**
* @author Mark Fisher
*/
public interface FunctionRegistry {
<T, R> Function<T, R> lookup(String name);
void register(String name, Function<?, ?> function);
void register(String name, String function);
}

35
spring-cloud-function-core/src/main/java/org/springframework/cloud/function/registry/FunctionRegistrySupport.java Normal file
View File

@@ -0,0 +1,35 @@
/*
* Copyright 2016 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
*
* http://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.registry;
import java.util.function.Function;
import org.springframework.cloud.function.compiler.FunctionCompiler;
/**
* @author Mark Fisher
*/
public abstract class FunctionRegistrySupport implements FunctionRegistry {
private final FunctionCompiler compiler = new FunctionCompiler();
@Override
public void register(String name, String code) {
Function<?, ?> function = compiler.compile(code);
this.register(name, function);
}
}

38
spring-cloud-function-core/src/main/java/org/springframework/cloud/function/registry/InMemoryFunctionRegistry.java Normal file
View File

@@ -0,0 +1,38 @@
/*
* Copyright 2016 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
*
* http://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.registry;
import java.util.concurrent.ConcurrentHashMap;
import java.util.function.Function;
/**
* @author Mark Fisher
*/
public class InMemoryFunctionRegistry extends FunctionRegistrySupport {
private final ConcurrentHashMap<String, Function<?, ?>> map = new ConcurrentHashMap<>();
@Override
public Function<?, ?> lookup(String name) {
return this.map.get(name);
}
@Override
public void register(String name, Function<?, ?> function) {
this.map.put(name, function);
}
}