org.datacleaner.panels.HadoopClusterPanel Maven / Gradle / Ivy

Go to download
/**
 * DataCleaner (community edition)
 * Copyright (C) 2014 Free Software Foundation, Inc.
 *
 * 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, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY 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
 * along with this distribution; if not, write to:
 * Free Software Foundation, Inc.
 * 51 Franklin Street, Fifth Floor
 * Boston, MA  02110-1301  USA
 */
package org.datacleaner.panels;

import java.awt.GridBagConstraints;

import javax.swing.JButton;
import javax.swing.JOptionPane;

import org.datacleaner.bootstrap.WindowContext;
import org.datacleaner.configuration.ServerInformation;
import org.datacleaner.server.DirectConnectionHadoopClusterInformation;
import org.datacleaner.server.DirectoryBasedHadoopClusterInformation;
import org.datacleaner.server.EnvironmentBasedHadoopClusterInformation;
import org.datacleaner.user.MutableServerInformationCatalog;
import org.datacleaner.util.IconUtils;
import org.datacleaner.util.WidgetFactory;
import org.datacleaner.util.WidgetUtils;
import org.datacleaner.widgets.DCLabel;
import org.datacleaner.windows.DirectConnectionHadoopClusterDialog;

public class HadoopClusterPanel extends DCPanel {

    private static final long serialVersionUID = 1L;
    private final MutableServerInformationCatalog _serverInformationCatalog;
    private final ServerInformation _serverInformation;
    private final WindowContext _windowContext;

    public HadoopClusterPanel(final WindowContext windowContext, final ServerInformation serverInformation,
            final MutableServerInformationCatalog serverInformationCatalog) {
        _serverInformationCatalog = serverInformationCatalog;
        _serverInformation = serverInformation;
        _windowContext = windowContext;

        final String serverName;
        if (serverInformation.getClass().equals(EnvironmentBasedHadoopClusterInformation.class)) {
            serverName = "(default)";
        } else {
            serverName = serverInformation.getName();
        }

        final String description;
        if (serverInformation.getClass().equals(EnvironmentBasedHadoopClusterInformation.class)) {
            description = "Use HADOOP_CONF_DIR and YARN_CONF_DIR";
        } else {
            final String serverDescription = serverInformation.getDescription();
            if (serverDescription != null) {
                description = serverDescription;
            } else {
                description = "";
            }
        }

        final DCLabel label = DCLabel.dark("" + serverName + "" + "
" + description + "");

        final JButton editButton = createEditButton();
        final JButton removeButton = createRemoveButton(serverInformation);
        if (serverInformation.getClass().equals(EnvironmentBasedHadoopClusterInformation.class)) {
            editButton.setEnabled(false);
            removeButton.setEnabled(false);
        }

        setBorder(WidgetUtils.BORDER_LIST_ITEM_SUBTLE);

        WidgetUtils.addToGridBag(DCPanel.flow(label), this, 1, 0, GridBagConstraints.WEST, 1.0, 1.0);
        WidgetUtils.addToGridBag(editButton, this, 2, 0, GridBagConstraints.EAST);
        WidgetUtils.addToGridBag(removeButton, this, 3, 0, GridBagConstraints.EAST);
    }

    private JButton createEditButton() {
        final JButton editButton = WidgetFactory.createDefaultButton("Edit", IconUtils.ACTION_EDIT);
        editButton.setToolTipText("Edit Hadoop cluster");
        editButton.addActionListener(e -> {
            if (_serverInformation instanceof DirectConnectionHadoopClusterInformation) {
                final DirectConnectionHadoopClusterDialog hadoopConnectionToNamenodeDialog =
                        new DirectConnectionHadoopClusterDialog(_windowContext,
                                (DirectConnectionHadoopClusterInformation) _serverInformation,
                                _serverInformationCatalog);
                hadoopConnectionToNamenodeDialog.setVisible(true);
            } else if (_serverInformation.getClass().equals(DirectoryBasedHadoopClusterInformation.class)) {
                final DirectoryBasedHadoopClusterDialog hadoopDirectoryConfigurationDialog =
                        new DirectoryBasedHadoopClusterDialog(_windowContext,
                                (DirectoryBasedHadoopClusterInformation) _serverInformation, _serverInformationCatalog);
                hadoopDirectoryConfigurationDialog.setVisible(true);
            }
        });
        return editButton;
    }

    private JButton createRemoveButton(final ServerInformation serverInformation) {
        final JButton removeButton = WidgetFactory.createDefaultButton("Remove", IconUtils.ACTION_REMOVE_DARK);
        removeButton.setToolTipText("Remove Hadoop cluster");
        removeButton.addActionListener(e -> {
            final int result = JOptionPane.showConfirmDialog(HadoopClusterPanel.this,
                    "Are you sure you wish to remove the Hadoop cluster '" + serverInformation.getName() + "'?",
                    "Confirm remove", JOptionPane.YES_NO_OPTION);
            if (result == JOptionPane.YES_OPTION) {
                _serverInformationCatalog.removeServer(_serverInformation);
            }
        });
        return removeButton;
    }
}