Skip to content

Commit

Permalink
Configures ivy to download jar libraries only from remote
Browse files Browse the repository at this point in the history
  • Loading branch information
Reamer committed Oct 21, 2022
1 parent 7a370ea commit b74631a
Show file tree
Hide file tree
Showing 2 changed files with 24 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@

package org.apache.zeppelin.integration;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.apache.hadoop.yarn.api.protocolrecords.GetApplicationsRequest;
import org.apache.hadoop.yarn.api.protocolrecords.GetApplicationsResponse;
Expand Down Expand Up @@ -46,6 +47,7 @@
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.EnumSet;
import java.util.List;
import java.util.stream.Collectors;
Expand Down Expand Up @@ -101,6 +103,18 @@ protected void setUpSparkInterpreterSetting(InterpreterSetting interpreterSettin
// sub class can customize spark interpreter setting.
}

/**
* Configures ivy to download jar libraries only from remote.
*
* @param interpreterSetting
* @throws IOException
*/
private void setupIvySettings(InterpreterSetting interpreterSetting) throws IOException {
File ivysettings = new File(zeppelin.getZeppelinConfDir(), "ivysettings.xml");
FileUtils.copyToFile(SparkIntegrationTest.class.getResourceAsStream("/ivysettings.xml"), ivysettings);
interpreterSetting.setProperty("spark.jars.ivySettings", ivysettings.getAbsolutePath());
}

private boolean isHadoopVersionMatch() {
String version = org.apache.hadoop.util.VersionInfo.getVersion();
String majorVersion = version.split("\\.")[0];
Expand Down Expand Up @@ -184,6 +198,7 @@ public void testLocalMode() throws IOException, YarnException, InterpreterExcept

try {
setUpSparkInterpreterSetting(sparkInterpreterSetting);
setupIvySettings(sparkInterpreterSetting);
testInterpreterBasics();

// no yarn application launched
Expand Down Expand Up @@ -215,6 +230,7 @@ public void testYarnClientMode() throws IOException, YarnException, InterruptedE

try {
setUpSparkInterpreterSetting(sparkInterpreterSetting);
setupIvySettings(sparkInterpreterSetting);
testInterpreterBasics();

// 1 yarn application launched
Expand Down Expand Up @@ -270,6 +286,7 @@ public void testYarnClusterMode() throws IOException, YarnException, Interrupted
String yarnAppId = null;
try {
setUpSparkInterpreterSetting(sparkInterpreterSetting);
setupIvySettings(sparkInterpreterSetting);
testInterpreterBasics();

// 1 yarn application launched
Expand Down Expand Up @@ -366,6 +383,6 @@ private String getPythonExec() throws IOException, InterruptedException {
if (process.waitFor() != 0) {
throw new RuntimeException("Fail to run command: which python.");
}
return IOUtils.toString(process.getInputStream()).trim();
return IOUtils.toString(process.getInputStream(), StandardCharsets.UTF_8).trim();
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
<ivysettings>
<settings defaultResolver="central"/>
<resolvers>
<ibiblio name="central" m2compatible="true" root="https://repo1.maven.org/maven2/"/>
</resolvers>
</ivysettings>

0 comments on commit b74631a

Please sign in to comment.