|
23 | 23 | import com.google.common.base.Strings; |
24 | 24 | import com.google.common.collect.ImmutableList; |
25 | 25 | import com.google.common.collect.Iterables; |
| 26 | +import com.google.common.io.Files; |
26 | 27 | import com.google.common.util.concurrent.ThreadFactoryBuilder; |
27 | 28 | import com.google.common.util.concurrent.Uninterruptibles; |
28 | 29 | import io.cdap.cdap.api.spark.SparkSpecification; |
|
35 | 36 | import org.slf4j.Logger; |
36 | 37 | import org.slf4j.LoggerFactory; |
37 | 38 |
|
| 39 | +import java.io.File; |
38 | 40 | import java.net.URI; |
39 | 41 | import java.util.Arrays; |
40 | 42 | import java.util.Collections; |
@@ -244,13 +246,30 @@ private List<String> createSubmitArguments(SparkRuntimeContext runtimeContext, M |
244 | 246 | BiConsumer<String, String> confAdder = (k, v) -> builder.add("--conf").add(k + "=" + v); |
245 | 247 | configs.forEach(confAdder); |
246 | 248 |
|
| 249 | + |
| 250 | + String artifactTry = null; |
247 | 251 | LOG.warn("SANKET : createSubmitArguments : ALL archives "); |
248 | 252 | for (LocalizeResource lr : archivesIterable){ |
249 | 253 | 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 | + } |
250 | 264 | } |
251 | 265 |
|
252 | 266 | String archives = Joiner.on(',').join(Iterables.transform(archivesIterable, |
253 | 267 | getLocalizeResourceToURIFunc())); |
| 268 | + |
| 269 | + if (artifactTry != null){ |
| 270 | + archives = archives + ",file:" +artifactTry; |
| 271 | + } |
| 272 | + |
254 | 273 | String files = Joiner.on(',').join(Iterables.transform(filesIterable, getLocalizeResourceToURIFunc())); |
255 | 274 |
|
256 | 275 | if (!Strings.isNullOrEmpty(archives)) { |
|
0 commit comments