org.pkl.thirdparty.truffle.polyglot.PolyglotThreadLocalActions Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of pkl-tools Show documentation
Show all versions of pkl-tools Show documentation
Fat Jar containing pkl-cli, pkl-codegen-java, pkl-codegen-kotlin, pkl-config-java, pkl-core, pkl-doc, and their shaded third-party dependencies.
/*
* Copyright (c) 2021, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* The Universal Permissive License (UPL), Version 1.0
*
* Subject to the condition set forth below, permission is hereby granted to any
* person obtaining a copy of this software, associated documentation and/or
* data (collectively the "Software"), free of charge and under any and all
* copyright rights in the Software, and any and all patent rights owned or
* freely licensable by each licensor hereunder covering either (i) the
* unmodified Software as contributed to or provided by such licensor, or (ii)
* the Larger Works (as defined below), to deal in both
*
* (a) the Software, and
*
* (b) any piece of software and/or hardware listed in the lrgrwrks.txt file if
* one is included with the Software each a "Larger Work" to which the Software
* is contributed by such licensors),
*
* without restriction, including without limitation the rights to copy, create
* derivative works of, display, perform, and distribute the Software and make,
* use, sell, offer for sale, import, export, have made, and have sold the
* Software and the Larger Work(s), and to sublicense the foregoing rights on
* either these or other terms.
*
* This license is subject to the following condition:
*
* The above copyright notice and either this complete permission notice or at a
* minimum a reference to the UPL must be included in all copies or substantial
* portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
package org.pkl.thirdparty.truffle.polyglot;
import java.lang.management.ManagementFactory;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashSet;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.LongSummaryStatistics;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.Timer;
import java.util.TimerTask;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
import java.util.function.Consumer;
import java.util.logging.Level;
import org.pkl.thirdparty.truffle.api.CompilerDirectives.CompilationFinal;
import org.pkl.thirdparty.truffle.api.CompilerDirectives.TruffleBoundary;
import org.pkl.thirdparty.truffle.api.ThreadLocalAction;
import org.pkl.thirdparty.truffle.api.TruffleLogger;
import org.pkl.thirdparty.truffle.api.TruffleSafepoint;
import org.pkl.thirdparty.truffle.api.impl.ThreadLocalHandshake;
import org.pkl.thirdparty.truffle.api.interop.InteropLibrary;
import org.pkl.thirdparty.truffle.api.nodes.Node;
import com.sun.management.ThreadMXBean;
final class PolyglotThreadLocalActions {
private static final Future COMPLETED_FUTURE = CompletableFuture.completedFuture(null);
static final ThreadLocalHandshake TL_HANDSHAKE = EngineAccessor.ACCESSOR.runtimeSupport().getThreadLocalHandshake();
private final PolyglotContextImpl context;
private final Map activeEvents = new LinkedHashMap<>();
@CompilationFinal private TruffleLogger logger;
private long idCounter;
@CompilationFinal private boolean traceActions;
private List statistics; // final after context patching
private Timer intervalTimer; // final after context patching
PolyglotThreadLocalActions(PolyglotContextImpl context) {
this.context = context;
initialize();
}
void prepareContextStore() {
if (intervalTimer != null) {
intervalTimer.cancel();
intervalTimer = null;
}
}
void onContextPatch() {
initialize();
}
boolean hasActiveEvents() {
assert Thread.holdsLock(context);
return !activeEvents.isEmpty();
}
private void initialize() {
OptionValuesImpl options = this.context.engine.getEngineOptionValues();
if (options.get(PolyglotEngineOptions.SafepointALot)) {
statistics = new ArrayList<>();
} else {
statistics = null;
}
this.traceActions = options.get(PolyglotEngineOptions.TraceThreadLocalActions);
long interval = options.get(PolyglotEngineOptions.TraceStackTraceInterval);
if (interval > 0) {
intervalTimer = new Timer(true);
setupIntervalTimer(interval);
} else {
intervalTimer = null;
}
if (statistics != null || traceActions || interval > 0) {
logger = this.context.engine.getEngineLogger();
} else {
logger = null;
}
}
private void setupIntervalTimer(long interval) {
intervalTimer.schedule(new TimerTask() {
@Override
public void run() {
submit(null, PolyglotEngineImpl.ENGINE_ID, new PrintStackTraceAction(false, false), true);
}
}, interval, interval);
}
/**
* Invoked when a thread is newly entered for a context for the first time.
*/
void notifyEnterCreatedThread() {
assert Thread.holdsLock(context);
/*
* This potentially initialized fast thread locals. Before that no events must be submitted.
* Since this hold the context lock it is not possible to do that.
*/
TL_HANDSHAKE.ensureThreadInitialized();
if (statistics != null) {
PolyglotStatisticsAction collector = new PolyglotStatisticsAction(Thread.currentThread());
statistics.add(collector);
submit(new Thread[]{Thread.currentThread()}, PolyglotEngineImpl.ENGINE_ID, collector, false);
}
}
void notifyContextClosed() {
assert Thread.holdsLock(context);
assert !context.isActive() || context.state == PolyglotContextImpl.State.CLOSED_CANCELLED ||
context.state == PolyglotContextImpl.State.CLOSED_EXITED : "context is still active, cannot flush safepoints";
if (intervalTimer != null) {
intervalTimer.cancel();
}
if (!activeEvents.isEmpty()) {
/*
* The set can be modified during the subsequent iteration.
*/
ArrayList activeEventsList = new ArrayList<>(activeEvents.keySet());
boolean pendingThreadLocalAction = false;
for (AbstractTLHandshake handshake : activeEventsList) {
Future future = handshake.future;
if (!future.isDone()) {
if (context.state == PolyglotContextImpl.State.CLOSED_CANCELLED || context.state == PolyglotContextImpl.State.CLOSED_EXITED) {
// we allow cancellation for cancelled or exited contexts
future.cancel(true);
pendingThreadLocalAction = true;
} else {
/*
* otherwise this should not happen as leaving the context before close
* should perform all events.
*/
throw new AssertionError("Pending thread local actions found. Did the actions not process on last leave? Pending action: " + handshake.action);
}
}
}
if (!pendingThreadLocalAction) {
/*
* We have to keep pending events as threads can still leave after close is
* completed in which case the active events still need to be deactivated otherwise
* the waiters can be blocked forever.
*/
activeEvents.clear();
}
}
if (statistics != null) {
logStatistics();
}
}
private void logStatistics() {
LongSummaryStatistics all = new LongSummaryStatistics();
StringBuilder s = new StringBuilder();
s.append(String.format("Safepoint Statistics %n"));
s.append(String.format(" -------------------------------------------------------------------------------------- %n"));
s.append(String.format(" Thread Name Safepoints | Interval Avg Min Max %n"));
s.append(String.format(" -------------------------------------------------------------------------------------- %n"));
for (PolyglotStatisticsAction statistic : statistics) {
all.combine(statistic.intervalStatistics);
formatStatisticLine(s, " " + statistic.threadName, statistic.intervalStatistics);
}
s.append(String.format(" ------------------------------------------------------------------------------------- %n"));
formatStatisticLine(s, " All threads", all);
logger.log(Level.INFO, s.toString());
statistics.clear();
}
private static void formatStatisticLine(StringBuilder s, String label, LongSummaryStatistics statistics) {
s.append(String.format(" %-20s %10d | %16.3f us %12.1f us %12.1f us%n", label,
statistics.getCount() + 1, // intervals + 1 is the number of safepoints
statistics.getAverage() / 1000,
statistics.getMin() / 1000d,
statistics.getMax() / 1000d));
}
Future submit(Thread[] threads, String originId, ThreadLocalAction action, boolean needsEnter) {
boolean sync = EngineAccessor.LANGUAGE.isSynchronousTLAction(action);
return submit(threads, originId, action, new HandshakeConfig(needsEnter, sync, sync, false));
}
Future submit(Thread[] threads, String originId, ThreadLocalAction action, HandshakeConfig config) {
return submit(threads, originId, action, config, null);
}
Future submit(Thread[] threads, String originId, ThreadLocalAction action, HandshakeConfig config, RecurringFuture existingFuture) {
TL_HANDSHAKE.testSupport();
Objects.requireNonNull(action);
if (threads != null) {
for (int i = 0; i < threads.length; i++) {
Objects.requireNonNull(threads[i]);
}
}
// lock to stop new threads
synchronized (context) {
// send enter/leave to slow-path
context.setCachedThreadInfo(PolyglotThreadInfo.NULL);
if (context.state.isClosed() && !config.ignoreContextClosed) {
return COMPLETED_FUTURE;
}
Set filterThreads = null;
if (threads != null) {
filterThreads = new HashSet<>(Arrays.asList(threads));
}
boolean recurring = EngineAccessor.LANGUAGE.isRecurringTLAction(action);
assert existingFuture == null || recurring : "recurring invariant";
boolean sync = EngineAccessor.LANGUAGE.isSynchronousTLAction(action);
boolean sideEffect = EngineAccessor.LANGUAGE.isSideEffectingTLAction(action);
List activePolyglotThreads = new ArrayList<>();
for (PolyglotThreadInfo info : context.getSeenThreads().values()) {
Thread t = info.getThread();
if (info.isActive() && (filterThreads == null || filterThreads.contains(t))) {
if (info.isCurrent() && sync && info.isSafepointActive()) {
throw new IllegalStateException(
"Recursive synchronous thread local action detected. " +
"They are disallowed as they may cause deadlocks. " +
"Schedule an asynchronous thread local action instead.");
}
activePolyglotThreads.add(t);
}
}
Thread[] activeThreads = activePolyglotThreads.toArray(new Thread[0]);
AbstractTLHandshake handshake;
if (sync) {
assert config.syncStartOfEvent || config.syncEndOfEvent : "No synchronization requested for sync event!";
handshake = new SyncEvent(context, threads, originId, action, config);
} else {
assert !config.syncStartOfEvent : "Start of event sync requested for async event!";
assert !config.syncEndOfEvent : "End of event sync requested for async event!";
handshake = new AsyncEvent(context, threads, originId, action, config);
}
if (traceActions) {
String threadLabel;
if (threads == null) {
threadLabel = "all-threads";
} else if (threads.length == 1) {
threadLabel = "single-thread";
} else {
threadLabel = "multiple-threads-" + threads.length;
}
threadLabel += "[alive=" + activePolyglotThreads.size() + "]";
String sideEffectLabel = sideEffect ? "side-effecting " : "side-effect-free";
String syncLabel = sync ? "synchronous " : "asynchronous";
String recurringLabel = recurring ? "recurring" : "one-shot";
handshake.debugId = idCounter++;
log("submit", handshake, String.format("%-25s %s %s %s", threadLabel, sideEffectLabel, syncLabel, recurringLabel));
}
Future future;
if (activeThreads.length > 0) {
future = TL_HANDSHAKE.runThreadLocal(activeThreads, handshake,
AbstractTLHandshake::notifyDone, EngineAccessor.LANGUAGE.isSideEffectingTLAction(action), config.syncStartOfEvent, config.syncEndOfEvent);
this.activeEvents.put(handshake, null);
} else {
future = COMPLETED_FUTURE;
if (recurring) {
// make sure recurring events are registered
this.activeEvents.put(handshake, null);
}
}
if (recurring) {
if (existingFuture != null) {
existingFuture.setCurrentFuture(future);
future = existingFuture;
} else {
future = new RecurringFuture(future);
}
}
handshake.future = future;
return future;
}
}
private static final class RecurringFuture implements Future {
private volatile Future firstFuture;
private volatile Future currentFuture;
volatile boolean cancelled;
RecurringFuture(Future f) {
Objects.requireNonNull(f);
this.firstFuture = f;
this.currentFuture = f;
}
public boolean cancel(boolean mayInterruptIfRunning) {
cancelled = true;
return currentFuture.cancel(mayInterruptIfRunning);
}
public Void get() throws InterruptedException, ExecutionException {
if (cancelled) {
return null;
}
Future first = firstFuture;
if (first == null) {
return null;
}
return first.get();
}
public Void get(long timeout, TimeUnit unit) throws InterruptedException, ExecutionException, TimeoutException {
if (cancelled) {
return null;
}
Future first = firstFuture;
if (first == null) {
return null;
}
return first.get(timeout, unit);
}
Future getCurrentFuture() {
return currentFuture;
}
void setCurrentFuture(Future currentFuture) {
assert !(currentFuture instanceof RecurringFuture) : "no recursive recurring futures";
assert currentFuture != null;
this.firstFuture = null;
this.currentFuture = currentFuture;
}
public boolean isCancelled() {
return cancelled;
}
public boolean isDone() {
if (cancelled) {
return true;
}
Future first = firstFuture;
if (first == null) {
return true;
}
return first.isDone();
}
}
private void log(String action, AbstractTLHandshake handshake, String details) {
if (traceActions) {
logger.log(Level.INFO,
String.format("[tl] %-18s %8d %-30s %-10s %-30s %s", action,
handshake.debugId,
"thread[" + Thread.currentThread().getName() + "]",
handshake.originId,
"action[" + handshake.action.toString() + "]", details));
}
}
Set notifyThreadActivation(PolyglotThreadInfo info, boolean active) {
assert !active || info.getEnteredCount() == 1 : "must be currently entered successfully";
assert Thread.holdsLock(context);
if (activeEvents.isEmpty()) {
// fast common path
return Collections.emptySet();
}
Set updatedActions = new HashSet<>();
// we cannot process the events while the context lock is held
// so we need to collect them first.
TruffleSafepoint s = TruffleSafepoint.getCurrent();
/*
* The set can be modified during the subsequent iteration.
*/
ArrayList activeEventsList = new ArrayList<>(activeEvents.keySet());
for (AbstractTLHandshake handshake : activeEventsList) {
if (!handshake.isEnabledForThread(Thread.currentThread())) {
continue;
}
Future f = handshake.future;
if (f instanceof RecurringFuture) {
f = ((RecurringFuture) f).getCurrentFuture();
assert f != null : "current future must never be null";
}
if (active) {
if (traceActions) {
log("activate", handshake, "");
}
if (f == COMPLETED_FUTURE) {
assert handshake.future instanceof RecurringFuture;
handshake.resubmitRecurring();
} else {
if (TL_HANDSHAKE.activateThread(s, f)) {
updatedActions.add(handshake.action);
}
}
} else {
if (traceActions) {
log("deactivate", handshake, "");
}
if (f == COMPLETED_FUTURE) {
assert handshake.future instanceof RecurringFuture;
// nothing to do, wait for reactivation
} else {
if (TL_HANDSHAKE.deactivateThread(s, f)) {
updatedActions.add(handshake.action);
}
}
}
}
return updatedActions;
}
void notifyLastDone(AbstractTLHandshake handshake) {
assert Thread.holdsLock(context);
if (activeEvents.remove(handshake, null)) {
// this might actually be called multiple times due to a race condition.
// in onDone notification in ThreadLocalHandshake.
if (traceActions) {
if (handshake.future.isCancelled()) {
log("cancelled", handshake, "");
} else {
log("done", handshake, "");
}
}
// important to remove and resubmit recurring events in the same lock
// otherwise we might race with entering and leaving the thread.
handshake.resubmitRecurring();
}
}
private final class PrintStackTraceAction extends ThreadLocalAction {
PrintStackTraceAction(boolean hasSideEffects, boolean synchronous) {
super(hasSideEffects, synchronous);
}
@Override
protected void perform(Access access) {
logger.log(Level.INFO, String.format("Stack Trace Thread %s: %s",
Thread.currentThread().getName(),
PolyglotExceptionImpl.printStackToString(context.getHostContext(), access.getLocation())));
}
}
static final class PolyglotTLAccess extends ThreadLocalAction.Access {
final Thread thread;
final Node location;
volatile boolean invalid;
PolyglotTLAccess(Thread thread, Node location) {
super(PolyglotImpl.getInstance());
this.thread = thread;
this.location = location;
}
@Override
public Node getLocation() {
checkInvalid();
return location;
}
@Override
public Thread getThread() {
checkInvalid();
return Thread.currentThread();
}
private void checkInvalid() {
if (thread != Thread.currentThread()) {
throw new IllegalStateException("ThreadLocalAccess used on the wrong thread.");
} else if (invalid) {
throw new IllegalStateException("ThreadLocalAccess is no longer valid.");
}
}
}
static final class HandshakeConfig {
final boolean needsEnter;
final boolean syncStartOfEvent;
final boolean syncEndOfEvent;
final boolean ignoreContextClosed;
HandshakeConfig(boolean needsEnter, boolean syncStartOfEvent, boolean syncEndOfEvent, boolean ignoreContextClosed) {
this.needsEnter = needsEnter;
this.syncStartOfEvent = syncStartOfEvent;
this.syncEndOfEvent = syncEndOfEvent;
this.ignoreContextClosed = ignoreContextClosed;
}
}
abstract static class AbstractTLHandshake implements Consumer {
private final String originId;
final ThreadLocalAction action;
long debugId;
protected final PolyglotContextImpl context;
final HandshakeConfig config;
final Thread[] filterThreads;
Future future;
AbstractTLHandshake(PolyglotContextImpl context, Thread[] filterThreads, String originId, ThreadLocalAction action, HandshakeConfig config) {
this.action = action;
this.originId = originId;
this.context = context;
this.config = config;
this.filterThreads = filterThreads;
}
protected final void resubmitRecurring() {
if (future instanceof RecurringFuture) {
RecurringFuture f = (RecurringFuture) future;
if (!f.cancelled) {
context.threadLocalActions.submit(filterThreads, originId, action, config, f);
}
}
}
final boolean isEnabledForThread(Thread currentThread) {
if (filterThreads == null) {
return true;
} else {
for (Thread filterThread : filterThreads) {
if (filterThread == currentThread) {
return true;
}
}
return false;
}
}
final void notifyDone() {
synchronized (context) {
this.context.threadLocalActions.notifyLastDone(this);
}
}
public final void accept(Node location) {
Object prev = null;
if (config.needsEnter) {
prev = context.engine.enterIfNeeded(context, false);
}
try {
notifyStart();
PolyglotTLAccess access = new PolyglotTLAccess(Thread.currentThread(), location);
try {
acceptImpl(access);
} finally {
access.invalid = true;
}
notifySuccess();
} catch (Throwable t) {
if (!EngineAccessor.LANGUAGE.isSideEffectingTLAction(action)) {
// no truffle exceptions allowed in non side-effecting events.
if (InteropLibrary.getUncached().isException(t)) {
AssertionError e = new AssertionError("Throwing Truffle exception is disallowed in non-side-effecting thread local actions.", t);
notifyFailed(e);
throw e;
}
}
notifyFailed(t);
throw t;
} finally {
if (config.needsEnter) {
context.engine.leaveIfNeeded(prev, context);
}
}
}
private void notifyStart() {
context.threadLocalActions.log(" perform-start", this, "");
}
private void notifySuccess() {
context.threadLocalActions.log(" perform-done", this, "");
}
private void notifyFailed(Throwable t) {
if (context.threadLocalActions.traceActions) {
context.threadLocalActions.log(" perform-failed", this, " exception: " + t.toString());
}
}
protected abstract void acceptImpl(PolyglotTLAccess access);
}
private static final class AsyncEvent extends AbstractTLHandshake {
AsyncEvent(PolyglotContextImpl context, Thread[] filerThreads, String originId, ThreadLocalAction action, HandshakeConfig config) {
super(context, filerThreads, originId, action, config);
}
@Override
protected void acceptImpl(PolyglotTLAccess access) {
EngineAccessor.LANGUAGE.performTLAction(action, access);
}
}
private static final class SyncEvent extends AbstractTLHandshake {
SyncEvent(PolyglotContextImpl context, Thread[] filterThreads, String originId, ThreadLocalAction action, HandshakeConfig config) {
super(context, filterThreads, originId, action, config);
}
@Override
protected void acceptImpl(PolyglotTLAccess access) {
PolyglotThreadInfo thread;
synchronized (context) {
thread = context.getCurrentThreadInfo();
}
thread.setSafepointActive(true);
try {
EngineAccessor.LANGUAGE.performTLAction(action, access);
} finally {
thread.setSafepointActive(false);
}
}
}
private static final class PolyglotStatisticsAction extends ThreadLocalAction {
private static volatile ThreadMXBean threadBean;
private long prevTime = 0;
private final LongSummaryStatistics intervalStatistics = new LongSummaryStatistics();
private final String threadName;
PolyglotStatisticsAction(Thread thread) {
// no side-effects, async, recurring
super(false, false, true);
this.threadName = thread.getName();
}
@Override
protected void perform(Access access) {
long prev = this.prevTime;
if (prev != 0) {
long now = System.nanoTime();
intervalStatistics.accept(now - prev);
}
this.prevTime = System.nanoTime();
}
@TruffleBoundary
static long getCurrentCPUTime() {
ThreadMXBean bean = threadBean;
if (bean == null) {
/*
* getThreadMXBean is synchronized so better cache in a local volatile field to
* avoid contention.
*/
threadBean = bean = (ThreadMXBean) ManagementFactory.getThreadMXBean();
}
return bean.getCurrentThreadCpuTime();
}
@Override
public String toString() {
return "PolyglotStatisticsAction@" + Integer.toHexString(hashCode());
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy