linyiqun commented on a change in pull request #1503: URL: https://github.com/apache/hadoop-ozone/pull/1503#discussion_r506792006
########## File path: hadoop-ozone/ozone-manager/src/main/java/org/apache/hadoop/ozone/om/request/file/OMFileRequest.java ########## @@ -583,4 +588,112 @@ public static OmKeyInfo getOmKeyInfoFromFileTable(boolean openFileTable, return dbOmKeyInfo; } + /** + * Gets OmKeyInfo if exists for the given key name in the DB. + * + * @param omMetadataMgr metadata manager + * @param volumeName volume name + * @param bucketName bucket name + * @param keyName key name + * @param scmBlockSize scm block size + * @return OzoneFileStatus + * @throws IOException DB failure + */ + @Nullable + public static OzoneFileStatus getOMKeyInfoIfExists( + OMMetadataManager omMetadataMgr, String volumeName, String bucketName, + String keyName, long scmBlockSize) throws IOException { + + Path keyPath = Paths.get(keyName); + Iterator<Path> elements = keyPath.iterator(); + String bucketKey = omMetadataMgr.getBucketKey(volumeName, bucketName); + OmBucketInfo omBucketInfo = + omMetadataMgr.getBucketTable().get(bucketKey); + + long lastKnownParentId = omBucketInfo.getObjectID(); + OmDirectoryInfo omDirInfo = null; + while (elements.hasNext()) { + String fileName = elements.next().toString(); + + // For example, /vol1/buck1/a/b/c/d/e/file1.txt + // 1. Do lookup path component on directoryTable starting from bucket + // 'buck1' to the leaf node component, which is 'file1.txt'. + // 2. If there is no dir exists for the leaf node component 'file1.txt' + // then do look it on fileTable. + String dbNodeName = omMetadataMgr.getOzonePathKey( + lastKnownParentId, fileName); + omDirInfo = omMetadataMgr.getDirectoryTable().get(dbNodeName); + + if (omDirInfo != null) { + lastKnownParentId = omDirInfo.getObjectID(); + } else if (!elements.hasNext()) { + // reached last path component. Check file exists for the given path. + OmKeyInfo omKeyInfo = OMFileRequest.getOmKeyInfoFromFileTable(false, + omMetadataMgr, dbNodeName, keyName); + if (omKeyInfo != null) { + return new OzoneFileStatus(omKeyInfo, scmBlockSize, false); + } + } else { + // Missing intermediate directory and just return null; + // key not found in DB + return null; + } + } + + if (omDirInfo != null) { + OmKeyInfo omKeyInfo = getOmKeyInfo(volumeName, bucketName, omDirInfo, + keyName); + return new OzoneFileStatus(omKeyInfo, scmBlockSize, true); + } + + // key not found in DB + return null; + } + + /** + * Prepare OmKeyInfo from OmDirectoryInfo. + * + * @param volumeName volume name + * @param bucketName bucket name + * @param dirInfo directory info + * @param keyName usewr given key name + * @return OmKeyInfo object + */ + @NotNull + public static OmKeyInfo getOmKeyInfo(String volumeName, String bucketName, + OmDirectoryInfo dirInfo, String keyName) { + + OmKeyInfo.Builder builder = new OmKeyInfo.Builder(); + builder.setParentObjectID(dirInfo.getParentObjectID()); + builder.setKeyName(keyName); + builder.setAcls(dirInfo.getAcls()); + builder.addAllMetadata(dirInfo.getMetadata()); + builder.setVolumeName(volumeName); + builder.setBucketName(bucketName); + builder.setCreationTime(dirInfo.getCreationTime()); + builder.setModificationTime(dirInfo.getModificationTime()); + builder.setObjectID(dirInfo.getObjectID()); + builder.setUpdateID(dirInfo.getUpdateID()); + builder.setFileName(dirInfo.getName()); + builder.setReplicationType(HddsProtos.ReplicationType.RATIS); + builder.setReplicationFactor(HddsProtos.ReplicationFactor.ONE); Review comment: Can we use default replication type/factor: * OzoneConfigKeys.OZONE_REPLICATION_TYPE_DEFAULT * OzoneConfigKeys.OZONE_REPLICATION_DEFAULT ---------------------------------------------------------------- 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: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: ozone-issues-unsubscr...@hadoop.apache.org For additional commands, e-mail: ozone-issues-h...@hadoop.apache.org