Examples with LRAParticipant io.narayana.lra.client.internal.proxy.nonjaxrs.LRAParticipant used on opensource projects

Example 1 with LRAParticipant

use of io.narayana.lra.client.internal.proxy.nonjaxrs.LRAParticipant in project quarkus by quarkusio.

the class NarayanaLRARecorder method setParticipantTypes.

public void setParticipantTypes(List<String> classNames) {
    Map<String, LRAParticipant> nonJaxParticipants = new HashMap<>();
    for (String className : classNames) {
        try {
            Class<?> clazz = Class.forName(className, false, Thread.currentThread().getContextClassLoader());
            LRAParticipant lraParticipant = new LRAParticipant(clazz);
            nonJaxParticipants.put(className, lraParticipant);
        } catch (ClassNotFoundException e) {
            log.errorf("Unable to proxy class %s (%s)", className, e.getMessage());
        }
    }
    registry = new LRAParticipantRegistry(nonJaxParticipants);
}

Example 2 with LRAParticipant

use of io.narayana.lra.client.internal.proxy.nonjaxrs.LRAParticipant in project narayana by jbosstm.

the class ServerLRAFilter method filter.

@Override
public void filter(ContainerRequestContext containerRequestContext) {
    // Note that this filter uses abortWith instead of throwing exceptions on encountering exceptional
    // conditions. This facilitates async because filters for asynchronous JAX-RS methods are
    // not allowed to throw exceptions.
    Method method = resourceInfo.getResourceMethod();
    MultivaluedMap<String, String> headers = containerRequestContext.getHeaders();
    LRA.Type type = null;
    LRA transactional = AnnotationResolver.resolveAnnotation(LRA.class, method);
    URI lraId;
    URI newLRA = null;
    Long timeout = null;
    URI suspendedLRA = null;
    URI incommingLRA = null;
    URI recoveryUrl;
    boolean isLongRunning = false;
    boolean requiresActiveLRA = false;
    ArrayList<Progress> progress = null;
    if (transactional == null) {
        transactional = method.getDeclaringClass().getDeclaredAnnotation(LRA.class);
    }
    if (transactional != null) {
        type = transactional.value();
        isLongRunning = !transactional.end();
        Response.Status.Family[] cancel0nFamily = transactional.cancelOnFamily();
        Response.Status[] cancel0n = transactional.cancelOn();
        if (cancel0nFamily.length != 0) {
            containerRequestContext.setProperty(CANCEL_ON_FAMILY_PROP, cancel0nFamily);
        }
        if (cancel0n.length != 0) {
            containerRequestContext.setProperty(CANCEL_ON_PROP, cancel0n);
        }
        if (transactional.timeLimit() != 0) {
            timeout = Duration.of(transactional.timeLimit(), transactional.timeUnit()).toMillis();
        }
    }
    boolean endAnnotation = AnnotationResolver.isAnnotationPresent(Complete.class, method) || AnnotationResolver.isAnnotationPresent(Compensate.class, method) || AnnotationResolver.isAnnotationPresent(Leave.class, method) || AnnotationResolver.isAnnotationPresent(Status.class, method) || AnnotationResolver.isAnnotationPresent(Forget.class, method) || AnnotationResolver.isAnnotationPresent(AfterLRA.class, method);
    if (headers.containsKey(LRA_HTTP_CONTEXT_HEADER)) {
        try {
            incommingLRA = new URI(Current.getLast(headers.get(LRA_HTTP_CONTEXT_HEADER)));
        } catch (URISyntaxException e) {
            String msg = String.format("header %s contains an invalid URL %s", LRA_HTTP_CONTEXT_HEADER, Current.getLast(headers.get(LRA_HTTP_CONTEXT_HEADER)));
            abortWith(containerRequestContext, null, Response.Status.PRECONDITION_FAILED.getStatusCode(), msg, null);
            // user error, bail out
            return;
        }
        if (AnnotationResolver.isAnnotationPresent(Leave.class, method)) {
            // leave the LRA
            Map<String, String> terminateURIs = NarayanaLRAClient.getTerminationUris(resourceInfo.getResourceClass(), containerRequestContext.getUriInfo(), timeout);
            String compensatorId = terminateURIs.get("Link");
            if (compensatorId == null) {
                abortWith(containerRequestContext, incommingLRA.toASCIIString(), Response.Status.BAD_REQUEST.getStatusCode(), "Missing complete or compensate annotations", null);
                // user error, bail out
                return;
            }
            try {
                getLRAClient().leaveLRA(incommingLRA, compensatorId);
                // leave succeeded
                progress = updateProgress(progress, ProgressStep.Left, null);
            } catch (WebApplicationException e) {
                // leave may have failed
                progress = updateProgress(progress, ProgressStep.LeaveFailed, e.getMessage());
                abortWith(containerRequestContext, incommingLRA.toASCIIString(), e.getResponse().getStatus(), e.getMessage(), progress);
                // the error will be handled or reported via the response filter
                return;
            } catch (ProcessingException e) {
                // a remote coordinator was unavailable
                // leave may have failed
                progress = updateProgress(progress, ProgressStep.LeaveFailed, e.getMessage());
                abortWith(containerRequestContext, incommingLRA.toASCIIString(), Response.Status.INTERNAL_SERVER_ERROR.getStatusCode(), e.getMessage(), progress);
                // the error will be handled or reported via the response filter
                return;
            }
        // let the participant know which lra he left by leaving the header intact
        }
    }
    if (type == null) {
        if (!endAnnotation) {
            Current.clearContext(headers);
        }
        if (incommingLRA != null) {
            Current.push(incommingLRA);
            containerRequestContext.setProperty(SUSPENDED_LRA_PROP, incommingLRA);
        }
        // not transactional
        return;
    }
    // check the incoming request for an LRA context
    if (!headers.containsKey(LRA_HTTP_CONTEXT_HEADER)) {
        Object lraContext = containerRequestContext.getProperty(LRA_HTTP_CONTEXT_HEADER);
        if (lraContext != null) {
            incommingLRA = (URI) lraContext;
        }
    }
    if (endAnnotation && incommingLRA == null) {
        return;
    }
    if (incommingLRA != null) {
        // set the parent context header
        try {
            headers.putSingle(LRA_HTTP_PARENT_CONTEXT_HEADER, Current.getFirstParent(incommingLRA));
        } catch (UnsupportedEncodingException e) {
            abortWith(containerRequestContext, incommingLRA.toASCIIString(), Response.Status.PRECONDITION_FAILED.getStatusCode(), String.format("incoming LRA %s contains an invalid parent: %s", incommingLRA, e.getMessage()), progress);
            // any previous actions (the leave request) will be reported via the response filter
            return;
        }
    }
    switch(type) {
        case // a txn must be present
        MANDATORY:
            if (isTxInvalid(containerRequestContext, type, incommingLRA, true, progress)) {
                // any previous actions (eg the leave request) will be reported via the response filter
                return;
            }
            lraId = incommingLRA;
            requiresActiveLRA = true;
            break;
        case // a txn must not be present
        NEVER:
            if (isTxInvalid(containerRequestContext, type, incommingLRA, false, progress)) {
                // any previous actions (the leave request) will be reported via the response filter
                return;
            }
            // must not run with any context
            lraId = null;
            break;
        case NOT_SUPPORTED:
            suspendedLRA = incommingLRA;
            // must not run with any context
            lraId = null;
            break;
        case NESTED:
        // FALLTHRU
        case REQUIRED:
            if (incommingLRA != null) {
                if (type == NESTED) {
                    headers.putSingle(LRA_HTTP_PARENT_CONTEXT_HEADER, incommingLRA.toASCIIString());
                    // if there is an LRA present nest a new LRA under it
                    suspendedLRA = incommingLRA;
                    if (progress == null) {
                        progress = new ArrayList<>();
                    }
                    newLRA = lraId = startLRA(containerRequestContext, incommingLRA, method, timeout, progress);
                    if (newLRA == null) {
                        // the failure plus any previous actions (the leave request) will be reported via the response filter
                        return;
                    }
                } else {
                    lraId = incommingLRA;
                    // incomingLRA will be resumed
                    requiresActiveLRA = true;
                }
            } else {
                if (progress == null) {
                    // NB my IDE seems to think this check is redundant
                    progress = new ArrayList<>();
                }
                newLRA = lraId = startLRA(containerRequestContext, null, method, timeout, progress);
                if (newLRA == null) {
                    // the failure and any previous actions (the leave request) will be reported via the response filter
                    return;
                }
            }
            break;
        case REQUIRES_NEW:
            // previous = AtomicAction.suspend();
            suspendedLRA = incommingLRA;
            if (progress == null) {
                progress = new ArrayList<>();
            }
            newLRA = lraId = startLRA(containerRequestContext, null, method, timeout, progress);
            if (newLRA == null) {
                // the failure and any previous actions (the leave request) will be reported via the response filter
                return;
            }
            break;
        case SUPPORTS:
            lraId = incommingLRA;
            break;
        default:
            lraId = incommingLRA;
    }
    if (lraId == null) {
        // the method call needs to run without a transaction
        Current.clearContext(headers);
        if (suspendedLRA != null) {
            containerRequestContext.setProperty(SUSPENDED_LRA_PROP, suspendedLRA);
        }
        // non transactional
        return;
    }
    if (!isLongRunning) {
        containerRequestContext.setProperty(TERMINAL_LRA_PROP, lraId);
    }
    // store state with the current thread. TODO for the async version use containerRequestContext.setProperty("lra", Current.peek());
    // make the current LRA available to the called method
    Current.updateLRAContext(lraId, headers);
    if (newLRA != null) {
        if (suspendedLRA != null) {
            containerRequestContext.setProperty(SUSPENDED_LRA_PROP, incommingLRA);
        }
        containerRequestContext.setProperty(NEW_LRA_PROP, newLRA);
    }
    Current.push(lraId);
    try {
        // make the current LRA available to the called method
        getLRAClient().setCurrentLRA(lraId);
    } catch (Exception e) {
        // should not happen since lraId has already been validated
        // (perhaps we should not use the client API to set the context)
        abortWith(containerRequestContext, lraId.toASCIIString(), Response.Status.BAD_REQUEST.getStatusCode(), e.getMessage(), progress);
        // any previous actions (such as leave and start requests) will be reported via the response filter
        return;
    }
    // TODO make sure it is possible to do compensations inside a new LRA
    if (!endAnnotation) {
        // don't enlist for methods marked with Compensate, Complete or Leave
        URI baseUri = containerRequestContext.getUriInfo().getBaseUri();
        Map<String, String> terminateURIs = NarayanaLRAClient.getTerminationUris(resourceInfo.getResourceClass(), containerRequestContext.getUriInfo(), timeout);
        String timeLimitStr = terminateURIs.get(TIMELIMIT_PARAM_NAME);
        long timeLimit = timeLimitStr == null ? DEFAULT_TIMEOUT_MILLIS : Long.parseLong(timeLimitStr);
        LRAParticipant participant = lraParticipantRegistry != null ? lraParticipantRegistry.getParticipant(resourceInfo.getResourceClass().getName()) : null;
        if (terminateURIs.containsKey("Link") || participant != null) {
            try {
                if (participant != null) {
                    participant.augmentTerminationURIs(terminateURIs, baseUri);
                }
                recoveryUrl = getLRAClient().joinLRA(lraId, timeLimit, toURI(terminateURIs.get(COMPENSATE)), toURI(terminateURIs.get(COMPLETE)), toURI(terminateURIs.get(FORGET)), toURI(terminateURIs.get(LEAVE)), toURI(terminateURIs.get(AFTER)), toURI(terminateURIs.get(STATUS)), null);
                progress = updateProgress(progress, ProgressStep.Joined, null);
                headers.putSingle(LRA_HTTP_RECOVERY_HEADER, START_END_QUOTES_PATTERN.matcher(recoveryUrl.toASCIIString()).replaceAll(""));
            } catch (WebApplicationException e) {
                progress = updateProgress(progress, ProgressStep.JoinFailed, e.getMessage());
                abortWith(containerRequestContext, lraId.toASCIIString(), e.getResponse().getStatus(), String.format("%s: %s", e.getClass().getSimpleName(), e.getMessage()), progress);
            // the failure plus any previous actions (such as leave and start requests) will be reported via the response filter
            } catch (URISyntaxException e) {
                // one or more of the participant end points was invalid
                progress = updateProgress(progress, ProgressStep.JoinFailed, e.getMessage());
                abortWith(containerRequestContext, lraId.toASCIIString(), Response.Status.BAD_REQUEST.getStatusCode(), String.format("%s %s: %s", lraId, e.getClass().getSimpleName(), e.getMessage()), progress);
            // the failure plus any previous actions (such as leave and start requests) will be reported via the response filter
            } catch (ProcessingException e) {
                // a remote coordinator was unavailable
                progress = updateProgress(progress, ProgressStep.JoinFailed, e.getMessage());
                abortWith(containerRequestContext, lraId.toASCIIString(), Response.Status.INTERNAL_SERVER_ERROR.getStatusCode(), String.format("%s %s,", e.getClass().getSimpleName(), e.getMessage()), progress);
            // the failure plus any previous actions (such as leave and start requests) will be reported via the response filter
            }
        } else if (requiresActiveLRA && getLRAClient().getStatus(lraId) != LRAStatus.Active) {
            Current.clearContext(headers);
            Current.pop(lraId);
            containerRequestContext.removeProperty(SUSPENDED_LRA_PROP);
            if (type == MANDATORY) {
                abortWith(containerRequestContext, lraId.toASCIIString(), Response.Status.PRECONDITION_FAILED.getStatusCode(), "LRA should have been active: ", progress);
            // any previous actions (such as leave and start requests) will be reported via the response filter
            }
        }
    }
}