aljoscha commented on a change in pull request #11991: URL: https://github.com/apache/flink/pull/11991#discussion_r419939292
########## File path: flink-yarn/src/main/java/org/apache/flink/yarn/YarnApplicationFileUploader.java ########## @@ -0,0 +1,283 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.flink.yarn; + +import org.apache.flink.api.java.tuple.Tuple2; +import org.apache.flink.util.IOUtils; + +import org.apache.hadoop.fs.FileStatus; +import org.apache.hadoop.fs.FileSystem; +import org.apache.hadoop.fs.Path; +import org.apache.hadoop.fs.permission.FsAction; +import org.apache.hadoop.fs.permission.FsPermission; +import org.apache.hadoop.yarn.api.records.ApplicationId; +import org.apache.hadoop.yarn.api.records.LocalResource; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +import java.io.File; +import java.io.FileNotFoundException; +import java.io.IOException; +import java.nio.file.FileVisitResult; +import java.nio.file.Files; +import java.nio.file.SimpleFileVisitor; +import java.nio.file.attribute.BasicFileAttributes; +import java.util.ArrayList; +import java.util.Collection; +import java.util.HashSet; +import java.util.List; +import java.util.Map; +import java.util.Set; +import java.util.concurrent.TimeUnit; + +import static org.apache.flink.util.Preconditions.checkArgument; +import static org.apache.flink.util.Preconditions.checkNotNull; + +/** + * A class with utilities for uploading files + * related to the deployment of a single application. + */ +class YarnApplicationFileUploader implements AutoCloseable { + + private static final Logger LOG = LoggerFactory.getLogger(YarnApplicationFileUploader.class); + + private final FileSystem fileSystem; + + private final ApplicationId applicationId; + + private final Path homeDir; + + private final Path applicationDir; + + private YarnApplicationFileUploader( + final FileSystem fileSystem, + final Path homeDir, + final ApplicationId applicationId) throws IOException { + this.fileSystem = checkNotNull(fileSystem); + this.homeDir = checkNotNull(homeDir); + this.applicationId = checkNotNull(applicationId); + this.applicationDir = getApplicationDir(applicationId); + } + + Path getHomeDir() { + return homeDir; + } + + Path getApplicationDir() { + return applicationDir; + } + + @Override + public void close() { + IOUtils.closeQuietly(fileSystem); + } + + /** + * Uploads and registers a single resource and adds it to <tt>localResources</tt>. + * @param key the key to add the resource under + * @param localSrcPath local path to the file + * @param localResources map of resources + * @param relativeDstPath the relative path at the target location + * (this will be prefixed by the application-specific directory) + * @param replicationFactor number of replications of a remote file to be created + * @return the remote path to the uploaded resource + */ + Path setupSingleLocalResource( + final String key, + final Path localSrcPath, + final Map<String, LocalResource> localResources, + final String relativeDstPath, + final int replicationFactor) throws IOException { + + File localFile = new File(localSrcPath.toUri().getPath()); + Tuple2<Path, Long> remoteFileInfo = uploadLocalFileToRemote(localSrcPath, relativeDstPath, replicationFactor); + LocalResource resource = Utils.registerLocalResource(remoteFileInfo.f0, localFile.length(), remoteFileInfo.f1); + localResources.put(key, resource); + return remoteFileInfo.f0; + } + + Tuple2<Path, Long> uploadLocalFileToRemote( + final Path localSrcPath, + final String relativeDstPath, + final int replicationFactor) throws IOException { + + final File localFile = new File(localSrcPath.toUri().getPath()); + checkArgument(!localFile.isDirectory(), "File to copy cannot be a directory: " + localSrcPath); + + final Path dst = copyToRemoteApplicationDir(localSrcPath, relativeDstPath, replicationFactor); + + // Note: If we directly used registerLocalResource(FileSystem, Path) here, we would access the remote + // file once again which has problems with eventually consistent read-after-write file + // systems. Instead, we decide to wait until the remote file be available. + + final FileStatus[] fss = waitForTransferToComplete(dst); + if (fss == null || fss.length <= 0) { + LOG.debug("Failed to fetch remote modification time from {}, using local timestamp {}", dst, localFile.lastModified()); + return Tuple2.of(dst, localFile.lastModified()); + } + + LOG.debug("Got modification time {} from remote path {}", fss[0].getModificationTime(), dst); + return Tuple2.of(dst, fss[0].getModificationTime()); + } + + /** + * Recursively uploads (and registers) any (user and system) files in <tt>shipFiles</tt> except + * for files matching "<tt>flink-dist*.jar</tt>" which should be uploaded separately. + * + * @param shipFiles + * files to upload + * @param remotePaths + * paths of the remote resources (uploaded resources will be added) + * @param localResources + * map of resources (uploaded resources will be added) + * @param localResourcesDirectory + * the directory the localResources are uploaded to + * @param envShipFileList + * list of shipped files in a format understood by {@link Utils#createTaskExecutorContext} + * @param replicationFactor + * number of replications of a remote file to be created + * + * @return list of class paths with the the proper resource keys from the registration + */ + List<String> setupMultipleLocalResources( + final Collection<File> shipFiles, + final List<Path> remotePaths, + final Map<String, LocalResource> localResources, + final String localResourcesDirectory, + final StringBuilder envShipFileList, + final int replicationFactor) throws IOException { + + checkArgument(replicationFactor >= 1); + final List<Path> localPaths = new ArrayList<>(); + final List<Path> relativePaths = new ArrayList<>(); + for (File shipFile : shipFiles) { + if (shipFile.isDirectory()) { + // add directories to the classpath + final java.nio.file.Path shipPath = shipFile.toPath(); + final java.nio.file.Path parentPath = shipPath.getParent(); + Files.walkFileTree(shipPath, new SimpleFileVisitor<java.nio.file.Path>() { + @Override + public FileVisitResult visitFile(java.nio.file.Path file, BasicFileAttributes attrs) { + localPaths.add(new Path(file.toUri())); + relativePaths.add(new Path(localResourcesDirectory, parentPath.relativize(file).toString())); + return FileVisitResult.CONTINUE; + } + }); + } else { + localPaths.add(new Path(shipFile.toURI())); + relativePaths.add(new Path(localResourcesDirectory, shipFile.getName())); + } + } + + final Set<String> archives = new HashSet<>(); + final Set<String> resources = new HashSet<>(); + for (int i = 0; i < localPaths.size(); i++) { + final Path localPath = localPaths.get(i); + final Path relativePath = relativePaths.get(i); + if (!isFlinkDistJar(relativePath.getName())) { + final String key = relativePath.toString(); + final Path remotePath = setupSingleLocalResource( + key, + localPath, + localResources, + relativePath.getParent().toString(), + replicationFactor); + remotePaths.add(remotePath); + envShipFileList.append(key).append("=").append(remotePath).append(","); + // add files to the classpath + if (key.endsWith("jar")) { + archives.add(relativePath.toString()); + } else { + resources.add(relativePath.getParent().toString()); + } + } + } + + // construct classpath, we always want resource directories to go first, we also sort + // both resources and archives in order to make classpath deterministic + final ArrayList<String> classPaths = new ArrayList<>(); + resources.stream().sorted().forEach(classPaths::add); + archives.stream().sorted().forEach(classPaths::add); + return classPaths; + } + + static YarnApplicationFileUploader initialize( Review comment: shouldn't this be `create()` or `of()`? ########## File path: flink-yarn/src/main/java/org/apache/flink/yarn/YarnClusterDescriptor.java ########## @@ -901,10 +885,6 @@ private ApplicationReport startAppMaster( } } - final Path yarnFilesDir = getYarnFilesDir(appId); Review comment: Why don't we need the permission setting anymore? ########## File path: flink-yarn/src/main/java/org/apache/flink/yarn/YarnClusterDescriptor.java ########## @@ -1105,160 +1076,6 @@ private ApplicationReport startAppMaster( return report; } - /** - * Returns the Path where the YARN application files should be uploaded to. - * - * @param appId YARN application id - */ - private Path getYarnFilesDir(final ApplicationId appId) throws IOException { - final FileSystem fileSystem = FileSystem.get(yarnConfiguration); - final Path homeDir = fileSystem.getHomeDirectory(); - return new Path(homeDir, ".flink/" + appId + '/'); - } - - /** - * Uploads and registers a single resource and adds it to <tt>localResources</tt>. - * - * @param key - * the key to add the resource under - * @param fs - * the remote file system to upload to - * @param appId - * application ID - * @param localSrcPath - * local path to the file - * @param localResources - * map of resources - * @param replication - * number of replications of a remote file to be created - * - * @return the remote path to the uploaded resource - */ - private static Path setupSingleLocalResource( - String key, - FileSystem fs, - ApplicationId appId, - Path localSrcPath, - Map<String, LocalResource> localResources, - Path targetHomeDir, - String relativeTargetPath, - int replication) throws IOException { - Tuple2<Path, LocalResource> resource = Utils.setupLocalResource( - fs, - appId.toString(), - localSrcPath, - targetHomeDir, - relativeTargetPath, - replication); - - localResources.put(key, resource.f1); - - return resource.f0; - } - - /** - * Match file name for "<tt>flink-dist*.jar</tt>" pattern. - * - * @param fileName file name to check - * @return true if file is a dist jar - */ - private static boolean isDistJar(String fileName) { - return fileName.startsWith("flink-dist") && fileName.endsWith("jar"); - } - - /** - * Recursively uploads (and registers) any (user and system) files in <tt>shipFiles</tt> except - * for files matching "<tt>flink-dist*.jar</tt>" which should be uploaded separately. - * - * @param shipFiles - * files to upload - * @param fs - * file system to upload to - * @param targetHomeDir - * remote home directory to upload to - * @param appId - * application ID - * @param remotePaths - * paths of the remote resources (uploaded resources will be added) - * @param localResources - * map of resources (uploaded resources will be added) - * @param localResourcesDirectory - * the directory the localResources are uploaded to - * @param envShipFileList - * list of shipped files in a format understood by {@link Utils#createTaskExecutorContext} - * @param replication - * number of replications of a remote file to be created - * - * @return list of class paths with the the proper resource keys from the registration - */ - static List<String> uploadAndRegisterFiles( - Collection<File> shipFiles, - FileSystem fs, - Path targetHomeDir, - ApplicationId appId, - List<Path> remotePaths, - Map<String, LocalResource> localResources, - String localResourcesDirectory, - StringBuilder envShipFileList, - int replication) throws IOException { - - checkArgument(replication >= 1); - final List<Path> localPaths = new ArrayList<>(); - final List<Path> relativePaths = new ArrayList<>(); - for (File shipFile : shipFiles) { - if (shipFile.isDirectory()) { - // add directories to the classpath - final java.nio.file.Path shipPath = shipFile.toPath(); - final java.nio.file.Path parentPath = shipPath.getParent(); - Files.walkFileTree(shipPath, new SimpleFileVisitor<java.nio.file.Path>() { - @Override - public FileVisitResult visitFile(java.nio.file.Path file, BasicFileAttributes attrs) { - localPaths.add(new Path(file.toUri())); - relativePaths.add(new Path(localResourcesDirectory, parentPath.relativize(file).toString())); - return FileVisitResult.CONTINUE; - } - }); - } else { - localPaths.add(new Path(shipFile.toURI())); - relativePaths.add(new Path(localResourcesDirectory, shipFile.getName())); - } - } - - final Set<String> archives = new HashSet<>(); - final Set<String> resources = new HashSet<>(); - for (int i = 0; i < localPaths.size(); i++) { - final Path localPath = localPaths.get(i); - final Path relativePath = relativePaths.get(i); - if (!isDistJar(relativePath.getName())) { Review comment: Why don't we need this "dist jar" business anymore? ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
