/*- * ============LICENSE_START======================================================= * ONAP * ================================================================================ * Copyright (C) 2020 AT&T Intellectual Property. All rights reserved. * ================================================================================ * 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. * ============LICENSE_END========================================================= */ package org.onap.policy.controlloop.actorserviceprovider.impl; import java.util.ArrayDeque; import java.util.ArrayList; import java.util.Arrays; import java.util.LinkedHashMap; import java.util.LinkedList; import java.util.List; import java.util.Map; import java.util.Queue; import java.util.concurrent.CompletableFuture; import java.util.concurrent.CompletionException; import java.util.concurrent.Executor; import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeoutException; import java.util.function.BiConsumer; import java.util.function.Function; import java.util.function.Supplier; import java.util.function.UnaryOperator; import lombok.Getter; import org.onap.policy.common.endpoints.event.comm.Topic.CommInfrastructure; import org.onap.policy.common.endpoints.utils.NetLoggerUtil; import org.onap.policy.common.endpoints.utils.NetLoggerUtil.EventType; import org.onap.policy.common.utils.coder.Coder; import org.onap.policy.common.utils.coder.CoderException; import org.onap.policy.common.utils.coder.StandardCoder; import org.onap.policy.controlloop.ControlLoopOperation; import org.onap.policy.controlloop.actorserviceprovider.CallbackManager; import org.onap.policy.controlloop.actorserviceprovider.Operation; import org.onap.policy.controlloop.actorserviceprovider.OperationOutcome; import org.onap.policy.controlloop.actorserviceprovider.parameters.ControlLoopOperationParams; import org.onap.policy.controlloop.actorserviceprovider.parameters.OperatorConfig; import org.onap.policy.controlloop.actorserviceprovider.pipeline.PipelineControllerFuture; import org.onap.policy.controlloop.policy.PolicyResult; import org.slf4j.Logger; import org.slf4j.LoggerFactory; /** * Partial implementation of an operator. In general, it's preferable that subclasses * would override {@link #startOperationAsync(int, OperationOutcome) * startOperationAsync()}. However, if that proves to be too difficult, then they can * simply override {@link #doOperation(int, OperationOutcome) doOperation()}. In addition, * if the operation requires any preprocessor steps, the subclass may choose to override * {@link #startPreprocessorAsync()}. *

* The futures returned by the methods within this class can be canceled, and will * propagate the cancellation to any subtasks. Thus it is also expected that any futures * returned by overridden methods will do the same. Of course, if a class overrides * {@link #doOperation(int, OperationOutcome) doOperation()}, then there's little that can * be done to cancel that particular operation. */ public abstract class OperationPartial implements Operation { private static final Logger logger = LoggerFactory.getLogger(OperationPartial.class); private static final Coder coder = new StandardCoder(); public static final String GUARD_ACTOR_NAME = "GUARD"; public static final String GUARD_OPERATION_NAME = "Decision"; public static final long DEFAULT_RETRY_WAIT_MS = 1000L; private final OperatorConfig config; /** * Operation parameters. */ protected final ControlLoopOperationParams params; @Getter private final String fullName; /** * Constructs the object. * * @param params operation parameters * @param config configuration for this operation */ public OperationPartial(ControlLoopOperationParams params, OperatorConfig config) { this.params = params; this.config = config; this.fullName = params.getActor() + "." + params.getOperation(); } public Executor getBlockingExecutor() { return config.getBlockingExecutor(); } public String getActorName() { return params.getActor(); } public String getName() { return params.getOperation(); } @Override public final CompletableFuture start() { // allocate a controller for the entire operation final PipelineControllerFuture controller = new PipelineControllerFuture<>(); CompletableFuture preproc = startPreprocessorAsync(); if (preproc == null) { // no preprocessor required - just start the operation return startOperationAttempt(controller, 1); } /* * Do preprocessor first and then, if successful, start the operation. Note: * operations create their own outcome, ignoring the outcome from any previous * steps. * * Wrap the preprocessor to ensure "stop" is propagated to it. */ // @formatter:off controller.wrap(preproc) .exceptionally(fromException("preprocessor of operation")) .thenCompose(handlePreprocessorFailure(controller)) .thenCompose(unusedOutcome -> startOperationAttempt(controller, 1)) .whenCompleteAsync(controller.delayedComplete(), params.getExecutor()); // @formatter:on return controller; } /** * Handles a failure in the preprocessor pipeline. If a failure occurred, then it * invokes the call-backs, marks the controller complete, and returns an incomplete * future, effectively halting the pipeline. Otherwise, it returns the outcome that it * received. *

* Assumes that no callbacks have been invoked yet. * * @param controller pipeline controller * @return a function that checks the outcome status and continues, if successful, or * indicates a failure otherwise */ private Function> handlePreprocessorFailure( PipelineControllerFuture controller) { return outcome -> { if (outcome != null && isSuccess(outcome)) { logger.info("{}: preprocessor succeeded for {}", getFullName(), params.getRequestId()); return CompletableFuture.completedFuture(outcome); } logger.warn("preprocessor failed, discontinuing operation {} for {}", getFullName(), params.getRequestId()); final Executor executor = params.getExecutor(); final CallbackManager callbacks = new CallbackManager(); // propagate "stop" to the callbacks controller.add(callbacks); final OperationOutcome outcome2 = params.makeOutcome(); // TODO need a FAILURE_MISSING_DATA (e.g., A&AI) outcome2.setFinalOutcome(true); outcome2.setResult(PolicyResult.FAILURE_GUARD); outcome2.setMessage(outcome != null ? outcome.getMessage() : null); // @formatter:off CompletableFuture.completedFuture(outcome2) .whenCompleteAsync(callbackStarted(callbacks), executor) .whenCompleteAsync(callbackCompleted(callbacks), executor) .whenCompleteAsync(controller.delayedComplete(), executor); // @formatter:on return new CompletableFuture<>(); }; } /** * Invokes the operation's preprocessor step(s) as a "future". This method simply * returns {@code null}. *

* This method assumes the following: *

* * @return a function that will start the preprocessor and returns its outcome, or * {@code null} if this operation needs no preprocessor */ protected CompletableFuture startPreprocessorAsync() { return null; } /** * Invokes the operation's guard step(s) as a "future". *

* This method assumes the following: *

* * @return a function that will start the guard checks and returns its outcome, or * {@code null} if this operation has no guard */ protected CompletableFuture startGuardAsync() { // get the guard payload Map guardPayload = makeGuardPayload(); // wrap it in a "resource" Map resource = new LinkedHashMap<>(); resource.put("guard", guardPayload); Map payload = new LinkedHashMap<>(); payload.put("resource", resource); /* * Note: can't use constants from actor.guard, because that would create a * circular dependency. */ return params.toBuilder().actor(GUARD_ACTOR_NAME).operation(GUARD_OPERATION_NAME).retry(null).timeoutSec(null) .payload(payload).build().start(); } /** * Creates a payload to execute a guard operation. * * @return a new guard payload */ protected Map makeGuardPayload() { Map guard = new LinkedHashMap<>(); guard.put("actor", params.getActor()); guard.put("recipe", params.getOperation()); guard.put("target", params.getTargetEntity()); guard.put("requestId", params.getRequestId()); String clname = params.getContext().getEvent().getClosedLoopControlName(); if (clname != null) { guard.put("clname", clname); } return guard; } /** * Starts the operation attempt, with no preprocessor. When all retries complete, it * will complete the controller. * * @param controller controller for all operation attempts * @param attempt attempt number, typically starting with 1 * @return a future that will return the final result of all attempts */ private CompletableFuture startOperationAttempt( PipelineControllerFuture controller, int attempt) { // propagate "stop" to the operation attempt controller.wrap(startAttemptWithoutRetries(attempt)).thenCompose(retryOnFailure(controller, attempt)) .whenCompleteAsync(controller.delayedComplete(), params.getExecutor()); return controller; } /** * Starts the operation attempt, without doing any retries. * * @param params operation parameters * @param attempt attempt number, typically starting with 1 * @return a future that will return the result of a single operation attempt */ private CompletableFuture startAttemptWithoutRetries(int attempt) { logger.info("{}: start operation attempt {} for {}", getFullName(), attempt, params.getRequestId()); final Executor executor = params.getExecutor(); final OperationOutcome outcome = params.makeOutcome(); final CallbackManager callbacks = new CallbackManager(); // this operation attempt gets its own controller final PipelineControllerFuture controller = new PipelineControllerFuture<>(); // propagate "stop" to the callbacks controller.add(callbacks); // @formatter:off CompletableFuture future = CompletableFuture.completedFuture(outcome) .whenCompleteAsync(callbackStarted(callbacks), executor) .thenCompose(controller.wrap(outcome2 -> startOperationAsync(attempt, outcome2))); // @formatter:on // handle timeouts, if specified long timeoutMillis = getTimeoutMs(params.getTimeoutSec()); if (timeoutMillis > 0) { logger.info("{}: set timeout to {}ms for {}", getFullName(), timeoutMillis, params.getRequestId()); future = future.orTimeout(timeoutMillis, TimeUnit.MILLISECONDS); } /* * Note: we re-invoke callbackStarted() just to be sure the callback is invoked * before callbackCompleted() is invoked. * * Note: no need to remove "callbacks" from the pipeline, as we're going to stop * the pipeline as the last step anyway. */ // @formatter:off future.exceptionally(fromException("operation")) .thenApply(setRetryFlag(attempt)) .whenCompleteAsync(callbackStarted(callbacks), executor) .whenCompleteAsync(callbackCompleted(callbacks), executor) .whenCompleteAsync(controller.delayedComplete(), executor); // @formatter:on return controller; } /** * Determines if the outcome was successful. * * @param outcome outcome to examine * @return {@code true} if the outcome was successful */ protected boolean isSuccess(OperationOutcome outcome) { return (outcome.getResult() == PolicyResult.SUCCESS); } /** * Determines if the outcome was a failure for this operator. * * @param outcome outcome to examine, or {@code null} * @return {@code true} if the outcome is not {@code null} and was a failure * and was associated with this operator, {@code false} otherwise */ protected boolean isActorFailed(OperationOutcome outcome) { return (isSameOperation(outcome) && outcome.getResult() == PolicyResult.FAILURE); } /** * Determines if the given outcome is for this operation. * * @param outcome outcome to examine * @return {@code true} if the outcome is for this operation, {@code false} otherwise */ protected boolean isSameOperation(OperationOutcome outcome) { return OperationOutcome.isFor(outcome, getActorName(), getName()); } /** * Invokes the operation as a "future". This method simply invokes * {@link #doOperation()} using the {@link #blockingExecutor "blocking executor"}, * returning the result via a "future". *

* Note: if the operation uses blocking I/O, then it should not be run using * the executor in the "params", as that may bring the background thread pool to a * grinding halt. The {@link #blockingExecutor "blocking executor"} should be used * instead. *

* This method assumes the following: *

    *
  • the operator is alive
  • *
  • verifyRunning() has been invoked
  • *
  • callbackStarted() has been invoked
  • *
  • the invoker will perform appropriate timeout checks
  • *
  • exceptions generated within the pipeline will be handled by the invoker
  • *
* * @param attempt attempt number, typically starting with 1 * @return a function that will start the operation and return its result when * complete */ protected CompletableFuture startOperationAsync(int attempt, OperationOutcome outcome) { return CompletableFuture.supplyAsync(() -> doOperation(attempt, outcome), getBlockingExecutor()); } /** * Low-level method that performs the operation. This can make the same assumptions * that are made by {@link #doOperationAsFuture()}. This particular method simply * throws an {@link UnsupportedOperationException}. * * @param attempt attempt number, typically starting with 1 * @param operation the operation being performed * @return the outcome of the operation */ protected OperationOutcome doOperation(int attempt, OperationOutcome operation) { throw new UnsupportedOperationException("start operation " + getFullName()); } /** * Sets the outcome status to FAILURE_RETRIES, if the current operation outcome is * FAILURE, assuming the policy specifies retries and the retry count has been * exhausted. * * @param attempt latest attempt number, starting with 1 * @return a function to get the next future to execute */ private Function setRetryFlag(int attempt) { return origOutcome -> { // ensure we have a non-null outcome OperationOutcome outcome; if (origOutcome != null) { outcome = origOutcome; } else { logger.warn("{}: null outcome; treating as a failure for {}", getFullName(), params.getRequestId()); outcome = this.setOutcome(params.makeOutcome(), PolicyResult.FAILURE); } // ensure correct actor/operation outcome.setActor(getActorName()); outcome.setOperation(getName()); // determine if we should retry, based on the result if (outcome.getResult() != PolicyResult.FAILURE) { // do not retry success or other failure types (e.g., exception) outcome.setFinalOutcome(true); return outcome; } int retry = getRetry(params.getRetry()); if (retry <= 0) { // no retries were specified outcome.setFinalOutcome(true); } else if (attempt <= retry) { // have more retries - not the final outcome outcome.setFinalOutcome(false); } else { /* * retries were specified and we've already tried them all - change to * FAILURE_RETRIES */ logger.info("operation {} retries exhausted for {}", getFullName(), params.getRequestId()); outcome.setResult(PolicyResult.FAILURE_RETRIES); outcome.setFinalOutcome(true); } return outcome; }; } /** * Restarts the operation if it was a FAILURE. Assumes that {@link #setRetryFlag(int)} * was previously invoked, and thus that the "operation" is not {@code null}. * * @param controller controller for all of the retries * @param attempt latest attempt number, starting with 1 * @return a function to get the next future to execute */ private Function> retryOnFailure( PipelineControllerFuture controller, int attempt) { return operation -> { if (!isActorFailed(operation)) { // wrong type or wrong operation - just leave it as is logger.info("not retrying operation {} for {}", getFullName(), params.getRequestId()); controller.complete(operation); return new CompletableFuture<>(); } if (getRetry(params.getRetry()) <= 0) { // no retries - already marked as FAILURE, so just return it logger.info("operation {} no retries for {}", getFullName(), params.getRequestId()); controller.complete(operation); return new CompletableFuture<>(); } /* * Retry the operation. */ long waitMs = getRetryWaitMs(); logger.info("retry operation {} in {}ms for {}", getFullName(), waitMs, params.getRequestId()); return sleep(waitMs, TimeUnit.MILLISECONDS) .thenCompose(unused -> startOperationAttempt(controller, attempt + 1)); }; } /** * Convenience method that starts a sleep(), running via a future. * * @param sleepTime time to sleep * @param unit time unit * @return a future that will complete when the sleep completes */ protected CompletableFuture sleep(long sleepTime, TimeUnit unit) { if (sleepTime <= 0) { return CompletableFuture.completedFuture(null); } return new CompletableFuture().completeOnTimeout(null, sleepTime, unit); } /** * Converts an exception into an operation outcome, returning a copy of the outcome to * prevent background jobs from changing it. * * @param type type of item throwing the exception * @return a function that will convert an exception into an operation outcome */ private Function fromException(String type) { return thrown -> { OperationOutcome outcome = params.makeOutcome(); logger.warn("exception throw by {} {}.{} for {}", type, outcome.getActor(), outcome.getOperation(), params.getRequestId(), thrown); return setOutcome(outcome, thrown); }; } /** * Similar to {@link CompletableFuture#anyOf(CompletableFuture...)}, but it cancels * any outstanding futures when one completes. * * @param futureMakers function to make a future. If the function returns * {@code null}, then no future is created for that function. On the other * hand, if the function throws an exception, then the previously created * functions are canceled and the exception is re-thrown * @return a future to cancel or await an outcome, or {@code null} if no futures were * created. If this future is canceled, then all of the futures will be * canceled */ protected CompletableFuture anyOf( @SuppressWarnings("unchecked") Supplier>... futureMakers) { return anyOf(Arrays.asList(futureMakers)); } /** * Similar to {@link CompletableFuture#anyOf(CompletableFuture...)}, but it cancels * any outstanding futures when one completes. * * @param futureMakers function to make a future. If the function returns * {@code null}, then no future is created for that function. On the other * hand, if the function throws an exception, then the previously created * functions are canceled and the exception is re-thrown * @return a future to cancel or await an outcome, or {@code null} if no futures were * created. If this future is canceled, then all of the futures will be * canceled. Similarly, when this future completes, any incomplete futures * will be canceled */ protected CompletableFuture anyOf( List>> futureMakers) { PipelineControllerFuture controller = new PipelineControllerFuture<>(); CompletableFuture[] futures = attachFutures(controller, futureMakers, UnaryOperator.identity()); if (futures.length == 0) { // no futures were started return null; } if (futures.length == 1) { return futures[0]; } CompletableFuture.anyOf(futures).thenApply(outcome -> (OperationOutcome) outcome) .whenCompleteAsync(controller.delayedComplete(), params.getExecutor()); return controller; } /** * Similar to {@link CompletableFuture#allOf(CompletableFuture...)}. * * @param futureMakers function to make a future. If the function returns * {@code null}, then no future is created for that function. On the other * hand, if the function throws an exception, then the previously created * functions are canceled and the exception is re-thrown * @return a future to cancel or await an outcome, or {@code null} if no futures were * created. If this future is canceled, then all of the futures will be * canceled */ protected CompletableFuture allOf( @SuppressWarnings("unchecked") Supplier>... futureMakers) { return allOf(Arrays.asList(futureMakers)); } /** * Similar to {@link CompletableFuture#allOf(CompletableFuture...)}. * * @param futureMakers function to make a future. If the function returns * {@code null}, then no future is created for that function. On the other * hand, if the function throws an exception, then the previously created * functions are canceled and the exception is re-thrown * @return a future to cancel or await an outcome, or {@code null} if no futures were * created. If this future is canceled, then all of the futures will be * canceled. Similarly, when this future completes, any incomplete futures * will be canceled */ protected CompletableFuture allOf( List>> futureMakers) { PipelineControllerFuture controller = new PipelineControllerFuture<>(); Queue outcomes = new LinkedList<>(); CompletableFuture[] futures = attachFutures(controller, futureMakers, future -> future.thenApply(outcome -> { synchronized (outcomes) { outcomes.add(outcome); } return outcome; })); if (futures.length == 0) { // no futures were started return null; } if (futures.length == 1) { return futures[0]; } // @formatter:off CompletableFuture.allOf(futures) .thenApply(unused -> combineOutcomes(outcomes)) .whenCompleteAsync(controller.delayedComplete(), params.getExecutor()); // @formatter:on return controller; } /** * Invokes the functions to create the futures and attaches them to the controller. * * @param controller master controller for all of the futures * @param futureMakers futures to be attached to the controller * @param adorn function that "adorns" the future, possible adding onto its pipeline. * Returns the adorned future * @return an array of futures, possibly zero-length. If the array is of size one, * then that one item should be returned instead of the controller */ private CompletableFuture[] attachFutures(PipelineControllerFuture controller, List>> futureMakers, UnaryOperator> adorn) { if (futureMakers.isEmpty()) { @SuppressWarnings("unchecked") CompletableFuture[] result = new CompletableFuture[0]; return result; } // the last, unadorned future that is created CompletableFuture lastFuture = null; List> futures = new ArrayList<>(futureMakers.size()); // make each future for (var maker : futureMakers) { try { CompletableFuture future = maker.get(); if (future == null) { continue; } // propagate "stop" to the future controller.add(future); futures.add(adorn.apply(future)); lastFuture = future; } catch (RuntimeException e) { logger.warn("{}: exception creating 'future' for {}", getFullName(), params.getRequestId()); controller.cancel(false); throw e; } } @SuppressWarnings("unchecked") CompletableFuture[] result = new CompletableFuture[futures.size()]; if (result.length == 1) { // special case - return the unadorned future result[0] = lastFuture; return result; } return futures.toArray(result); } /** * Combines the outcomes from a set of tasks. * * @param outcomes outcomes to be examined * @return the combined outcome */ private OperationOutcome combineOutcomes(Queue outcomes) { // identify the outcome with the highest priority OperationOutcome outcome = outcomes.remove(); int priority = detmPriority(outcome); for (OperationOutcome outcome2 : outcomes) { int priority2 = detmPriority(outcome2); if (priority2 > priority) { outcome = outcome2; priority = priority2; } } logger.info("{}: combined outcome of tasks is {} for {}", getFullName(), (outcome == null ? null : outcome.getResult()), params.getRequestId()); return outcome; } /** * Determines the priority of an outcome based on its result. * * @param outcome outcome to examine, or {@code null} * @return the outcome's priority */ protected int detmPriority(OperationOutcome outcome) { if (outcome == null || outcome.getResult() == null) { return 1; } switch (outcome.getResult()) { case SUCCESS: return 0; case FAILURE_GUARD: return 2; case FAILURE_RETRIES: return 3; case FAILURE: return 4; case FAILURE_TIMEOUT: return 5; case FAILURE_EXCEPTION: default: return 6; } } /** * Performs a sequence of tasks, stopping if a task fails. A given task's future is * not created until the previous task completes. The pipeline returns the outcome of * the last task executed. * * @param futureMakers functions to make the futures * @return a future to cancel the sequence or await the outcome */ protected CompletableFuture sequence( @SuppressWarnings("unchecked") Supplier>... futureMakers) { return sequence(Arrays.asList(futureMakers)); } /** * Performs a sequence of tasks, stopping if a task fails. A given task's future is * not created until the previous task completes. The pipeline returns the outcome of * the last task executed. * * @param futureMakers functions to make the futures * @return a future to cancel the sequence or await the outcome, or {@code null} if * there were no tasks to perform */ protected CompletableFuture sequence( List>> futureMakers) { Queue>> queue = new ArrayDeque<>(futureMakers); CompletableFuture nextTask = getNextTask(queue); if (nextTask == null) { // no tasks return null; } if (queue.isEmpty()) { // only one task - just return it rather than wrapping it in a controller return nextTask; } /* * multiple tasks - need a controller to stop whichever task is currently * executing */ final PipelineControllerFuture controller = new PipelineControllerFuture<>(); final Executor executor = params.getExecutor(); // @formatter:off controller.wrap(nextTask) .thenComposeAsync(nextTaskOnSuccess(controller, queue), executor) .whenCompleteAsync(controller.delayedComplete(), executor); // @formatter:on return controller; } /** * Executes the next task in the queue, if the previous outcome was successful. * * @param controller pipeline controller * @param taskQueue queue of tasks to be performed * @return a future to execute the remaining tasks, or the current outcome, if it's a * failure, or if there are no more tasks */ private Function> nextTaskOnSuccess( PipelineControllerFuture controller, Queue>> taskQueue) { return outcome -> { if (!isSuccess(outcome)) { // return the failure return CompletableFuture.completedFuture(outcome); } CompletableFuture nextTask = getNextTask(taskQueue); if (nextTask == null) { // no tasks - just return the success return CompletableFuture.completedFuture(outcome); } // @formatter:off return controller .wrap(nextTask) .thenComposeAsync(nextTaskOnSuccess(controller, taskQueue), params.getExecutor()); // @formatter:on }; } /** * Gets the next task from the queue, skipping those that are {@code null}. * * @param taskQueue task queue * @return the next task, or {@code null} if the queue is now empty */ private CompletableFuture getNextTask( Queue>> taskQueue) { Supplier> maker; while ((maker = taskQueue.poll()) != null) { CompletableFuture future = maker.get(); if (future != null) { return future; } } return null; } /** * Sets the start time of the operation and invokes the callback to indicate that the * operation has started. Does nothing if the pipeline has been stopped. *

* This assumes that the "outcome" is not {@code null}. * * @param callbacks used to determine if the start callback can be invoked * @return a function that sets the start time and invokes the callback */ private BiConsumer callbackStarted(CallbackManager callbacks) { return (outcome, thrown) -> { if (callbacks.canStart()) { outcome.setStart(callbacks.getStartTime()); outcome.setEnd(null); // pass a copy to the callback OperationOutcome outcome2 = new OperationOutcome(outcome); outcome2.setFinalOutcome(false); params.callbackStarted(outcome2); } }; } /** * Sets the end time of the operation and invokes the callback to indicate that the * operation has completed. Does nothing if the pipeline has been stopped. *

* This assumes that the "outcome" is not {@code null}. *

* Note: the start time must be a reference rather than a plain value, because it's * value must be gotten on-demand, when the returned function is executed at a later * time. * * @param callbacks used to determine if the end callback can be invoked * @return a function that sets the end time and invokes the callback */ private BiConsumer callbackCompleted(CallbackManager callbacks) { return (outcome, thrown) -> { if (callbacks.canEnd()) { outcome.setStart(callbacks.getStartTime()); outcome.setEnd(callbacks.getEndTime()); // pass a copy to the callback params.callbackCompleted(new OperationOutcome(outcome)); } }; } /** * Sets an operation's outcome and message, based on a throwable. * * @param operation operation to be updated * @return the updated operation */ protected OperationOutcome setOutcome(OperationOutcome operation, Throwable thrown) { PolicyResult result = (isTimeout(thrown) ? PolicyResult.FAILURE_TIMEOUT : PolicyResult.FAILURE_EXCEPTION); return setOutcome(operation, result); } /** * Sets an operation's outcome and default message based on the result. * * @param operation operation to be updated * @param result result of the operation * @return the updated operation */ public OperationOutcome setOutcome(OperationOutcome operation, PolicyResult result) { logger.trace("{}: set outcome {} for {}", getFullName(), result, params.getRequestId()); operation.setResult(result); operation.setMessage(result == PolicyResult.SUCCESS ? ControlLoopOperation.SUCCESS_MSG : ControlLoopOperation.FAILED_MSG); return operation; } /** * Determines if a throwable is due to a timeout. * * @param thrown throwable of interest * @return {@code true} if the throwable is due to a timeout, {@code false} otherwise */ protected boolean isTimeout(Throwable thrown) { if (thrown instanceof CompletionException) { thrown = thrown.getCause(); } return (thrown instanceof TimeoutException); } /** * Logs a response. If the response is not of type, String, then it attempts to * pretty-print it into JSON before logging. * * @param direction IN or OUT * @param infra communication infrastructure on which it was published * @param source source name (e.g., the URL or Topic name) * @param response response to be logged * @return the JSON text that was logged */ public String logMessage(EventType direction, CommInfrastructure infra, String source, T response) { String json; try { if (response == null) { json = null; } else if (response instanceof String) { json = response.toString(); } else { json = makeCoder().encode(response, true); } } catch (CoderException e) { String type = (direction == EventType.IN ? "response" : "request"); logger.warn("cannot pretty-print {}", type, e); json = response.toString(); } logger.info("[{}|{}|{}|]{}{}", direction, infra, source, NetLoggerUtil.SYSTEM_LS, json); return json; } // these may be overridden by subclasses or junit tests /** * Gets the retry count. * * @param retry retry, extracted from the parameters, or {@code null} * @return the number of retries, or {@code 0} if no retries were specified */ protected int getRetry(Integer retry) { return (retry == null ? 0 : retry); } /** * Gets the retry wait, in milliseconds. * * @return the retry wait, in milliseconds */ protected long getRetryWaitMs() { return DEFAULT_RETRY_WAIT_MS; } /** * Gets the operation timeout. * * @param timeoutSec timeout, in seconds, extracted from the parameters, or * {@code null} * @return the operation timeout, in milliseconds, or {@code 0} if no timeout was * specified */ protected long getTimeoutMs(Integer timeoutSec) { return (timeoutSec == null ? 0 : TimeUnit.MILLISECONDS.convert(timeoutSec, TimeUnit.SECONDS)); } // these may be overridden by junit tests protected Coder makeCoder() { return coder; } }