Skip to content

Commit 6419e1c

Browse files
authored
[VMware to KVM Migration] Fix unused convert env vars (#11947)
* Fix unused convert env vars * Address review comments * Small fix to invoke internal method
1 parent b74f21b commit 6419e1c

File tree

3 files changed

+34
-9
lines changed

3 files changed

+34
-9
lines changed

plugins/hypervisors/kvm/src/main/java/com/cloud/hypervisor/kvm/resource/LibvirtComputingResource.java

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -882,7 +882,7 @@ protected enum HealthCheckResult {
882882
protected StorageSubsystemCommandHandler storageHandler;
883883

884884
private boolean convertInstanceVerboseMode = false;
885-
private String[] convertInstanceEnv = null;
885+
private Map<String, String> convertInstanceEnv = null;
886886
protected boolean dpdkSupport = false;
887887
protected String dpdkOvsPath;
888888
protected String directDownloadTemporaryDownloadPath;
@@ -947,7 +947,7 @@ public boolean isConvertInstanceVerboseModeEnabled() {
947947
return convertInstanceVerboseMode;
948948
}
949949

950-
public String[] getConvertInstanceEnv() {
950+
public Map<String, String> getConvertInstanceEnv() {
951951
return convertInstanceEnv;
952952
}
953953

@@ -1437,14 +1437,14 @@ private void setConvertInstanceEnv(String convertEnvTmpDir, String convertEnvVir
14371437
return;
14381438
}
14391439
if (StringUtils.isNotBlank(convertEnvTmpDir) && StringUtils.isNotBlank(convertEnvVirtv2vTmpDir)) {
1440-
convertInstanceEnv = new String[2];
1441-
convertInstanceEnv[0] = String.format("%s=%s", "TMPDIR", convertEnvTmpDir);
1442-
convertInstanceEnv[1] = String.format("%s=%s", "VIRT_V2V_TMPDIR", convertEnvVirtv2vTmpDir);
1440+
convertInstanceEnv = new HashMap<>(2);
1441+
convertInstanceEnv.put("TMPDIR", convertEnvTmpDir);
1442+
convertInstanceEnv.put("VIRT_V2V_TMPDIR", convertEnvVirtv2vTmpDir);
14431443
} else {
1444-
convertInstanceEnv = new String[1];
1444+
convertInstanceEnv = new HashMap<>(1);
14451445
String key = StringUtils.isNotBlank(convertEnvTmpDir) ? "TMPDIR" : "VIRT_V2V_TMPDIR";
14461446
String value = StringUtils.isNotBlank(convertEnvTmpDir) ? convertEnvTmpDir : convertEnvVirtv2vTmpDir;
1447-
convertInstanceEnv[0] = String.format("%s=%s", key, value);
1447+
convertInstanceEnv.put(key, value);
14481448
}
14491449
}
14501450

plugins/hypervisors/kvm/src/main/java/com/cloud/hypervisor/kvm/resource/wrapper/LibvirtConvertInstanceCommandWrapper.java

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,11 @@
2222
import java.nio.charset.Charset;
2323
import java.util.Arrays;
2424
import java.util.List;
25+
import java.util.Map;
2526
import java.util.UUID;
2627

2728
import org.apache.cloudstack.storage.to.PrimaryDataStoreTO;
29+
import org.apache.commons.collections4.MapUtils;
2830
import org.apache.commons.lang3.StringUtils;
2931

3032
import com.cloud.agent.api.Answer;
@@ -244,7 +246,12 @@ protected boolean performInstanceConversion(String originalVMName, String source
244246

245247
String logPrefix = String.format("(%s) virt-v2v ovf source: %s progress", originalVMName, sourceOVFDirPath);
246248
OutputInterpreter.LineByLineOutputLogger outputLogger = new OutputInterpreter.LineByLineOutputLogger(logger, logPrefix);
247-
script.execute(outputLogger);
249+
Map<String, String> convertInstanceEnv = serverResource.getConvertInstanceEnv();
250+
if (MapUtils.isEmpty(convertInstanceEnv)) {
251+
script.execute(outputLogger);
252+
} else {
253+
script.execute(outputLogger, convertInstanceEnv);
254+
}
248255
int exitValue = script.getExitValue();
249256
return exitValue == 0;
250257
}

utils/src/main/java/com/cloud/utils/script/Script.java

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
import java.util.Arrays;
3232
import java.util.HashSet;
3333
import java.util.List;
34+
import java.util.Map;
3435
import java.util.Properties;
3536
import java.util.Set;
3637
import java.util.concurrent.Callable;
@@ -44,6 +45,7 @@
4445
import java.util.concurrent.atomic.AtomicReference;
4546
import java.util.stream.Collectors;
4647

48+
import org.apache.commons.collections4.MapUtils;
4749
import org.apache.commons.collections.CollectionUtils;
4850
import org.apache.commons.io.IOUtils;
4951
import org.apache.logging.log4j.LogManager;
@@ -243,6 +245,14 @@ static String stackTraceAsString(Throwable throwable) {
243245
}
244246

245247
public String execute(OutputInterpreter interpreter) {
248+
return executeInternal(interpreter, null);
249+
}
250+
251+
public String execute(OutputInterpreter interpreter, Map<String, String> environment) {
252+
return executeInternal(interpreter, environment);
253+
}
254+
255+
private String executeInternal(OutputInterpreter interpreter, Map<String, String> environment) {
246256
String[] command = _command.toArray(new String[_command.size()]);
247257
String commandLine = buildCommandLine(command);
248258
if (_logger.isDebugEnabled() ) {
@@ -254,11 +264,19 @@ public String execute(OutputInterpreter interpreter) {
254264

255265
ProcessBuilder pb = new ProcessBuilder(command);
256266
pb.redirectErrorStream(true);
257-
if (_workDir != null)
267+
268+
if (MapUtils.isNotEmpty(environment)) {
269+
Map<String, String> processEnvironment = pb.environment();
270+
processEnvironment.putAll(environment);
271+
}
272+
273+
if (_workDir != null) {
258274
pb.directory(new File(_workDir));
275+
}
259276

260277
_logger.trace(String.format("Starting process for command [%s].", commandLine));
261278
_process = pb.start();
279+
262280
if (_process == null) {
263281
_logger.warn(String.format("Unable to execute command [%s] because no process was created.", commandLine));
264282
return "Unable to execute the command: " + command[0];

0 commit comments

Comments
 (0)