raw.runtime.truffle.ast.expressions.tryable.TryableNullableFlatMapNode Maven / Gradle / Ivy
The newest version!
/*
* Copyright 2023 RAW Labs S.A.
*
* Use of this software is governed by the Business Source License
* included in the file licenses/BSL.txt.
*
* As of the Change Date specified in that file, in accordance with
* the Business Source License, use of this software will be governed
* by the Apache License, Version 2.0, included in the file
* licenses/APL.txt.
*/
package raw.runtime.truffle.ast.expressions.tryable;
import com.oracle.truffle.api.CompilerDirectives;
import com.oracle.truffle.api.dsl.NodeChild;
import com.oracle.truffle.api.dsl.Specialization;
import com.oracle.truffle.api.library.CachedLibrary;
import com.oracle.truffle.api.nodes.NodeInfo;
import raw.runtime.truffle.ExpressionNode;
import raw.runtime.truffle.runtime.function.Closure;
import raw.runtime.truffle.runtime.option.EmptyOption;
import raw.runtime.truffle.runtime.option.OptionLibrary;
import raw.runtime.truffle.runtime.tryable.ObjectTryable;
import raw.runtime.truffle.runtime.tryable.TryableLibrary;
@NodeInfo(shortName = "TryableNullable.FlatMap")
@NodeChild("tryable")
@NodeChild("function")
public abstract class TryableNullableFlatMapNode extends ExpressionNode {
private final Object[] argumentValues = new Object[1];
private final ObjectTryable successNone = ObjectTryable.BuildSuccess(new EmptyOption());
@Specialization(limit = "1")
@CompilerDirectives.TruffleBoundary
protected Object doTryable(
Object tryable,
Closure closure,
@CachedLibrary("tryable") TryableLibrary tryables,
@CachedLibrary(limit = "1") OptionLibrary nullables) {
if (tryables.isSuccess(tryable)) {
Object n = tryables.success(tryable);
if (nullables.isDefined(n)) {
Object v = nullables.get(n);
argumentValues[0] = v;
return closure.call(argumentValues);
} else {
return successNone;
}
} else {
return tryable;
}
}
}