design/designer-realize/src/main/java/com/fr/design/mainframe/socketio/DesignerSocketIO.java

package com.fr.design.mainframe.socketio;

import com.fr.config.RemoteConfigEvent;
import com.fr.decision.webservice.utils.DecisionServiceConstants;
import com.fr.design.mainframe.DesignerContext;
import com.fr.design.mainframe.TemplatePane;
import com.fr.design.mainframe.loghandler.DesignerLogHandler;
import com.fr.event.EventDispatcher;
import com.fr.log.FineLoggerFactory;
import com.fr.serialization.SerializerHelper;
import com.fr.stable.ArrayUtils;
import com.fr.third.apache.log4j.spi.LoggingEvent;
import com.fr.third.guava.base.Optional;
import com.fr.workspace.WorkContext;
import com.fr.workspace.Workspace;
import com.fr.workspace.base.WorkspaceConstants;
import com.fr.workspace.engine.server.rpc.netty.RemoteCallClient;
import com.fr.workspace.server.socket.SocketInfoOperator;
import io.socket.client.IO;
import io.socket.client.Socket;
import io.socket.emitter.Emitter;

import javax.swing.JOptionPane;
import javax.swing.SwingUtilities;
import javax.swing.UIManager;
import java.io.IOException;
import java.net.URI;
import java.net.URL;

public class DesignerSocketIO {

    enum Status {
        Connected,
        Disconnected,
        Disconnecting
    }

    private static Optional<Socket> socketIO = Optional.absent();
    public static Status status = Status.Disconnected;

    private static final Emitter.Listener printLog = new Emitter.Listener() {
        @Override
        public void call(Object... objects) {
            if (ArrayUtils.isNotEmpty(objects)) {
                try {
                    LoggingEvent event = SerializerHelper.deserialize((byte[]) objects[0]);
                    DesignerLogHandler.getInstance().printLoggingEvent(event);
                } catch (Exception e) {
                    FineLoggerFactory.getLogger().error(e.getMessage(), e);
                }
            }
        }
    };

    public static void close() {
        if (socketIO.isPresent()) {
            status = Status.Disconnecting;
            socketIO.get().close();
            socketIO = Optional.absent();
        }
    }

    public static void update() {
        Workspace current = WorkContext.getCurrent();
        if (current.isLocal()) {
            return;
        }
        try {
            String uri = getSocketUri(current);
            socketIO = Optional.of(IO.socket(new URI(uri)));
            socketIO.get().on(WorkspaceConstants.WS_LOGRECORD, printLog);
            socketIO.get().on(WorkspaceConstants.CONFIG_MODIFY, new Emitter.Listener() {
                @Override
                public void call(Object... objects) {
                    assert objects != null && objects.length == 1;
                    String param = (String) objects[0];
                    EventDispatcher.fire(RemoteConfigEvent.EDIT, param);
                }
            });
            socketIO.get().on(Socket.EVENT_DISCONNECT, new Emitter.Listener() {
                @Override
                public void call(Object... objects) {
                    if (status != Status.Disconnecting) {
                        try {
                            SwingUtilities.invokeAndWait(new Runnable() {
                                public void run() {
                                    JOptionPane.showMessageDialog(DesignerContext.getDesignerFrame(), com.fr.design.i18n.Toolkit.i18nText("Fine-Design_Basic_Remote_Disconnected"),
                                            null, 0, UIManager.getIcon("OptionPane.errorIcon"));
                                    TemplatePane.getInstance().editItems();
                                }
                            });
                        } catch (Exception e) {
                            FineLoggerFactory.getLogger().error(e.getMessage(), e);
                        }
                    }
                    status = Status.Disconnected;
                }
            });
            socketIO.get().connect();
            status = Status.Connected;
        } catch (Exception e) {
            FineLoggerFactory.getLogger().error(e.getMessage(), e);
        }
    }

    private static String getSocketUri(Workspace current) throws IOException {
        URL url = new URL(current.getPath());
        int port = WorkContext.getCurrent().get(SocketInfoOperator.class).getPort();
        return String.format("%s://%s:%s%s?%s=%s",
                url.getProtocol(),
                url.getHost(),
                port,
                WorkspaceConstants.WS_NAMESPACE,
                DecisionServiceConstants.WEB_SOCKET_TOKEN_NAME,
                RemoteCallClient.getInstance().getToken());
    }
}