477 lines
11 KiB
Java
477 lines
11 KiB
Java
package PSEditor;
|
|
|
|
import java.awt.BorderLayout;
|
|
import java.awt.Color;
|
|
import java.awt.FlowLayout;
|
|
|
|
import javax.swing.JButton;
|
|
import javax.swing.JDialog;
|
|
import javax.swing.JPanel;
|
|
import javax.swing.SwingWorker;
|
|
import javax.swing.border.EmptyBorder;
|
|
import javax.swing.GroupLayout;
|
|
import javax.swing.GroupLayout.Alignment;
|
|
import javax.swing.JLabel;
|
|
import java.awt.Font;
|
|
import javax.swing.JScrollPane;
|
|
import javax.swing.LayoutStyle.ComponentPlacement;
|
|
import java.awt.event.ActionListener;
|
|
import java.awt.event.ActionEvent;
|
|
import java.io.BufferedInputStream;
|
|
import java.io.BufferedReader;
|
|
import java.io.IOException;
|
|
import java.io.InputStreamReader;
|
|
import java.util.HashSet;
|
|
import java.util.List;
|
|
import java.util.ArrayList;
|
|
import java.util.Set;
|
|
import java.util.concurrent.ExecutionException;
|
|
|
|
import javax.swing.JTextArea;
|
|
|
|
public class CheckMsgDlg extends JDialog {
|
|
|
|
static private final String newline = "\n";
|
|
private final JPanel contentPanel = new JPanel();
|
|
private JTextArea msgText;
|
|
private JButton okButton;
|
|
|
|
private List<DataServer> cluster;
|
|
private String userName;
|
|
private boolean ns4m, selfClose;
|
|
private boolean available;
|
|
private boolean chkHadoop = false;
|
|
private List<Integer> hdpPorts;
|
|
|
|
|
|
/**
|
|
* Create the dialog.
|
|
*/
|
|
public CheckMsgDlg() {
|
|
|
|
available = false;
|
|
setTitle("Cluster Checking");
|
|
setBounds(100, 100, 450, 300);
|
|
getContentPane().setLayout(new BorderLayout());
|
|
contentPanel.setBorder(new EmptyBorder(5, 5, 5, 5));
|
|
getContentPane().add(contentPanel, BorderLayout.CENTER);
|
|
|
|
JLabel lblNewLabel = new JLabel("Checking the defined cluster now: ");
|
|
lblNewLabel.setFont(new Font("Lucida Grande", Font.PLAIN, 15));
|
|
|
|
JScrollPane scrollPane = new JScrollPane();
|
|
scrollPane.setVerticalScrollBarPolicy(
|
|
JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED);
|
|
|
|
GroupLayout gl_contentPanel = new GroupLayout(contentPanel);
|
|
gl_contentPanel.setHorizontalGroup(
|
|
gl_contentPanel.createParallelGroup(Alignment.LEADING)
|
|
.addGroup(gl_contentPanel.createSequentialGroup()
|
|
.addContainerGap()
|
|
.addGroup(gl_contentPanel.createParallelGroup(Alignment.LEADING)
|
|
.addComponent(scrollPane, GroupLayout.PREFERRED_SIZE, 429, GroupLayout.PREFERRED_SIZE)
|
|
.addComponent(lblNewLabel))
|
|
.addContainerGap(GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE))
|
|
);
|
|
gl_contentPanel.setVerticalGroup(
|
|
gl_contentPanel.createParallelGroup(Alignment.LEADING)
|
|
.addGroup(gl_contentPanel.createSequentialGroup()
|
|
.addContainerGap()
|
|
.addComponent(lblNewLabel)
|
|
.addPreferredGap(ComponentPlacement.RELATED)
|
|
.addComponent(scrollPane, GroupLayout.DEFAULT_SIZE, 192, Short.MAX_VALUE)
|
|
.addContainerGap())
|
|
);
|
|
|
|
msgText = new JTextArea();
|
|
msgText.setEditable(false);
|
|
msgText.setLineWrap(true);
|
|
msgText.setForeground(Color.BLACK);
|
|
|
|
scrollPane.setViewportView(msgText);
|
|
|
|
|
|
contentPanel.setLayout(gl_contentPanel);
|
|
{
|
|
JPanel buttonPane = new JPanel();
|
|
buttonPane.setLayout(new FlowLayout(FlowLayout.RIGHT));
|
|
getContentPane().add(buttonPane, BorderLayout.SOUTH);
|
|
{
|
|
okButton = new JButton("OK");
|
|
okButton.addActionListener(new ActionListener() {
|
|
public void actionPerformed(ActionEvent arg0) {
|
|
setVisible(false);
|
|
dispose();
|
|
}
|
|
});
|
|
okButton.setActionCommand("OK");
|
|
buttonPane.add(okButton);
|
|
getRootPane().setDefaultButton(okButton);
|
|
}
|
|
}
|
|
}
|
|
|
|
public boolean CheckCluster(String uname,
|
|
ArrayList<DataServer> c,
|
|
SCReader hdPara, // Hadoop parameters
|
|
boolean n, //ns4m
|
|
boolean sc) //self close the window
|
|
{
|
|
setVisible(true);
|
|
|
|
if (c == null)
|
|
return false;
|
|
|
|
if (c.isEmpty()){
|
|
msgText.append("The cluster is empty!");
|
|
return false;
|
|
}
|
|
|
|
cluster = new ArrayList<DataServer>(c.size());
|
|
for (DataServer ds : c){
|
|
cluster.add(ds);
|
|
}
|
|
userName = uname;
|
|
ns4m = n;
|
|
selfClose = sc;
|
|
|
|
if (hdPara != null)
|
|
{
|
|
ArrayList<String> hdpTitles = hdPara.getTitles("Hadoop");
|
|
|
|
if (!hdpTitles.isEmpty())
|
|
{
|
|
chkHadoop = true;
|
|
hdpPorts = new ArrayList<Integer>();
|
|
for (String t : hdpTitles)
|
|
{
|
|
String value = hdPara.get("Hadoop", t);
|
|
if (value.contains(":"))
|
|
{
|
|
String sport = value.substring(value.lastIndexOf(":") + 1);
|
|
try
|
|
{
|
|
int port = Integer.parseInt(sport);
|
|
hdpPorts.add(port);
|
|
} catch (NumberFormatException e){
|
|
System.err.println("Warning!! Invalid port number " + sport);
|
|
}
|
|
}
|
|
}
|
|
if (hdpPorts.isEmpty())
|
|
chkHadoop = false;
|
|
}
|
|
}
|
|
|
|
okButton.setEnabled(false);
|
|
|
|
CheckWorker worker = new CheckWorker();
|
|
worker.execute();
|
|
|
|
return isAvailable();
|
|
}
|
|
|
|
public boolean isAvailable()
|
|
{
|
|
return available;
|
|
}
|
|
|
|
public boolean willCreateFile()
|
|
{
|
|
return selfClose;
|
|
}
|
|
|
|
|
|
class CheckWorker extends SwingWorker<Boolean, String>
|
|
{
|
|
private boolean running = true;
|
|
boolean chkStatus = true;
|
|
|
|
|
|
@Override
|
|
protected Boolean doInBackground() throws Exception{
|
|
running = true;
|
|
chkStatus = true;
|
|
|
|
ArrayList<String> nodeIPs = new ArrayList<String>();
|
|
for (DataServer ds : cluster)
|
|
{
|
|
String ip = ds.getIp().trim();
|
|
if (!nodeIPs.contains(ip))
|
|
nodeIPs.add(ip);
|
|
}
|
|
|
|
String cmd;
|
|
boolean rtn;
|
|
try{
|
|
|
|
boolean mfound = false; //find the master
|
|
int slaveCnt = 0; //slave counter
|
|
for (DataServer ds: cluster)
|
|
{
|
|
if (ds.getRole() == DataServer.ROLES.SLAVE){
|
|
slaveCnt++;
|
|
}
|
|
else
|
|
{
|
|
if (ds.getRole() == DataServer.ROLES.MASLAVE)
|
|
slaveCnt++;
|
|
|
|
if (!mfound)
|
|
mfound = true;
|
|
else
|
|
{
|
|
publish("ERROR! Multiple master Data Server is set.");
|
|
chkStatus = false;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if (!mfound){
|
|
publish("ERROR! No Master Data Server is defined. ");
|
|
chkStatus = false;
|
|
}
|
|
if (slaveCnt == 0){
|
|
publish("ERROR! There is no one slave Data Server defined.");
|
|
chkStatus = false;
|
|
}
|
|
if (!chkStatus){
|
|
throw new Exception("Check failed");
|
|
}
|
|
|
|
publish("------------------------------------------------");
|
|
|
|
//Check the accessibility of all cluster nodes
|
|
for (String ip : nodeIPs)
|
|
{
|
|
cmd = "ssh -q " +
|
|
"-o BatchMode=yes " +
|
|
"-o ConnectTimeout=3 " +
|
|
userName + "@" + ip + " echo 1>/dev/null";
|
|
|
|
rtn = runCommand(cmd);
|
|
if (rtn){
|
|
publish("Node " + ip + " is accessible.");
|
|
}
|
|
else{
|
|
publish("ERROR! Node " + ip +
|
|
" is NOT accessible without passphrase.");
|
|
chkStatus = false;
|
|
break;
|
|
}
|
|
}
|
|
if (!chkStatus){
|
|
throw new Exception("Check failed");
|
|
}
|
|
|
|
publish("------------------------------------------------");
|
|
|
|
//Check screen installed on every machine
|
|
for (String ip : nodeIPs)
|
|
{
|
|
cmd = "ssh -q " +
|
|
"-o BatchMode=yes " +
|
|
"-o ConnectTimeout=3 " +
|
|
userName + "@" + ip +
|
|
" which screen >/dev/null";
|
|
|
|
rtn = runCommand(cmd);
|
|
if (rtn){
|
|
publish("Node " + ip + " has the program 'screen'.");
|
|
}
|
|
else{
|
|
publish("ERROR! Node " + ip +
|
|
" NOT have the program 'screen'.");
|
|
chkStatus = false;
|
|
break;
|
|
}
|
|
if (!chkStatus)
|
|
return chkStatus;
|
|
}
|
|
|
|
publish("------------------------------------------------");
|
|
|
|
// Check paths exist on all nodes
|
|
// TODO also check the R/W grant
|
|
for (DataServer ds : cluster)
|
|
{
|
|
cmd = "ssh -q " + "-o BatchMode=yes " + "-o ConnectTimeout=3 "
|
|
+ userName + "@" + ds.getIp() + " ls " + ds.getPath()
|
|
+ " 2>&1 1>/dev/null";
|
|
|
|
rtn = runCommand(cmd);
|
|
|
|
if (rtn) {
|
|
publish("Check Data Server " + ds.getIp() + ":" + ds.getPath()
|
|
+ " SUCCESS.");
|
|
} else {
|
|
publish("Check Data Server " + ds.getIp() + ":" + ds.getPath()
|
|
+ " FAILS. ");
|
|
chkStatus = false;
|
|
break;
|
|
}
|
|
}
|
|
if (!chkStatus){
|
|
throw new Exception("Check failed");
|
|
}
|
|
|
|
|
|
publish("------------------------------------------------");
|
|
|
|
//Check the ports are not used by any running nodes
|
|
for (DataServer ds : cluster)
|
|
{
|
|
cmd = "ssh -q " +
|
|
"-o BatchMode=yes " +
|
|
"-o ConnectTimeout=3 " +
|
|
userName + "@" + ds.getIp() +
|
|
" lsof -i:" + ds.getPort() + " 2>&1 1>/dev/null";
|
|
|
|
//Here if the port is taken then return TRUE, or else return FALSE
|
|
rtn = runCommand(cmd);
|
|
|
|
if (!rtn){
|
|
publish("Port " + ds.getPort() + " is available on node " +
|
|
ds.getIp());
|
|
}
|
|
else{
|
|
publish("Port " + ds.getPort() + " is NOT available on node " +
|
|
ds.getIp());
|
|
chkStatus = false;
|
|
break;
|
|
}
|
|
}
|
|
if (!chkStatus){
|
|
throw new Exception("Check failed");
|
|
}
|
|
|
|
// Check Hadoop ports
|
|
if (chkHadoop)
|
|
{
|
|
publish("------------------------------------------------");
|
|
Set<Integer> set = new HashSet<Integer>(hdpPorts);
|
|
if (set.size() < hdpPorts.size()){
|
|
publish("There exist duplicated Hadoop port setting.");
|
|
chkStatus = false;
|
|
}
|
|
if (!chkStatus){
|
|
throw new Exception("Check failed");
|
|
}
|
|
|
|
for (String ip : nodeIPs)
|
|
{
|
|
for ( int port : hdpPorts )
|
|
{
|
|
cmd = "ssh -q " +
|
|
"-o BatchMode=yes " +
|
|
"-o ConnectTimeout=3 " +
|
|
userName + "@" + ip +
|
|
" lsof -i:" + port + " 2>&1 1>/dev/null";
|
|
|
|
rtn = runCommand(cmd);
|
|
if (!rtn){
|
|
publish("Port " + port + " is available on node " + ip);
|
|
}
|
|
else{
|
|
publish("Port " + port + " is NOT available on node " + ip);
|
|
chkStatus = false;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!chkStatus){
|
|
throw new Exception("Check failed");
|
|
}
|
|
}
|
|
}
|
|
|
|
publish("------------------------------------------------");
|
|
publish("All Checks are done, " +
|
|
"and the cluster is available to install Parallel SECONDO on it.");
|
|
|
|
} catch (Exception e){
|
|
publish(e.toString());
|
|
}
|
|
finally {
|
|
running = false;
|
|
return chkStatus;
|
|
}
|
|
}
|
|
|
|
public boolean isRunning()
|
|
{
|
|
return running;
|
|
}
|
|
|
|
public boolean checkResult()
|
|
{
|
|
return chkStatus;
|
|
}
|
|
|
|
protected boolean runCommand (String cmd) throws InterruptedException, IOException
|
|
{
|
|
Runtime run = Runtime.getRuntime();
|
|
Process p;
|
|
BufferedInputStream in;
|
|
BufferedReader inBr;
|
|
String rtnLine;
|
|
boolean result;
|
|
|
|
p = run.exec(cmd);
|
|
in = new BufferedInputStream(p.getInputStream());
|
|
inBr = new BufferedReader(new InputStreamReader(in));
|
|
|
|
while ((rtnLine = inBr.readLine()) != null){
|
|
publish(rtnLine);
|
|
}
|
|
|
|
if (p.waitFor() != 0){
|
|
result = false;
|
|
}
|
|
else {
|
|
result = true;
|
|
}
|
|
|
|
inBr.close();
|
|
in.close();
|
|
return result;
|
|
}
|
|
|
|
@Override
|
|
protected void process(List<String> chunks){
|
|
for (String chunk : chunks){
|
|
msgText.append(chunk + newline);
|
|
msgText.setCaretPosition(msgText.getText().length() - 1);
|
|
}
|
|
}
|
|
|
|
/* (non-Javadoc)
|
|
* @see javax.swing.SwingWorker#done()
|
|
*/
|
|
@Override
|
|
protected void done() {
|
|
// TODO Auto-generated method stub
|
|
try {
|
|
available = get();
|
|
okButton.setEnabled(true);
|
|
if (selfClose && isAvailable()){
|
|
setVisible(false);
|
|
dispose();
|
|
}
|
|
|
|
|
|
} catch (InterruptedException e) {
|
|
// TODO Auto-generated catch block
|
|
e.printStackTrace();
|
|
} catch (ExecutionException e) {
|
|
// TODO Auto-generated catch block
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|