Exemplo n.º 1
0
 private void trace(Type type, String id, ModelNode operation) {
   if (!trackInvocations) return;
   if (Type.BEGIN.equals(type)) {
     diagnostics.logRpc(type.getClassifier(), id, System.currentTimeMillis(), getToken(operation));
   } else {
     diagnostics.logRpc(type.getClassifier(), id, System.currentTimeMillis());
   }
 }
Exemplo n.º 2
0
/**
 * @author Heiko Braun
 * @date 3/17/11
 */
@SuppressWarnings("GwtClientClassFromNonInheritedModule")
public class DMRHandler implements ActionHandler<DMRAction, DMRResponse> {

  private static final String HEADER_CONTENT_TYPE = "Content-Type";
  private static final String HEADER_ACCEPT = "Accept";
  private static final String DMR_ENCODED = "application/dmr-encoded";
  private static final String HEADER_CONNECTION = "Connection";
  private static final String KEEP_ALIVE = "Keep-Alive";

  /**
   * The read resource description supports the following parameters: recursive, proxies,
   * operations, inherited plus one not documented: locale. See
   * https://docs.jboss.org/author/display/AS72/Global+operations#Globaloperations-readresourcedescription
   * for a more detailed description
   */
  private static final String[] READ_RESOURCE_DESCRIPTION_OPTIONAL_PARAMETERS =
      new String[] {RECURSIVE, PROXIES, OPERATIONS, INHERITED, LOCALE};

  private static long idCounter = 0;

  private RequestBuilder prb;
  private Diagnostics diagnostics = GWT.create(Diagnostics.class);
  private boolean trackInvocations = diagnostics.isEnabled();
  private DMREndpointConfig endpointConfig = GWT.create(DMREndpointConfig.class);
  private ResourceAccessLog resourceLog = ResourceAccessLog.INSTANCE;

  private RequestBuilder postRequestBuilder() {
    // lazy init, because endpointConfig.getUrl() is not initialized at construction time
    if (prb == null) {
      prb = new RequestBuilder(RequestBuilder.POST, endpointConfig.getUrl());
      prb.setHeader(HEADER_ACCEPT, DMR_ENCODED);
      prb.setHeader(HEADER_CONTENT_TYPE, DMR_ENCODED);
      prb.setIncludeCredentials(true);
    }
    return prb;
  }

  private static native void redirect(String url) /*-{
        $wnd.location = url;
    }-*/;

  private static String getToken(ModelNode operation) {
    StringBuffer sb = new StringBuffer();
    if (operation.get(OP).asString().equals(COMPOSITE)) {
      for (ModelNode step : operation.get(STEPS).asList()) {
        sb.append(" _").append(getOpToken(step));
      }
    } else {
      sb.append(getOpToken(operation));
    }
    return sb.toString();
  }

  private static String getOpToken(ModelNode operation) {
    StringBuffer sb = new StringBuffer();
    sb.append(operation.get(ADDRESS).asString())
        .append(": ")
        .append(operation.get(OP))
        .append("; ")
        .append(operation.get(CHILD_TYPE).asString())
        .append("; ");

    if (operation.get(NAME).isDefined()) {
      sb.append(operation.get(NAME).asString());
    }
    return sb.toString();
  }

  @Override
  public DispatchRequest execute(
      DMRAction action,
      final AsyncCallback<DMRResponse> resultCallback,
      Map<String, String> properties) {
    assert action.getOperation() != null;
    final ModelNode operation = action.getOperation();

    // diagnostics, development only
    if (!GWT.isScript()) {
      Scheduler.get()
          .scheduleDeferred(
              new Scheduler.ScheduledCommand() {
                @Override
                public void execute() {
                  decomposeAndLog(operation);
                }
              });
    }

    // Request request = executeRequest(resultCallback, GWT.isScript() ? operation :
    // runAsRole(operation));
    // TODO: https://issues.jboss.org/browse/HAL-100
    Request request = executeRequest(resultCallback, runAsRole(operation, properties));
    return new DispatchRequestHandle(request);
  }

  private ModelNode runAsRole(final ModelNode operation, final Map<String, String> properties) {

    String role = properties.get("run_as");
    if (role != null
        && !operation.get(OP).asString().equals("whoami")) // otherwise we get the replacement role
    {
      operation.get("operation-headers").get("roles").set(role);
    }
    return operation;
  }

  private void decomposeAndLog(ModelNode operation) {
    if (operation.get(OP).asString().equals(COMPOSITE)) {
      List<ModelNode> steps = operation.get(STEPS).asList();
      for (ModelNode step : steps) logAtomicOperation(step);
    } else {
      logAtomicOperation(operation);
    }
  }

  private void logAtomicOperation(ModelNode operation) {
    if (operation.get(OP).asString().equals(COMPOSITE)) // nested composite ops?
    {
      Log.error("Failed to to log resources access", operation.toString());
    } else if (operation.hasDefined(CHILD_TYPE)) {
      // ModelNode address = operation.get(ADDRESS).clone();
      // address.add(operation.get(CHILD_TYPE).toString(), "*");
      resourceLog.log(
          Window.Location.getHash(),
          operation.get(ADDRESS).toString()
              + " : "
              + operation.get(OP).asString()
              + "(child-type="
              + operation.get(CHILD_TYPE)
              + ")");
    } else {
      resourceLog.log(
          Window.Location.getHash(),
          operation.get(ADDRESS).toString() + " : " + operation.get(OP).asString());
    }
  }

  @Override
  public DispatchRequest undo(DMRAction action, DMRResponse result, AsyncCallback<Void> callback) {
    throw new RuntimeException("Not implemented yet.");
  }

  private Request executeRequest(
      final AsyncCallback<DMRResponse> resultCallback, final ModelNode operation) {
    if (idCounter == Long.MAX_VALUE) {
      idCounter = 0;
    }

    Request request = null;
    try {
      final String id = String.valueOf(idCounter++);
      trace(Type.BEGIN, id, operation);

      final RequestBuilder requestBuilder = chooseRequestBuilder(operation);
      trace(Type.SERIALIZED, id, operation);

      final RequestCallback requestCallback =
          new RequestCallback() {
            @Override
            public void onResponseReceived(Request request, Response response) {
              trace(Type.RECEIVE, id, operation);

              int statusCode = response.getStatusCode();
              if (200 == statusCode) {
                resultCallback.onSuccess(
                    new DMRResponse(
                        requestBuilder.getHTTPMethod(),
                        response.getText(),
                        response.getHeader(HEADER_CONTENT_TYPE)));
              } else if (401 == statusCode || 0 == statusCode) {
                resultCallback.onFailure(new DispatchError("Authentication required.", statusCode));
              } else if (403 == statusCode) {
                resultCallback.onFailure(new DispatchError("Authentication required.", statusCode));
              } else if (307 == statusCode) {
                String location = response.getHeader("Location");
                Log.error("Redirect '" + location + "'. Could not execute " + operation.toString());
                redirect(location);
              } else if (503 == statusCode) {
                resultCallback.onFailure(
                    new DispatchError(
                        "Service temporarily unavailable. Is the server still booting?",
                        statusCode));
              } else {
                StringBuilder sb = new StringBuilder();
                sb.append("Unexpected HTTP response").append(": ").append(statusCode);
                sb.append("\n\n");
                sb.append("Request\n");
                sb.append(operation.toString());
                sb.append("\n\nResponse\n\n");
                sb.append(response.getStatusText()).append("\n");
                String payload =
                    response.getText().equals("")
                        ? "No details"
                        : ModelNode.fromBase64(response.getText()).toString();
                sb.append(payload);
                resultCallback.onFailure(new DispatchError(sb.toString(), statusCode));
              }
              trace(Type.END, id, operation);
            }

            @Override
            public void onError(Request request, Throwable e) {
              trace(Type.RECEIVE, id, operation);
              resultCallback.onFailure(e);
              trace(Type.END, id, operation);
            }
          };
      requestBuilder.setCallback(requestCallback);
      request = requestBuilder.send();
      trace(Type.SEND, id, operation);
    } catch (RequestException e) {
      resultCallback.onFailure(e);
    }
    return request;
  }

  static final String[] COLLECTION_OPS = {READ_CHILDREN_RESOURCES_OPERATION};

  private static boolean expectCollectionResponse(ModelNode operation) {
    for (String op : COLLECTION_OPS) {
      if (op.equals(operation.get(OP).asString())) {
        return true;
      }
    }
    return false;
  }

  private RequestBuilder chooseRequestBuilder(final ModelNode operation) {
    RequestBuilder requestBuilder;
    final String op = operation.get(OP).asString();
    if (READ_RESOURCE_DESCRIPTION_OPERATION.equals(op)) {
      String endpoint = endpointConfig.getUrl();
      if (endpoint.endsWith("/")) {
        endpoint = endpoint.substring(0, endpoint.length() - 1);
      }
      String descriptionUrl = endpoint + descriptionOperationToUrl(operation);
      requestBuilder =
          new RequestBuilder(
              RequestBuilder.GET, com.google.gwt.http.client.URL.encode(descriptionUrl));
      requestBuilder.setHeader(HEADER_ACCEPT, DMR_ENCODED);
      requestBuilder.setHeader(HEADER_CONTENT_TYPE, DMR_ENCODED);
      requestBuilder.setIncludeCredentials(true);
      requestBuilder.setRequestData(null);
    } else {
      requestBuilder = postRequestBuilder();
      requestBuilder.setRequestData(operation.toBase64String());
    }
    return requestBuilder;
  }

  private String descriptionOperationToUrl(final ModelNode operation) {
    StringBuilder url = new StringBuilder();
    final List<Property> address = operation.get(ADDRESS).asPropertyList();
    for (Property property : address) {
      url.append("/").append(property.getName()).append("/").append(property.getValue().asString());
    }

    url.append("?operation=").append("resource-description");
    for (String parameter : READ_RESOURCE_DESCRIPTION_OPTIONAL_PARAMETERS) {
      if (operation.has(parameter)) {
        url.append("&").append(parameter).append("=").append(operation.get(parameter).asString());
      }
    }
    return url.toString();
  }

  private void trace(Type type, String id, ModelNode operation) {
    if (!trackInvocations) return;
    if (Type.BEGIN.equals(type)) {
      diagnostics.logRpc(type.getClassifier(), id, System.currentTimeMillis(), getToken(operation));
    } else {
      diagnostics.logRpc(type.getClassifier(), id, System.currentTimeMillis());
    }
  }

  private static enum Type {
    BEGIN("begin"),
    END("end"),
    SEND("requestSent"),
    RECEIVE("responseReceived"),
    SERIALIZED("requestSerialized"),
    DESERIALIZED("responseDeserialized");
    private String classifier;

    private Type(String classifier) {
      this.classifier = classifier;
    }

    public String getClassifier() {
      return classifier;
    }
  }

  class DispatchRequestHandle implements DispatchRequest {
    private Request delegate;

    DispatchRequestHandle(Request delegate) {
      this.delegate = delegate;
    }

    @Override
    public void cancel() {
      if (delegate != null) {
        delegate.cancel();
      }
    }

    @Override
    public boolean isPending() {
      return delegate != null ? delegate.isPending() : false;
    }
  }
}