blob: 16f0f2f09e006809ee92fa37fc8129116c1b59c0 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
/*******************************************************************************
* Copyright (c) 2006-2011 Gluster, Inc. <http://www.gluster.com>
* This file is part of Gluster Management Console.
*
* Gluster Management Console is free software; you can redistribute
* it and/or modify it under the terms of the GNU General Public
* License as published by the Free Software Foundation; either
* version 3 of the License, or (at your option) any later version.
*
* Gluster Management Console 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see
* <http://www.gnu.org/licenses/>.
*******************************************************************************/
package org.gluster.storage.management.core.utils;
import static org.gluster.storage.management.core.constants.CoreConstants.NEWLINE;
import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import org.gluster.storage.management.core.exceptions.GlusterRuntimeException;
/**
* Utility class for creating processes (foreground/background) with given
* command and checking the output/exit status
*/
public class ProcessUtil {
public static ProcessResult executeCommand(List<String> command) {
return executeCommand(true, command);
}
/**
* Waits till all the threads in given list are dead
* @param threads Threads to watch
* @throws InterruptedException
*/
public static void waitForThreads(List<Thread> threads) throws InterruptedException {
// Wait till all servers have been processed
for (int i = threads.size() - 1; i >= 0; i--) {
if (threads.get(i).isAlive()) {
// thread alive. sleep for half a second and check again.
Thread.sleep(500);
i++; // check the same thread in next iteration
}
}
}
/**
* Executes given command in a separate process in FOREGROUND
* @param command
* @return {@link ProcessResult} object
*/
public static ProcessResult executeCommand(String... command) {
ArrayList<String> commandList = new ArrayList<String>();
for (String part : command) {
commandList.add(part);
}
return executeCommand(commandList);
}
/**
* Executes given command in foreground/background
* @param runInForeground Boolean flag indicating whether the command should
* be executed in foreground
* @param command
* @return {@link ProcessResult} object
*/
public static ProcessResult executeCommand(boolean runInForeground, List<String> command) {
StringBuilder output = new StringBuilder();
try {
Process process = new ProcessBuilder(command).redirectErrorStream(true).start();
if (runInForeground) {
process.waitFor(); // Wait for process to finish
InputStream is = process.getInputStream();
InputStreamReader isr = new InputStreamReader(is);
BufferedReader br = new BufferedReader(isr);
String line;
while ((line = br.readLine()) != null) {
output.append(line);
output.append(NEWLINE);
}
br.close();
isr.close();
is.close();
} else {
output.append("Command [");
output.append(command);
output.append("] triggerred in background.");
}
return new ProcessResult(process.exitValue(), output.toString());
} catch (Throwable e) {
throw new GlusterRuntimeException("Exception while executing command [" + command + "] : ["
+ e.getMessage() + "]", e);
}
}
}
|
