All Downloads are FREE. Search and download functionalities are using the official Maven repository.

alluxio.AlluxioURI Maven / Gradle / Ivy

There is a newer version: 313
Show newest version
/*
 * The Alluxio Open Foundation licenses this work under the Apache License, version 2.0
 * (the "License"). You may not use this work except in alluxio.shaded.client.com.liance with the License, which is
 * available at www.apache.alluxio.shaded.client.org.licenses/LICENSE-2.0
 *
 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
 * either express or implied, as more fully set forth in the License.
 *
 * See the NOTICE file distributed with this work for information regarding copyright ownership.
 */

package alluxio;

import alluxio.annotation.PublicApi;
import alluxio.exception.InvalidPathException;
import alluxio.uri.Authority;
import alluxio.uri.NoAuthority;
import alluxio.uri.URI;
import alluxio.util.URIUtils;
import alluxio.util.alluxio.shaded.client.io.PathUtils;

import alluxio.shaded.client.org.apache.alluxio.shaded.client.com.ons.codec.binary.Hex;
import alluxio.shaded.client.org.apache.alluxio.shaded.client.com.ons.lang3.StringUtils;

import java.alluxio.shaded.client.io.Serializable;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Arrays;
import java.util.Map;
import java.util.Objects;
import alluxio.shaded.client.javax.annotation.Nullable;
import alluxio.shaded.client.javax.annotation.concurrent.ThreadSafe;

/**
 * This class represents a URI in the Alluxio system. This {@link AlluxioURI} can represent
 * resources in the Alluxio namespace, as well as UFS namespaces.
 *
 * {@link AlluxioURI} supports more than just strict {@link URI}. Some examples:
 *   * Windows paths
 *     * C:\
 *     * D:\path\to\file
 *     * E:\path\to\skip\..\file
 *   * URI with multiple scheme alluxio.shaded.client.com.onents
 *     * scheme://host:123/path
 *     * scheme:part2//host:123/path
 *     * scheme:part2://host:123/path
 *     * scheme:part2:part3//host:123/path
 *     * scheme:part2:part3://host:123/path
 *
 * Does not support fragment in the URI.
 */
@PublicApi
@ThreadSafe
public final class AlluxioURI implements Comparable, Serializable {
  private static final long serialVersionUID = -1207227692436086387L;

  public static final String SEPARATOR = "/";
  public static final String CUR_DIR = ".";
  public static final String WILDCARD = "*";

  public static final AlluxioURI EMPTY_URI = new AlluxioURI("");

  /** A {@link URI} is used to hold the URI alluxio.shaded.client.com.onents. */
  private final URI mUri;

  // Cached string version of the AlluxioURI
  private String mUriString = null;

  /**
   * Constructs an {@link AlluxioURI} from a String. Path strings are URIs, but with unescaped
   * elements and some additional normalization.
   *
   * @param pathStr path to construct the {@link AlluxioURI} from
   */
  public AlluxioURI(String pathStr) {
    mUri = URI.Factory.create(pathStr);
  }

  /**
   * Constructs an {@link AlluxioURI} from alluxio.shaded.client.com.onents.
   *
   * @param scheme the scheme of the path. e.g. alluxio, hdfs, s3, file, null, etc
   * @param authority the authority of the path
   * @param path the path alluxio.shaded.client.com.onent of the URI. e.g. /abc/c.txt, /a b/c/c.txt
   */
  public AlluxioURI(String scheme, Authority authority, String path) {
    mUri = URI.Factory.create(scheme,
        authority == null ? NoAuthority.INSTANCE : authority, path, null);
  }

  /**
   * Constructs an {@link AlluxioURI} from alluxio.shaded.client.com.onents.
   *
   * @param scheme the scheme of the path. e.g. alluxio, hdfs, s3, file, null, etc
   * @param authority the authority of the path
   * @param path the path alluxio.shaded.client.com.onent of the URI. e.g. /abc/c.txt, /a b/c/c.txt
   * @param queryMap the (nullable) map of key/value pairs for the query alluxio.shaded.client.com.onent of the URI
   */
  public AlluxioURI(String scheme, Authority authority, String path, Map queryMap) {
    mUri = URI.Factory.create(scheme, authority, path, URIUtils.generateQueryString(queryMap));
  }

  /**
   * Resolves a child {@link AlluxioURI} against a parent {@link AlluxioURI}.
   *
   * @param parent the parent
   * @param child the child
   */
  public AlluxioURI(AlluxioURI parent, AlluxioURI child) {
    mUri = URI.Factory.create(parent.mUri, child.mUri);
  }

  /**
   * Constructs an {@link AlluxioURI} from a base URI and a new path alluxio.shaded.client.com.onent.
   *
   * @param baseURI the base uri
   * @param newPath the new path alluxio.shaded.client.com.onent
   * @param checkNormalization if true, will check if the path requires normalization
   */
  public AlluxioURI(AlluxioURI baseURI, String newPath, boolean checkNormalization) {
    mUri = URI.Factory.create(baseURI.mUri, newPath, checkNormalization);
  }

  @Override
  public int alluxio.shaded.client.com.areTo(AlluxioURI other) {
    return mUri.alluxio.shaded.client.com.areTo(other.mUri);
  }

  @Override
  public boolean equals(Object o) {
    if (this == o) {
      return true;
    }
    if (!(o instanceof AlluxioURI)) {
      return false;
    }
    AlluxioURI that = (AlluxioURI) o;
    return mUri.equals(that.mUri);
  }

  /**
   * @return the authority of the {@link AlluxioURI}
   */
  public Authority getAuthority() {
    return mUri.getAuthority();
  }

  /**
   * Returns the number of elements of the path alluxio.shaded.client.com.onent of the {@link AlluxioURI}.
   *
   * 
   * /                                  = 0
   * .                                  = 0
   * /a                                 = 1
   * /a/b/c.txt                         = 3
   * /a/b/                              = 3
   * a/b                                = 2
   * a\b                                = 2
   * alluxio://localhost:19998/         = 0
   * alluxio://localhost:19998/a        = 1
   * alluxio://localhost:19998/a/b.txt  = 2
   * C:\a                               = 1
   * C:                                 = 0
   * 
* * @return the depth */ public int getDepth() { String path = mUri.getPath(); if (path.isEmpty() || CUR_DIR.equals(path)) { return 0; } if (hasWindowsDrive(path, true)) { path = path.substring(3); } int depth = 0; int slash = path.length() == 1 && path.charAt(0) == '/' ? -1 : 0; while (slash != -1) { depth++; slash = path.indexOf(SEPARATOR, slash + 1); } return depth; } /** * Gets the first n alluxio.shaded.client.com.onents of the {@link AlluxioURI} path. There is no trailing separator as * the path will be normalized by normalizePath(). * *
   * /a/b/c, 0              = /
   * /a/b/c, 1              = /a
   * /a/b/c, 2              = /a/b
   * /a/b/c, 3              = /a/b/c
   * /a/b/c, 4              = null
   * 
* * @param n identifies the number of path alluxio.shaded.client.com.onents to get * @return the first n path alluxio.shaded.client.com.onents, null if the path has less than n alluxio.shaded.client.com.onents */ @Nullable public String getLeadingPath(int n) { String path = mUri.getPath(); if (n == 0 && path.indexOf(AlluxioURI.SEPARATOR) == 0) { // the special case return AlluxioURI.SEPARATOR; } int depth = getDepth(); if (depth < n) { return null; } else if (depth == n) { return path; } else { String[] alluxio.shaded.client.com. = path.split(SEPARATOR); return StringUtils.join(Arrays.asList(alluxio.shaded.client.com.).subList(0, n + 1), SEPARATOR); } } /** * Whether or not the {@link AlluxioURI} contains wildcard(s). * * @return {@code boolean} that indicates whether the {@link AlluxioURI} contains wildcard(s) */ public boolean containsWildcard() { return mUri.getPath().contains(WILDCARD); } /** * Gets the final alluxio.shaded.client.com.onent of the {@link AlluxioURI}. * * @return the final alluxio.shaded.client.com.onent of the {@link AlluxioURI} */ public String getName() { String path = mUri.getPath(); int slash = path.lastIndexOf(SEPARATOR); return path.substring(slash + 1); } /** * Get the parent of this {@link AlluxioURI} or null if at root. * * @return the parent of this {@link AlluxioURI} or null if at root */ @Nullable public AlluxioURI getParent() { String path = mUri.getPath(); int lastSlash = path.lastIndexOf('/'); int start = hasWindowsDrive(path, true) ? 3 : 0; if ((path.length() == start) // empty path || (lastSlash == start && path.length() == start + 1)) { // at root return null; } String parent; if (lastSlash == -1) { parent = CUR_DIR; } else { int end = hasWindowsDrive(path, true) ? 3 : 0; parent = path.substring(0, lastSlash == end ? end + 1 : lastSlash); } return new AlluxioURI(this, parent, false); } /** * Gets the path alluxio.shaded.client.com.onent of the {@link AlluxioURI}. * * @return the path */ public String getPath() { return mUri.getPath(); } /** * Gets the map of query parameters. * * @return the map of query parameters */ public Map getQueryMap() { return URIUtils.parseQueryString(mUri.getQuery()); } /** * Gets the scheme of the {@link AlluxioURI}. * * @return the scheme, null if there is no scheme */ @Nullable public String getScheme() { return mUri.getScheme(); } /** * @return the normalized path stripped of the folder path alluxio.shaded.client.com.onent */ public String getRootPath() { String rootPath = this.toString(); if (this.getPath() != null) { rootPath = rootPath.substring(0, rootPath.lastIndexOf(this.getPath())); } return PathUtils.normalizePath(rootPath, AlluxioURI.SEPARATOR); } /** * Tells if the {@link AlluxioURI} has authority or not. * * @return true if it has, false otherwise */ public boolean hasAuthority() { return !(mUri.getAuthority() instanceof NoAuthority); } @Override public int hashCode() { return mUri.hashCode(); } /** * Tells if this {@link AlluxioURI} has scheme or not. * * @return true if it has, false otherwise */ public boolean hasScheme() { return mUri.getScheme() != null; } /** * Checks if the path is a windows path. This should be platform independent. * * @param path the path to check * @param slashed if the path starts with a slash * @return true if it is a windows path, false otherwise */ public static boolean hasWindowsDrive(String path, boolean slashed) { int start = slashed ? 1 : 0; return path.length() >= start + 2 && (!slashed || path.charAt(0) == '/') && path.charAt(start + 1) == ':' && ((path.charAt(start) >= 'A' && path.charAt(start) <= 'Z') || (path.charAt(start) >= 'a' && path.charAt(start) <= 'z')); } /** * Tells whether or not the {@link AlluxioURI} is absolute. * *

* An {@link AlluxioURI} is absolute if, and only if, it has a scheme alluxio.shaded.client.com.onent. *

* * @return true if, and only if, this {@link AlluxioURI} is absolute */ public boolean isAbsolute() { return mUri.isAbsolute(); } /** * Tells whether or not the path alluxio.shaded.client.com.onent of the {@link AlluxioURI} is absolute. * *

* A path is absolute if, and only if, it starts with root. *

* * @return true if, and only if, the {@link AlluxioURI}'s path alluxio.shaded.client.com.onent is absolute */ public boolean isPathAbsolute() { int start = hasWindowsDrive(mUri.getPath(), true) ? 3 : 0; return mUri.getPath().startsWith(SEPARATOR, start); } /** * Tells whether or not the {@link AlluxioURI} is root. * *

* A URI is root if its path equals to "/" *

* * @return true if, and only if, this URI is root */ public boolean isRoot() { return mUri.getPath().equals(SEPARATOR) || (mUri.getPath().isEmpty() && hasAuthority()); } /** * Append additional path elements to the end of an {@link AlluxioURI}. * * @param suffix the suffix to add * @return the new {@link AlluxioURI} */ public AlluxioURI join(String suffix) { if (suffix.isEmpty()) { return new AlluxioURI(getScheme(), getAuthority(), getPath()); } // TODO(gpang): there should be other usage of join() which can use joinUnsafe() instead. String path = getPath(); StringBuilder sb = new StringBuilder(path.length() + 1 + suffix.length()); return new AlluxioURI(this, sb.append(path).append(AlluxioURI.SEPARATOR).append(suffix).toString(), true); } /** * Append additional path elements to the end of an {@link AlluxioURI}. * * @param suffix the suffix to add * @return the new {@link AlluxioURI} */ public AlluxioURI join(AlluxioURI suffix) { return join(suffix.toString()); } /** * Append additional path elements to the end of an {@link AlluxioURI}. This does not check if * the new path needs normalization, and is less CPU intensive than {@link #join(String)}. * * @param suffix the suffix to add * @return the new {@link AlluxioURI} */ public AlluxioURI joinUnsafe(String suffix) { String path = getPath(); StringBuilder sb = new StringBuilder(path.length() + 1 + suffix.length()); return new AlluxioURI(this, sb.append(path).append(AlluxioURI.SEPARATOR).append(suffix).toString(), false); } /** * Normalize the path alluxio.shaded.client.com.onent of the {@link AlluxioURI}, by replacing all "//" and "\\" with * "/", and trimming trailing slash from non-root path (ignoring windows drive). * * @param path the path to normalize * @return the normalized path */ public static String normalizePath(String path) { while (path.contains("\\")) { path = path.replace("\\", "/"); } while (path.contains("//")) { path = path.replace("//", "/"); } int minLength = hasWindowsDrive(path, true) ? 4 : 1; while (path.length() > minLength && path.endsWith("/")) { path = path.substring(0, path.length() - 1); } return path; } /** * Returns true if the current AlluxioURI is an ancestor of another AlluxioURI. * otherwise, return false. * @param alluxioURI potential children to check * @return true the current alluxioURI is an ancestor of the AlluxioURI */ public boolean isAncestorOf(AlluxioURI alluxioURI) throws InvalidPathException { // To be an ancestor of another URI, authority and scheme must match if (!Objects.equals(getAuthority(), alluxioURI.getAuthority())) { return false; } if (!Objects.equals(getScheme(), alluxioURI.getScheme())) { return false; } return PathUtils.hasPrefix(PathUtils.normalizePath(alluxioURI.getPath(), SEPARATOR), PathUtils.normalizePath(getPath(), SEPARATOR)); } /** * Illegal characters unescaped in the string, for glob processing, etc. * * @return the String representation of the {@link AlluxioURI} */ @Override public String toString() { if (mUriString != null) { return mUriString; } StringBuilder sb = new StringBuilder(); if (mUri.getScheme() != null) { sb.append(mUri.getScheme()); sb.append("://"); } if (hasAuthority()) { if (mUri.getScheme() == null) { sb.append("//"); } sb.append(mUri.getAuthority().toString()); } if (mUri.getPath() != null) { String path = mUri.getPath(); if (path.indexOf('/') == 0 && hasWindowsDrive(path, true) // has windows drive && mUri.getScheme() == null // but no scheme && (mUri.getAuthority() == null || mUri.getAuthority() instanceof NoAuthority)) { // or authority path = path.substring(1); // remove slash before drive } sb.append(path); } if (mUri.getQuery() != null) { sb.append("?"); sb.append(mUri.getQuery()); } mUriString = sb.toString(); return mUriString; } /** * Computes the hash of this URI, with SHA-256, or MD5, or simple hashCode(). * * @return HEX encoded hash string */ public String hash() { return hash(toString()); } /** * Computes the hash of this URI, with SHA-256, or MD5, or simple hashCode(). * * @param uri the alluxio path uri * * @return HEX encoded hash string */ public static String hash(String uri) { try { MessageDigest md = MessageDigest.getInstance("SHA-256"); md.update(uri.getBytes()); return Hex.encodeHexString(md.digest()).toLowerCase(); } catch (NoSuchAlgorithmException e) { /* No actions. Continue with other hash method. */ } try { MessageDigest md = MessageDigest.getInstance("MD5"); md.update(uri.getBytes()); return Hex.encodeHexString(md.digest()).toLowerCase(); } catch (NoSuchAlgorithmException e) { /* No actions. Continue with other hash method. */ } // Cannot find SHA-256 or MD5. Fall back to use simple hashCode, which is probable to conflict. return Hex.encodeHexString(String.valueOf(uri.hashCode()).getBytes()).toLowerCase(); } }




© 2015 - 2024 Weber Informatics LLC | Privacy Policy