org.jboss.as.console.client.shared.dispatch.impl.DMRHandler.java Source code

Java tutorial

Introduction

Here is the source code for org.jboss.as.console.client.shared.dispatch.impl.DMRHandler.java

Source

/*
 * JBoss, Home of Professional Open Source
 * Copyright 2011 Red Hat Inc. and/or its affiliates and other contributors
 * as indicated by the @author tags. All rights reserved.
 * See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * This copyrighted material is made available to anyone wishing to use,
 * modify, copy, or redistribute it subject to the terms and conditions
 * of the GNU Lesser General Public License, v. 2.1.
 * This program is distributed in the hope that it will be useful, but WITHOUT A
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
 * PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more details.
 * You should have received a copy of the GNU Lesser General Public License,
 * v.2.1 along with this distribution; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA  02110-1301, USA.
 */

package org.jboss.as.console.client.shared.dispatch.impl;

import com.allen_sauer.gwt.log.client.Log;
import com.google.gwt.http.client.Request;
import com.google.gwt.http.client.RequestBuilder;
import com.google.gwt.http.client.RequestCallback;
import com.google.gwt.http.client.RequestException;
import com.google.gwt.http.client.Response;
import com.google.gwt.user.client.rpc.AsyncCallback;
import com.google.inject.Inject;
import org.jboss.as.console.client.core.BootstrapContext;
import org.jboss.as.console.client.core.UIConstants;
import org.jboss.as.console.client.debug.Diagnostics;
import org.jboss.as.console.client.shared.dispatch.ActionHandler;
import org.jboss.as.console.client.shared.dispatch.DispatchRequest;
import org.jboss.dmr.client.ModelNode;

import static org.jboss.dmr.client.ModelDescriptionConstants.*;

/**
 * @author Heiko Braun
 * @date 3/17/11
 */
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";

    private final RequestBuilder requestBuilder;

    private boolean trackInvocations = Diagnostics.isEnabled();

    private UIConstants constants;

    private static long idCounter = 0;

    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;
        }
    }

    @Inject
    public DMRHandler(BootstrapContext bootstrap, UIConstants constants) {

        this.constants = constants;

        requestBuilder = new RequestBuilder(RequestBuilder.POST,
                bootstrap.getProperty(BootstrapContext.DOMAIN_API));

        requestBuilder.setHeader(HEADER_ACCEPT, DMR_ENCODED);
        requestBuilder.setHeader(HEADER_CONTENT_TYPE, DMR_ENCODED);

    }

    @Override
    public DispatchRequest execute(DMRAction action, final AsyncCallback<DMRResponse> resultCallback) {

        assert action.getOperation() != null;

        final ModelNode operation = action.getOperation();

        Request request = executeRequest(resultCallback, operation);
        DispatchRequest handle = new DispatchRequestHandle(request);

        return handle;

    }

    private Request executeRequest(final AsyncCallback<DMRResponse> resultCallback, final ModelNode operation) {

        if (idCounter == Long.MAX_VALUE)
            idCounter = 0;

        Request requestHandle = null;
        try {

            final String id = String.valueOf(idCounter++);
            trace(Type.BEGIN, id, operation);

            String requestData = operation.toBase64String();
            trace(Type.SERIALIZED, id, operation);

            requestHandle = requestBuilder.sendRequest(requestData, 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(response.getText(), response.getHeader(HEADER_CONTENT_TYPE)));
                    } else if (401 == statusCode || 0 == statusCode) {
                        resultCallback.onFailure(new Exception("Authentication required."));

                    } 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 Exception("Service temporarily unavailable. Is the server is still booting?"));
                    } else {
                        StringBuilder sb = new StringBuilder();
                        sb.append(constants.common_error_unexpectedHttpResponse()).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("") ? constants.common_error_detailsMissing()
                                : ModelNode.fromBase64(response.getText()).toString();

                        sb.append(payload);
                        resultCallback.onFailure(new Exception(sb.toString()));
                    }

                    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);
                }
            });

            trace(Type.SEND, id, operation);

        } catch (RequestException e) {
            resultCallback.onFailure(e);
        }
        return requestHandle;
    }

    private void trace(Type type, String id, ModelNode operation) {

        if (!trackInvocations)
            return;

        //String token = getToken(operation);
        if (Type.BEGIN.equals(type)) {
            Diagnostics.logRpc(type.getClassifier(), id, System.currentTimeMillis(), getToken(operation));
        } else {
            Diagnostics.logRpc(type.getClassifier(), id, System.currentTimeMillis());
        }

    }

    public 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 undo(DMRAction action, DMRResponse result, AsyncCallback<Void> callback) {
        throw new RuntimeException("Not implemented yet.");
    }

    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;
        }
    }

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