Skip to content

Commit 78488d4

Browse files
committed
Manually renaming artifacts archive
1 parent 993e665 commit 78488d4

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

cdap-spark-core-base/src/main/java/io/cdap/cdap/app/runtime/spark/submit/AbstractSparkSubmitter.java

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
import com.google.common.base.Strings;
2424
import com.google.common.collect.ImmutableList;
2525
import com.google.common.collect.Iterables;
26+
import com.google.common.io.Files;
2627
import com.google.common.util.concurrent.ThreadFactoryBuilder;
2728
import com.google.common.util.concurrent.Uninterruptibles;
2829
import io.cdap.cdap.api.spark.SparkSpecification;
@@ -35,6 +36,7 @@
3536
import org.slf4j.Logger;
3637
import org.slf4j.LoggerFactory;
3738

39+
import java.io.File;
3840
import java.net.URI;
3941
import java.util.Arrays;
4042
import java.util.Collections;
@@ -244,13 +246,30 @@ private List<String> createSubmitArguments(SparkRuntimeContext runtimeContext, M
244246
BiConsumer<String, String> confAdder = (k, v) -> builder.add("--conf").add(k + "=" + v);
245247
configs.forEach(confAdder);
246248

249+
250+
String artifactTry = null;
247251
LOG.warn("SANKET : createSubmitArguments : ALL archives ");
248252
for (LocalizeResource lr : archivesIterable){
249253
LOG.warn("SANKET : archivesIterable : " + lr.getURI());
254+
if (lr.getURI().getPath().contains("artifacts_archive")){
255+
LOG.warn("SANKET : archivesIterable : COPYING : " + lr.getURI());
256+
File tmpDir = Files.createTempDir();
257+
File artifacts_archive_jar = tmpDir.toPath().resolve("artifacts_archive.jar").toFile();
258+
File file = new File(lr.getURI());
259+
Files.copy(file, artifacts_archive_jar);
260+
LOG.warn("SANKET : archivesIterable : COPIED to : " + artifacts_archive_jar.getAbsolutePath());
261+
artifactTry = artifacts_archive_jar.getAbsolutePath();
262+
263+
}
250264
}
251265

252266
String archives = Joiner.on(',').join(Iterables.transform(archivesIterable,
253267
getLocalizeResourceToURIFunc()));
268+
269+
if (artifactTry != null){
270+
archives = archives + ",file:" +artifactTry;
271+
}
272+
254273
String files = Joiner.on(',').join(Iterables.transform(filesIterable, getLocalizeResourceToURIFunc()));
255274

256275
if (!Strings.isNullOrEmpty(archives)) {

0 commit comments

Comments
 (0)