com.linkedin.playparseq.trace.j.renderers.ParSeqTraceRendererImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of play-parseq-trace_2.11 Show documentation
play-parseq-trace
The newest version!
/*
 * Copyright 2015 LinkedIn Corp.
 *
 * 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.
 */
package com.linkedin.playparseq.trace.j.renderers;

import com.linkedin.parseq.trace.ShallowTrace;
import com.linkedin.parseq.trace.Trace;
import com.linkedin.parseq.trace.TraceRelationship;
import com.linkedin.playparseq.j.stores.ParSeqTaskStore;
import com.linkedin.playparseq.trace.utils.ParSeqTraceBaseVisualizer;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionStage;
import javax.inject.Inject;
import javax.inject.Singleton;
import play.api.http.HttpConfiguration;
import play.Environment;
import play.libs.concurrent.HttpExecutionContext;
import play.mvc.Http;
import play.mvc.Result;
import play.mvc.Results;


/**
 * The class ParSeqTraceRendererImpl is an implementation of the interface {@link ParSeqTraceRenderer} with the
 * help from the class {@link ParSeqTraceBaseVisualizer}.
 *
 * @author Yinan Ding ([email protected])
 */
@Singleton
public class ParSeqTraceRendererImpl extends ParSeqTraceBaseVisualizer implements ParSeqTraceRenderer {

  /**
   * The field _environment is the injected Environment.
   */
  private final Environment _environment;

  /**
   * The field _httpConfiguration is the injected HttpConfiguration.
   */
  private final HttpConfiguration _httpConfiguration;

  /**
   * The field _httpExecutionContext is a {@link HttpExecutionContext} for setting Java async task's executor.
   */
  private final HttpExecutionContext _httpExecutionContext;

  /**
   * The constructor injects the Environment and the HttpConfiguration.
   *
   * @param environment The injected Environment component
   * @param httpConfiguration The injected HttpConfiguration component
   * @param httpExecutionContext The injected HttpExecutionContext component
   */
  @Inject
  public ParSeqTraceRendererImpl(final Environment environment, final HttpConfiguration httpConfiguration,
      final HttpExecutionContext httpExecutionContext) {
    _environment = environment;
    _httpConfiguration = httpConfiguration;
    _httpExecutionContext = httpExecutionContext;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public CompletionStage render(final Http.Context context, final ParSeqTaskStore parSeqTaskStore) {
    return CompletableFuture.supplyAsync(() -> {
      Map traceMap = new HashMap<>();
      Set relationships = new HashSet<>();
      // Get all Tasks from the request out of the store and combine all Trace information
      parSeqTaskStore.get(context).forEach(task -> {
        Trace trace = task.getTrace();
        traceMap.putAll(trace.getTraceMap());
        relationships.addAll(trace.getRelationships());
      });
      // Generate Result of ParSeq Trace
      return Optional.ofNullable(
          showTrace(new Trace(traceMap, relationships), _environment.asScala(), _httpConfiguration))
          .map(s -> Results.ok(s).as("text/html")).orElse(Results.internalServerError("Can't show Trace."));
    }, _httpExecutionContext.current());
  }

}