merge from trunk r1598430
git-svn-id: https://svn.apache.org/repos/asf/hadoop/common/branches/fs-encryption@1598435 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
commit
333562b767
|
@ -18,8 +18,6 @@
|
||||||
|
|
||||||
Hadoop Auth, Java HTTP SPNEGO ${project.version} - Building It
|
Hadoop Auth, Java HTTP SPNEGO ${project.version} - Building It
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
||||||
* Requirements
|
* Requirements
|
||||||
|
|
||||||
* Java 6+
|
* Java 6+
|
||||||
|
@ -70,6 +68,3 @@ $ mvn package -Pdocs
|
||||||
|
|
||||||
The generated documentation is available at
|
The generated documentation is available at
|
||||||
<<<hadoop-auth/target/site/>>>.
|
<<<hadoop-auth/target/site/>>>.
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
||||||
|
|
|
@ -20,8 +20,6 @@
|
||||||
Hadoop Auth, Java HTTP SPNEGO ${project.version} - Server Side
|
Hadoop Auth, Java HTTP SPNEGO ${project.version} - Server Side
|
||||||
Configuration
|
Configuration
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
||||||
* Server Side Configuration Setup
|
* Server Side Configuration Setup
|
||||||
|
|
||||||
The AuthenticationFilter filter is Hadoop Auth's server side component.
|
The AuthenticationFilter filter is Hadoop Auth's server side component.
|
||||||
|
@ -241,5 +239,3 @@ Configuration
|
||||||
...
|
...
|
||||||
</web-app>
|
</web-app>
|
||||||
+---+
|
+---+
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
|
@ -18,8 +18,6 @@
|
||||||
|
|
||||||
Hadoop Auth, Java HTTP SPNEGO ${project.version} - Examples
|
Hadoop Auth, Java HTTP SPNEGO ${project.version} - Examples
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
||||||
* Accessing a Hadoop Auth protected URL Using a browser
|
* Accessing a Hadoop Auth protected URL Using a browser
|
||||||
|
|
||||||
<<IMPORTANT:>> The browser must support HTTP Kerberos SPNEGO. For example,
|
<<IMPORTANT:>> The browser must support HTTP Kerberos SPNEGO. For example,
|
||||||
|
@ -133,5 +131,3 @@ You are: user[tucu] principal[tucu@LOCALHOST]
|
||||||
....
|
....
|
||||||
|
|
||||||
+---+
|
+---+
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
|
@ -334,6 +334,9 @@ Trunk (Unreleased)
|
||||||
|
|
||||||
HADOOP-10586. KeyShell doesn't allow setting Options via CLI. (clamb via tucu)
|
HADOOP-10586. KeyShell doesn't allow setting Options via CLI. (clamb via tucu)
|
||||||
|
|
||||||
|
HADOOP-10625. Trim configuration names when putting/getting them
|
||||||
|
to properties. (Wangda Tan via xgong)
|
||||||
|
|
||||||
OPTIMIZATIONS
|
OPTIMIZATIONS
|
||||||
|
|
||||||
HADOOP-7761. Improve the performance of raw comparisons. (todd)
|
HADOOP-7761. Improve the performance of raw comparisons. (todd)
|
||||||
|
@ -413,6 +416,12 @@ Release 2.5.0 - UNRELEASED
|
||||||
HADOOP-10614. CBZip2InputStream is not threadsafe (Xiangrui Meng via
|
HADOOP-10614. CBZip2InputStream is not threadsafe (Xiangrui Meng via
|
||||||
Sandy Ryza)
|
Sandy Ryza)
|
||||||
|
|
||||||
|
HADOOP-10618. Remove SingleNodeSetup.apt.vm. (Akira Ajisaka via
|
||||||
|
Arpit Agarwal)
|
||||||
|
|
||||||
|
HADOOP-10448. Support pluggable mechanism to specify proxy user settings.
|
||||||
|
(Benoy Antony via Arpit Agarwal)
|
||||||
|
|
||||||
OPTIMIZATIONS
|
OPTIMIZATIONS
|
||||||
|
|
||||||
BUG FIXES
|
BUG FIXES
|
||||||
|
@ -504,6 +513,12 @@ Release 2.5.0 - UNRELEASED
|
||||||
HADOOP-10489. UserGroupInformation#getTokens and UserGroupInformation
|
HADOOP-10489. UserGroupInformation#getTokens and UserGroupInformation
|
||||||
#addToken can lead to ConcurrentModificationException (Robert Kanter via atm)
|
#addToken can lead to ConcurrentModificationException (Robert Kanter via atm)
|
||||||
|
|
||||||
|
HADOOP-10602. Documentation has broken "Go Back" hyperlinks.
|
||||||
|
(Akira AJISAKA via cnauroth)
|
||||||
|
|
||||||
|
HADOOP-10639. FileBasedKeyStoresFactory initialization is not using default
|
||||||
|
for SSL_REQUIRE_CLIENT_CERT_KEY. (tucu)
|
||||||
|
|
||||||
Release 2.4.1 - UNRELEASED
|
Release 2.4.1 - UNRELEASED
|
||||||
|
|
||||||
INCOMPATIBLE CHANGES
|
INCOMPATIBLE CHANGES
|
||||||
|
|
|
@ -566,6 +566,9 @@ public class Configuration implements Iterable<Map.Entry<String,String>>,
|
||||||
*/
|
*/
|
||||||
private String[] handleDeprecation(DeprecationContext deprecations,
|
private String[] handleDeprecation(DeprecationContext deprecations,
|
||||||
String name) {
|
String name) {
|
||||||
|
if (null != name) {
|
||||||
|
name = name.trim();
|
||||||
|
}
|
||||||
ArrayList<String > names = new ArrayList<String>();
|
ArrayList<String > names = new ArrayList<String>();
|
||||||
if (isDeprecated(name)) {
|
if (isDeprecated(name)) {
|
||||||
DeprecatedKeyInfo keyInfo = deprecations.getDeprecatedKeyMap().get(name);
|
DeprecatedKeyInfo keyInfo = deprecations.getDeprecatedKeyMap().get(name);
|
||||||
|
@ -843,12 +846,12 @@ public class Configuration implements Iterable<Map.Entry<String,String>>,
|
||||||
/**
|
/**
|
||||||
* Get the value of the <code>name</code> property, <code>null</code> if
|
* Get the value of the <code>name</code> property, <code>null</code> if
|
||||||
* no such property exists. If the key is deprecated, it returns the value of
|
* no such property exists. If the key is deprecated, it returns the value of
|
||||||
* the first key which replaces the deprecated key and is not null
|
* the first key which replaces the deprecated key and is not null.
|
||||||
*
|
*
|
||||||
* Values are processed for <a href="#VariableExpansion">variable expansion</a>
|
* Values are processed for <a href="#VariableExpansion">variable expansion</a>
|
||||||
* before being returned.
|
* before being returned.
|
||||||
*
|
*
|
||||||
* @param name the property name.
|
* @param name the property name, will be trimmed before get value.
|
||||||
* @return the value of the <code>name</code> or its replacing property,
|
* @return the value of the <code>name</code> or its replacing property,
|
||||||
* or null if no such property exists.
|
* or null if no such property exists.
|
||||||
*/
|
*/
|
||||||
|
@ -952,7 +955,8 @@ public class Configuration implements Iterable<Map.Entry<String,String>>,
|
||||||
/**
|
/**
|
||||||
* Set the <code>value</code> of the <code>name</code> property. If
|
* Set the <code>value</code> of the <code>name</code> property. If
|
||||||
* <code>name</code> is deprecated or there is a deprecated name associated to it,
|
* <code>name</code> is deprecated or there is a deprecated name associated to it,
|
||||||
* it sets the value to both names.
|
* it sets the value to both names. Name will be trimmed before put into
|
||||||
|
* configuration.
|
||||||
*
|
*
|
||||||
* @param name property name.
|
* @param name property name.
|
||||||
* @param value property value.
|
* @param value property value.
|
||||||
|
@ -964,7 +968,8 @@ public class Configuration implements Iterable<Map.Entry<String,String>>,
|
||||||
/**
|
/**
|
||||||
* Set the <code>value</code> of the <code>name</code> property. If
|
* Set the <code>value</code> of the <code>name</code> property. If
|
||||||
* <code>name</code> is deprecated, it also sets the <code>value</code> to
|
* <code>name</code> is deprecated, it also sets the <code>value</code> to
|
||||||
* the keys that replace the deprecated key.
|
* the keys that replace the deprecated key. Name will be trimmed before put
|
||||||
|
* into configuration.
|
||||||
*
|
*
|
||||||
* @param name property name.
|
* @param name property name.
|
||||||
* @param value property value.
|
* @param value property value.
|
||||||
|
@ -979,6 +984,7 @@ public class Configuration implements Iterable<Map.Entry<String,String>>,
|
||||||
Preconditions.checkArgument(
|
Preconditions.checkArgument(
|
||||||
value != null,
|
value != null,
|
||||||
"The value of property " + name + " must not be null");
|
"The value of property " + name + " must not be null");
|
||||||
|
name = name.trim();
|
||||||
DeprecationContext deprecations = deprecationContext.get();
|
DeprecationContext deprecations = deprecationContext.get();
|
||||||
if (deprecations.getDeprecatedKeyMap().isEmpty()) {
|
if (deprecations.getDeprecatedKeyMap().isEmpty()) {
|
||||||
getProps();
|
getProps();
|
||||||
|
@ -1064,7 +1070,7 @@ public class Configuration implements Iterable<Map.Entry<String,String>>,
|
||||||
* If no such property exists,
|
* If no such property exists,
|
||||||
* then <code>defaultValue</code> is returned.
|
* then <code>defaultValue</code> is returned.
|
||||||
*
|
*
|
||||||
* @param name property name.
|
* @param name property name, will be trimmed before get value.
|
||||||
* @param defaultValue default value.
|
* @param defaultValue default value.
|
||||||
* @return property value, or <code>defaultValue</code> if the property
|
* @return property value, or <code>defaultValue</code> if the property
|
||||||
* doesn't exist.
|
* doesn't exist.
|
||||||
|
|
|
@ -293,5 +293,8 @@ public class CommonConfigurationKeysPublic {
|
||||||
"hadoop.security.crypto.buffer.size";
|
"hadoop.security.crypto.buffer.size";
|
||||||
/** Defalt value for HADOOP_SECURITY_CRYPTO_BUFFER_SIZE_KEY */
|
/** Defalt value for HADOOP_SECURITY_CRYPTO_BUFFER_SIZE_KEY */
|
||||||
public static final int HADOOP_SECURITY_CRYPTO_BUFFER_SIZE_DEFAULT = 8192;
|
public static final int HADOOP_SECURITY_CRYPTO_BUFFER_SIZE_DEFAULT = 8192;
|
||||||
|
/** Class to override Impersonation provider */
|
||||||
|
public static final String HADOOP_SECURITY_IMPERSONATION_PROVIDER_CLASS =
|
||||||
|
"hadoop.security.impersonation.provider.class";
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -2140,6 +2140,18 @@ public abstract class FileSystem extends Configured implements Closeable {
|
||||||
* in the corresponding FileSystem.
|
* in the corresponding FileSystem.
|
||||||
*/
|
*/
|
||||||
public FileChecksum getFileChecksum(Path f) throws IOException {
|
public FileChecksum getFileChecksum(Path f) throws IOException {
|
||||||
|
return getFileChecksum(f, Long.MAX_VALUE);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Get the checksum of a file, from the beginning of the file till the
|
||||||
|
* specific length.
|
||||||
|
* @param f The file path
|
||||||
|
* @param length The length of the file range for checksum calculation
|
||||||
|
* @return The file checksum.
|
||||||
|
*/
|
||||||
|
public FileChecksum getFileChecksum(Path f, final long length)
|
||||||
|
throws IOException {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -31,7 +31,6 @@ import org.apache.hadoop.conf.Configuration;
|
||||||
import org.apache.hadoop.fs.permission.AclEntry;
|
import org.apache.hadoop.fs.permission.AclEntry;
|
||||||
import org.apache.hadoop.fs.permission.AclStatus;
|
import org.apache.hadoop.fs.permission.AclStatus;
|
||||||
import org.apache.hadoop.fs.permission.FsPermission;
|
import org.apache.hadoop.fs.permission.FsPermission;
|
||||||
import org.apache.hadoop.fs.ContentSummary;
|
|
||||||
import org.apache.hadoop.fs.Options.ChecksumOpt;
|
import org.apache.hadoop.fs.Options.ChecksumOpt;
|
||||||
import org.apache.hadoop.security.AccessControlException;
|
import org.apache.hadoop.security.AccessControlException;
|
||||||
import org.apache.hadoop.util.Progressable;
|
import org.apache.hadoop.util.Progressable;
|
||||||
|
@ -429,6 +428,11 @@ public class FilterFileSystem extends FileSystem {
|
||||||
return fs.getFileChecksum(f);
|
return fs.getFileChecksum(f);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public FileChecksum getFileChecksum(Path f, long length) throws IOException {
|
||||||
|
return fs.getFileChecksum(f, length);
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void setVerifyChecksum(boolean verifyChecksum) {
|
public void setVerifyChecksum(boolean verifyChecksum) {
|
||||||
fs.setVerifyChecksum(verifyChecksum);
|
fs.setVerifyChecksum(verifyChecksum);
|
||||||
|
|
|
@ -687,7 +687,7 @@ public class HarFileSystem extends FileSystem {
|
||||||
* @return null since no checksum algorithm is implemented.
|
* @return null since no checksum algorithm is implemented.
|
||||||
*/
|
*/
|
||||||
@Override
|
@Override
|
||||||
public FileChecksum getFileChecksum(Path f) {
|
public FileChecksum getFileChecksum(Path f, long length) {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,210 @@
|
||||||
|
/**
|
||||||
|
* 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.hadoop.security.authorize;
|
||||||
|
|
||||||
|
import java.net.InetAddress;
|
||||||
|
import java.net.UnknownHostException;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.Collection;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Map.Entry;
|
||||||
|
|
||||||
|
import org.apache.hadoop.conf.Configuration;
|
||||||
|
import org.apache.hadoop.security.Groups;
|
||||||
|
import org.apache.hadoop.security.UserGroupInformation;
|
||||||
|
import org.apache.hadoop.util.StringUtils;
|
||||||
|
|
||||||
|
import com.google.common.annotations.VisibleForTesting;
|
||||||
|
|
||||||
|
public class DefaultImpersonationProvider implements ImpersonationProvider {
|
||||||
|
private static final String CONF_HOSTS = ".hosts";
|
||||||
|
private static final String CONF_USERS = ".users";
|
||||||
|
private static final String CONF_GROUPS = ".groups";
|
||||||
|
private static final String CONF_HADOOP_PROXYUSER = "hadoop.proxyuser.";
|
||||||
|
private static final String CONF_HADOOP_PROXYUSER_RE = "hadoop\\.proxyuser\\.";
|
||||||
|
// list of users, groups and hosts per proxyuser
|
||||||
|
private Map<String, Collection<String>> proxyUsers =
|
||||||
|
new HashMap<String, Collection<String>>();
|
||||||
|
private Map<String, Collection<String>> proxyGroups =
|
||||||
|
new HashMap<String, Collection<String>>();
|
||||||
|
private Map<String, Collection<String>> proxyHosts =
|
||||||
|
new HashMap<String, Collection<String>>();
|
||||||
|
private Configuration conf;
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void setConf(Configuration conf) {
|
||||||
|
this.conf = conf;
|
||||||
|
|
||||||
|
// get all the new keys for users
|
||||||
|
String regex = CONF_HADOOP_PROXYUSER_RE+"[^.]*\\"+CONF_USERS;
|
||||||
|
Map<String,String> allMatchKeys = conf.getValByRegex(regex);
|
||||||
|
for(Entry<String, String> entry : allMatchKeys.entrySet()) {
|
||||||
|
Collection<String> users = StringUtils.getTrimmedStringCollection(entry.getValue());
|
||||||
|
proxyUsers.put(entry.getKey(), users);
|
||||||
|
}
|
||||||
|
|
||||||
|
// get all the new keys for groups
|
||||||
|
regex = CONF_HADOOP_PROXYUSER_RE+"[^.]*\\"+CONF_GROUPS;
|
||||||
|
allMatchKeys = conf.getValByRegex(regex);
|
||||||
|
for(Entry<String, String> entry : allMatchKeys.entrySet()) {
|
||||||
|
Collection<String> groups = StringUtils.getTrimmedStringCollection(entry.getValue());
|
||||||
|
proxyGroups.put(entry.getKey(), groups);
|
||||||
|
//cache the groups. This is needed for NetGroups
|
||||||
|
Groups.getUserToGroupsMappingService(conf).cacheGroupsAdd(
|
||||||
|
new ArrayList<String>(groups));
|
||||||
|
}
|
||||||
|
|
||||||
|
// now hosts
|
||||||
|
regex = CONF_HADOOP_PROXYUSER_RE+"[^.]*\\"+CONF_HOSTS;
|
||||||
|
allMatchKeys = conf.getValByRegex(regex);
|
||||||
|
for(Entry<String, String> entry : allMatchKeys.entrySet()) {
|
||||||
|
proxyHosts.put(entry.getKey(),
|
||||||
|
StringUtils.getTrimmedStringCollection(entry.getValue()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Configuration getConf() {
|
||||||
|
return conf;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void authorize(UserGroupInformation user,
|
||||||
|
String remoteAddress) throws AuthorizationException {
|
||||||
|
|
||||||
|
if (user.getRealUser() == null) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
boolean userAuthorized = false;
|
||||||
|
boolean ipAuthorized = false;
|
||||||
|
UserGroupInformation superUser = user.getRealUser();
|
||||||
|
|
||||||
|
Collection<String> allowedUsers = proxyUsers.get(
|
||||||
|
getProxySuperuserUserConfKey(superUser.getShortUserName()));
|
||||||
|
|
||||||
|
if (isWildcardList(allowedUsers)) {
|
||||||
|
userAuthorized = true;
|
||||||
|
} else if (allowedUsers != null && !allowedUsers.isEmpty()) {
|
||||||
|
if (allowedUsers.contains(user.getShortUserName())) {
|
||||||
|
userAuthorized = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!userAuthorized){
|
||||||
|
Collection<String> allowedUserGroups = proxyGroups.get(
|
||||||
|
getProxySuperuserGroupConfKey(superUser.getShortUserName()));
|
||||||
|
|
||||||
|
if (isWildcardList(allowedUserGroups)) {
|
||||||
|
userAuthorized = true;
|
||||||
|
} else if (allowedUserGroups != null && !allowedUserGroups.isEmpty()) {
|
||||||
|
for (String group : user.getGroupNames()) {
|
||||||
|
if (allowedUserGroups.contains(group)) {
|
||||||
|
userAuthorized = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!userAuthorized) {
|
||||||
|
throw new AuthorizationException("User: " + superUser.getUserName()
|
||||||
|
+ " is not allowed to impersonate " + user.getUserName());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Collection<String> ipList = proxyHosts.get(
|
||||||
|
getProxySuperuserIpConfKey(superUser.getShortUserName()));
|
||||||
|
|
||||||
|
if (isWildcardList(ipList)) {
|
||||||
|
ipAuthorized = true;
|
||||||
|
} else if (ipList != null && !ipList.isEmpty()) {
|
||||||
|
for (String allowedHost : ipList) {
|
||||||
|
InetAddress hostAddr;
|
||||||
|
try {
|
||||||
|
hostAddr = InetAddress.getByName(allowedHost);
|
||||||
|
} catch (UnknownHostException e) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if (hostAddr.getHostAddress().equals(remoteAddress)) {
|
||||||
|
// Authorization is successful
|
||||||
|
ipAuthorized = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if(!ipAuthorized) {
|
||||||
|
throw new AuthorizationException("Unauthorized connection for super-user: "
|
||||||
|
+ superUser.getUserName() + " from IP " + remoteAddress);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Return true if the configuration specifies the special configuration value
|
||||||
|
* "*", indicating that any group or host list is allowed to use this configuration.
|
||||||
|
*/
|
||||||
|
private boolean isWildcardList(Collection<String> list) {
|
||||||
|
return (list != null) &&
|
||||||
|
(list.size() == 1) &&
|
||||||
|
(list.contains("*"));
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns configuration key for effective usergroups allowed for a superuser
|
||||||
|
*
|
||||||
|
* @param userName name of the superuser
|
||||||
|
* @return configuration key for superuser usergroups
|
||||||
|
*/
|
||||||
|
public static String getProxySuperuserUserConfKey(String userName) {
|
||||||
|
return CONF_HADOOP_PROXYUSER+userName+CONF_USERS;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns configuration key for effective groups allowed for a superuser
|
||||||
|
*
|
||||||
|
* @param userName name of the superuser
|
||||||
|
* @return configuration key for superuser groups
|
||||||
|
*/
|
||||||
|
public static String getProxySuperuserGroupConfKey(String userName) {
|
||||||
|
return CONF_HADOOP_PROXYUSER+userName+CONF_GROUPS;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Return configuration key for superuser ip addresses
|
||||||
|
*
|
||||||
|
* @param userName name of the superuser
|
||||||
|
* @return configuration key for superuser ip-addresses
|
||||||
|
*/
|
||||||
|
public static String getProxySuperuserIpConfKey(String userName) {
|
||||||
|
return CONF_HADOOP_PROXYUSER+userName+CONF_HOSTS;
|
||||||
|
}
|
||||||
|
|
||||||
|
@VisibleForTesting
|
||||||
|
public Map<String, Collection<String>> getProxyUsers() {
|
||||||
|
return proxyUsers;
|
||||||
|
}
|
||||||
|
|
||||||
|
@VisibleForTesting
|
||||||
|
public Map<String, Collection<String>> getProxyGroups() {
|
||||||
|
return proxyGroups;
|
||||||
|
}
|
||||||
|
|
||||||
|
@VisibleForTesting
|
||||||
|
public Map<String, Collection<String>> getProxyHosts() {
|
||||||
|
return proxyHosts;
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,34 @@
|
||||||
|
/**
|
||||||
|
* 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.hadoop.security.authorize;
|
||||||
|
|
||||||
|
import org.apache.hadoop.conf.Configurable;
|
||||||
|
import org.apache.hadoop.security.UserGroupInformation;
|
||||||
|
|
||||||
|
public interface ImpersonationProvider extends Configurable {
|
||||||
|
/**
|
||||||
|
* Authorize the superuser which is doing doAs
|
||||||
|
*
|
||||||
|
* @param user ugi of the effective or proxy user which contains a real user
|
||||||
|
* @param remoteAddress the ip address of client
|
||||||
|
* @throws AuthorizationException
|
||||||
|
*/
|
||||||
|
public void authorize(UserGroupInformation user, String remoteAddress)
|
||||||
|
throws AuthorizationException;
|
||||||
|
}
|
|
@ -18,42 +18,35 @@
|
||||||
|
|
||||||
package org.apache.hadoop.security.authorize;
|
package org.apache.hadoop.security.authorize;
|
||||||
|
|
||||||
import java.net.InetAddress;
|
|
||||||
import java.net.UnknownHostException;
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.Collection;
|
|
||||||
import java.util.HashMap;
|
|
||||||
import java.util.Map;
|
|
||||||
import java.util.Map.Entry;
|
|
||||||
|
|
||||||
import org.apache.hadoop.classification.InterfaceAudience;
|
import org.apache.hadoop.classification.InterfaceAudience;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
import org.apache.hadoop.security.Groups;
|
import org.apache.hadoop.fs.CommonConfigurationKeysPublic;
|
||||||
import org.apache.hadoop.security.UserGroupInformation;
|
import org.apache.hadoop.security.UserGroupInformation;
|
||||||
import org.apache.hadoop.util.StringUtils;
|
import org.apache.hadoop.util.ReflectionUtils;
|
||||||
|
|
||||||
import com.google.common.annotations.VisibleForTesting;
|
import com.google.common.annotations.VisibleForTesting;
|
||||||
|
|
||||||
@InterfaceAudience.LimitedPrivate({"HDFS", "MapReduce", "HBase", "Hive"})
|
@InterfaceAudience.LimitedPrivate({"HDFS", "MapReduce", "HBase", "Hive"})
|
||||||
public class ProxyUsers {
|
public class ProxyUsers {
|
||||||
|
|
||||||
private static final String CONF_HOSTS = ".hosts";
|
private static volatile ImpersonationProvider sip ;
|
||||||
private static final String CONF_USERS = ".users";
|
|
||||||
private static final String CONF_GROUPS = ".groups";
|
|
||||||
private static final String CONF_HADOOP_PROXYUSER = "hadoop.proxyuser.";
|
|
||||||
private static final String CONF_HADOOP_PROXYUSER_RE = "hadoop\\.proxyuser\\.";
|
|
||||||
|
|
||||||
private static boolean init = false;
|
|
||||||
//list of users, groups and hosts per proxyuser
|
|
||||||
private static Map<String, Collection<String>> proxyUsers =
|
|
||||||
new HashMap<String, Collection<String>>();
|
|
||||||
private static Map<String, Collection<String>> proxyGroups =
|
|
||||||
new HashMap<String, Collection<String>>();
|
|
||||||
private static Map<String, Collection<String>> proxyHosts =
|
|
||||||
new HashMap<String, Collection<String>>();
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* reread the conf and get new values for "hadoop.proxyuser.*.groups/users/hosts"
|
* Returns an instance of ImpersonationProvider.
|
||||||
|
* Looks up the configuration to see if there is custom class specified.
|
||||||
|
* @param conf
|
||||||
|
* @return ImpersonationProvider
|
||||||
|
*/
|
||||||
|
private static ImpersonationProvider getInstance(Configuration conf) {
|
||||||
|
Class<? extends ImpersonationProvider> clazz =
|
||||||
|
conf.getClass(
|
||||||
|
CommonConfigurationKeysPublic.HADOOP_SECURITY_IMPERSONATION_PROVIDER_CLASS,
|
||||||
|
DefaultImpersonationProvider.class, ImpersonationProvider.class);
|
||||||
|
return ReflectionUtils.newInstance(clazz, conf);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* refresh Impersonation rules
|
||||||
*/
|
*/
|
||||||
public static void refreshSuperUserGroupsConfiguration() {
|
public static void refreshSuperUserGroupsConfiguration() {
|
||||||
//load server side configuration;
|
//load server side configuration;
|
||||||
|
@ -64,73 +57,13 @@ public class ProxyUsers {
|
||||||
* refresh configuration
|
* refresh configuration
|
||||||
* @param conf
|
* @param conf
|
||||||
*/
|
*/
|
||||||
public static synchronized void refreshSuperUserGroupsConfiguration(Configuration conf) {
|
public static void refreshSuperUserGroupsConfiguration(Configuration conf) {
|
||||||
|
// sip is volatile. Any assignment to it as well as the object's state
|
||||||
// remove all existing stuff
|
// will be visible to all the other threads.
|
||||||
proxyGroups.clear();
|
sip = getInstance(conf);
|
||||||
proxyHosts.clear();
|
|
||||||
proxyUsers.clear();
|
|
||||||
|
|
||||||
// get all the new keys for users
|
|
||||||
String regex = CONF_HADOOP_PROXYUSER_RE+"[^.]*\\"+CONF_USERS;
|
|
||||||
Map<String,String> allMatchKeys = conf.getValByRegex(regex);
|
|
||||||
for(Entry<String, String> entry : allMatchKeys.entrySet()) {
|
|
||||||
Collection<String> users = StringUtils.getTrimmedStringCollection(entry.getValue());
|
|
||||||
proxyUsers.put(entry.getKey(), users);
|
|
||||||
}
|
|
||||||
|
|
||||||
// get all the new keys for groups
|
|
||||||
regex = CONF_HADOOP_PROXYUSER_RE+"[^.]*\\"+CONF_GROUPS;
|
|
||||||
allMatchKeys = conf.getValByRegex(regex);
|
|
||||||
for(Entry<String, String> entry : allMatchKeys.entrySet()) {
|
|
||||||
Collection<String> groups = StringUtils.getTrimmedStringCollection(entry.getValue());
|
|
||||||
proxyGroups.put(entry.getKey(), groups );
|
|
||||||
//cache the groups. This is needed for NetGroups
|
|
||||||
Groups.getUserToGroupsMappingService(conf).cacheGroupsAdd(
|
|
||||||
new ArrayList<String>(groups));
|
|
||||||
}
|
|
||||||
|
|
||||||
// now hosts
|
|
||||||
regex = CONF_HADOOP_PROXYUSER_RE+"[^.]*\\"+CONF_HOSTS;
|
|
||||||
allMatchKeys = conf.getValByRegex(regex);
|
|
||||||
for(Entry<String, String> entry : allMatchKeys.entrySet()) {
|
|
||||||
proxyHosts.put(entry.getKey(),
|
|
||||||
StringUtils.getTrimmedStringCollection(entry.getValue()));
|
|
||||||
}
|
|
||||||
init = true;
|
|
||||||
ProxyServers.refresh(conf);
|
ProxyServers.refresh(conf);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns configuration key for effective users allowed for a superuser
|
|
||||||
*
|
|
||||||
* @param userName name of the superuser
|
|
||||||
* @return configuration key for superuser users
|
|
||||||
*/
|
|
||||||
public static String getProxySuperuserUserConfKey(String userName) {
|
|
||||||
return ProxyUsers.CONF_HADOOP_PROXYUSER+userName+ProxyUsers.CONF_USERS;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns configuration key for effective user groups allowed for a superuser
|
|
||||||
*
|
|
||||||
* @param userName name of the superuser
|
|
||||||
* @return configuration key for superuser groups
|
|
||||||
*/
|
|
||||||
public static String getProxySuperuserGroupConfKey(String userName) {
|
|
||||||
return ProxyUsers.CONF_HADOOP_PROXYUSER+userName+ProxyUsers.CONF_GROUPS;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Return configuration key for superuser ip addresses
|
|
||||||
*
|
|
||||||
* @param userName name of the superuser
|
|
||||||
* @return configuration key for superuser ip-addresses
|
|
||||||
*/
|
|
||||||
public static String getProxySuperuserIpConfKey(String userName) {
|
|
||||||
return ProxyUsers.CONF_HADOOP_PROXYUSER+userName+ProxyUsers.CONF_HOSTS;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Authorize the superuser which is doing doAs
|
* Authorize the superuser which is doing doAs
|
||||||
*
|
*
|
||||||
|
@ -138,75 +71,14 @@ public class ProxyUsers {
|
||||||
* @param remoteAddress the ip address of client
|
* @param remoteAddress the ip address of client
|
||||||
* @throws AuthorizationException
|
* @throws AuthorizationException
|
||||||
*/
|
*/
|
||||||
public static synchronized void authorize(UserGroupInformation user,
|
public static void authorize(UserGroupInformation user,
|
||||||
String remoteAddress) throws AuthorizationException {
|
String remoteAddress) throws AuthorizationException {
|
||||||
|
if (sip==null) {
|
||||||
if(!init) {
|
// In a race situation, It is possible for multiple threads to satisfy this condition.
|
||||||
|
// The last assignment will prevail.
|
||||||
refreshSuperUserGroupsConfiguration();
|
refreshSuperUserGroupsConfiguration();
|
||||||
}
|
}
|
||||||
|
sip.authorize(user, remoteAddress);
|
||||||
if (user.getRealUser() == null) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
boolean userAuthorized = false;
|
|
||||||
boolean ipAuthorized = false;
|
|
||||||
UserGroupInformation superUser = user.getRealUser();
|
|
||||||
|
|
||||||
Collection<String> allowedUsers = proxyUsers.get(
|
|
||||||
getProxySuperuserUserConfKey(superUser.getShortUserName()));
|
|
||||||
|
|
||||||
if (isWildcardList(allowedUsers)) {
|
|
||||||
userAuthorized = true;
|
|
||||||
} else if (allowedUsers != null && !allowedUsers.isEmpty()) {
|
|
||||||
if (allowedUsers.contains(user.getShortUserName())) {
|
|
||||||
userAuthorized = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!userAuthorized) {
|
|
||||||
Collection<String> allowedUserGroups = proxyGroups.get(
|
|
||||||
getProxySuperuserGroupConfKey(superUser.getShortUserName()));
|
|
||||||
|
|
||||||
if (isWildcardList(allowedUserGroups)) {
|
|
||||||
userAuthorized = true;
|
|
||||||
} else if (allowedUserGroups != null && !allowedUserGroups.isEmpty()) {
|
|
||||||
for (String group : user.getGroupNames()) {
|
|
||||||
if (allowedUserGroups.contains(group)) {
|
|
||||||
userAuthorized = true;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!userAuthorized) {
|
|
||||||
throw new AuthorizationException("User: " + superUser.getUserName()
|
|
||||||
+ " is not allowed to impersonate " + user.getUserName());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Collection<String> ipList = proxyHosts.get(
|
|
||||||
getProxySuperuserIpConfKey(superUser.getShortUserName()));
|
|
||||||
|
|
||||||
if (isWildcardList(ipList)) {
|
|
||||||
ipAuthorized = true;
|
|
||||||
} else if (ipList != null && !ipList.isEmpty()) {
|
|
||||||
for (String allowedHost : ipList) {
|
|
||||||
InetAddress hostAddr;
|
|
||||||
try {
|
|
||||||
hostAddr = InetAddress.getByName(allowedHost);
|
|
||||||
} catch (UnknownHostException e) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if (hostAddr.getHostAddress().equals(remoteAddress)) {
|
|
||||||
// Authorization is successful
|
|
||||||
ipAuthorized = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (!ipAuthorized) {
|
|
||||||
throw new AuthorizationException("Unauthorized connection for super-user: "
|
|
||||||
+ superUser.getUserName() + " from IP " + remoteAddress);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -218,33 +90,14 @@ public class ProxyUsers {
|
||||||
* @deprecated use {@link #authorize(UserGroupInformation, String) instead.
|
* @deprecated use {@link #authorize(UserGroupInformation, String) instead.
|
||||||
*/
|
*/
|
||||||
@Deprecated
|
@Deprecated
|
||||||
public static synchronized void authorize(UserGroupInformation user,
|
public static void authorize(UserGroupInformation user,
|
||||||
String remoteAddress, Configuration conf) throws AuthorizationException {
|
String remoteAddress, Configuration conf) throws AuthorizationException {
|
||||||
authorize(user,remoteAddress);
|
authorize(user,remoteAddress);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
@VisibleForTesting
|
||||||
* Return true if the configuration specifies the special configuration value
|
public static DefaultImpersonationProvider getDefaultImpersonationProvider() {
|
||||||
* "*", indicating that any group or host list is allowed to use this configuration.
|
return ((DefaultImpersonationProvider)sip);
|
||||||
*/
|
|
||||||
private static boolean isWildcardList(Collection<String> list) {
|
|
||||||
return (list != null) &&
|
|
||||||
(list.size() == 1) &&
|
|
||||||
(list.contains("*"));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@VisibleForTesting
|
|
||||||
public static Map<String, Collection<String>> getProxyUsers() {
|
|
||||||
return proxyUsers;
|
|
||||||
}
|
|
||||||
|
|
||||||
@VisibleForTesting
|
|
||||||
public static Map<String, Collection<String>> getProxyGroups() {
|
|
||||||
return proxyGroups;
|
|
||||||
}
|
|
||||||
|
|
||||||
@VisibleForTesting
|
|
||||||
public static Map<String, Collection<String>> getProxyHosts() {
|
|
||||||
return proxyHosts;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -131,7 +131,8 @@ public class FileBasedKeyStoresFactory implements KeyStoresFactory {
|
||||||
throws IOException, GeneralSecurityException {
|
throws IOException, GeneralSecurityException {
|
||||||
|
|
||||||
boolean requireClientCert =
|
boolean requireClientCert =
|
||||||
conf.getBoolean(SSLFactory.SSL_REQUIRE_CLIENT_CERT_KEY, true);
|
conf.getBoolean(SSLFactory.SSL_REQUIRE_CLIENT_CERT_KEY,
|
||||||
|
SSLFactory.DEFAULT_SSL_REQUIRE_CLIENT_CERT);
|
||||||
|
|
||||||
// certificate store
|
// certificate store
|
||||||
String keystoreType =
|
String keystoreType =
|
||||||
|
|
|
@ -722,6 +722,17 @@
|
||||||
|
|
||||||
<!-- Proxy Configuration -->
|
<!-- Proxy Configuration -->
|
||||||
|
|
||||||
|
<property>
|
||||||
|
<name>hadoop.security.impersonation.provider.class</name>
|
||||||
|
<value></value>
|
||||||
|
<description>A class which implements ImpersonationProvider interface, used to
|
||||||
|
authorize whether one user can impersonate a specific user.
|
||||||
|
If not specified, the DefaultImpersonationProvider will be used.
|
||||||
|
If a class is specified, then that class will be used to determine
|
||||||
|
the impersonation capability.
|
||||||
|
</description>
|
||||||
|
</property>
|
||||||
|
|
||||||
<property>
|
<property>
|
||||||
<name>hadoop.rpc.socket.factory.class.default</name>
|
<name>hadoop.rpc.socket.factory.class.default</name>
|
||||||
<value>org.apache.hadoop.net.StandardSocketFactory</value>
|
<value>org.apache.hadoop.net.StandardSocketFactory</value>
|
||||||
|
|
|
@ -18,210 +18,7 @@
|
||||||
|
|
||||||
Single Node Setup
|
Single Node Setup
|
||||||
|
|
||||||
%{toc|section=1|fromDepth=0}
|
This page will be removed in the next major release.
|
||||||
|
|
||||||
* Purpose
|
See {{{./SingleCluster.html}Single Cluster Setup}} to set up and configure a
|
||||||
|
single-node Hadoop installation.
|
||||||
This document describes how to set up and configure a single-node
|
|
||||||
Hadoop installation so that you can quickly perform simple operations
|
|
||||||
using Hadoop MapReduce and the Hadoop Distributed File System (HDFS).
|
|
||||||
|
|
||||||
* Prerequisites
|
|
||||||
|
|
||||||
** Supported Platforms
|
|
||||||
|
|
||||||
* GNU/Linux is supported as a development and production platform.
|
|
||||||
Hadoop has been demonstrated on GNU/Linux clusters with 2000 nodes.
|
|
||||||
|
|
||||||
* Windows is also a supported platform.
|
|
||||||
|
|
||||||
** Required Software
|
|
||||||
|
|
||||||
Required software for Linux and Windows include:
|
|
||||||
|
|
||||||
[[1]] Java^TM 1.6.x, preferably from Sun, must be installed.
|
|
||||||
|
|
||||||
[[2]] ssh must be installed and sshd must be running to use the Hadoop
|
|
||||||
scripts that manage remote Hadoop daemons.
|
|
||||||
|
|
||||||
** Installing Software
|
|
||||||
|
|
||||||
If your cluster doesn't have the requisite software you will need to
|
|
||||||
install it.
|
|
||||||
|
|
||||||
For example on Ubuntu Linux:
|
|
||||||
|
|
||||||
----
|
|
||||||
$ sudo apt-get install ssh
|
|
||||||
$ sudo apt-get install rsync
|
|
||||||
----
|
|
||||||
|
|
||||||
* Download
|
|
||||||
|
|
||||||
To get a Hadoop distribution, download a recent stable release from one
|
|
||||||
of the Apache Download Mirrors.
|
|
||||||
|
|
||||||
* Prepare to Start the Hadoop Cluster
|
|
||||||
|
|
||||||
Unpack the downloaded Hadoop distribution. In the distribution, edit
|
|
||||||
the file <<<conf/hadoop-env.sh>>> to define at least <<<JAVA_HOME>>> to be the root
|
|
||||||
of your Java installation.
|
|
||||||
|
|
||||||
Try the following command:
|
|
||||||
|
|
||||||
----
|
|
||||||
$ bin/hadoop
|
|
||||||
----
|
|
||||||
|
|
||||||
This will display the usage documentation for the hadoop script.
|
|
||||||
|
|
||||||
Now you are ready to start your Hadoop cluster in one of the three
|
|
||||||
supported modes:
|
|
||||||
|
|
||||||
* Local (Standalone) Mode
|
|
||||||
|
|
||||||
* Pseudo-Distributed Mode
|
|
||||||
|
|
||||||
* Fully-Distributed Mode
|
|
||||||
|
|
||||||
* Standalone Operation
|
|
||||||
|
|
||||||
By default, Hadoop is configured to run in a non-distributed mode, as a
|
|
||||||
single Java process. This is useful for debugging.
|
|
||||||
|
|
||||||
The following example copies the unpacked conf directory to use as
|
|
||||||
input and then finds and displays every match of the given regular
|
|
||||||
expression. Output is written to the given output directory.
|
|
||||||
|
|
||||||
----
|
|
||||||
$ mkdir input
|
|
||||||
$ cp conf/*.xml input
|
|
||||||
$ bin/hadoop jar hadoop-*-examples.jar grep input output 'dfs[a-z.]+'
|
|
||||||
$ cat output/*
|
|
||||||
---
|
|
||||||
|
|
||||||
* Pseudo-Distributed Operation
|
|
||||||
|
|
||||||
Hadoop can also be run on a single-node in a pseudo-distributed mode
|
|
||||||
where each Hadoop daemon runs in a separate Java process.
|
|
||||||
|
|
||||||
** Configuration
|
|
||||||
|
|
||||||
Use the following:
|
|
||||||
|
|
||||||
conf/core-site.xml:
|
|
||||||
|
|
||||||
----
|
|
||||||
<configuration>
|
|
||||||
<property>
|
|
||||||
<name>fs.defaultFS</name>
|
|
||||||
<value>hdfs://localhost:9000</value>
|
|
||||||
</property>
|
|
||||||
</configuration>
|
|
||||||
----
|
|
||||||
|
|
||||||
conf/hdfs-site.xml:
|
|
||||||
|
|
||||||
----
|
|
||||||
<configuration>
|
|
||||||
<property>
|
|
||||||
<name>dfs.replication</name>
|
|
||||||
<value>1</value>
|
|
||||||
</property>
|
|
||||||
</configuration>
|
|
||||||
----
|
|
||||||
|
|
||||||
conf/mapred-site.xml:
|
|
||||||
|
|
||||||
----
|
|
||||||
<configuration>
|
|
||||||
<property>
|
|
||||||
<name>mapred.job.tracker</name>
|
|
||||||
<value>localhost:9001</value>
|
|
||||||
</property>
|
|
||||||
</configuration>
|
|
||||||
----
|
|
||||||
|
|
||||||
** Setup passphraseless ssh
|
|
||||||
|
|
||||||
Now check that you can ssh to the localhost without a passphrase:
|
|
||||||
|
|
||||||
----
|
|
||||||
$ ssh localhost
|
|
||||||
----
|
|
||||||
|
|
||||||
If you cannot ssh to localhost without a passphrase, execute the
|
|
||||||
following commands:
|
|
||||||
|
|
||||||
----
|
|
||||||
$ ssh-keygen -t dsa -P '' -f ~/.ssh/id_dsa
|
|
||||||
$ cat ~/.ssh/id_dsa.pub >> ~/.ssh/authorized_keys
|
|
||||||
----
|
|
||||||
|
|
||||||
** Execution
|
|
||||||
|
|
||||||
Format a new distributed-filesystem:
|
|
||||||
|
|
||||||
----
|
|
||||||
$ bin/hadoop namenode -format
|
|
||||||
----
|
|
||||||
|
|
||||||
Start the hadoop daemons:
|
|
||||||
|
|
||||||
----
|
|
||||||
$ bin/start-all.sh
|
|
||||||
----
|
|
||||||
|
|
||||||
The hadoop daemon log output is written to the <<<${HADOOP_LOG_DIR}>>>
|
|
||||||
directory (defaults to <<<${HADOOP_PREFIX}/logs>>>).
|
|
||||||
|
|
||||||
Browse the web interface for the NameNode and the JobTracker; by
|
|
||||||
default they are available at:
|
|
||||||
|
|
||||||
* NameNode - <<<http://localhost:50070/>>>
|
|
||||||
|
|
||||||
* JobTracker - <<<http://localhost:50030/>>>
|
|
||||||
|
|
||||||
Copy the input files into the distributed filesystem:
|
|
||||||
|
|
||||||
----
|
|
||||||
$ bin/hadoop fs -put conf input
|
|
||||||
----
|
|
||||||
|
|
||||||
Run some of the examples provided:
|
|
||||||
|
|
||||||
----
|
|
||||||
$ bin/hadoop jar hadoop-*-examples.jar grep input output 'dfs[a-z.]+'
|
|
||||||
----
|
|
||||||
|
|
||||||
Examine the output files:
|
|
||||||
|
|
||||||
Copy the output files from the distributed filesystem to the local
|
|
||||||
filesytem and examine them:
|
|
||||||
|
|
||||||
----
|
|
||||||
$ bin/hadoop fs -get output output
|
|
||||||
$ cat output/*
|
|
||||||
----
|
|
||||||
|
|
||||||
or
|
|
||||||
|
|
||||||
View the output files on the distributed filesystem:
|
|
||||||
|
|
||||||
----
|
|
||||||
$ bin/hadoop fs -cat output/*
|
|
||||||
----
|
|
||||||
|
|
||||||
When you're done, stop the daemons with:
|
|
||||||
|
|
||||||
----
|
|
||||||
$ bin/stop-all.sh
|
|
||||||
----
|
|
||||||
|
|
||||||
* Fully-Distributed Operation
|
|
||||||
|
|
||||||
For information on setting up fully-distributed, non-trivial clusters
|
|
||||||
see {{{./ClusterSetup.html}Cluster Setup}}.
|
|
||||||
|
|
||||||
Java and JNI are trademarks or registered trademarks of Sun
|
|
||||||
Microsystems, Inc. in the United States and other countries.
|
|
||||||
|
|
|
@ -1003,6 +1003,14 @@ public class TestConfiguration extends TestCase {
|
||||||
String resource;
|
String resource;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public void testGetSetTrimmedNames() throws IOException {
|
||||||
|
Configuration conf = new Configuration(false);
|
||||||
|
conf.set(" name", "value");
|
||||||
|
assertEquals("value", conf.get("name"));
|
||||||
|
assertEquals("value", conf.get(" name"));
|
||||||
|
assertEquals("value", conf.getRaw(" name "));
|
||||||
|
}
|
||||||
|
|
||||||
public void testDumpConfiguration () throws IOException {
|
public void testDumpConfiguration () throws IOException {
|
||||||
StringWriter outWriter = new StringWriter();
|
StringWriter outWriter = new StringWriter();
|
||||||
Configuration.dumpConfiguration(conf, outWriter);
|
Configuration.dumpConfiguration(conf, outWriter);
|
||||||
|
|
|
@ -139,6 +139,7 @@ public class TestHarFileSystem {
|
||||||
public int getDefaultPort();
|
public int getDefaultPort();
|
||||||
public String getCanonicalServiceName();
|
public String getCanonicalServiceName();
|
||||||
public Token<?> getDelegationToken(String renewer) throws IOException;
|
public Token<?> getDelegationToken(String renewer) throws IOException;
|
||||||
|
public FileChecksum getFileChecksum(Path f) throws IOException;
|
||||||
public boolean deleteOnExit(Path f) throws IOException;
|
public boolean deleteOnExit(Path f) throws IOException;
|
||||||
public boolean cancelDeleteOnExit(Path f) throws IOException;
|
public boolean cancelDeleteOnExit(Path f) throws IOException;
|
||||||
public Token<?>[] addDelegationTokens(String renewer, Credentials creds)
|
public Token<?>[] addDelegationTokens(String renewer, Credentials creds)
|
||||||
|
@ -223,10 +224,16 @@ public class TestHarFileSystem {
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testFileChecksum() {
|
public void testFileChecksum() throws Exception {
|
||||||
final Path p = new Path("har://file-localhost/foo.har/file1");
|
final Path p = new Path("har://file-localhost/foo.har/file1");
|
||||||
final HarFileSystem harfs = new HarFileSystem();
|
final HarFileSystem harfs = new HarFileSystem();
|
||||||
|
try {
|
||||||
Assert.assertEquals(null, harfs.getFileChecksum(p));
|
Assert.assertEquals(null, harfs.getFileChecksum(p));
|
||||||
|
} finally {
|
||||||
|
if (harfs != null) {
|
||||||
|
harfs.close();
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
|
@ -35,6 +35,7 @@ import org.apache.hadoop.net.NetUtils;
|
||||||
import org.apache.hadoop.security.KerberosInfo;
|
import org.apache.hadoop.security.KerberosInfo;
|
||||||
import org.apache.hadoop.security.SecurityUtil;
|
import org.apache.hadoop.security.SecurityUtil;
|
||||||
import org.apache.hadoop.security.UserGroupInformation;
|
import org.apache.hadoop.security.UserGroupInformation;
|
||||||
|
import org.apache.hadoop.security.authorize.DefaultImpersonationProvider;
|
||||||
import org.apache.hadoop.security.authorize.ProxyUsers;
|
import org.apache.hadoop.security.authorize.ProxyUsers;
|
||||||
import org.apache.hadoop.security.token.Token;
|
import org.apache.hadoop.security.token.Token;
|
||||||
import org.apache.hadoop.security.token.TokenInfo;
|
import org.apache.hadoop.security.token.TokenInfo;
|
||||||
|
@ -326,7 +327,7 @@ public class MiniRPCBenchmark {
|
||||||
String shortUserName =
|
String shortUserName =
|
||||||
UserGroupInformation.createRemoteUser(user).getShortUserName();
|
UserGroupInformation.createRemoteUser(user).getShortUserName();
|
||||||
try {
|
try {
|
||||||
conf.setStrings(ProxyUsers.getProxySuperuserGroupConfKey(shortUserName),
|
conf.setStrings(DefaultImpersonationProvider.getProxySuperuserGroupConfKey(shortUserName),
|
||||||
GROUP_NAME_1);
|
GROUP_NAME_1);
|
||||||
configureSuperUserIPAddresses(conf, shortUserName);
|
configureSuperUserIPAddresses(conf, shortUserName);
|
||||||
// start the server
|
// start the server
|
||||||
|
@ -410,7 +411,7 @@ public class MiniRPCBenchmark {
|
||||||
}
|
}
|
||||||
builder.append("127.0.1.1,");
|
builder.append("127.0.1.1,");
|
||||||
builder.append(InetAddress.getLocalHost().getCanonicalHostName());
|
builder.append(InetAddress.getLocalHost().getCanonicalHostName());
|
||||||
conf.setStrings(ProxyUsers.getProxySuperuserIpConfKey(superUserShortName),
|
conf.setStrings(DefaultImpersonationProvider.getProxySuperuserIpConfKey(superUserShortName),
|
||||||
builder.toString());
|
builder.toString());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -35,6 +35,7 @@ import org.apache.hadoop.ipc.Server;
|
||||||
import org.apache.hadoop.ipc.VersionedProtocol;
|
import org.apache.hadoop.ipc.VersionedProtocol;
|
||||||
import org.apache.hadoop.net.NetUtils;
|
import org.apache.hadoop.net.NetUtils;
|
||||||
import org.apache.hadoop.security.UserGroupInformation.AuthenticationMethod;
|
import org.apache.hadoop.security.UserGroupInformation.AuthenticationMethod;
|
||||||
|
import org.apache.hadoop.security.authorize.DefaultImpersonationProvider;
|
||||||
import org.apache.hadoop.security.authorize.ProxyUsers;
|
import org.apache.hadoop.security.authorize.ProxyUsers;
|
||||||
import org.apache.hadoop.security.token.Token;
|
import org.apache.hadoop.security.token.Token;
|
||||||
import org.apache.hadoop.security.token.TokenInfo;
|
import org.apache.hadoop.security.token.TokenInfo;
|
||||||
|
@ -100,7 +101,7 @@ public class TestDoAsEffectiveUser {
|
||||||
builder.append("127.0.1.1,");
|
builder.append("127.0.1.1,");
|
||||||
builder.append(InetAddress.getLocalHost().getCanonicalHostName());
|
builder.append(InetAddress.getLocalHost().getCanonicalHostName());
|
||||||
LOG.info("Local Ip addresses: "+builder.toString());
|
LOG.info("Local Ip addresses: "+builder.toString());
|
||||||
conf.setStrings(ProxyUsers.getProxySuperuserIpConfKey(superUserShortName),
|
conf.setStrings(DefaultImpersonationProvider.getProxySuperuserIpConfKey(superUserShortName),
|
||||||
builder.toString());
|
builder.toString());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -180,7 +181,7 @@ public class TestDoAsEffectiveUser {
|
||||||
@Test(timeout=4000)
|
@Test(timeout=4000)
|
||||||
public void testRealUserSetup() throws IOException {
|
public void testRealUserSetup() throws IOException {
|
||||||
final Configuration conf = new Configuration();
|
final Configuration conf = new Configuration();
|
||||||
conf.setStrings(ProxyUsers
|
conf.setStrings(DefaultImpersonationProvider
|
||||||
.getProxySuperuserGroupConfKey(REAL_USER_SHORT_NAME), "group1");
|
.getProxySuperuserGroupConfKey(REAL_USER_SHORT_NAME), "group1");
|
||||||
configureSuperUserIPAddresses(conf, REAL_USER_SHORT_NAME);
|
configureSuperUserIPAddresses(conf, REAL_USER_SHORT_NAME);
|
||||||
Server server = new RPC.Builder(conf).setProtocol(TestProtocol.class)
|
Server server = new RPC.Builder(conf).setProtocol(TestProtocol.class)
|
||||||
|
@ -213,7 +214,7 @@ public class TestDoAsEffectiveUser {
|
||||||
public void testRealUserAuthorizationSuccess() throws IOException {
|
public void testRealUserAuthorizationSuccess() throws IOException {
|
||||||
final Configuration conf = new Configuration();
|
final Configuration conf = new Configuration();
|
||||||
configureSuperUserIPAddresses(conf, REAL_USER_SHORT_NAME);
|
configureSuperUserIPAddresses(conf, REAL_USER_SHORT_NAME);
|
||||||
conf.setStrings(ProxyUsers.getProxySuperuserGroupConfKey(REAL_USER_SHORT_NAME),
|
conf.setStrings(DefaultImpersonationProvider.getProxySuperuserGroupConfKey(REAL_USER_SHORT_NAME),
|
||||||
"group1");
|
"group1");
|
||||||
Server server = new RPC.Builder(conf).setProtocol(TestProtocol.class)
|
Server server = new RPC.Builder(conf).setProtocol(TestProtocol.class)
|
||||||
.setInstance(new TestImpl()).setBindAddress(ADDRESS).setPort(0)
|
.setInstance(new TestImpl()).setBindAddress(ADDRESS).setPort(0)
|
||||||
|
@ -247,9 +248,9 @@ public class TestDoAsEffectiveUser {
|
||||||
@Test
|
@Test
|
||||||
public void testRealUserIPAuthorizationFailure() throws IOException {
|
public void testRealUserIPAuthorizationFailure() throws IOException {
|
||||||
final Configuration conf = new Configuration();
|
final Configuration conf = new Configuration();
|
||||||
conf.setStrings(ProxyUsers.getProxySuperuserIpConfKey(REAL_USER_SHORT_NAME),
|
conf.setStrings(DefaultImpersonationProvider.getProxySuperuserIpConfKey(REAL_USER_SHORT_NAME),
|
||||||
"20.20.20.20"); //Authorized IP address
|
"20.20.20.20"); //Authorized IP address
|
||||||
conf.setStrings(ProxyUsers.getProxySuperuserGroupConfKey(REAL_USER_SHORT_NAME),
|
conf.setStrings(DefaultImpersonationProvider.getProxySuperuserGroupConfKey(REAL_USER_SHORT_NAME),
|
||||||
"group1");
|
"group1");
|
||||||
Server server = new RPC.Builder(conf).setProtocol(TestProtocol.class)
|
Server server = new RPC.Builder(conf).setProtocol(TestProtocol.class)
|
||||||
.setInstance(new TestImpl()).setBindAddress(ADDRESS).setPort(0)
|
.setInstance(new TestImpl()).setBindAddress(ADDRESS).setPort(0)
|
||||||
|
@ -292,7 +293,7 @@ public class TestDoAsEffectiveUser {
|
||||||
@Test
|
@Test
|
||||||
public void testRealUserIPNotSpecified() throws IOException {
|
public void testRealUserIPNotSpecified() throws IOException {
|
||||||
final Configuration conf = new Configuration();
|
final Configuration conf = new Configuration();
|
||||||
conf.setStrings(ProxyUsers
|
conf.setStrings(DefaultImpersonationProvider
|
||||||
.getProxySuperuserGroupConfKey(REAL_USER_SHORT_NAME), "group1");
|
.getProxySuperuserGroupConfKey(REAL_USER_SHORT_NAME), "group1");
|
||||||
Server server = new RPC.Builder(conf).setProtocol(TestProtocol.class)
|
Server server = new RPC.Builder(conf).setProtocol(TestProtocol.class)
|
||||||
.setInstance(new TestImpl()).setBindAddress(ADDRESS).setPort(0)
|
.setInstance(new TestImpl()).setBindAddress(ADDRESS).setPort(0)
|
||||||
|
@ -376,7 +377,7 @@ public class TestDoAsEffectiveUser {
|
||||||
public void testRealUserGroupAuthorizationFailure() throws IOException {
|
public void testRealUserGroupAuthorizationFailure() throws IOException {
|
||||||
final Configuration conf = new Configuration();
|
final Configuration conf = new Configuration();
|
||||||
configureSuperUserIPAddresses(conf, REAL_USER_SHORT_NAME);
|
configureSuperUserIPAddresses(conf, REAL_USER_SHORT_NAME);
|
||||||
conf.setStrings(ProxyUsers.getProxySuperuserGroupConfKey(REAL_USER_SHORT_NAME),
|
conf.setStrings(DefaultImpersonationProvider.getProxySuperuserGroupConfKey(REAL_USER_SHORT_NAME),
|
||||||
"group3");
|
"group3");
|
||||||
Server server = new RPC.Builder(conf).setProtocol(TestProtocol.class)
|
Server server = new RPC.Builder(conf).setProtocol(TestProtocol.class)
|
||||||
.setInstance(new TestImpl()).setBindAddress(ADDRESS).setPort(0)
|
.setInstance(new TestImpl()).setBindAddress(ADDRESS).setPort(0)
|
||||||
|
|
|
@ -17,6 +17,9 @@
|
||||||
*/
|
*/
|
||||||
package org.apache.hadoop.security.authorize;
|
package org.apache.hadoop.security.authorize;
|
||||||
|
|
||||||
|
import static org.junit.Assert.assertEquals;
|
||||||
|
import static org.junit.Assert.fail;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
|
@ -25,13 +28,11 @@ import org.apache.commons.logging.Log;
|
||||||
import org.apache.commons.logging.LogFactory;
|
import org.apache.commons.logging.LogFactory;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
import org.apache.hadoop.fs.CommonConfigurationKeysPublic;
|
import org.apache.hadoop.fs.CommonConfigurationKeysPublic;
|
||||||
import org.apache.hadoop.util.NativeCodeLoader;
|
|
||||||
import org.apache.hadoop.util.StringUtils;
|
|
||||||
import org.apache.hadoop.security.Groups;
|
import org.apache.hadoop.security.Groups;
|
||||||
import org.apache.hadoop.security.UserGroupInformation;
|
import org.apache.hadoop.security.UserGroupInformation;
|
||||||
|
import org.apache.hadoop.util.NativeCodeLoader;
|
||||||
|
import org.apache.hadoop.util.StringUtils;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
import static org.junit.Assert.*;
|
|
||||||
|
|
||||||
|
|
||||||
public class TestProxyUsers {
|
public class TestProxyUsers {
|
||||||
|
@ -46,6 +47,8 @@ public class TestProxyUsers {
|
||||||
new String[] { "@foo_group" };
|
new String[] { "@foo_group" };
|
||||||
private static final String[] OTHER_GROUP_NAMES =
|
private static final String[] OTHER_GROUP_NAMES =
|
||||||
new String[] { "bar_group" };
|
new String[] { "bar_group" };
|
||||||
|
private static final String[] SUDO_GROUP_NAMES =
|
||||||
|
new String[] { "sudo_proxied_user" };
|
||||||
private static final String PROXY_IP = "1.2.3.4";
|
private static final String PROXY_IP = "1.2.3.4";
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -106,10 +109,10 @@ public class TestProxyUsers {
|
||||||
groupMappingClassName);
|
groupMappingClassName);
|
||||||
|
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserGroupConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserGroupConfKey(REAL_USER_NAME),
|
||||||
StringUtils.join(",", Arrays.asList(NETGROUP_NAMES)));
|
StringUtils.join(",", Arrays.asList(NETGROUP_NAMES)));
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
||||||
PROXY_IP);
|
PROXY_IP);
|
||||||
|
|
||||||
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
||||||
|
@ -130,10 +133,10 @@ public class TestProxyUsers {
|
||||||
public void testProxyUsers() throws Exception {
|
public void testProxyUsers() throws Exception {
|
||||||
Configuration conf = new Configuration();
|
Configuration conf = new Configuration();
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserGroupConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserGroupConfKey(REAL_USER_NAME),
|
||||||
StringUtils.join(",", Arrays.asList(GROUP_NAMES)));
|
StringUtils.join(",", Arrays.asList(GROUP_NAMES)));
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
||||||
PROXY_IP);
|
PROXY_IP);
|
||||||
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
||||||
|
|
||||||
|
@ -164,10 +167,10 @@ public class TestProxyUsers {
|
||||||
public void testProxyUsersWithUserConf() throws Exception {
|
public void testProxyUsersWithUserConf() throws Exception {
|
||||||
Configuration conf = new Configuration();
|
Configuration conf = new Configuration();
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserUserConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserUserConfKey(REAL_USER_NAME),
|
||||||
StringUtils.join(",", Arrays.asList(AUTHORIZED_PROXY_USER_NAME)));
|
StringUtils.join(",", Arrays.asList(AUTHORIZED_PROXY_USER_NAME)));
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
||||||
PROXY_IP);
|
PROXY_IP);
|
||||||
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
||||||
|
|
||||||
|
@ -198,10 +201,10 @@ public class TestProxyUsers {
|
||||||
public void testWildcardGroup() {
|
public void testWildcardGroup() {
|
||||||
Configuration conf = new Configuration();
|
Configuration conf = new Configuration();
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserGroupConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserGroupConfKey(REAL_USER_NAME),
|
||||||
"*");
|
"*");
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
||||||
PROXY_IP);
|
PROXY_IP);
|
||||||
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
||||||
|
|
||||||
|
@ -232,10 +235,10 @@ public class TestProxyUsers {
|
||||||
public void testWildcardUser() {
|
public void testWildcardUser() {
|
||||||
Configuration conf = new Configuration();
|
Configuration conf = new Configuration();
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserUserConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserUserConfKey(REAL_USER_NAME),
|
||||||
"*");
|
"*");
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
||||||
PROXY_IP);
|
PROXY_IP);
|
||||||
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
||||||
|
|
||||||
|
@ -266,10 +269,10 @@ public class TestProxyUsers {
|
||||||
public void testWildcardIP() {
|
public void testWildcardIP() {
|
||||||
Configuration conf = new Configuration();
|
Configuration conf = new Configuration();
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserGroupConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserGroupConfKey(REAL_USER_NAME),
|
||||||
StringUtils.join(",", Arrays.asList(GROUP_NAMES)));
|
StringUtils.join(",", Arrays.asList(GROUP_NAMES)));
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
||||||
"*");
|
"*");
|
||||||
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
||||||
|
|
||||||
|
@ -297,15 +300,16 @@ public class TestProxyUsers {
|
||||||
public void testWithDuplicateProxyGroups() throws Exception {
|
public void testWithDuplicateProxyGroups() throws Exception {
|
||||||
Configuration conf = new Configuration();
|
Configuration conf = new Configuration();
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserGroupConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserGroupConfKey(REAL_USER_NAME),
|
||||||
StringUtils.join(",", Arrays.asList(GROUP_NAMES,GROUP_NAMES)));
|
StringUtils.join(",", Arrays.asList(GROUP_NAMES,GROUP_NAMES)));
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
||||||
PROXY_IP);
|
PROXY_IP);
|
||||||
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
||||||
|
|
||||||
Collection<String> groupsToBeProxied = ProxyUsers.getProxyGroups().get(
|
Collection<String> groupsToBeProxied =
|
||||||
ProxyUsers.getProxySuperuserGroupConfKey(REAL_USER_NAME));
|
ProxyUsers.getDefaultImpersonationProvider().getProxyGroups().get(
|
||||||
|
DefaultImpersonationProvider.getProxySuperuserGroupConfKey(REAL_USER_NAME));
|
||||||
|
|
||||||
assertEquals (1,groupsToBeProxied.size());
|
assertEquals (1,groupsToBeProxied.size());
|
||||||
}
|
}
|
||||||
|
@ -314,19 +318,52 @@ public class TestProxyUsers {
|
||||||
public void testWithDuplicateProxyHosts() throws Exception {
|
public void testWithDuplicateProxyHosts() throws Exception {
|
||||||
Configuration conf = new Configuration();
|
Configuration conf = new Configuration();
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserGroupConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserGroupConfKey(REAL_USER_NAME),
|
||||||
StringUtils.join(",", Arrays.asList(GROUP_NAMES)));
|
StringUtils.join(",", Arrays.asList(GROUP_NAMES)));
|
||||||
conf.set(
|
conf.set(
|
||||||
ProxyUsers.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
DefaultImpersonationProvider.getProxySuperuserIpConfKey(REAL_USER_NAME),
|
||||||
StringUtils.join(",", Arrays.asList(PROXY_IP,PROXY_IP)));
|
StringUtils.join(",", Arrays.asList(PROXY_IP,PROXY_IP)));
|
||||||
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
||||||
|
|
||||||
Collection<String> hosts = ProxyUsers.getProxyHosts().get(
|
Collection<String> hosts =
|
||||||
ProxyUsers.getProxySuperuserIpConfKey(REAL_USER_NAME));
|
ProxyUsers.getDefaultImpersonationProvider().getProxyHosts().get(
|
||||||
|
DefaultImpersonationProvider.getProxySuperuserIpConfKey(REAL_USER_NAME));
|
||||||
|
|
||||||
assertEquals (1,hosts.size());
|
assertEquals (1,hosts.size());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testProxyUsersWithProviderOverride() throws Exception {
|
||||||
|
Configuration conf = new Configuration();
|
||||||
|
conf.set(
|
||||||
|
CommonConfigurationKeysPublic.HADOOP_SECURITY_IMPERSONATION_PROVIDER_CLASS,
|
||||||
|
"org.apache.hadoop.security.authorize.TestProxyUsers$TestDummyImpersonationProvider");
|
||||||
|
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
||||||
|
|
||||||
|
// First try proxying a group that's allowed
|
||||||
|
UserGroupInformation realUserUgi = UserGroupInformation
|
||||||
|
.createUserForTesting(REAL_USER_NAME, SUDO_GROUP_NAMES);
|
||||||
|
UserGroupInformation proxyUserUgi = UserGroupInformation.createProxyUserForTesting(
|
||||||
|
PROXY_USER_NAME, realUserUgi, GROUP_NAMES);
|
||||||
|
|
||||||
|
// From good IP
|
||||||
|
assertAuthorized(proxyUserUgi, "1.2.3.4");
|
||||||
|
// From bad IP
|
||||||
|
assertAuthorized(proxyUserUgi, "1.2.3.5");
|
||||||
|
|
||||||
|
// Now try proxying a group that's not allowed
|
||||||
|
realUserUgi = UserGroupInformation
|
||||||
|
.createUserForTesting(REAL_USER_NAME, GROUP_NAMES);
|
||||||
|
proxyUserUgi = UserGroupInformation.createProxyUserForTesting(
|
||||||
|
PROXY_USER_NAME, realUserUgi, GROUP_NAMES);
|
||||||
|
|
||||||
|
// From good IP
|
||||||
|
assertNotAuthorized(proxyUserUgi, "1.2.3.4");
|
||||||
|
// From bad IP
|
||||||
|
assertNotAuthorized(proxyUserUgi, "1.2.3.5");
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
private void assertNotAuthorized(UserGroupInformation proxyUgi, String host) {
|
private void assertNotAuthorized(UserGroupInformation proxyUgi, String host) {
|
||||||
try {
|
try {
|
||||||
ProxyUsers.authorize(proxyUgi, host);
|
ProxyUsers.authorize(proxyUgi, host);
|
||||||
|
@ -343,4 +380,32 @@ public class TestProxyUsers {
|
||||||
fail("Did not allow authorization of " + proxyUgi + " from " + host);
|
fail("Did not allow authorization of " + proxyUgi + " from " + host);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static class TestDummyImpersonationProvider implements ImpersonationProvider {
|
||||||
|
/**
|
||||||
|
* Authorize a user (superuser) to impersonate another user (user1) if the
|
||||||
|
* superuser belongs to the group "sudo_user1" .
|
||||||
|
*/
|
||||||
|
|
||||||
|
public void authorize(UserGroupInformation user,
|
||||||
|
String remoteAddress) throws AuthorizationException{
|
||||||
|
UserGroupInformation superUser = user.getRealUser();
|
||||||
|
|
||||||
|
String sudoGroupName = "sudo_" + user.getShortUserName();
|
||||||
|
if (!Arrays.asList(superUser.getGroupNames()).contains(sudoGroupName)){
|
||||||
|
throw new AuthorizationException("User: " + superUser.getUserName()
|
||||||
|
+ " is not allowed to impersonate " + user.getUserName());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void setConf(Configuration conf) {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Configuration getConf() {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -272,4 +272,17 @@ public class TestSSLFactory {
|
||||||
sslFactory.destroy();
|
sslFactory.destroy();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testNoClientCertsInitialization() throws Exception {
|
||||||
|
Configuration conf = createConfiguration(false);
|
||||||
|
conf.unset(SSLFactory.SSL_REQUIRE_CLIENT_CERT_KEY);
|
||||||
|
SSLFactory sslFactory = new SSLFactory(SSLFactory.Mode.CLIENT, conf);
|
||||||
|
try {
|
||||||
|
sslFactory.init();
|
||||||
|
} finally {
|
||||||
|
sslFactory.destroy();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -483,5 +483,3 @@ Content-Type: application/json
|
||||||
...
|
...
|
||||||
]
|
]
|
||||||
+---+
|
+---+
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
|
@ -18,8 +18,14 @@
|
||||||
package org.apache.hadoop.nfs.nfs3;
|
package org.apache.hadoop.nfs.nfs3;
|
||||||
|
|
||||||
import java.io.BufferedReader;
|
import java.io.BufferedReader;
|
||||||
|
import java.io.File;
|
||||||
|
import java.io.FileInputStream;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.InputStreamReader;
|
import java.io.InputStreamReader;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.regex.Matcher;
|
||||||
|
import java.util.regex.Pattern;
|
||||||
|
|
||||||
import org.apache.commons.logging.Log;
|
import org.apache.commons.logging.Log;
|
||||||
import org.apache.commons.logging.LogFactory;
|
import org.apache.commons.logging.LogFactory;
|
||||||
|
@ -44,6 +50,17 @@ public class IdUserGroup {
|
||||||
static final String MAC_GET_ALL_USERS_CMD = "dscl . -list /Users UniqueID";
|
static final String MAC_GET_ALL_USERS_CMD = "dscl . -list /Users UniqueID";
|
||||||
static final String MAC_GET_ALL_GROUPS_CMD = "dscl . -list /Groups PrimaryGroupID";
|
static final String MAC_GET_ALL_GROUPS_CMD = "dscl . -list /Groups PrimaryGroupID";
|
||||||
|
|
||||||
|
// Used for finding the configured static mapping file.
|
||||||
|
static final String NFS_STATIC_MAPPING_FILE_KEY = "dfs.nfs.static.mapping.file";
|
||||||
|
private static final String NFS_STATIC_MAPPING_FILE_DEFAULT = "/etc/nfs.map";
|
||||||
|
private final File staticMappingFile;
|
||||||
|
|
||||||
|
// Used for parsing the static mapping file.
|
||||||
|
private static final Pattern EMPTY_LINE = Pattern.compile("^\\s*$");
|
||||||
|
private static final Pattern COMMENT_LINE = Pattern.compile("^\\s*#.*$");
|
||||||
|
private static final Pattern MAPPING_LINE =
|
||||||
|
Pattern.compile("^(uid|gid)\\s+(\\d+)\\s+(\\d+)\\s*(#.*)?$");
|
||||||
|
|
||||||
// Do update every 15 minutes by default
|
// Do update every 15 minutes by default
|
||||||
final static long TIMEOUT_DEFAULT = 15 * 60 * 1000; // ms
|
final static long TIMEOUT_DEFAULT = 15 * 60 * 1000; // ms
|
||||||
final static long TIMEOUT_MIN = 1 * 60 * 1000; // ms
|
final static long TIMEOUT_MIN = 1 * 60 * 1000; // ms
|
||||||
|
@ -58,6 +75,7 @@ public class IdUserGroup {
|
||||||
|
|
||||||
public IdUserGroup() throws IOException {
|
public IdUserGroup() throws IOException {
|
||||||
timeout = TIMEOUT_DEFAULT;
|
timeout = TIMEOUT_DEFAULT;
|
||||||
|
staticMappingFile = new File(NFS_STATIC_MAPPING_FILE_DEFAULT);
|
||||||
updateMaps();
|
updateMaps();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -71,6 +89,11 @@ public class IdUserGroup {
|
||||||
} else {
|
} else {
|
||||||
timeout = updateTime;
|
timeout = updateTime;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
String staticFilePath = conf.get(NFS_STATIC_MAPPING_FILE_KEY,
|
||||||
|
NFS_STATIC_MAPPING_FILE_DEFAULT);
|
||||||
|
staticMappingFile = new File(staticFilePath);
|
||||||
|
|
||||||
updateMaps();
|
updateMaps();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -137,7 +160,8 @@ public class IdUserGroup {
|
||||||
*/
|
*/
|
||||||
@VisibleForTesting
|
@VisibleForTesting
|
||||||
public static void updateMapInternal(BiMap<Integer, String> map, String mapName,
|
public static void updateMapInternal(BiMap<Integer, String> map, String mapName,
|
||||||
String command, String regex) throws IOException {
|
String command, String regex, Map<Integer, Integer> staticMapping)
|
||||||
|
throws IOException {
|
||||||
BufferedReader br = null;
|
BufferedReader br = null;
|
||||||
try {
|
try {
|
||||||
Process process = Runtime.getRuntime().exec(
|
Process process = Runtime.getRuntime().exec(
|
||||||
|
@ -151,7 +175,7 @@ public class IdUserGroup {
|
||||||
}
|
}
|
||||||
LOG.debug("add to " + mapName + "map:" + nameId[0] + " id:" + nameId[1]);
|
LOG.debug("add to " + mapName + "map:" + nameId[0] + " id:" + nameId[1]);
|
||||||
// HDFS can't differentiate duplicate names with simple authentication
|
// HDFS can't differentiate duplicate names with simple authentication
|
||||||
final Integer key = parseId(nameId[1]);
|
final Integer key = staticMapping.get(parseId(nameId[1]));
|
||||||
final String value = nameId[0];
|
final String value = nameId[0];
|
||||||
if (map.containsKey(key)) {
|
if (map.containsKey(key)) {
|
||||||
final String prevValue = map.get(key);
|
final String prevValue = map.get(key);
|
||||||
|
@ -200,13 +224,27 @@ public class IdUserGroup {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
StaticMapping staticMapping = new StaticMapping(
|
||||||
|
new HashMap<Integer, Integer>(), new HashMap<Integer, Integer>());
|
||||||
|
if (staticMappingFile.exists()) {
|
||||||
|
LOG.info("Using '" + staticMappingFile + "' for static UID/GID mapping...");
|
||||||
|
staticMapping = parseStaticMap(staticMappingFile);
|
||||||
|
} else {
|
||||||
|
LOG.info("Not doing static UID/GID mapping because '" + staticMappingFile
|
||||||
|
+ "' does not exist.");
|
||||||
|
}
|
||||||
|
|
||||||
if (OS.startsWith("Linux")) {
|
if (OS.startsWith("Linux")) {
|
||||||
updateMapInternal(uMap, "user", LINUX_GET_ALL_USERS_CMD, ":");
|
updateMapInternal(uMap, "user", LINUX_GET_ALL_USERS_CMD, ":",
|
||||||
updateMapInternal(gMap, "group", LINUX_GET_ALL_GROUPS_CMD, ":");
|
staticMapping.uidMapping);
|
||||||
|
updateMapInternal(gMap, "group", LINUX_GET_ALL_GROUPS_CMD, ":",
|
||||||
|
staticMapping.gidMapping);
|
||||||
} else {
|
} else {
|
||||||
// Mac
|
// Mac
|
||||||
updateMapInternal(uMap, "user", MAC_GET_ALL_USERS_CMD, "\\s+");
|
updateMapInternal(uMap, "user", MAC_GET_ALL_USERS_CMD, "\\s+",
|
||||||
updateMapInternal(gMap, "group", MAC_GET_ALL_GROUPS_CMD, "\\s+");
|
staticMapping.uidMapping);
|
||||||
|
updateMapInternal(gMap, "group", MAC_GET_ALL_GROUPS_CMD, "\\s+",
|
||||||
|
staticMapping.gidMapping);
|
||||||
}
|
}
|
||||||
|
|
||||||
uidNameMap = uMap;
|
uidNameMap = uMap;
|
||||||
|
@ -214,6 +252,87 @@ public class IdUserGroup {
|
||||||
lastUpdateTime = Time.monotonicNow();
|
lastUpdateTime = Time.monotonicNow();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@SuppressWarnings("serial")
|
||||||
|
static final class PassThroughMap<K> extends HashMap<K, K> {
|
||||||
|
|
||||||
|
public PassThroughMap() {
|
||||||
|
this(new HashMap<K, K>());
|
||||||
|
}
|
||||||
|
|
||||||
|
public PassThroughMap(Map<K, K> mapping) {
|
||||||
|
super();
|
||||||
|
for (Map.Entry<K, K> entry : mapping.entrySet()) {
|
||||||
|
super.put(entry.getKey(), entry.getValue());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@SuppressWarnings("unchecked")
|
||||||
|
@Override
|
||||||
|
public K get(Object key) {
|
||||||
|
if (super.containsKey(key)) {
|
||||||
|
return super.get(key);
|
||||||
|
} else {
|
||||||
|
return (K) key;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@VisibleForTesting
|
||||||
|
static final class StaticMapping {
|
||||||
|
final Map<Integer, Integer> uidMapping;
|
||||||
|
final Map<Integer, Integer> gidMapping;
|
||||||
|
|
||||||
|
public StaticMapping(Map<Integer, Integer> uidMapping,
|
||||||
|
Map<Integer, Integer> gidMapping) {
|
||||||
|
this.uidMapping = new PassThroughMap<Integer>(uidMapping);
|
||||||
|
this.gidMapping = new PassThroughMap<Integer>(gidMapping);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static StaticMapping parseStaticMap(File staticMapFile)
|
||||||
|
throws IOException {
|
||||||
|
|
||||||
|
Map<Integer, Integer> uidMapping = new HashMap<Integer, Integer>();
|
||||||
|
Map<Integer, Integer> gidMapping = new HashMap<Integer, Integer>();
|
||||||
|
|
||||||
|
BufferedReader in = new BufferedReader(new InputStreamReader(
|
||||||
|
new FileInputStream(staticMapFile)));
|
||||||
|
|
||||||
|
try {
|
||||||
|
String line = null;
|
||||||
|
while ((line = in.readLine()) != null) {
|
||||||
|
// Skip entirely empty and comment lines.
|
||||||
|
if (EMPTY_LINE.matcher(line).matches() ||
|
||||||
|
COMMENT_LINE.matcher(line).matches()) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
Matcher lineMatcher = MAPPING_LINE.matcher(line);
|
||||||
|
if (!lineMatcher.matches()) {
|
||||||
|
LOG.warn("Could not parse line '" + line + "'. Lines should be of " +
|
||||||
|
"the form '[uid|gid] [remote id] [local id]'. Blank lines and " +
|
||||||
|
"everything following a '#' on a line will be ignored.");
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// We know the line is fine to parse without error checking like this
|
||||||
|
// since it matched the regex above.
|
||||||
|
String firstComponent = lineMatcher.group(1);
|
||||||
|
int remoteId = Integer.parseInt(lineMatcher.group(2));
|
||||||
|
int localId = Integer.parseInt(lineMatcher.group(3));
|
||||||
|
if (firstComponent.equals("uid")) {
|
||||||
|
uidMapping.put(localId, remoteId);
|
||||||
|
} else {
|
||||||
|
gidMapping.put(localId, remoteId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} finally {
|
||||||
|
in.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
return new StaticMapping(uidMapping, gidMapping);
|
||||||
|
}
|
||||||
|
|
||||||
synchronized public int getUid(String user) throws IOException {
|
synchronized public int getUid(String user) throws IOException {
|
||||||
checkAndUpdateMaps();
|
checkAndUpdateMaps();
|
||||||
|
|
||||||
|
|
|
@ -46,10 +46,12 @@ public class ACCESS3Response extends NFS3Response {
|
||||||
@Override
|
@Override
|
||||||
public XDR writeHeaderAndResponse(XDR out, int xid, Verifier verifier) {
|
public XDR writeHeaderAndResponse(XDR out, int xid, Verifier verifier) {
|
||||||
super.writeHeaderAndResponse(out, xid, verifier);
|
super.writeHeaderAndResponse(out, xid, verifier);
|
||||||
|
if (this.getStatus() == Nfs3Status.NFS3_OK) {
|
||||||
out.writeBoolean(true);
|
out.writeBoolean(true);
|
||||||
postOpAttr.serialize(out);
|
postOpAttr.serialize(out);
|
||||||
if (this.getStatus() == Nfs3Status.NFS3_OK) {
|
|
||||||
out.writeInt(access);
|
out.writeInt(access);
|
||||||
|
} else {
|
||||||
|
out.writeBoolean(false);
|
||||||
}
|
}
|
||||||
return out;
|
return out;
|
||||||
}
|
}
|
||||||
|
|
|
@ -19,9 +19,16 @@ package org.apache.hadoop.nfs.nfs3;
|
||||||
|
|
||||||
import static org.junit.Assert.assertEquals;
|
import static org.junit.Assert.assertEquals;
|
||||||
import static org.junit.Assert.assertTrue;
|
import static org.junit.Assert.assertTrue;
|
||||||
|
|
||||||
|
import java.io.File;
|
||||||
|
import java.io.FileOutputStream;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
|
import java.io.OutputStream;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
|
import org.apache.hadoop.nfs.nfs3.IdUserGroup.PassThroughMap;
|
||||||
|
import org.apache.hadoop.nfs.nfs3.IdUserGroup.StaticMapping;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
|
|
||||||
import com.google.common.collect.BiMap;
|
import com.google.common.collect.BiMap;
|
||||||
|
@ -29,6 +36,81 @@ import com.google.common.collect.HashBiMap;
|
||||||
|
|
||||||
public class TestIdUserGroup {
|
public class TestIdUserGroup {
|
||||||
|
|
||||||
|
private static final Map<Integer, Integer> EMPTY_PASS_THROUGH_MAP =
|
||||||
|
new PassThroughMap<Integer>();
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testStaticMapParsing() throws IOException {
|
||||||
|
File tempStaticMapFile = File.createTempFile("nfs-", ".map");
|
||||||
|
final String staticMapFileContents =
|
||||||
|
"uid 10 100\n" +
|
||||||
|
"gid 10 200\n" +
|
||||||
|
"uid 11 201 # comment at the end of a line\n" +
|
||||||
|
"uid 12 301\n" +
|
||||||
|
"# Comment at the beginning of a line\n" +
|
||||||
|
" # Comment that starts late in the line\n" +
|
||||||
|
"uid 10000 10001# line without whitespace before comment\n" +
|
||||||
|
"uid 13 302\n" +
|
||||||
|
"gid\t11\t201\n" + // Tabs instead of spaces.
|
||||||
|
"\n" + // Entirely empty line.
|
||||||
|
"gid 12 202";
|
||||||
|
OutputStream out = new FileOutputStream(tempStaticMapFile);
|
||||||
|
out.write(staticMapFileContents.getBytes());
|
||||||
|
out.close();
|
||||||
|
StaticMapping parsedMap = IdUserGroup.parseStaticMap(tempStaticMapFile);
|
||||||
|
|
||||||
|
assertEquals(10, (int)parsedMap.uidMapping.get(100));
|
||||||
|
assertEquals(11, (int)parsedMap.uidMapping.get(201));
|
||||||
|
assertEquals(12, (int)parsedMap.uidMapping.get(301));
|
||||||
|
assertEquals(13, (int)parsedMap.uidMapping.get(302));
|
||||||
|
assertEquals(10, (int)parsedMap.gidMapping.get(200));
|
||||||
|
assertEquals(11, (int)parsedMap.gidMapping.get(201));
|
||||||
|
assertEquals(12, (int)parsedMap.gidMapping.get(202));
|
||||||
|
assertEquals(10000, (int)parsedMap.uidMapping.get(10001));
|
||||||
|
// Ensure pass-through of unmapped IDs works.
|
||||||
|
assertEquals(1000, (int)parsedMap.uidMapping.get(1000));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testStaticMapping() throws IOException {
|
||||||
|
Map<Integer, Integer> uidStaticMap = new PassThroughMap<Integer>();
|
||||||
|
Map<Integer, Integer> gidStaticMap = new PassThroughMap<Integer>();
|
||||||
|
|
||||||
|
uidStaticMap.put(11501, 10);
|
||||||
|
gidStaticMap.put(497, 200);
|
||||||
|
|
||||||
|
// Maps for id to name map
|
||||||
|
BiMap<Integer, String> uMap = HashBiMap.create();
|
||||||
|
BiMap<Integer, String> gMap = HashBiMap.create();
|
||||||
|
|
||||||
|
String GET_ALL_USERS_CMD =
|
||||||
|
"echo \"atm:x:1000:1000:Aaron T. Myers,,,:/home/atm:/bin/bash\n"
|
||||||
|
+ "hdfs:x:11501:10787:Grid Distributed File System:/home/hdfs:/bin/bash\""
|
||||||
|
+ " | cut -d: -f1,3";
|
||||||
|
|
||||||
|
String GET_ALL_GROUPS_CMD = "echo \"hdfs:*:11501:hrt_hdfs\n"
|
||||||
|
+ "mapred:x:497\n"
|
||||||
|
+ "mapred2:x:498\""
|
||||||
|
+ " | cut -d: -f1,3";
|
||||||
|
|
||||||
|
IdUserGroup.updateMapInternal(uMap, "user", GET_ALL_USERS_CMD, ":",
|
||||||
|
uidStaticMap);
|
||||||
|
IdUserGroup.updateMapInternal(gMap, "group", GET_ALL_GROUPS_CMD, ":",
|
||||||
|
gidStaticMap);
|
||||||
|
|
||||||
|
assertEquals("hdfs", uMap.get(10));
|
||||||
|
assertEquals(10, (int)uMap.inverse().get("hdfs"));
|
||||||
|
assertEquals("atm", uMap.get(1000));
|
||||||
|
assertEquals(1000, (int)uMap.inverse().get("atm"));
|
||||||
|
|
||||||
|
assertEquals("hdfs", gMap.get(11501));
|
||||||
|
assertEquals(11501, (int)gMap.inverse().get("hdfs"));
|
||||||
|
assertEquals("mapred", gMap.get(200));
|
||||||
|
assertEquals(200, (int)gMap.inverse().get("mapred"));
|
||||||
|
assertEquals("mapred2", gMap.get(498));
|
||||||
|
assertEquals(498, (int)gMap.inverse().get("mapred2"));
|
||||||
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testDuplicates() throws IOException {
|
public void testDuplicates() throws IOException {
|
||||||
String GET_ALL_USERS_CMD = "echo \"root:x:0:0:root:/root:/bin/bash\n"
|
String GET_ALL_USERS_CMD = "echo \"root:x:0:0:root:/root:/bin/bash\n"
|
||||||
|
@ -51,15 +133,17 @@ public class TestIdUserGroup {
|
||||||
BiMap<Integer, String> uMap = HashBiMap.create();
|
BiMap<Integer, String> uMap = HashBiMap.create();
|
||||||
BiMap<Integer, String> gMap = HashBiMap.create();
|
BiMap<Integer, String> gMap = HashBiMap.create();
|
||||||
|
|
||||||
IdUserGroup.updateMapInternal(uMap, "user", GET_ALL_USERS_CMD, ":");
|
IdUserGroup.updateMapInternal(uMap, "user", GET_ALL_USERS_CMD, ":",
|
||||||
assertTrue(uMap.size() == 5);
|
EMPTY_PASS_THROUGH_MAP);
|
||||||
|
assertEquals(5, uMap.size());
|
||||||
assertEquals("root", uMap.get(0));
|
assertEquals("root", uMap.get(0));
|
||||||
assertEquals("hdfs", uMap.get(11501));
|
assertEquals("hdfs", uMap.get(11501));
|
||||||
assertEquals("hdfs2",uMap.get(11502));
|
assertEquals("hdfs2",uMap.get(11502));
|
||||||
assertEquals("bin", uMap.get(2));
|
assertEquals("bin", uMap.get(2));
|
||||||
assertEquals("daemon", uMap.get(1));
|
assertEquals("daemon", uMap.get(1));
|
||||||
|
|
||||||
IdUserGroup.updateMapInternal(gMap, "group", GET_ALL_GROUPS_CMD, ":");
|
IdUserGroup.updateMapInternal(gMap, "group", GET_ALL_GROUPS_CMD, ":",
|
||||||
|
EMPTY_PASS_THROUGH_MAP);
|
||||||
assertTrue(gMap.size() == 3);
|
assertTrue(gMap.size() == 3);
|
||||||
assertEquals("hdfs",gMap.get(11501));
|
assertEquals("hdfs",gMap.get(11501));
|
||||||
assertEquals("mapred", gMap.get(497));
|
assertEquals("mapred", gMap.get(497));
|
||||||
|
@ -90,7 +174,8 @@ public class TestIdUserGroup {
|
||||||
BiMap<Integer, String> uMap = HashBiMap.create();
|
BiMap<Integer, String> uMap = HashBiMap.create();
|
||||||
BiMap<Integer, String> gMap = HashBiMap.create();
|
BiMap<Integer, String> gMap = HashBiMap.create();
|
||||||
|
|
||||||
IdUserGroup.updateMapInternal(uMap, "user", GET_ALL_USERS_CMD, ":");
|
IdUserGroup.updateMapInternal(uMap, "user", GET_ALL_USERS_CMD, ":",
|
||||||
|
EMPTY_PASS_THROUGH_MAP);
|
||||||
assertTrue(uMap.size() == 7);
|
assertTrue(uMap.size() == 7);
|
||||||
assertEquals("nfsnobody", uMap.get(-2));
|
assertEquals("nfsnobody", uMap.get(-2));
|
||||||
assertEquals("nfsnobody1", uMap.get(-1));
|
assertEquals("nfsnobody1", uMap.get(-1));
|
||||||
|
@ -100,7 +185,8 @@ public class TestIdUserGroup {
|
||||||
assertEquals("hdfs",uMap.get(11501));
|
assertEquals("hdfs",uMap.get(11501));
|
||||||
assertEquals("daemon", uMap.get(2));
|
assertEquals("daemon", uMap.get(2));
|
||||||
|
|
||||||
IdUserGroup.updateMapInternal(gMap, "group", GET_ALL_GROUPS_CMD, ":");
|
IdUserGroup.updateMapInternal(gMap, "group", GET_ALL_GROUPS_CMD, ":",
|
||||||
|
EMPTY_PASS_THROUGH_MAP);
|
||||||
assertTrue(gMap.size() == 7);
|
assertTrue(gMap.size() == 7);
|
||||||
assertEquals("hdfs",gMap.get(11501));
|
assertEquals("hdfs",gMap.get(11501));
|
||||||
assertEquals("rpcuser", gMap.get(29));
|
assertEquals("rpcuser", gMap.get(29));
|
||||||
|
|
|
@ -18,8 +18,6 @@
|
||||||
|
|
||||||
Hadoop HDFS over HTTP ${project.version} - Server Setup
|
Hadoop HDFS over HTTP ${project.version} - Server Setup
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
||||||
This page explains how to quickly setup HttpFS with Pseudo authentication
|
This page explains how to quickly setup HttpFS with Pseudo authentication
|
||||||
against a Hadoop cluster with Pseudo authentication.
|
against a Hadoop cluster with Pseudo authentication.
|
||||||
|
|
||||||
|
@ -159,5 +157,3 @@ $ keytool -genkey -alias tomcat -keyalg RSA
|
||||||
<<<swebhdfs://>>> scheme. Make sure the JVM is picking up the truststore
|
<<<swebhdfs://>>> scheme. Make sure the JVM is picking up the truststore
|
||||||
containing the public key of the SSL certificate if using a self-signed
|
containing the public key of the SSL certificate if using a self-signed
|
||||||
certificate.
|
certificate.
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
|
@ -18,8 +18,6 @@
|
||||||
|
|
||||||
Hadoop HDFS over HTTP ${project.version} - Using HTTP Tools
|
Hadoop HDFS over HTTP ${project.version} - Using HTTP Tools
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
||||||
* Security
|
* Security
|
||||||
|
|
||||||
Out of the box HttpFS supports both pseudo authentication and Kerberos HTTP
|
Out of the box HttpFS supports both pseudo authentication and Kerberos HTTP
|
||||||
|
@ -87,5 +85,3 @@ $ curl --negotiate -u foo -c ~/.httpfsauth "http://<HTTPFS_HOST>:14000/webhdfs/v
|
||||||
+---+
|
+---+
|
||||||
$ curl -b ~/.httpfsauth "http://<HTTPFS_HOST>:14000/webhdfs/v1?op=liststatus"
|
$ curl -b ~/.httpfsauth "http://<HTTPFS_HOST>:14000/webhdfs/v1?op=liststatus"
|
||||||
+---+
|
+---+
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
|
@ -54,6 +54,7 @@ import org.apache.hadoop.nfs.nfs3.response.WccData;
|
||||||
import org.apache.hadoop.oncrpc.XDR;
|
import org.apache.hadoop.oncrpc.XDR;
|
||||||
import org.apache.hadoop.oncrpc.security.VerifierNone;
|
import org.apache.hadoop.oncrpc.security.VerifierNone;
|
||||||
import org.apache.hadoop.util.Daemon;
|
import org.apache.hadoop.util.Daemon;
|
||||||
|
import org.apache.hadoop.util.Time;
|
||||||
import org.jboss.netty.channel.Channel;
|
import org.jboss.netty.channel.Channel;
|
||||||
|
|
||||||
import com.google.common.annotations.VisibleForTesting;
|
import com.google.common.annotations.VisibleForTesting;
|
||||||
|
@ -136,7 +137,7 @@ class OpenFileCtx {
|
||||||
this.channel = channel;
|
this.channel = channel;
|
||||||
this.xid = xid;
|
this.xid = xid;
|
||||||
this.preOpAttr = preOpAttr;
|
this.preOpAttr = preOpAttr;
|
||||||
this.startTime = System.currentTimeMillis();
|
this.startTime = Time.monotonicNow();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
@ -158,11 +159,11 @@ class OpenFileCtx {
|
||||||
private Daemon dumpThread;
|
private Daemon dumpThread;
|
||||||
|
|
||||||
private void updateLastAccessTime() {
|
private void updateLastAccessTime() {
|
||||||
lastAccessTime = System.currentTimeMillis();
|
lastAccessTime = Time.monotonicNow();
|
||||||
}
|
}
|
||||||
|
|
||||||
private boolean checkStreamTimeout(long streamTimeout) {
|
private boolean checkStreamTimeout(long streamTimeout) {
|
||||||
return System.currentTimeMillis() - lastAccessTime > streamTimeout;
|
return Time.monotonicNow() - lastAccessTime > streamTimeout;
|
||||||
}
|
}
|
||||||
|
|
||||||
long getLastAccessTime() {
|
long getLastAccessTime() {
|
||||||
|
@ -696,7 +697,7 @@ class OpenFileCtx {
|
||||||
+ " updating the mtime, then return success");
|
+ " updating the mtime, then return success");
|
||||||
Nfs3FileAttributes postOpAttr = null;
|
Nfs3FileAttributes postOpAttr = null;
|
||||||
try {
|
try {
|
||||||
dfsClient.setTimes(path, System.currentTimeMillis(), -1);
|
dfsClient.setTimes(path, Time.monotonicNow(), -1);
|
||||||
postOpAttr = Nfs3Utils.getFileAttr(dfsClient, path, iug);
|
postOpAttr = Nfs3Utils.getFileAttr(dfsClient, path, iug);
|
||||||
} catch (IOException e) {
|
} catch (IOException e) {
|
||||||
LOG.info("Got error when processing perfect overwrite, path=" + path
|
LOG.info("Got error when processing perfect overwrite, path=" + path
|
||||||
|
@ -997,7 +998,7 @@ class OpenFileCtx {
|
||||||
|
|
||||||
if (LOG.isDebugEnabled()) {
|
if (LOG.isDebugEnabled()) {
|
||||||
LOG.debug("FileId: " + latestAttr.getFileId() + " Service time:"
|
LOG.debug("FileId: " + latestAttr.getFileId() + " Service time:"
|
||||||
+ (System.currentTimeMillis() - commit.getStartTime())
|
+ (Time.monotonicNow() - commit.getStartTime())
|
||||||
+ "ms. Sent response for commit:" + commit);
|
+ "ms. Sent response for commit:" + commit);
|
||||||
}
|
}
|
||||||
entry = pendingCommits.firstEntry();
|
entry = pendingCommits.firstEntry();
|
||||||
|
|
|
@ -28,6 +28,7 @@ import org.apache.hadoop.conf.Configuration;
|
||||||
import org.apache.hadoop.nfs.nfs3.FileHandle;
|
import org.apache.hadoop.nfs.nfs3.FileHandle;
|
||||||
import org.apache.hadoop.nfs.nfs3.Nfs3Constant;
|
import org.apache.hadoop.nfs.nfs3.Nfs3Constant;
|
||||||
import org.apache.hadoop.util.Daemon;
|
import org.apache.hadoop.util.Daemon;
|
||||||
|
import org.apache.hadoop.util.Time;
|
||||||
|
|
||||||
import com.google.common.annotations.VisibleForTesting;
|
import com.google.common.annotations.VisibleForTesting;
|
||||||
import com.google.common.base.Preconditions;
|
import com.google.common.base.Preconditions;
|
||||||
|
@ -99,7 +100,7 @@ class OpenFileCtxCache {
|
||||||
LOG.warn("No eviction candidate. All streams have pending work.");
|
LOG.warn("No eviction candidate. All streams have pending work.");
|
||||||
return null;
|
return null;
|
||||||
} else {
|
} else {
|
||||||
long idleTime = System.currentTimeMillis()
|
long idleTime = Time.monotonicNow()
|
||||||
- idlest.getValue().getLastAccessTime();
|
- idlest.getValue().getLastAccessTime();
|
||||||
if (idleTime < Nfs3Constant.OUTPUT_STREAM_TIMEOUT_MIN_DEFAULT) {
|
if (idleTime < Nfs3Constant.OUTPUT_STREAM_TIMEOUT_MIN_DEFAULT) {
|
||||||
if (LOG.isDebugEnabled()) {
|
if (LOG.isDebugEnabled()) {
|
||||||
|
@ -250,7 +251,7 @@ class OpenFileCtxCache {
|
||||||
|
|
||||||
// Check if it can sleep
|
// Check if it can sleep
|
||||||
try {
|
try {
|
||||||
long workedTime = System.currentTimeMillis() - lastWakeupTime;
|
long workedTime = Time.monotonicNow() - lastWakeupTime;
|
||||||
if (workedTime < rotation) {
|
if (workedTime < rotation) {
|
||||||
if (LOG.isTraceEnabled()) {
|
if (LOG.isTraceEnabled()) {
|
||||||
LOG.trace("StreamMonitor can still have a sleep:"
|
LOG.trace("StreamMonitor can still have a sleep:"
|
||||||
|
@ -258,7 +259,7 @@ class OpenFileCtxCache {
|
||||||
}
|
}
|
||||||
Thread.sleep(rotation - workedTime);
|
Thread.sleep(rotation - workedTime);
|
||||||
}
|
}
|
||||||
lastWakeupTime = System.currentTimeMillis();
|
lastWakeupTime = Time.monotonicNow();
|
||||||
|
|
||||||
} catch (InterruptedException e) {
|
} catch (InterruptedException e) {
|
||||||
LOG.info("StreamMonitor got interrupted");
|
LOG.info("StreamMonitor got interrupted");
|
||||||
|
|
|
@ -44,7 +44,6 @@ import org.apache.hadoop.hdfs.client.HdfsDataOutputStream;
|
||||||
import org.apache.hadoop.hdfs.protocol.DirectoryListing;
|
import org.apache.hadoop.hdfs.protocol.DirectoryListing;
|
||||||
import org.apache.hadoop.hdfs.protocol.HdfsConstants;
|
import org.apache.hadoop.hdfs.protocol.HdfsConstants;
|
||||||
import org.apache.hadoop.hdfs.protocol.HdfsFileStatus;
|
import org.apache.hadoop.hdfs.protocol.HdfsFileStatus;
|
||||||
import org.apache.hadoop.hdfs.server.namenode.NameNode;
|
|
||||||
import org.apache.hadoop.ipc.RemoteException;
|
import org.apache.hadoop.ipc.RemoteException;
|
||||||
import org.apache.hadoop.nfs.AccessPrivilege;
|
import org.apache.hadoop.nfs.AccessPrivilege;
|
||||||
import org.apache.hadoop.nfs.NfsExports;
|
import org.apache.hadoop.nfs.NfsExports;
|
||||||
|
@ -124,6 +123,7 @@ import org.apache.hadoop.oncrpc.security.VerifierNone;
|
||||||
import org.apache.hadoop.security.AccessControlException;
|
import org.apache.hadoop.security.AccessControlException;
|
||||||
import org.apache.hadoop.security.SecurityUtil;
|
import org.apache.hadoop.security.SecurityUtil;
|
||||||
import org.apache.hadoop.security.UserGroupInformation;
|
import org.apache.hadoop.security.UserGroupInformation;
|
||||||
|
import org.apache.hadoop.security.authorize.AuthorizationException;
|
||||||
import org.jboss.netty.buffer.ChannelBuffer;
|
import org.jboss.netty.buffer.ChannelBuffer;
|
||||||
import org.jboss.netty.buffer.ChannelBuffers;
|
import org.jboss.netty.buffer.ChannelBuffers;
|
||||||
import org.jboss.netty.channel.Channel;
|
import org.jboss.netty.channel.Channel;
|
||||||
|
@ -151,13 +151,6 @@ public class RpcProgramNfs3 extends RpcProgram implements Nfs3Interface {
|
||||||
|
|
||||||
private final NfsExports exports;
|
private final NfsExports exports;
|
||||||
|
|
||||||
/**
|
|
||||||
* superUserClient should always impersonate HDFS file system owner to send
|
|
||||||
* requests which requires supergroup privilege. This requires the same user
|
|
||||||
* to start HDFS and NFS.
|
|
||||||
*/
|
|
||||||
private final DFSClient superUserClient;
|
|
||||||
|
|
||||||
private final short replication;
|
private final short replication;
|
||||||
private final long blockSize;
|
private final long blockSize;
|
||||||
private final int bufferSize;
|
private final int bufferSize;
|
||||||
|
@ -179,7 +172,6 @@ public class RpcProgramNfs3 extends RpcProgram implements Nfs3Interface {
|
||||||
exports = NfsExports.getInstance(config);
|
exports = NfsExports.getInstance(config);
|
||||||
writeManager = new WriteManager(iug, config);
|
writeManager = new WriteManager(iug, config);
|
||||||
clientCache = new DFSClientCache(config);
|
clientCache = new DFSClientCache(config);
|
||||||
superUserClient = new DFSClient(NameNode.getAddress(config), config);
|
|
||||||
replication = (short) config.getInt(DFSConfigKeys.DFS_REPLICATION_KEY,
|
replication = (short) config.getInt(DFSConfigKeys.DFS_REPLICATION_KEY,
|
||||||
DFSConfigKeys.DFS_REPLICATION_DEFAULT);
|
DFSConfigKeys.DFS_REPLICATION_DEFAULT);
|
||||||
blockSize = config.getLong(DFSConfigKeys.DFS_BLOCK_SIZE_KEY,
|
blockSize = config.getLong(DFSConfigKeys.DFS_BLOCK_SIZE_KEY,
|
||||||
|
@ -270,6 +262,17 @@ public class RpcProgramNfs3 extends RpcProgram implements Nfs3Interface {
|
||||||
Nfs3FileAttributes attrs = null;
|
Nfs3FileAttributes attrs = null;
|
||||||
try {
|
try {
|
||||||
attrs = writeManager.getFileAttr(dfsClient, handle, iug);
|
attrs = writeManager.getFileAttr(dfsClient, handle, iug);
|
||||||
|
} catch (RemoteException r) {
|
||||||
|
LOG.warn("Exception ", r);
|
||||||
|
IOException io = r.unwrapRemoteException();
|
||||||
|
/**
|
||||||
|
* AuthorizationException can be thrown if the user can't be proxy'ed.
|
||||||
|
*/
|
||||||
|
if (io instanceof AuthorizationException) {
|
||||||
|
return new GETATTR3Response(Nfs3Status.NFS3ERR_ACCES);
|
||||||
|
} else {
|
||||||
|
return new GETATTR3Response(Nfs3Status.NFS3ERR_IO);
|
||||||
|
}
|
||||||
} catch (IOException e) {
|
} catch (IOException e) {
|
||||||
LOG.info("Can't get file attribute, fileId=" + handle.getFileId(), e);
|
LOG.info("Can't get file attribute, fileId=" + handle.getFileId(), e);
|
||||||
response.setStatus(Nfs3Status.NFS3ERR_IO);
|
response.setStatus(Nfs3Status.NFS3ERR_IO);
|
||||||
|
@ -499,6 +502,17 @@ public class RpcProgramNfs3 extends RpcProgram implements Nfs3Interface {
|
||||||
securityHandler.getUid(), securityHandler.getGid(), attrs);
|
securityHandler.getUid(), securityHandler.getGid(), attrs);
|
||||||
|
|
||||||
return new ACCESS3Response(Nfs3Status.NFS3_OK, attrs, access);
|
return new ACCESS3Response(Nfs3Status.NFS3_OK, attrs, access);
|
||||||
|
} catch (RemoteException r) {
|
||||||
|
LOG.warn("Exception ", r);
|
||||||
|
IOException io = r.unwrapRemoteException();
|
||||||
|
/**
|
||||||
|
* AuthorizationException can be thrown if the user can't be proxy'ed.
|
||||||
|
*/
|
||||||
|
if (io instanceof AuthorizationException) {
|
||||||
|
return new ACCESS3Response(Nfs3Status.NFS3ERR_ACCES);
|
||||||
|
} else {
|
||||||
|
return new ACCESS3Response(Nfs3Status.NFS3ERR_IO);
|
||||||
|
}
|
||||||
} catch (IOException e) {
|
} catch (IOException e) {
|
||||||
LOG.warn("Exception ", e);
|
LOG.warn("Exception ", e);
|
||||||
return new ACCESS3Response(Nfs3Status.NFS3ERR_IO);
|
return new ACCESS3Response(Nfs3Status.NFS3ERR_IO);
|
||||||
|
@ -1658,8 +1672,7 @@ public class RpcProgramNfs3 extends RpcProgram implements Nfs3Interface {
|
||||||
}
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
// Use superUserClient to get file system status
|
FsStatus fsStatus = dfsClient.getDiskStatus();
|
||||||
FsStatus fsStatus = superUserClient.getDiskStatus();
|
|
||||||
long totalBytes = fsStatus.getCapacity();
|
long totalBytes = fsStatus.getCapacity();
|
||||||
long freeBytes = fsStatus.getRemaining();
|
long freeBytes = fsStatus.getRemaining();
|
||||||
|
|
||||||
|
@ -1680,6 +1693,17 @@ public class RpcProgramNfs3 extends RpcProgram implements Nfs3Interface {
|
||||||
|
|
||||||
return new FSSTAT3Response(Nfs3Status.NFS3_OK, attrs, totalBytes,
|
return new FSSTAT3Response(Nfs3Status.NFS3_OK, attrs, totalBytes,
|
||||||
freeBytes, freeBytes, maxFsObjects, maxFsObjects, maxFsObjects, 0);
|
freeBytes, freeBytes, maxFsObjects, maxFsObjects, maxFsObjects, 0);
|
||||||
|
} catch (RemoteException r) {
|
||||||
|
LOG.warn("Exception ", r);
|
||||||
|
IOException io = r.unwrapRemoteException();
|
||||||
|
/**
|
||||||
|
* AuthorizationException can be thrown if the user can't be proxy'ed.
|
||||||
|
*/
|
||||||
|
if (io instanceof AuthorizationException) {
|
||||||
|
return new FSSTAT3Response(Nfs3Status.NFS3ERR_ACCES);
|
||||||
|
} else {
|
||||||
|
return new FSSTAT3Response(Nfs3Status.NFS3ERR_IO);
|
||||||
|
}
|
||||||
} catch (IOException e) {
|
} catch (IOException e) {
|
||||||
LOG.warn("Exception ", e);
|
LOG.warn("Exception ", e);
|
||||||
return new FSSTAT3Response(Nfs3Status.NFS3ERR_IO);
|
return new FSSTAT3Response(Nfs3Status.NFS3ERR_IO);
|
||||||
|
|
|
@ -22,7 +22,6 @@ import static org.junit.Assert.assertTrue;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.net.InetAddress;
|
import java.net.InetAddress;
|
||||||
import java.util.Arrays;
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
|
@ -41,9 +40,8 @@ import org.apache.hadoop.nfs.nfs3.response.READDIRPLUS3Response;
|
||||||
import org.apache.hadoop.nfs.nfs3.response.READDIRPLUS3Response.EntryPlus3;
|
import org.apache.hadoop.nfs.nfs3.response.READDIRPLUS3Response.EntryPlus3;
|
||||||
import org.apache.hadoop.oncrpc.XDR;
|
import org.apache.hadoop.oncrpc.XDR;
|
||||||
import org.apache.hadoop.oncrpc.security.SecurityHandler;
|
import org.apache.hadoop.oncrpc.security.SecurityHandler;
|
||||||
import org.apache.hadoop.security.UserGroupInformation;
|
import org.apache.hadoop.security.authorize.DefaultImpersonationProvider;
|
||||||
import org.apache.hadoop.security.authorize.ProxyUsers;
|
import org.apache.hadoop.security.authorize.ProxyUsers;
|
||||||
import org.apache.hadoop.util.StringUtils;
|
|
||||||
import org.junit.AfterClass;
|
import org.junit.AfterClass;
|
||||||
import org.junit.Before;
|
import org.junit.Before;
|
||||||
import org.junit.BeforeClass;
|
import org.junit.BeforeClass;
|
||||||
|
@ -67,10 +65,10 @@ public class TestReaddir {
|
||||||
public static void setup() throws Exception {
|
public static void setup() throws Exception {
|
||||||
String currentUser = System.getProperty("user.name");
|
String currentUser = System.getProperty("user.name");
|
||||||
config.set(
|
config.set(
|
||||||
ProxyUsers.getProxySuperuserGroupConfKey(currentUser),
|
DefaultImpersonationProvider.getProxySuperuserGroupConfKey(currentUser),
|
||||||
"*");
|
"*");
|
||||||
config.set(
|
config.set(
|
||||||
ProxyUsers.getProxySuperuserIpConfKey(currentUser),
|
DefaultImpersonationProvider.getProxySuperuserIpConfKey(currentUser),
|
||||||
"*");
|
"*");
|
||||||
ProxyUsers.refreshSuperUserGroupsConfiguration(config);
|
ProxyUsers.refreshSuperUserGroupsConfiguration(config);
|
||||||
cluster = new MiniDFSCluster.Builder(config).numDataNodes(1).build();
|
cluster = new MiniDFSCluster.Builder(config).numDataNodes(1).build();
|
||||||
|
|
|
@ -50,6 +50,7 @@ import org.apache.hadoop.nfs.nfs3.response.CREATE3Response;
|
||||||
import org.apache.hadoop.nfs.nfs3.response.READ3Response;
|
import org.apache.hadoop.nfs.nfs3.response.READ3Response;
|
||||||
import org.apache.hadoop.oncrpc.XDR;
|
import org.apache.hadoop.oncrpc.XDR;
|
||||||
import org.apache.hadoop.oncrpc.security.SecurityHandler;
|
import org.apache.hadoop.oncrpc.security.SecurityHandler;
|
||||||
|
import org.apache.hadoop.security.authorize.DefaultImpersonationProvider;
|
||||||
import org.apache.hadoop.security.authorize.ProxyUsers;
|
import org.apache.hadoop.security.authorize.ProxyUsers;
|
||||||
import org.jboss.netty.channel.Channel;
|
import org.jboss.netty.channel.Channel;
|
||||||
import org.junit.Assert;
|
import org.junit.Assert;
|
||||||
|
@ -288,10 +289,10 @@ public class TestWrites {
|
||||||
System.getProperty("user.name"));
|
System.getProperty("user.name"));
|
||||||
String currentUser = System.getProperty("user.name");
|
String currentUser = System.getProperty("user.name");
|
||||||
config.set(
|
config.set(
|
||||||
ProxyUsers.getProxySuperuserGroupConfKey(currentUser),
|
DefaultImpersonationProvider.getProxySuperuserGroupConfKey(currentUser),
|
||||||
"*");
|
"*");
|
||||||
config.set(
|
config.set(
|
||||||
ProxyUsers.getProxySuperuserIpConfKey(currentUser),
|
DefaultImpersonationProvider.getProxySuperuserIpConfKey(currentUser),
|
||||||
"*");
|
"*");
|
||||||
ProxyUsers.refreshSuperUserGroupsConfiguration(config);
|
ProxyUsers.refreshSuperUserGroupsConfiguration(config);
|
||||||
|
|
||||||
|
|
|
@ -317,6 +317,9 @@ Trunk (Unreleased)
|
||||||
HDFS-6374. setXAttr should require the user to be the owner of the file
|
HDFS-6374. setXAttr should require the user to be the owner of the file
|
||||||
or directory (Charles Lamb via wang)
|
or directory (Charles Lamb via wang)
|
||||||
|
|
||||||
|
HDFS-6110 adding more slow action log in critical write path
|
||||||
|
(Liang Xie via stack)
|
||||||
|
|
||||||
Release 2.5.0 - UNRELEASED
|
Release 2.5.0 - UNRELEASED
|
||||||
|
|
||||||
INCOMPATIBLE CHANGES
|
INCOMPATIBLE CHANGES
|
||||||
|
@ -440,6 +443,17 @@ Release 2.5.0 - UNRELEASED
|
||||||
HDFS-6396. Remove support for ACL feature from INodeSymlink.
|
HDFS-6396. Remove support for ACL feature from INodeSymlink.
|
||||||
(Charles Lamb via wang)
|
(Charles Lamb via wang)
|
||||||
|
|
||||||
|
HDFS-6435. Add support for specifying a static uid/gid mapping for the NFS
|
||||||
|
gateway. (atm via wang)
|
||||||
|
|
||||||
|
HDFS-6416. Use Time#monotonicNow in OpenFileCtx and OpenFileCtxCatch to
|
||||||
|
avoid system clock bugs (Abhiraj Butala via brandonli)
|
||||||
|
|
||||||
|
HDFS-6356. Fix typo in DatanodeLayoutVersion. (Tulasi G via wang)
|
||||||
|
|
||||||
|
HDFS-6447. balancer should timestamp the completion message.
|
||||||
|
(Juan Yu via wang).
|
||||||
|
|
||||||
OPTIMIZATIONS
|
OPTIMIZATIONS
|
||||||
|
|
||||||
HDFS-6214. Webhdfs has poor throughput for files >2GB (daryn)
|
HDFS-6214. Webhdfs has poor throughput for files >2GB (daryn)
|
||||||
|
@ -583,6 +597,26 @@ Release 2.5.0 - UNRELEASED
|
||||||
HDFS-6423. Diskspace quota usage should be updated when appending data to
|
HDFS-6423. Diskspace quota usage should be updated when appending data to
|
||||||
partial block. (jing9)
|
partial block. (jing9)
|
||||||
|
|
||||||
|
HDFS-6443. Fix MiniQJMHACluster related test failures. (Zesheng Wu via
|
||||||
|
Arpit Agarwal)
|
||||||
|
|
||||||
|
HDFS-6227. ShortCircuitCache#unref should purge ShortCircuitReplicas whose
|
||||||
|
streams have been closed by java interrupts. (Colin Patrick McCabe via jing9)
|
||||||
|
|
||||||
|
HDFS-6442. Fix TestEditLogAutoroll and TestStandbyCheckpoints failure
|
||||||
|
caused by port conficts. (Zesheng Wu via Arpit Agarwal)
|
||||||
|
|
||||||
|
HDFS-6448. BlockReaderLocalLegacy should set socket timeout based on
|
||||||
|
conf.socketTimeout (liangxie via cmccabe)
|
||||||
|
|
||||||
|
HDFS-6453. Use Time#monotonicNow to avoid system clock reset.
|
||||||
|
(Liang Xie via wang)
|
||||||
|
|
||||||
|
HDFS-6461. Use Time#monotonicNow to compute duration in DataNode#shutDown.
|
||||||
|
(James Thomas via wang)
|
||||||
|
|
||||||
|
HDFS-6462. NFS: fsstat request fails with the secure hdfs (brandonli)
|
||||||
|
|
||||||
Release 2.4.1 - UNRELEASED
|
Release 2.4.1 - UNRELEASED
|
||||||
|
|
||||||
INCOMPATIBLE CHANGES
|
INCOMPATIBLE CHANGES
|
||||||
|
@ -666,6 +700,9 @@ Release 2.4.1 - UNRELEASED
|
||||||
HDFS-6397. NN shows inconsistent value in deadnode count.
|
HDFS-6397. NN shows inconsistent value in deadnode count.
|
||||||
(Mohammad Kamrul Islam via kihwal)
|
(Mohammad Kamrul Islam via kihwal)
|
||||||
|
|
||||||
|
HDFS-6411. nfs-hdfs-gateway mount raises I/O error and hangs when a
|
||||||
|
unauthorized user attempts to access it (brandonli)
|
||||||
|
|
||||||
Release 2.4.0 - 2014-04-07
|
Release 2.4.0 - 2014-04-07
|
||||||
|
|
||||||
INCOMPATIBLE CHANGES
|
INCOMPATIBLE CHANGES
|
||||||
|
|
|
@ -116,7 +116,7 @@ public class Hdfs extends AbstractFileSystem {
|
||||||
@Override
|
@Override
|
||||||
public FileChecksum getFileChecksum(Path f)
|
public FileChecksum getFileChecksum(Path f)
|
||||||
throws IOException, UnresolvedLinkException {
|
throws IOException, UnresolvedLinkException {
|
||||||
return dfs.getFileChecksum(getUriPath(f));
|
return dfs.getFileChecksum(getUriPath(f), Long.MAX_VALUE);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
|
|
@ -187,7 +187,7 @@ class BlockReaderLocalLegacy implements BlockReader {
|
||||||
userGroupInformation = UserGroupInformation.getCurrentUser();
|
userGroupInformation = UserGroupInformation.getCurrentUser();
|
||||||
}
|
}
|
||||||
pathinfo = getBlockPathInfo(userGroupInformation, blk, node,
|
pathinfo = getBlockPathInfo(userGroupInformation, blk, node,
|
||||||
configuration, conf.hdfsTimeout, token,
|
configuration, conf.socketTimeout, token,
|
||||||
conf.connectToDnViaHostname);
|
conf.connectToDnViaHostname);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -276,6 +276,7 @@ public class DFSClient implements java.io.Closeable, RemotePeerFactory {
|
||||||
final int retryTimesForGetLastBlockLength;
|
final int retryTimesForGetLastBlockLength;
|
||||||
final int retryIntervalForGetLastBlockLength;
|
final int retryIntervalForGetLastBlockLength;
|
||||||
final long datanodeRestartTimeout;
|
final long datanodeRestartTimeout;
|
||||||
|
final long dfsclientSlowIoWarningThresholdMs;
|
||||||
|
|
||||||
final boolean useLegacyBlockReader;
|
final boolean useLegacyBlockReader;
|
||||||
final boolean useLegacyBlockReaderLocal;
|
final boolean useLegacyBlockReaderLocal;
|
||||||
|
@ -430,6 +431,9 @@ public class DFSClient implements java.io.Closeable, RemotePeerFactory {
|
||||||
datanodeRestartTimeout = conf.getLong(
|
datanodeRestartTimeout = conf.getLong(
|
||||||
DFS_CLIENT_DATANODE_RESTART_TIMEOUT_KEY,
|
DFS_CLIENT_DATANODE_RESTART_TIMEOUT_KEY,
|
||||||
DFS_CLIENT_DATANODE_RESTART_TIMEOUT_DEFAULT) * 1000;
|
DFS_CLIENT_DATANODE_RESTART_TIMEOUT_DEFAULT) * 1000;
|
||||||
|
dfsclientSlowIoWarningThresholdMs = conf.getLong(
|
||||||
|
DFSConfigKeys.DFS_CLIENT_SLOW_IO_WARNING_THRESHOLD_KEY,
|
||||||
|
DFSConfigKeys.DFS_CLIENT_SLOW_IO_WARNING_THRESHOLD_DEFAULT);
|
||||||
}
|
}
|
||||||
|
|
||||||
public boolean isUseLegacyBlockReaderLocal() {
|
public boolean isUseLegacyBlockReaderLocal() {
|
||||||
|
@ -1801,15 +1805,19 @@ public class DFSClient implements java.io.Closeable, RemotePeerFactory {
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Get the checksum of a file.
|
* Get the checksum of the whole file of a range of the file. Note that the
|
||||||
|
* range always starts from the beginning of the file.
|
||||||
* @param src The file path
|
* @param src The file path
|
||||||
|
* @param length The length of the range
|
||||||
* @return The checksum
|
* @return The checksum
|
||||||
* @see DistributedFileSystem#getFileChecksum(Path)
|
* @see DistributedFileSystem#getFileChecksum(Path)
|
||||||
*/
|
*/
|
||||||
public MD5MD5CRC32FileChecksum getFileChecksum(String src) throws IOException {
|
public MD5MD5CRC32FileChecksum getFileChecksum(String src, long length)
|
||||||
|
throws IOException {
|
||||||
checkOpen();
|
checkOpen();
|
||||||
return getFileChecksum(src, clientName, namenode, socketFactory,
|
Preconditions.checkArgument(length >= 0);
|
||||||
dfsClientConf.socketTimeout, getDataEncryptionKey(),
|
return getFileChecksum(src, length, clientName, namenode,
|
||||||
|
socketFactory, dfsClientConf.socketTimeout, getDataEncryptionKey(),
|
||||||
dfsClientConf.connectToDnViaHostname);
|
dfsClientConf.connectToDnViaHostname);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1850,8 +1858,9 @@ public class DFSClient implements java.io.Closeable, RemotePeerFactory {
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Get the checksum of a file.
|
* Get the checksum of the whole file or a range of the file.
|
||||||
* @param src The file path
|
* @param src The file path
|
||||||
|
* @param length the length of the range, i.e., the range is [0, length]
|
||||||
* @param clientName the name of the client requesting the checksum.
|
* @param clientName the name of the client requesting the checksum.
|
||||||
* @param namenode the RPC proxy for the namenode
|
* @param namenode the RPC proxy for the namenode
|
||||||
* @param socketFactory to create sockets to connect to DNs
|
* @param socketFactory to create sockets to connect to DNs
|
||||||
|
@ -1861,12 +1870,13 @@ public class DFSClient implements java.io.Closeable, RemotePeerFactory {
|
||||||
* @return The checksum
|
* @return The checksum
|
||||||
*/
|
*/
|
||||||
private static MD5MD5CRC32FileChecksum getFileChecksum(String src,
|
private static MD5MD5CRC32FileChecksum getFileChecksum(String src,
|
||||||
String clientName,
|
long length, String clientName, ClientProtocol namenode,
|
||||||
ClientProtocol namenode, SocketFactory socketFactory, int socketTimeout,
|
SocketFactory socketFactory, int socketTimeout,
|
||||||
DataEncryptionKey encryptionKey, boolean connectToDnViaHostname)
|
DataEncryptionKey encryptionKey, boolean connectToDnViaHostname)
|
||||||
throws IOException {
|
throws IOException {
|
||||||
//get all block locations
|
//get block locations for the file range
|
||||||
LocatedBlocks blockLocations = callGetBlockLocations(namenode, src, 0, Long.MAX_VALUE);
|
LocatedBlocks blockLocations = callGetBlockLocations(namenode, src, 0,
|
||||||
|
length);
|
||||||
if (null == blockLocations) {
|
if (null == blockLocations) {
|
||||||
throw new FileNotFoundException("File does not exist: " + src);
|
throw new FileNotFoundException("File does not exist: " + src);
|
||||||
}
|
}
|
||||||
|
@ -1879,9 +1889,10 @@ public class DFSClient implements java.io.Closeable, RemotePeerFactory {
|
||||||
int lastRetriedIndex = -1;
|
int lastRetriedIndex = -1;
|
||||||
|
|
||||||
// get block checksum for each block
|
// get block checksum for each block
|
||||||
for(int i = 0; i < locatedblocks.size(); i++) {
|
long remaining = length;
|
||||||
|
for(int i = 0; i < locatedblocks.size() && remaining > 0; i++) {
|
||||||
if (refetchBlocks) { // refetch to get fresh tokens
|
if (refetchBlocks) { // refetch to get fresh tokens
|
||||||
blockLocations = callGetBlockLocations(namenode, src, 0, Long.MAX_VALUE);
|
blockLocations = callGetBlockLocations(namenode, src, 0, length);
|
||||||
if (null == blockLocations) {
|
if (null == blockLocations) {
|
||||||
throw new FileNotFoundException("File does not exist: " + src);
|
throw new FileNotFoundException("File does not exist: " + src);
|
||||||
}
|
}
|
||||||
|
@ -1890,6 +1901,10 @@ public class DFSClient implements java.io.Closeable, RemotePeerFactory {
|
||||||
}
|
}
|
||||||
LocatedBlock lb = locatedblocks.get(i);
|
LocatedBlock lb = locatedblocks.get(i);
|
||||||
final ExtendedBlock block = lb.getBlock();
|
final ExtendedBlock block = lb.getBlock();
|
||||||
|
if (remaining < block.getNumBytes()) {
|
||||||
|
block.setNumBytes(remaining);
|
||||||
|
}
|
||||||
|
remaining -= block.getNumBytes();
|
||||||
final DatanodeInfo[] datanodes = lb.getLocations();
|
final DatanodeInfo[] datanodes = lb.getLocations();
|
||||||
|
|
||||||
//try each datanode location of the block
|
//try each datanode location of the block
|
||||||
|
|
|
@ -646,4 +646,11 @@ public class DFSConfigKeys extends CommonConfigurationKeys {
|
||||||
public static final String DFS_NFS_ALLOW_INSECURE_PORTS_KEY = "dfs.nfs.allow.insecure.ports";
|
public static final String DFS_NFS_ALLOW_INSECURE_PORTS_KEY = "dfs.nfs.allow.insecure.ports";
|
||||||
public static final boolean DFS_NFS_ALLOW_INSECURE_PORTS_DEFAULT = true;
|
public static final boolean DFS_NFS_ALLOW_INSECURE_PORTS_DEFAULT = true;
|
||||||
|
|
||||||
|
// Slow io warning log threshold settings for dfsclient and datanode.
|
||||||
|
public static final String DFS_CLIENT_SLOW_IO_WARNING_THRESHOLD_KEY =
|
||||||
|
"dfs.client.slow.io.warning.threshold.ms";
|
||||||
|
public static final long DFS_CLIENT_SLOW_IO_WARNING_THRESHOLD_DEFAULT = 30000;
|
||||||
|
public static final String DFS_DATANODE_SLOW_IO_WARNING_THRESHOLD_KEY =
|
||||||
|
"dfs.datanode.slow.io.warning.threshold.ms";
|
||||||
|
public static final long DFS_DATANODE_SLOW_IO_WARNING_THRESHOLD_DEFAULT = 300;
|
||||||
}
|
}
|
||||||
|
|
|
@ -122,6 +122,7 @@ public class DFSOutputStream extends FSOutputSummer
|
||||||
implements Syncable, CanSetDropBehind {
|
implements Syncable, CanSetDropBehind {
|
||||||
private static final int MAX_PACKETS = 80; // each packet 64K, total 5MB
|
private static final int MAX_PACKETS = 80; // each packet 64K, total 5MB
|
||||||
private final DFSClient dfsClient;
|
private final DFSClient dfsClient;
|
||||||
|
private final long dfsclientSlowLogThresholdMs;
|
||||||
private Socket s;
|
private Socket s;
|
||||||
// closed is accessed by different threads under different locks.
|
// closed is accessed by different threads under different locks.
|
||||||
private volatile boolean closed = false;
|
private volatile boolean closed = false;
|
||||||
|
@ -788,8 +789,16 @@ public class DFSOutputStream extends FSOutputSummer
|
||||||
// process responses from datanodes.
|
// process responses from datanodes.
|
||||||
try {
|
try {
|
||||||
// read an ack from the pipeline
|
// read an ack from the pipeline
|
||||||
|
long begin = Time.monotonicNow();
|
||||||
ack.readFields(blockReplyStream);
|
ack.readFields(blockReplyStream);
|
||||||
if (DFSClient.LOG.isDebugEnabled()) {
|
long duration = Time.monotonicNow() - begin;
|
||||||
|
if (duration > dfsclientSlowLogThresholdMs
|
||||||
|
&& ack.getSeqno() != Packet.HEART_BEAT_SEQNO) {
|
||||||
|
DFSClient.LOG
|
||||||
|
.warn("Slow ReadProcessor read fields took " + duration
|
||||||
|
+ "ms (threshold=" + dfsclientSlowLogThresholdMs + "ms); ack: "
|
||||||
|
+ ack + ", targets: " + Arrays.asList(targets));
|
||||||
|
} else if (DFSClient.LOG.isDebugEnabled()) {
|
||||||
DFSClient.LOG.debug("DFSClient " + ack);
|
DFSClient.LOG.debug("DFSClient " + ack);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1570,6 +1579,8 @@ public class DFSOutputStream extends FSOutputSummer
|
||||||
|
|
||||||
}
|
}
|
||||||
this.checksum = checksum;
|
this.checksum = checksum;
|
||||||
|
this.dfsclientSlowLogThresholdMs =
|
||||||
|
dfsClient.getConf().dfsclientSlowIoWarningThresholdMs;
|
||||||
}
|
}
|
||||||
|
|
||||||
/** Construct a new output stream for creating a file. */
|
/** Construct a new output stream for creating a file. */
|
||||||
|
@ -2001,6 +2012,7 @@ public class DFSOutputStream extends FSOutputSummer
|
||||||
if (DFSClient.LOG.isDebugEnabled()) {
|
if (DFSClient.LOG.isDebugEnabled()) {
|
||||||
DFSClient.LOG.debug("Waiting for ack for: " + seqno);
|
DFSClient.LOG.debug("Waiting for ack for: " + seqno);
|
||||||
}
|
}
|
||||||
|
long begin = Time.monotonicNow();
|
||||||
try {
|
try {
|
||||||
synchronized (dataQueue) {
|
synchronized (dataQueue) {
|
||||||
while (!closed) {
|
while (!closed) {
|
||||||
|
@ -2020,6 +2032,11 @@ public class DFSOutputStream extends FSOutputSummer
|
||||||
checkClosed();
|
checkClosed();
|
||||||
} catch (ClosedChannelException e) {
|
} catch (ClosedChannelException e) {
|
||||||
}
|
}
|
||||||
|
long duration = Time.monotonicNow() - begin;
|
||||||
|
if (duration > dfsclientSlowLogThresholdMs) {
|
||||||
|
DFSClient.LOG.warn("Slow waitForAckedSeqno took " + duration
|
||||||
|
+ "ms (threshold=" + dfsclientSlowLogThresholdMs + "ms)");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private synchronized void start() {
|
private synchronized void start() {
|
||||||
|
|
|
@ -68,14 +68,12 @@ import org.apache.hadoop.hdfs.protocol.CachePoolEntry;
|
||||||
import org.apache.hadoop.hdfs.protocol.CachePoolInfo;
|
import org.apache.hadoop.hdfs.protocol.CachePoolInfo;
|
||||||
import org.apache.hadoop.hdfs.protocol.DatanodeInfo;
|
import org.apache.hadoop.hdfs.protocol.DatanodeInfo;
|
||||||
import org.apache.hadoop.hdfs.protocol.DirectoryListing;
|
import org.apache.hadoop.hdfs.protocol.DirectoryListing;
|
||||||
import org.apache.hadoop.hdfs.protocol.ExtendedBlock;
|
|
||||||
import org.apache.hadoop.hdfs.protocol.HdfsConstants;
|
import org.apache.hadoop.hdfs.protocol.HdfsConstants;
|
||||||
import org.apache.hadoop.hdfs.protocol.HdfsConstants.DatanodeReportType;
|
import org.apache.hadoop.hdfs.protocol.HdfsConstants.DatanodeReportType;
|
||||||
import org.apache.hadoop.hdfs.protocol.HdfsConstants.RollingUpgradeAction;
|
import org.apache.hadoop.hdfs.protocol.HdfsConstants.RollingUpgradeAction;
|
||||||
import org.apache.hadoop.hdfs.protocol.HdfsConstants.SafeModeAction;
|
import org.apache.hadoop.hdfs.protocol.HdfsConstants.SafeModeAction;
|
||||||
import org.apache.hadoop.hdfs.protocol.HdfsFileStatus;
|
import org.apache.hadoop.hdfs.protocol.HdfsFileStatus;
|
||||||
import org.apache.hadoop.hdfs.protocol.HdfsLocatedFileStatus;
|
import org.apache.hadoop.hdfs.protocol.HdfsLocatedFileStatus;
|
||||||
import org.apache.hadoop.hdfs.protocol.LocatedBlock;
|
|
||||||
import org.apache.hadoop.hdfs.protocol.RollingUpgradeInfo;
|
import org.apache.hadoop.hdfs.protocol.RollingUpgradeInfo;
|
||||||
import org.apache.hadoop.hdfs.protocol.SnapshotDiffReport;
|
import org.apache.hadoop.hdfs.protocol.SnapshotDiffReport;
|
||||||
import org.apache.hadoop.hdfs.protocol.SnapshottableDirectoryStatus;
|
import org.apache.hadoop.hdfs.protocol.SnapshottableDirectoryStatus;
|
||||||
|
@ -85,7 +83,6 @@ import org.apache.hadoop.hdfs.server.namenode.NameNode;
|
||||||
import org.apache.hadoop.io.Text;
|
import org.apache.hadoop.io.Text;
|
||||||
import org.apache.hadoop.net.NetUtils;
|
import org.apache.hadoop.net.NetUtils;
|
||||||
import org.apache.hadoop.security.AccessControlException;
|
import org.apache.hadoop.security.AccessControlException;
|
||||||
import org.apache.hadoop.security.token.SecretManager.InvalidToken;
|
|
||||||
import org.apache.hadoop.security.token.Token;
|
import org.apache.hadoop.security.token.Token;
|
||||||
import org.apache.hadoop.util.Progressable;
|
import org.apache.hadoop.util.Progressable;
|
||||||
|
|
||||||
|
@ -1142,7 +1139,7 @@ public class DistributedFileSystem extends FileSystem {
|
||||||
@Override
|
@Override
|
||||||
public FileChecksum doCall(final Path p)
|
public FileChecksum doCall(final Path p)
|
||||||
throws IOException, UnresolvedLinkException {
|
throws IOException, UnresolvedLinkException {
|
||||||
return dfs.getFileChecksum(getPathName(p));
|
return dfs.getFileChecksum(getPathName(p), Long.MAX_VALUE);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
|
@ -1153,6 +1150,32 @@ public class DistributedFileSystem extends FileSystem {
|
||||||
}.resolve(this, absF);
|
}.resolve(this, absF);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public FileChecksum getFileChecksum(Path f, final long length)
|
||||||
|
throws IOException {
|
||||||
|
statistics.incrementReadOps(1);
|
||||||
|
Path absF = fixRelativePart(f);
|
||||||
|
return new FileSystemLinkResolver<FileChecksum>() {
|
||||||
|
@Override
|
||||||
|
public FileChecksum doCall(final Path p)
|
||||||
|
throws IOException, UnresolvedLinkException {
|
||||||
|
return dfs.getFileChecksum(getPathName(p), length);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public FileChecksum next(final FileSystem fs, final Path p)
|
||||||
|
throws IOException {
|
||||||
|
if (fs instanceof DistributedFileSystem) {
|
||||||
|
return ((DistributedFileSystem) fs).getFileChecksum(p, length);
|
||||||
|
} else {
|
||||||
|
throw new UnsupportedFileSystemException(
|
||||||
|
"getFileChecksum(Path, long) is not supported by "
|
||||||
|
+ fs.getClass().getSimpleName());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}.resolve(this, absF);
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void setPermission(Path p, final FsPermission permission
|
public void setPermission(Path p, final FsPermission permission
|
||||||
) throws IOException {
|
) throws IOException {
|
||||||
|
|
|
@ -1552,6 +1552,7 @@ public class Balancer {
|
||||||
System.out.println(e + ". Exiting ...");
|
System.out.println(e + ". Exiting ...");
|
||||||
return ReturnStatus.INTERRUPTED.code;
|
return ReturnStatus.INTERRUPTED.code;
|
||||||
} finally {
|
} finally {
|
||||||
|
System.out.format("%-24s ", DateFormat.getDateTimeInstance().format(new Date()));
|
||||||
System.out.println("Balancing took " + time2Str(Time.now()-startTime));
|
System.out.println("Balancing took " + time2Str(Time.now()-startTime));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -69,7 +69,7 @@ class BlockReceiver implements Closeable {
|
||||||
|
|
||||||
@VisibleForTesting
|
@VisibleForTesting
|
||||||
static long CACHE_DROP_LAG_BYTES = 8 * 1024 * 1024;
|
static long CACHE_DROP_LAG_BYTES = 8 * 1024 * 1024;
|
||||||
|
private final long datanodeSlowLogThresholdMs;
|
||||||
private DataInputStream in = null; // from where data are read
|
private DataInputStream in = null; // from where data are read
|
||||||
private DataChecksum clientChecksum; // checksum used by client
|
private DataChecksum clientChecksum; // checksum used by client
|
||||||
private DataChecksum diskChecksum; // checksum we write to disk
|
private DataChecksum diskChecksum; // checksum we write to disk
|
||||||
|
@ -140,7 +140,7 @@ class BlockReceiver implements Closeable {
|
||||||
this.isDatanode = clientname.length() == 0;
|
this.isDatanode = clientname.length() == 0;
|
||||||
this.isClient = !this.isDatanode;
|
this.isClient = !this.isDatanode;
|
||||||
this.restartBudget = datanode.getDnConf().restartReplicaExpiry;
|
this.restartBudget = datanode.getDnConf().restartReplicaExpiry;
|
||||||
|
this.datanodeSlowLogThresholdMs = datanode.getDnConf().datanodeSlowIoWarningThresholdMs;
|
||||||
//for datanode, we have
|
//for datanode, we have
|
||||||
//1: clientName.length() == 0, and
|
//1: clientName.length() == 0, and
|
||||||
//2: stage == null or PIPELINE_SETUP_CREATE
|
//2: stage == null or PIPELINE_SETUP_CREATE
|
||||||
|
@ -335,6 +335,7 @@ class BlockReceiver implements Closeable {
|
||||||
*/
|
*/
|
||||||
void flushOrSync(boolean isSync) throws IOException {
|
void flushOrSync(boolean isSync) throws IOException {
|
||||||
long flushTotalNanos = 0;
|
long flushTotalNanos = 0;
|
||||||
|
long begin = Time.monotonicNow();
|
||||||
if (checksumOut != null) {
|
if (checksumOut != null) {
|
||||||
long flushStartNanos = System.nanoTime();
|
long flushStartNanos = System.nanoTime();
|
||||||
checksumOut.flush();
|
checksumOut.flush();
|
||||||
|
@ -363,6 +364,12 @@ class BlockReceiver implements Closeable {
|
||||||
datanode.metrics.incrFsyncCount();
|
datanode.metrics.incrFsyncCount();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
long duration = Time.monotonicNow() - begin;
|
||||||
|
if (duration > datanodeSlowLogThresholdMs) {
|
||||||
|
LOG.warn("Slow flushOrSync took " + duration + "ms (threshold="
|
||||||
|
+ datanodeSlowLogThresholdMs + "ms), isSync:" + isSync + ", flushTotalNanos="
|
||||||
|
+ flushTotalNanos + "ns");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -488,8 +495,14 @@ class BlockReceiver implements Closeable {
|
||||||
//First write the packet to the mirror:
|
//First write the packet to the mirror:
|
||||||
if (mirrorOut != null && !mirrorError) {
|
if (mirrorOut != null && !mirrorError) {
|
||||||
try {
|
try {
|
||||||
|
long begin = Time.monotonicNow();
|
||||||
packetReceiver.mirrorPacketTo(mirrorOut);
|
packetReceiver.mirrorPacketTo(mirrorOut);
|
||||||
mirrorOut.flush();
|
mirrorOut.flush();
|
||||||
|
long duration = Time.monotonicNow() - begin;
|
||||||
|
if (duration > datanodeSlowLogThresholdMs) {
|
||||||
|
LOG.warn("Slow BlockReceiver write packet to mirror took " + duration
|
||||||
|
+ "ms (threshold=" + datanodeSlowLogThresholdMs + "ms)");
|
||||||
|
}
|
||||||
} catch (IOException e) {
|
} catch (IOException e) {
|
||||||
handleMirrorOutError(e);
|
handleMirrorOutError(e);
|
||||||
}
|
}
|
||||||
|
@ -572,7 +585,13 @@ class BlockReceiver implements Closeable {
|
||||||
int numBytesToDisk = (int)(offsetInBlock-onDiskLen);
|
int numBytesToDisk = (int)(offsetInBlock-onDiskLen);
|
||||||
|
|
||||||
// Write data to disk.
|
// Write data to disk.
|
||||||
|
long begin = Time.monotonicNow();
|
||||||
out.write(dataBuf.array(), startByteToDisk, numBytesToDisk);
|
out.write(dataBuf.array(), startByteToDisk, numBytesToDisk);
|
||||||
|
long duration = Time.monotonicNow() - begin;
|
||||||
|
if (duration > datanodeSlowLogThresholdMs) {
|
||||||
|
LOG.warn("Slow BlockReceiver write data to disk cost:" + duration
|
||||||
|
+ "ms (threshold=" + datanodeSlowLogThresholdMs + "ms)");
|
||||||
|
}
|
||||||
|
|
||||||
// If this is a partial chunk, then verify that this is the only
|
// If this is a partial chunk, then verify that this is the only
|
||||||
// chunk in the packet. Calculate new crc for this chunk.
|
// chunk in the packet. Calculate new crc for this chunk.
|
||||||
|
@ -638,6 +657,7 @@ class BlockReceiver implements Closeable {
|
||||||
try {
|
try {
|
||||||
if (outFd != null &&
|
if (outFd != null &&
|
||||||
offsetInBlock > lastCacheManagementOffset + CACHE_DROP_LAG_BYTES) {
|
offsetInBlock > lastCacheManagementOffset + CACHE_DROP_LAG_BYTES) {
|
||||||
|
long begin = Time.monotonicNow();
|
||||||
//
|
//
|
||||||
// For SYNC_FILE_RANGE_WRITE, we want to sync from
|
// For SYNC_FILE_RANGE_WRITE, we want to sync from
|
||||||
// lastCacheManagementOffset to a position "two windows ago"
|
// lastCacheManagementOffset to a position "two windows ago"
|
||||||
|
@ -670,6 +690,11 @@ class BlockReceiver implements Closeable {
|
||||||
NativeIO.POSIX.POSIX_FADV_DONTNEED);
|
NativeIO.POSIX.POSIX_FADV_DONTNEED);
|
||||||
}
|
}
|
||||||
lastCacheManagementOffset = offsetInBlock;
|
lastCacheManagementOffset = offsetInBlock;
|
||||||
|
long duration = Time.monotonicNow() - begin;
|
||||||
|
if (duration > datanodeSlowLogThresholdMs) {
|
||||||
|
LOG.warn("Slow manageWriterOsCache took " + duration
|
||||||
|
+ "ms (threshold=" + datanodeSlowLogThresholdMs + "ms)");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
} catch (Throwable t) {
|
} catch (Throwable t) {
|
||||||
LOG.warn("Error managing cache for writer of block " + block, t);
|
LOG.warn("Error managing cache for writer of block " + block, t);
|
||||||
|
@ -1299,9 +1324,15 @@ class BlockReceiver implements Closeable {
|
||||||
replicaInfo.setBytesAcked(offsetInBlock);
|
replicaInfo.setBytesAcked(offsetInBlock);
|
||||||
}
|
}
|
||||||
// send my ack back to upstream datanode
|
// send my ack back to upstream datanode
|
||||||
|
long begin = Time.monotonicNow();
|
||||||
replyAck.write(upstreamOut);
|
replyAck.write(upstreamOut);
|
||||||
upstreamOut.flush();
|
upstreamOut.flush();
|
||||||
if (LOG.isDebugEnabled()) {
|
long duration = Time.monotonicNow() - begin;
|
||||||
|
if (duration > datanodeSlowLogThresholdMs) {
|
||||||
|
LOG.warn("Slow PacketResponder send ack to upstream took " + duration
|
||||||
|
+ "ms (threshold=" + datanodeSlowLogThresholdMs + "ms), " + myString
|
||||||
|
+ ", replyAck=" + replyAck);
|
||||||
|
} else if (LOG.isDebugEnabled()) {
|
||||||
LOG.debug(myString + ", replyAck=" + replyAck);
|
LOG.debug(myString + ", replyAck=" + replyAck);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -79,6 +79,8 @@ public class DNConf {
|
||||||
final long deleteReportInterval;
|
final long deleteReportInterval;
|
||||||
final long initialBlockReportDelay;
|
final long initialBlockReportDelay;
|
||||||
final long cacheReportInterval;
|
final long cacheReportInterval;
|
||||||
|
final long dfsclientSlowIoWarningThresholdMs;
|
||||||
|
final long datanodeSlowIoWarningThresholdMs;
|
||||||
final int writePacketSize;
|
final int writePacketSize;
|
||||||
|
|
||||||
final String minimumNameNodeVersion;
|
final String minimumNameNodeVersion;
|
||||||
|
@ -130,6 +132,13 @@ public class DNConf {
|
||||||
this.cacheReportInterval = conf.getLong(DFS_CACHEREPORT_INTERVAL_MSEC_KEY,
|
this.cacheReportInterval = conf.getLong(DFS_CACHEREPORT_INTERVAL_MSEC_KEY,
|
||||||
DFS_CACHEREPORT_INTERVAL_MSEC_DEFAULT);
|
DFS_CACHEREPORT_INTERVAL_MSEC_DEFAULT);
|
||||||
|
|
||||||
|
this.dfsclientSlowIoWarningThresholdMs = conf.getLong(
|
||||||
|
DFSConfigKeys.DFS_CLIENT_SLOW_IO_WARNING_THRESHOLD_KEY,
|
||||||
|
DFSConfigKeys.DFS_CLIENT_SLOW_IO_WARNING_THRESHOLD_DEFAULT);
|
||||||
|
this.datanodeSlowIoWarningThresholdMs = conf.getLong(
|
||||||
|
DFSConfigKeys.DFS_DATANODE_SLOW_IO_WARNING_THRESHOLD_KEY,
|
||||||
|
DFSConfigKeys.DFS_DATANODE_SLOW_IO_WARNING_THRESHOLD_DEFAULT);
|
||||||
|
|
||||||
long initBRDelay = conf.getLong(
|
long initBRDelay = conf.getLong(
|
||||||
DFS_BLOCKREPORT_INITIAL_DELAY_KEY,
|
DFS_BLOCKREPORT_INITIAL_DELAY_KEY,
|
||||||
DFS_BLOCKREPORT_INITIAL_DELAY_DEFAULT) * 1000L;
|
DFS_BLOCKREPORT_INITIAL_DELAY_DEFAULT) * 1000L;
|
||||||
|
|
|
@ -1221,7 +1221,7 @@ public class DataNode extends Configured
|
||||||
}
|
}
|
||||||
|
|
||||||
// Record the time of initial notification
|
// Record the time of initial notification
|
||||||
long timeNotified = Time.now();
|
long timeNotified = Time.monotonicNow();
|
||||||
|
|
||||||
if (localDataXceiverServer != null) {
|
if (localDataXceiverServer != null) {
|
||||||
((DataXceiverServer) this.localDataXceiverServer.getRunnable()).kill();
|
((DataXceiverServer) this.localDataXceiverServer.getRunnable()).kill();
|
||||||
|
@ -1254,7 +1254,8 @@ public class DataNode extends Configured
|
||||||
// When shutting down for restart, wait 2.5 seconds before forcing
|
// When shutting down for restart, wait 2.5 seconds before forcing
|
||||||
// termination of receiver threads.
|
// termination of receiver threads.
|
||||||
if (!this.shutdownForUpgrade ||
|
if (!this.shutdownForUpgrade ||
|
||||||
(this.shutdownForUpgrade && (Time.now() - timeNotified > 2500))) {
|
(this.shutdownForUpgrade && (Time.monotonicNow() - timeNotified
|
||||||
|
> 2500))) {
|
||||||
this.threadGroup.interrupt();
|
this.threadGroup.interrupt();
|
||||||
}
|
}
|
||||||
LOG.info("Waiting for threadgroup to exit, active threads is " +
|
LOG.info("Waiting for threadgroup to exit, active threads is " +
|
||||||
|
@ -2581,7 +2582,7 @@ public class DataNode extends Configured
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
synchronized(checkDiskErrorMutex) {
|
synchronized(checkDiskErrorMutex) {
|
||||||
lastDiskErrorCheck = System.currentTimeMillis();
|
lastDiskErrorCheck = Time.monotonicNow();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
try {
|
try {
|
||||||
|
|
|
@ -62,7 +62,7 @@ public class DataNodeLayoutVersion {
|
||||||
* </ul>
|
* </ul>
|
||||||
*/
|
*/
|
||||||
public static enum Feature implements LayoutFeature {
|
public static enum Feature implements LayoutFeature {
|
||||||
FIRST_LAYOUT(-55, -53, "First datenode layout", false);
|
FIRST_LAYOUT(-55, -53, "First datanode layout", false);
|
||||||
|
|
||||||
private final FeatureInfo info;
|
private final FeatureInfo info;
|
||||||
|
|
||||||
|
|
|
@ -42,6 +42,7 @@ import java.net.Socket;
|
||||||
import java.net.SocketException;
|
import java.net.SocketException;
|
||||||
import java.net.UnknownHostException;
|
import java.net.UnknownHostException;
|
||||||
import java.nio.channels.ClosedChannelException;
|
import java.nio.channels.ClosedChannelException;
|
||||||
|
import java.security.MessageDigest;
|
||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
|
|
||||||
import org.apache.commons.logging.Log;
|
import org.apache.commons.logging.Log;
|
||||||
|
@ -83,6 +84,7 @@ import org.apache.hadoop.security.token.SecretManager.InvalidToken;
|
||||||
import org.apache.hadoop.security.token.Token;
|
import org.apache.hadoop.security.token.Token;
|
||||||
import org.apache.hadoop.util.DataChecksum;
|
import org.apache.hadoop.util.DataChecksum;
|
||||||
|
|
||||||
|
import com.google.common.base.Preconditions;
|
||||||
import com.google.common.net.InetAddresses;
|
import com.google.common.net.InetAddresses;
|
||||||
import com.google.protobuf.ByteString;
|
import com.google.protobuf.ByteString;
|
||||||
|
|
||||||
|
@ -803,6 +805,43 @@ class DataXceiver extends Receiver implements Runnable {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private MD5Hash calcPartialBlockChecksum(ExtendedBlock block,
|
||||||
|
long requestLength, DataChecksum checksum, DataInputStream checksumIn)
|
||||||
|
throws IOException {
|
||||||
|
final int bytesPerCRC = checksum.getBytesPerChecksum();
|
||||||
|
final int csize = checksum.getChecksumSize();
|
||||||
|
final byte[] buffer = new byte[4*1024];
|
||||||
|
MessageDigest digester = MD5Hash.getDigester();
|
||||||
|
|
||||||
|
long remaining = requestLength / bytesPerCRC * csize;
|
||||||
|
for (int toDigest = 0; remaining > 0; remaining -= toDigest) {
|
||||||
|
toDigest = checksumIn.read(buffer, 0,
|
||||||
|
(int) Math.min(remaining, buffer.length));
|
||||||
|
if (toDigest < 0) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
digester.update(buffer, 0, toDigest);
|
||||||
|
}
|
||||||
|
|
||||||
|
int partialLength = (int) (requestLength % bytesPerCRC);
|
||||||
|
if (partialLength > 0) {
|
||||||
|
byte[] buf = new byte[partialLength];
|
||||||
|
final InputStream blockIn = datanode.data.getBlockInputStream(block,
|
||||||
|
requestLength - partialLength);
|
||||||
|
try {
|
||||||
|
// Get the CRC of the partialLength.
|
||||||
|
IOUtils.readFully(blockIn, buf, 0, partialLength);
|
||||||
|
} finally {
|
||||||
|
IOUtils.closeStream(blockIn);
|
||||||
|
}
|
||||||
|
checksum.update(buf, 0, partialLength);
|
||||||
|
byte[] partialCrc = new byte[csize];
|
||||||
|
checksum.writeValue(partialCrc, 0, true);
|
||||||
|
digester.update(partialCrc);
|
||||||
|
}
|
||||||
|
return new MD5Hash(digester.digest());
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void blockChecksum(final ExtendedBlock block,
|
public void blockChecksum(final ExtendedBlock block,
|
||||||
final Token<BlockTokenIdentifier> blockToken) throws IOException {
|
final Token<BlockTokenIdentifier> blockToken) throws IOException {
|
||||||
|
@ -810,25 +849,32 @@ class DataXceiver extends Receiver implements Runnable {
|
||||||
getOutputStream());
|
getOutputStream());
|
||||||
checkAccess(out, true, block, blockToken,
|
checkAccess(out, true, block, blockToken,
|
||||||
Op.BLOCK_CHECKSUM, BlockTokenSecretManager.AccessMode.READ);
|
Op.BLOCK_CHECKSUM, BlockTokenSecretManager.AccessMode.READ);
|
||||||
updateCurrentThreadName("Reading metadata for block " + block);
|
// client side now can specify a range of the block for checksum
|
||||||
final LengthInputStream metadataIn =
|
long requestLength = block.getNumBytes();
|
||||||
datanode.data.getMetaDataInputStream(block);
|
Preconditions.checkArgument(requestLength >= 0);
|
||||||
final DataInputStream checksumIn = new DataInputStream(new BufferedInputStream(
|
long visibleLength = datanode.data.getReplicaVisibleLength(block);
|
||||||
metadataIn, HdfsConstants.IO_FILE_BUFFER_SIZE));
|
boolean partialBlk = requestLength < visibleLength;
|
||||||
|
|
||||||
|
updateCurrentThreadName("Reading metadata for block " + block);
|
||||||
|
final LengthInputStream metadataIn = datanode.data
|
||||||
|
.getMetaDataInputStream(block);
|
||||||
|
|
||||||
|
final DataInputStream checksumIn = new DataInputStream(
|
||||||
|
new BufferedInputStream(metadataIn, HdfsConstants.IO_FILE_BUFFER_SIZE));
|
||||||
updateCurrentThreadName("Getting checksum for block " + block);
|
updateCurrentThreadName("Getting checksum for block " + block);
|
||||||
try {
|
try {
|
||||||
//read metadata file
|
//read metadata file
|
||||||
final BlockMetadataHeader header = BlockMetadataHeader.readHeader(checksumIn);
|
final BlockMetadataHeader header = BlockMetadataHeader
|
||||||
|
.readHeader(checksumIn);
|
||||||
final DataChecksum checksum = header.getChecksum();
|
final DataChecksum checksum = header.getChecksum();
|
||||||
|
final int csize = checksum.getChecksumSize();
|
||||||
final int bytesPerCRC = checksum.getBytesPerChecksum();
|
final int bytesPerCRC = checksum.getBytesPerChecksum();
|
||||||
final long crcPerBlock = checksum.getChecksumSize() > 0
|
final long crcPerBlock = csize <= 0 ? 0 :
|
||||||
? (metadataIn.getLength() - BlockMetadataHeader.getHeaderSize())/checksum.getChecksumSize()
|
(metadataIn.getLength() - BlockMetadataHeader.getHeaderSize()) / csize;
|
||||||
: 0;
|
|
||||||
|
|
||||||
//compute block checksum
|
|
||||||
final MD5Hash md5 = MD5Hash.digest(checksumIn);
|
|
||||||
|
|
||||||
|
final MD5Hash md5 = partialBlk && crcPerBlock > 0 ?
|
||||||
|
calcPartialBlockChecksum(block, requestLength, checksum, checksumIn)
|
||||||
|
: MD5Hash.digest(checksumIn);
|
||||||
if (LOG.isDebugEnabled()) {
|
if (LOG.isDebugEnabled()) {
|
||||||
LOG.debug("block=" + block + ", bytesPerCRC=" + bytesPerCRC
|
LOG.debug("block=" + block + ", bytesPerCRC=" + bytesPerCRC
|
||||||
+ ", crcPerBlock=" + crcPerBlock + ", md5=" + md5);
|
+ ", crcPerBlock=" + crcPerBlock + ", md5=" + md5);
|
||||||
|
@ -841,8 +887,7 @@ class DataXceiver extends Receiver implements Runnable {
|
||||||
.setBytesPerCrc(bytesPerCRC)
|
.setBytesPerCrc(bytesPerCRC)
|
||||||
.setCrcPerBlock(crcPerBlock)
|
.setCrcPerBlock(crcPerBlock)
|
||||||
.setMd5(ByteString.copyFrom(md5.getDigest()))
|
.setMd5(ByteString.copyFrom(md5.getDigest()))
|
||||||
.setCrcType(PBHelper.convert(checksum.getChecksumType()))
|
.setCrcType(PBHelper.convert(checksum.getChecksumType())))
|
||||||
)
|
|
||||||
.build()
|
.build()
|
||||||
.writeDelimitedTo(out);
|
.writeDelimitedTo(out);
|
||||||
out.flush();
|
out.flush();
|
||||||
|
|
|
@ -23,6 +23,7 @@ import java.util.*;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
import org.apache.hadoop.hdfs.server.datanode.fsdataset.FsVolumeSpi;
|
import org.apache.hadoop.hdfs.server.datanode.fsdataset.FsVolumeSpi;
|
||||||
import org.apache.hadoop.hdfs.server.datanode.fsdataset.VolumeChoosingPolicy;
|
import org.apache.hadoop.hdfs.server.datanode.fsdataset.VolumeChoosingPolicy;
|
||||||
|
import org.apache.hadoop.util.Time;
|
||||||
import org.apache.hadoop.util.DiskChecker.DiskErrorException;
|
import org.apache.hadoop.util.DiskChecker.DiskErrorException;
|
||||||
|
|
||||||
class FsVolumeList {
|
class FsVolumeList {
|
||||||
|
@ -97,7 +98,7 @@ class FsVolumeList {
|
||||||
}
|
}
|
||||||
|
|
||||||
void getAllVolumesMap(final String bpid, final ReplicaMap volumeMap) throws IOException {
|
void getAllVolumesMap(final String bpid, final ReplicaMap volumeMap) throws IOException {
|
||||||
long totalStartTime = System.currentTimeMillis();
|
long totalStartTime = Time.monotonicNow();
|
||||||
final List<IOException> exceptions = Collections.synchronizedList(
|
final List<IOException> exceptions = Collections.synchronizedList(
|
||||||
new ArrayList<IOException>());
|
new ArrayList<IOException>());
|
||||||
List<Thread> replicaAddingThreads = new ArrayList<Thread>();
|
List<Thread> replicaAddingThreads = new ArrayList<Thread>();
|
||||||
|
@ -107,9 +108,9 @@ class FsVolumeList {
|
||||||
try {
|
try {
|
||||||
FsDatasetImpl.LOG.info("Adding replicas to map for block pool " +
|
FsDatasetImpl.LOG.info("Adding replicas to map for block pool " +
|
||||||
bpid + " on volume " + v + "...");
|
bpid + " on volume " + v + "...");
|
||||||
long startTime = System.currentTimeMillis();
|
long startTime = Time.monotonicNow();
|
||||||
v.getVolumeMap(bpid, volumeMap);
|
v.getVolumeMap(bpid, volumeMap);
|
||||||
long timeTaken = System.currentTimeMillis() - startTime;
|
long timeTaken = Time.monotonicNow() - startTime;
|
||||||
FsDatasetImpl.LOG.info("Time to add replicas to map for block pool"
|
FsDatasetImpl.LOG.info("Time to add replicas to map for block pool"
|
||||||
+ " " + bpid + " on volume " + v + ": " + timeTaken + "ms");
|
+ " " + bpid + " on volume " + v + ": " + timeTaken + "ms");
|
||||||
} catch (IOException ioe) {
|
} catch (IOException ioe) {
|
||||||
|
@ -132,7 +133,7 @@ class FsVolumeList {
|
||||||
if (!exceptions.isEmpty()) {
|
if (!exceptions.isEmpty()) {
|
||||||
throw exceptions.get(0);
|
throw exceptions.get(0);
|
||||||
}
|
}
|
||||||
long totalTimeTaken = System.currentTimeMillis() - totalStartTime;
|
long totalTimeTaken = Time.monotonicNow() - totalStartTime;
|
||||||
FsDatasetImpl.LOG.info("Total time to add all replicas to map: "
|
FsDatasetImpl.LOG.info("Total time to add all replicas to map: "
|
||||||
+ totalTimeTaken + "ms");
|
+ totalTimeTaken + "ms");
|
||||||
}
|
}
|
||||||
|
@ -141,9 +142,9 @@ class FsVolumeList {
|
||||||
throws IOException {
|
throws IOException {
|
||||||
FsDatasetImpl.LOG.info("Adding replicas to map for block pool " + bpid +
|
FsDatasetImpl.LOG.info("Adding replicas to map for block pool " + bpid +
|
||||||
" on volume " + volume + "...");
|
" on volume " + volume + "...");
|
||||||
long startTime = System.currentTimeMillis();
|
long startTime = Time.monotonicNow();
|
||||||
volume.getVolumeMap(bpid, volumeMap);
|
volume.getVolumeMap(bpid, volumeMap);
|
||||||
long timeTaken = System.currentTimeMillis() - startTime;
|
long timeTaken = Time.monotonicNow() - startTime;
|
||||||
FsDatasetImpl.LOG.info("Time to add replicas to map for block pool " + bpid +
|
FsDatasetImpl.LOG.info("Time to add replicas to map for block pool " + bpid +
|
||||||
" on volume " + volume + ": " + timeTaken + "ms");
|
" on volume " + volume + ": " + timeTaken + "ms");
|
||||||
}
|
}
|
||||||
|
@ -195,7 +196,7 @@ class FsVolumeList {
|
||||||
|
|
||||||
|
|
||||||
void addBlockPool(final String bpid, final Configuration conf) throws IOException {
|
void addBlockPool(final String bpid, final Configuration conf) throws IOException {
|
||||||
long totalStartTime = System.currentTimeMillis();
|
long totalStartTime = Time.monotonicNow();
|
||||||
|
|
||||||
final List<IOException> exceptions = Collections.synchronizedList(
|
final List<IOException> exceptions = Collections.synchronizedList(
|
||||||
new ArrayList<IOException>());
|
new ArrayList<IOException>());
|
||||||
|
@ -206,9 +207,9 @@ class FsVolumeList {
|
||||||
try {
|
try {
|
||||||
FsDatasetImpl.LOG.info("Scanning block pool " + bpid +
|
FsDatasetImpl.LOG.info("Scanning block pool " + bpid +
|
||||||
" on volume " + v + "...");
|
" on volume " + v + "...");
|
||||||
long startTime = System.currentTimeMillis();
|
long startTime = Time.monotonicNow();
|
||||||
v.addBlockPool(bpid, conf);
|
v.addBlockPool(bpid, conf);
|
||||||
long timeTaken = System.currentTimeMillis() - startTime;
|
long timeTaken = Time.monotonicNow() - startTime;
|
||||||
FsDatasetImpl.LOG.info("Time taken to scan block pool " + bpid +
|
FsDatasetImpl.LOG.info("Time taken to scan block pool " + bpid +
|
||||||
" on " + v + ": " + timeTaken + "ms");
|
" on " + v + ": " + timeTaken + "ms");
|
||||||
} catch (IOException ioe) {
|
} catch (IOException ioe) {
|
||||||
|
@ -232,7 +233,7 @@ class FsVolumeList {
|
||||||
throw exceptions.get(0);
|
throw exceptions.get(0);
|
||||||
}
|
}
|
||||||
|
|
||||||
long totalTimeTaken = System.currentTimeMillis() - totalStartTime;
|
long totalTimeTaken = Time.monotonicNow() - totalStartTime;
|
||||||
FsDatasetImpl.LOG.info("Total time to scan all replicas for block pool " +
|
FsDatasetImpl.LOG.info("Total time to scan all replicas for block pool " +
|
||||||
bpid + ": " + totalTimeTaken + "ms");
|
bpid + ": " + totalTimeTaken + "ms");
|
||||||
}
|
}
|
||||||
|
|
|
@ -74,7 +74,6 @@ import org.apache.hadoop.hdfs.web.resources.PutOpParam;
|
||||||
import org.apache.hadoop.hdfs.web.resources.ReplicationParam;
|
import org.apache.hadoop.hdfs.web.resources.ReplicationParam;
|
||||||
import org.apache.hadoop.hdfs.web.resources.UriFsPathParam;
|
import org.apache.hadoop.hdfs.web.resources.UriFsPathParam;
|
||||||
import org.apache.hadoop.io.IOUtils;
|
import org.apache.hadoop.io.IOUtils;
|
||||||
import org.apache.hadoop.io.Text;
|
|
||||||
import org.apache.hadoop.security.SecurityUtil;
|
import org.apache.hadoop.security.SecurityUtil;
|
||||||
import org.apache.hadoop.security.UserGroupInformation;
|
import org.apache.hadoop.security.UserGroupInformation;
|
||||||
import org.apache.hadoop.security.token.Token;
|
import org.apache.hadoop.security.token.Token;
|
||||||
|
@ -452,7 +451,7 @@ public class DatanodeWebHdfsMethods {
|
||||||
MD5MD5CRC32FileChecksum checksum = null;
|
MD5MD5CRC32FileChecksum checksum = null;
|
||||||
DFSClient dfsclient = newDfsClient(nnId, conf);
|
DFSClient dfsclient = newDfsClient(nnId, conf);
|
||||||
try {
|
try {
|
||||||
checksum = dfsclient.getFileChecksum(fullpath);
|
checksum = dfsclient.getFileChecksum(fullpath, Long.MAX_VALUE);
|
||||||
dfsclient.close();
|
dfsclient.close();
|
||||||
dfsclient = null;
|
dfsclient = null;
|
||||||
} finally {
|
} finally {
|
||||||
|
|
|
@ -59,6 +59,7 @@ import org.apache.hadoop.hdfs.util.MD5FileUtils;
|
||||||
import org.apache.hadoop.io.MD5Hash;
|
import org.apache.hadoop.io.MD5Hash;
|
||||||
import org.apache.hadoop.io.compress.CompressionCodec;
|
import org.apache.hadoop.io.compress.CompressionCodec;
|
||||||
import org.apache.hadoop.io.compress.CompressorStream;
|
import org.apache.hadoop.io.compress.CompressorStream;
|
||||||
|
import org.apache.hadoop.util.Time;
|
||||||
|
|
||||||
import com.google.common.collect.Lists;
|
import com.google.common.collect.Lists;
|
||||||
import com.google.common.collect.Maps;
|
import com.google.common.collect.Maps;
|
||||||
|
@ -160,13 +161,13 @@ public final class FSImageFormatProtobuf {
|
||||||
}
|
}
|
||||||
|
|
||||||
void load(File file) throws IOException {
|
void load(File file) throws IOException {
|
||||||
long start = System.currentTimeMillis();
|
long start = Time.monotonicNow();
|
||||||
imgDigest = MD5FileUtils.computeMd5ForFile(file);
|
imgDigest = MD5FileUtils.computeMd5ForFile(file);
|
||||||
RandomAccessFile raFile = new RandomAccessFile(file, "r");
|
RandomAccessFile raFile = new RandomAccessFile(file, "r");
|
||||||
FileInputStream fin = new FileInputStream(file);
|
FileInputStream fin = new FileInputStream(file);
|
||||||
try {
|
try {
|
||||||
loadInternal(raFile, fin);
|
loadInternal(raFile, fin);
|
||||||
long end = System.currentTimeMillis();
|
long end = Time.monotonicNow();
|
||||||
LOG.info("Loaded FSImage in " + (end - start) / 1000 + " seconds.");
|
LOG.info("Loaded FSImage in " + (end - start) / 1000 + " seconds.");
|
||||||
} finally {
|
} finally {
|
||||||
fin.close();
|
fin.close();
|
||||||
|
|
|
@ -159,7 +159,7 @@ public class TransferFsImage {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
final long milliTime = System.currentTimeMillis();
|
final long milliTime = Time.monotonicNow();
|
||||||
String tmpFileName = NNStorage.getTemporaryEditsFileName(
|
String tmpFileName = NNStorage.getTemporaryEditsFileName(
|
||||||
log.getStartTxId(), log.getEndTxId(), milliTime);
|
log.getStartTxId(), log.getEndTxId(), milliTime);
|
||||||
List<File> tmpFiles = dstStorage.getFiles(NameNodeDirType.EDITS,
|
List<File> tmpFiles = dstStorage.getFiles(NameNodeDirType.EDITS,
|
||||||
|
|
|
@ -437,12 +437,23 @@ public class ShortCircuitCache implements Closeable {
|
||||||
void unref(ShortCircuitReplica replica) {
|
void unref(ShortCircuitReplica replica) {
|
||||||
lock.lock();
|
lock.lock();
|
||||||
try {
|
try {
|
||||||
// If the replica is stale, but we haven't purged it yet, let's do that.
|
// If the replica is stale or unusable, but we haven't purged it yet,
|
||||||
// It would be a shame to evict a non-stale replica so that we could put
|
// let's do that. It would be a shame to evict a non-stale replica so
|
||||||
// a stale one into the cache.
|
// that we could put a stale or unusable one into the cache.
|
||||||
if ((!replica.purged) && replica.isStale()) {
|
if (!replica.purged) {
|
||||||
|
String purgeReason = null;
|
||||||
|
if (!replica.getDataStream().getChannel().isOpen()) {
|
||||||
|
purgeReason = "purging replica because its data channel is closed.";
|
||||||
|
} else if (!replica.getMetaStream().getChannel().isOpen()) {
|
||||||
|
purgeReason = "purging replica because its meta channel is closed.";
|
||||||
|
} else if (replica.isStale()) {
|
||||||
|
purgeReason = "purging replica because it is stale.";
|
||||||
|
}
|
||||||
|
if (purgeReason != null) {
|
||||||
|
LOG.debug(this + ": " + purgeReason);
|
||||||
purge(replica);
|
purge(replica);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
String addedString = "";
|
String addedString = "";
|
||||||
boolean shouldTrimEvictionMaps = false;
|
boolean shouldTrimEvictionMaps = false;
|
||||||
int newRefCount = --replica.refCount;
|
int newRefCount = --replica.refCount;
|
||||||
|
|
|
@ -1930,4 +1930,22 @@
|
||||||
</description>
|
</description>
|
||||||
</property>
|
</property>
|
||||||
|
|
||||||
|
<property>
|
||||||
|
<name>dfs.client.slow.io.warning.threshold.ms</name>
|
||||||
|
<value>30000</value>
|
||||||
|
<description>The threshold in milliseconds at which we will log a slow
|
||||||
|
io warning in a dfsclient. By default, this parameter is set to 30000
|
||||||
|
milliseconds (30 seconds).
|
||||||
|
</description>
|
||||||
|
</property>
|
||||||
|
|
||||||
|
<property>
|
||||||
|
<name>dfs.datanode.slow.io.warning.threshold.ms</name>
|
||||||
|
<value>300</value>
|
||||||
|
<description>The threshold in milliseconds at which we will log a slow
|
||||||
|
io warning in a datanode. By default, this parameter is set to 300
|
||||||
|
milliseconds.
|
||||||
|
</description>
|
||||||
|
</property>
|
||||||
|
|
||||||
</configuration>
|
</configuration>
|
||||||
|
|
|
@ -18,8 +18,6 @@
|
||||||
|
|
||||||
Centralized Cache Management in HDFS
|
Centralized Cache Management in HDFS
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
||||||
%{toc|section=1|fromDepth=2|toDepth=4}
|
%{toc|section=1|fromDepth=2|toDepth=4}
|
||||||
|
|
||||||
* {Overview}
|
* {Overview}
|
||||||
|
|
|
@ -18,8 +18,6 @@
|
||||||
|
|
||||||
Extended Attributes in HDFS
|
Extended Attributes in HDFS
|
||||||
|
|
||||||
\[ {{{../../index.html}Go Back}} \]
|
|
||||||
|
|
||||||
%{toc|section=1|fromDepth=2|toDepth=4}
|
%{toc|section=1|fromDepth=2|toDepth=4}
|
||||||
|
|
||||||
* {Overview}
|
* {Overview}
|
||||||
|
|
|
@ -19,8 +19,6 @@
|
||||||
|
|
||||||
HDFS NFS Gateway
|
HDFS NFS Gateway
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
||||||
%{toc|section=1|fromDepth=0}
|
%{toc|section=1|fromDepth=0}
|
||||||
|
|
||||||
* {Overview}
|
* {Overview}
|
||||||
|
@ -338,8 +336,21 @@ HDFS NFS Gateway
|
||||||
The system administrator must ensure that the user on NFS client host has the same
|
The system administrator must ensure that the user on NFS client host has the same
|
||||||
name and UID as that on the NFS gateway host. This is usually not a problem if
|
name and UID as that on the NFS gateway host. This is usually not a problem if
|
||||||
the same user management system (e.g., LDAP/NIS) is used to create and deploy users on
|
the same user management system (e.g., LDAP/NIS) is used to create and deploy users on
|
||||||
HDFS nodes and NFS client node. In case the user account is created manually in different hosts, one might need to
|
HDFS nodes and NFS client node. In case the user account is created manually on different hosts, one might need to
|
||||||
modify UID (e.g., do "usermod -u 123 myusername") on either NFS client or NFS gateway host
|
modify UID (e.g., do "usermod -u 123 myusername") on either NFS client or NFS gateway host
|
||||||
in order to make it the same on both sides. More technical details of RPC AUTH_UNIX can be found
|
in order to make it the same on both sides. More technical details of RPC AUTH_UNIX can be found
|
||||||
in {{{http://tools.ietf.org/html/rfc1057}RPC specification}}.
|
in {{{http://tools.ietf.org/html/rfc1057}RPC specification}}.
|
||||||
|
|
||||||
|
Optionally, the system administrator can configure a custom static mapping
|
||||||
|
file in the event one wishes to access the HDFS NFS Gateway from a system with
|
||||||
|
a completely disparate set of UIDs/GIDs. By default this file is located at
|
||||||
|
"/etc/nfs.map", but a custom location can be configured by setting the
|
||||||
|
"dfs.nfs.static.mapping.file" property to the path of the static mapping file.
|
||||||
|
The format of the static mapping file is similar to what is described in the
|
||||||
|
exports(5) manual page, but roughly it is:
|
||||||
|
|
||||||
|
-------------------------
|
||||||
|
# Mapping for clients accessing the NFS gateway
|
||||||
|
uid 10 100 # Map the remote UID 10 the local UID 100
|
||||||
|
gid 11 101 # Map the remote GID 11 to the local GID 101
|
||||||
|
-------------------------
|
||||||
|
|
|
@ -18,8 +18,6 @@
|
||||||
|
|
||||||
ViewFs Guide
|
ViewFs Guide
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
||||||
%{toc|section=1|fromDepth=0}
|
%{toc|section=1|fromDepth=0}
|
||||||
|
|
||||||
* {Introduction}
|
* {Introduction}
|
||||||
|
|
|
@ -28,12 +28,15 @@ import static org.hamcrest.CoreMatchers.equalTo;
|
||||||
|
|
||||||
import java.io.File;
|
import java.io.File;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
|
import java.nio.channels.ClosedByInterruptException;
|
||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.concurrent.CountDownLatch;
|
import java.util.concurrent.CountDownLatch;
|
||||||
|
import java.util.concurrent.Semaphore;
|
||||||
import java.util.concurrent.TimeUnit;
|
import java.util.concurrent.TimeUnit;
|
||||||
import java.util.concurrent.atomic.AtomicBoolean;
|
import java.util.concurrent.atomic.AtomicBoolean;
|
||||||
|
import java.util.concurrent.atomic.AtomicInteger;
|
||||||
|
|
||||||
import org.apache.commons.logging.Log;
|
import org.apache.commons.logging.Log;
|
||||||
import org.apache.commons.logging.LogFactory;
|
import org.apache.commons.logging.LogFactory;
|
||||||
|
@ -409,4 +412,121 @@ public class TestBlockReaderFactory {
|
||||||
getDomainSocketWatcher().isClosed());
|
getDomainSocketWatcher().isClosed());
|
||||||
cluster.shutdown();
|
cluster.shutdown();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* When an InterruptedException is sent to a thread calling
|
||||||
|
* FileChannel#read, the FileChannel is immediately closed and the
|
||||||
|
* thread gets an exception. This effectively means that we might have
|
||||||
|
* someone asynchronously calling close() on the file descriptors we use
|
||||||
|
* in BlockReaderLocal. So when unreferencing a ShortCircuitReplica in
|
||||||
|
* ShortCircuitCache#unref, we should check if the FileChannel objects
|
||||||
|
* are still open. If not, we should purge the replica to avoid giving
|
||||||
|
* it out to any future readers.
|
||||||
|
*
|
||||||
|
* This is a regression test for HDFS-6227: Short circuit read failed
|
||||||
|
* due to ClosedChannelException.
|
||||||
|
*
|
||||||
|
* Note that you may still get ClosedChannelException errors if two threads
|
||||||
|
* are reading from the same replica and an InterruptedException is delivered
|
||||||
|
* to one of them.
|
||||||
|
*/
|
||||||
|
@Test(timeout=120000)
|
||||||
|
public void testPurgingClosedReplicas() throws Exception {
|
||||||
|
BlockReaderTestUtil.enableBlockReaderFactoryTracing();
|
||||||
|
final AtomicInteger replicasCreated = new AtomicInteger(0);
|
||||||
|
final AtomicBoolean testFailed = new AtomicBoolean(false);
|
||||||
|
DFSInputStream.tcpReadsDisabledForTesting = true;
|
||||||
|
BlockReaderFactory.createShortCircuitReplicaInfoCallback =
|
||||||
|
new ShortCircuitCache.ShortCircuitReplicaCreator() {
|
||||||
|
@Override
|
||||||
|
public ShortCircuitReplicaInfo createShortCircuitReplicaInfo() {
|
||||||
|
replicasCreated.incrementAndGet();
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
TemporarySocketDirectory sockDir = new TemporarySocketDirectory();
|
||||||
|
Configuration conf = createShortCircuitConf(
|
||||||
|
"testPurgingClosedReplicas", sockDir);
|
||||||
|
final MiniDFSCluster cluster =
|
||||||
|
new MiniDFSCluster.Builder(conf).numDataNodes(1).build();
|
||||||
|
cluster.waitActive();
|
||||||
|
final DistributedFileSystem dfs = cluster.getFileSystem();
|
||||||
|
final String TEST_FILE = "/test_file";
|
||||||
|
final int TEST_FILE_LEN = 4095;
|
||||||
|
final int SEED = 0xFADE0;
|
||||||
|
final DistributedFileSystem fs =
|
||||||
|
(DistributedFileSystem)FileSystem.get(cluster.getURI(0), conf);
|
||||||
|
DFSTestUtil.createFile(fs, new Path(TEST_FILE), TEST_FILE_LEN,
|
||||||
|
(short)1, SEED);
|
||||||
|
|
||||||
|
final Semaphore sem = new Semaphore(0);
|
||||||
|
final List<LocatedBlock> locatedBlocks =
|
||||||
|
cluster.getNameNode().getRpcServer().getBlockLocations(
|
||||||
|
TEST_FILE, 0, TEST_FILE_LEN).getLocatedBlocks();
|
||||||
|
final LocatedBlock lblock = locatedBlocks.get(0); // first block
|
||||||
|
final byte[] buf = new byte[TEST_FILE_LEN];
|
||||||
|
Runnable readerRunnable = new Runnable() {
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
try {
|
||||||
|
while (true) {
|
||||||
|
BlockReader blockReader = null;
|
||||||
|
try {
|
||||||
|
blockReader = BlockReaderTestUtil.
|
||||||
|
getBlockReader(cluster, lblock, 0, TEST_FILE_LEN);
|
||||||
|
sem.release();
|
||||||
|
try {
|
||||||
|
blockReader.readAll(buf, 0, TEST_FILE_LEN);
|
||||||
|
} finally {
|
||||||
|
sem.acquireUninterruptibly();
|
||||||
|
}
|
||||||
|
} catch (ClosedByInterruptException e) {
|
||||||
|
LOG.info("got the expected ClosedByInterruptException", e);
|
||||||
|
sem.release();
|
||||||
|
break;
|
||||||
|
} finally {
|
||||||
|
if (blockReader != null) blockReader.close();
|
||||||
|
}
|
||||||
|
LOG.info("read another " + TEST_FILE_LEN + " bytes.");
|
||||||
|
}
|
||||||
|
} catch (Throwable t) {
|
||||||
|
LOG.error("getBlockReader failure", t);
|
||||||
|
testFailed.set(true);
|
||||||
|
sem.release();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
Thread thread = new Thread(readerRunnable);
|
||||||
|
thread.start();
|
||||||
|
|
||||||
|
// While the thread is reading, send it interrupts.
|
||||||
|
// These should trigger a ClosedChannelException.
|
||||||
|
while (thread.isAlive()) {
|
||||||
|
sem.acquireUninterruptibly();
|
||||||
|
thread.interrupt();
|
||||||
|
sem.release();
|
||||||
|
}
|
||||||
|
Assert.assertFalse(testFailed.get());
|
||||||
|
|
||||||
|
// We should be able to read from the file without
|
||||||
|
// getting a ClosedChannelException.
|
||||||
|
BlockReader blockReader = null;
|
||||||
|
try {
|
||||||
|
blockReader = BlockReaderTestUtil.
|
||||||
|
getBlockReader(cluster, lblock, 0, TEST_FILE_LEN);
|
||||||
|
blockReader.readFully(buf, 0, TEST_FILE_LEN);
|
||||||
|
} finally {
|
||||||
|
if (blockReader != null) blockReader.close();
|
||||||
|
}
|
||||||
|
byte expected[] = DFSTestUtil.
|
||||||
|
calculateFileContentsFromSeed(SEED, TEST_FILE_LEN);
|
||||||
|
Assert.assertTrue(Arrays.equals(buf, expected));
|
||||||
|
|
||||||
|
// Another ShortCircuitReplica object should have been created.
|
||||||
|
Assert.assertEquals(2, replicasCreated.get());
|
||||||
|
|
||||||
|
dfs.close();
|
||||||
|
cluster.shutdown();
|
||||||
|
sockDir.close();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,75 @@
|
||||||
|
/**
|
||||||
|
* 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.hadoop.hdfs;
|
||||||
|
|
||||||
|
import org.apache.hadoop.conf.Configuration;
|
||||||
|
import org.apache.hadoop.fs.FileChecksum;
|
||||||
|
import org.apache.hadoop.fs.Path;
|
||||||
|
import org.junit.After;
|
||||||
|
import org.junit.Assert;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
public class TestGetFileChecksum {
|
||||||
|
private static final int BLOCKSIZE = 1024;
|
||||||
|
private static final short REPLICATION = 3;
|
||||||
|
|
||||||
|
private Configuration conf;
|
||||||
|
private MiniDFSCluster cluster;
|
||||||
|
private DistributedFileSystem dfs;
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void setUp() throws Exception {
|
||||||
|
conf = new Configuration();
|
||||||
|
conf.setLong(DFSConfigKeys.DFS_BLOCK_SIZE_KEY, BLOCKSIZE);
|
||||||
|
cluster = new MiniDFSCluster.Builder(conf).numDataNodes(REPLICATION)
|
||||||
|
.build();
|
||||||
|
cluster.waitActive();
|
||||||
|
dfs = cluster.getFileSystem();
|
||||||
|
}
|
||||||
|
|
||||||
|
@After
|
||||||
|
public void tearDown() throws Exception {
|
||||||
|
if (cluster != null) {
|
||||||
|
cluster.shutdown();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void testGetFileChecksum(final Path foo, final int appendLength)
|
||||||
|
throws Exception {
|
||||||
|
final int appendRounds = 16;
|
||||||
|
FileChecksum[] fc = new FileChecksum[appendRounds + 1];
|
||||||
|
DFSTestUtil.createFile(dfs, foo, appendLength, REPLICATION, 0L);
|
||||||
|
fc[0] = dfs.getFileChecksum(foo);
|
||||||
|
for (int i = 0; i < appendRounds; i++) {
|
||||||
|
DFSTestUtil.appendFile(dfs, foo, appendLength);
|
||||||
|
fc[i + 1] = dfs.getFileChecksum(foo);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int i = 0; i < appendRounds + 1; i++) {
|
||||||
|
FileChecksum checksum = dfs.getFileChecksum(foo, appendLength * (i+1));
|
||||||
|
Assert.assertTrue(checksum.equals(fc[i]));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testGetFileChecksum() throws Exception {
|
||||||
|
testGetFileChecksum(new Path("/foo"), BLOCKSIZE / 4);
|
||||||
|
testGetFileChecksum(new Path("/bar"), BLOCKSIZE / 4 - 1);
|
||||||
|
}
|
||||||
|
}
|
|
@ -22,8 +22,12 @@ import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_HA_NAMENODES_KEY_PREFIX;
|
||||||
import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_NAMENODE_RPC_ADDRESS_KEY;
|
import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_NAMENODE_RPC_ADDRESS_KEY;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
|
import java.net.BindException;
|
||||||
import java.net.URI;
|
import java.net.URI;
|
||||||
|
import java.util.Random;
|
||||||
|
|
||||||
|
import org.apache.commons.logging.Log;
|
||||||
|
import org.apache.commons.logging.LogFactory;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
import org.apache.hadoop.hdfs.DFSConfigKeys;
|
import org.apache.hadoop.hdfs.DFSConfigKeys;
|
||||||
import org.apache.hadoop.hdfs.DFSUtil;
|
import org.apache.hadoop.hdfs.DFSUtil;
|
||||||
|
@ -37,14 +41,13 @@ public class MiniQJMHACluster {
|
||||||
private MiniDFSCluster cluster;
|
private MiniDFSCluster cluster;
|
||||||
private MiniJournalCluster journalCluster;
|
private MiniJournalCluster journalCluster;
|
||||||
private final Configuration conf;
|
private final Configuration conf;
|
||||||
|
private static final Log LOG = LogFactory.getLog(MiniQJMHACluster.class);
|
||||||
|
|
||||||
public static final String NAMESERVICE = "ns1";
|
public static final String NAMESERVICE = "ns1";
|
||||||
private static final String NN1 = "nn1";
|
private static final String NN1 = "nn1";
|
||||||
private static final String NN2 = "nn2";
|
private static final String NN2 = "nn2";
|
||||||
private static final int NN1_IPC_PORT = 10000;
|
private static final Random RANDOM = new Random();
|
||||||
private static final int NN1_INFO_PORT = 10001;
|
private int basePort = 10000;
|
||||||
private static final int NN2_IPC_PORT = 10002;
|
|
||||||
private static final int NN2_INFO_PORT = 10003;
|
|
||||||
|
|
||||||
public static class Builder {
|
public static class Builder {
|
||||||
private final Configuration conf;
|
private final Configuration conf;
|
||||||
|
@ -69,24 +72,28 @@ public class MiniQJMHACluster {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public static MiniDFSNNTopology createDefaultTopology() {
|
public static MiniDFSNNTopology createDefaultTopology(int basePort) {
|
||||||
return new MiniDFSNNTopology()
|
return new MiniDFSNNTopology()
|
||||||
.addNameservice(new MiniDFSNNTopology.NSConf(NAMESERVICE).addNN(
|
.addNameservice(new MiniDFSNNTopology.NSConf(NAMESERVICE).addNN(
|
||||||
new MiniDFSNNTopology.NNConf("nn1").setIpcPort(NN1_IPC_PORT)
|
new MiniDFSNNTopology.NNConf("nn1").setIpcPort(basePort)
|
||||||
.setHttpPort(NN1_INFO_PORT)).addNN(
|
.setHttpPort(basePort + 1)).addNN(
|
||||||
new MiniDFSNNTopology.NNConf("nn2").setIpcPort(NN2_IPC_PORT)
|
new MiniDFSNNTopology.NNConf("nn2").setIpcPort(basePort + 2)
|
||||||
.setHttpPort(NN2_INFO_PORT)));
|
.setHttpPort(basePort + 3)));
|
||||||
}
|
}
|
||||||
|
|
||||||
private MiniQJMHACluster(Builder builder) throws IOException {
|
private MiniQJMHACluster(Builder builder) throws IOException {
|
||||||
this.conf = builder.conf;
|
this.conf = builder.conf;
|
||||||
|
int retryCount = 0;
|
||||||
|
while (true) {
|
||||||
|
try {
|
||||||
|
basePort = 10000 + RANDOM.nextInt(1000) * 4;
|
||||||
// start 3 journal nodes
|
// start 3 journal nodes
|
||||||
journalCluster = new MiniJournalCluster.Builder(conf).format(true)
|
journalCluster = new MiniJournalCluster.Builder(conf).format(true)
|
||||||
.build();
|
.build();
|
||||||
URI journalURI = journalCluster.getQuorumJournalURI(NAMESERVICE);
|
URI journalURI = journalCluster.getQuorumJournalURI(NAMESERVICE);
|
||||||
|
|
||||||
// start cluster with 2 NameNodes
|
// start cluster with 2 NameNodes
|
||||||
MiniDFSNNTopology topology = createDefaultTopology();
|
MiniDFSNNTopology topology = createDefaultTopology(basePort);
|
||||||
|
|
||||||
initHAConf(journalURI, builder.conf);
|
initHAConf(journalURI, builder.conf);
|
||||||
|
|
||||||
|
@ -106,14 +113,21 @@ public class MiniQJMHACluster {
|
||||||
|
|
||||||
// restart the cluster
|
// restart the cluster
|
||||||
cluster.restartNameNodes();
|
cluster.restartNameNodes();
|
||||||
|
++retryCount;
|
||||||
|
break;
|
||||||
|
} catch (BindException e) {
|
||||||
|
LOG.info("MiniQJMHACluster port conflicts, retried " +
|
||||||
|
retryCount + " times");
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private Configuration initHAConf(URI journalURI, Configuration conf) {
|
private Configuration initHAConf(URI journalURI, Configuration conf) {
|
||||||
conf.set(DFSConfigKeys.DFS_NAMENODE_SHARED_EDITS_DIR_KEY,
|
conf.set(DFSConfigKeys.DFS_NAMENODE_SHARED_EDITS_DIR_KEY,
|
||||||
journalURI.toString());
|
journalURI.toString());
|
||||||
|
|
||||||
String address1 = "127.0.0.1:" + NN1_IPC_PORT;
|
String address1 = "127.0.0.1:" + basePort;
|
||||||
String address2 = "127.0.0.1:" + NN2_IPC_PORT;
|
String address2 = "127.0.0.1:" + (basePort + 2);
|
||||||
conf.set(DFSUtil.addKeySuffixes(DFS_NAMENODE_RPC_ADDRESS_KEY,
|
conf.set(DFSUtil.addKeySuffixes(DFS_NAMENODE_RPC_ADDRESS_KEY,
|
||||||
NAMESERVICE, NN1), address1);
|
NAMESERVICE, NN1), address1);
|
||||||
conf.set(DFSUtil.addKeySuffixes(DFS_NAMENODE_RPC_ADDRESS_KEY,
|
conf.set(DFSUtil.addKeySuffixes(DFS_NAMENODE_RPC_ADDRESS_KEY,
|
||||||
|
|
|
@ -45,6 +45,7 @@ import org.apache.hadoop.hdfs.web.WebHdfsFileSystem;
|
||||||
import org.apache.hadoop.hdfs.web.WebHdfsTestUtil;
|
import org.apache.hadoop.hdfs.web.WebHdfsTestUtil;
|
||||||
import org.apache.hadoop.security.TestDoAsEffectiveUser;
|
import org.apache.hadoop.security.TestDoAsEffectiveUser;
|
||||||
import org.apache.hadoop.security.UserGroupInformation;
|
import org.apache.hadoop.security.UserGroupInformation;
|
||||||
|
import org.apache.hadoop.security.authorize.DefaultImpersonationProvider;
|
||||||
import org.apache.hadoop.security.authorize.ProxyUsers;
|
import org.apache.hadoop.security.authorize.ProxyUsers;
|
||||||
import org.apache.hadoop.security.token.Token;
|
import org.apache.hadoop.security.token.Token;
|
||||||
import org.junit.AfterClass;
|
import org.junit.AfterClass;
|
||||||
|
@ -88,7 +89,7 @@ public class TestDelegationTokenForProxyUser {
|
||||||
builder.append("127.0.1.1,");
|
builder.append("127.0.1.1,");
|
||||||
builder.append(InetAddress.getLocalHost().getCanonicalHostName());
|
builder.append(InetAddress.getLocalHost().getCanonicalHostName());
|
||||||
LOG.info("Local Ip addresses: " + builder.toString());
|
LOG.info("Local Ip addresses: " + builder.toString());
|
||||||
conf.setStrings(ProxyUsers.getProxySuperuserIpConfKey(superUserShortName),
|
conf.setStrings(DefaultImpersonationProvider.getProxySuperuserIpConfKey(superUserShortName),
|
||||||
builder.toString());
|
builder.toString());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -100,7 +101,7 @@ public class TestDelegationTokenForProxyUser {
|
||||||
DFSConfigKeys.DFS_NAMENODE_DELEGATION_TOKEN_MAX_LIFETIME_KEY, 10000);
|
DFSConfigKeys.DFS_NAMENODE_DELEGATION_TOKEN_MAX_LIFETIME_KEY, 10000);
|
||||||
config.setLong(
|
config.setLong(
|
||||||
DFSConfigKeys.DFS_NAMENODE_DELEGATION_TOKEN_RENEW_INTERVAL_KEY, 5000);
|
DFSConfigKeys.DFS_NAMENODE_DELEGATION_TOKEN_RENEW_INTERVAL_KEY, 5000);
|
||||||
config.setStrings(ProxyUsers.getProxySuperuserGroupConfKey(REAL_USER),
|
config.setStrings(DefaultImpersonationProvider.getProxySuperuserGroupConfKey(REAL_USER),
|
||||||
"group1");
|
"group1");
|
||||||
config.setBoolean(
|
config.setBoolean(
|
||||||
DFSConfigKeys.DFS_NAMENODE_DELEGATION_TOKEN_ALWAYS_USE_KEY, true);
|
DFSConfigKeys.DFS_NAMENODE_DELEGATION_TOKEN_ALWAYS_USE_KEY, true);
|
||||||
|
|
|
@ -30,6 +30,7 @@ import org.apache.hadoop.io.Text;
|
||||||
import org.apache.hadoop.security.UserGroupInformation;
|
import org.apache.hadoop.security.UserGroupInformation;
|
||||||
import org.apache.hadoop.security.UserGroupInformation.AuthenticationMethod;
|
import org.apache.hadoop.security.UserGroupInformation.AuthenticationMethod;
|
||||||
import org.apache.hadoop.security.authorize.AuthorizationException;
|
import org.apache.hadoop.security.authorize.AuthorizationException;
|
||||||
|
import org.apache.hadoop.security.authorize.DefaultImpersonationProvider;
|
||||||
import org.apache.hadoop.security.authorize.ProxyServers;
|
import org.apache.hadoop.security.authorize.ProxyServers;
|
||||||
import org.apache.hadoop.security.authorize.ProxyUsers;
|
import org.apache.hadoop.security.authorize.ProxyUsers;
|
||||||
import org.apache.hadoop.security.token.Token;
|
import org.apache.hadoop.security.token.Token;
|
||||||
|
@ -284,8 +285,8 @@ public class TestJspHelper {
|
||||||
String user = "TheNurse";
|
String user = "TheNurse";
|
||||||
conf.set(DFSConfigKeys.HADOOP_SECURITY_AUTHENTICATION, "kerberos");
|
conf.set(DFSConfigKeys.HADOOP_SECURITY_AUTHENTICATION, "kerberos");
|
||||||
|
|
||||||
conf.set(ProxyUsers.getProxySuperuserGroupConfKey(realUser), "*");
|
conf.set(DefaultImpersonationProvider.getProxySuperuserGroupConfKey(realUser), "*");
|
||||||
conf.set(ProxyUsers.getProxySuperuserIpConfKey(realUser), "*");
|
conf.set(DefaultImpersonationProvider.getProxySuperuserIpConfKey(realUser), "*");
|
||||||
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
ProxyUsers.refreshSuperUserGroupsConfiguration(conf);
|
||||||
UserGroupInformation.setConfiguration(conf);
|
UserGroupInformation.setConfiguration(conf);
|
||||||
UserGroupInformation ugi;
|
UserGroupInformation ugi;
|
||||||
|
|
|
@ -44,6 +44,7 @@ import org.apache.hadoop.hdfs.security.token.block.BlockTokenSecretManager;
|
||||||
import org.apache.hadoop.hdfs.server.datanode.fsdataset.FsVolumeSpi;
|
import org.apache.hadoop.hdfs.server.datanode.fsdataset.FsVolumeSpi;
|
||||||
import org.apache.hadoop.hdfs.server.namenode.NameNodeAdapter;
|
import org.apache.hadoop.hdfs.server.namenode.NameNodeAdapter;
|
||||||
import org.apache.hadoop.util.DataChecksum;
|
import org.apache.hadoop.util.DataChecksum;
|
||||||
|
import org.apache.hadoop.util.Time;
|
||||||
import org.junit.After;
|
import org.junit.After;
|
||||||
import org.junit.Before;
|
import org.junit.Before;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
|
@ -215,6 +216,6 @@ public class TestDiskError {
|
||||||
dataNode.checkDiskError();
|
dataNode.checkDiskError();
|
||||||
Thread.sleep(dataNode.checkDiskErrorInterval);
|
Thread.sleep(dataNode.checkDiskErrorInterval);
|
||||||
long lastDiskErrorCheck = dataNode.getLastDiskErrorCheck();
|
long lastDiskErrorCheck = dataNode.getLastDiskErrorCheck();
|
||||||
assertTrue("Disk Error check is not performed within " + dataNode.checkDiskErrorInterval + " ms", ((System.currentTimeMillis()-lastDiskErrorCheck) < (dataNode.checkDiskErrorInterval + slackTime)));
|
assertTrue("Disk Error check is not performed within " + dataNode.checkDiskErrorInterval + " ms", ((Time.monotonicNow()-lastDiskErrorCheck) < (dataNode.checkDiskErrorInterval + slackTime)));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -39,8 +39,8 @@ import org.apache.hadoop.hdfs.MiniDFSCluster;
|
||||||
import org.apache.hadoop.hdfs.web.resources.GetOpParam;
|
import org.apache.hadoop.hdfs.web.resources.GetOpParam;
|
||||||
import org.apache.hadoop.ipc.RemoteException;
|
import org.apache.hadoop.ipc.RemoteException;
|
||||||
import org.apache.hadoop.net.NetUtils;
|
import org.apache.hadoop.net.NetUtils;
|
||||||
import org.apache.hadoop.security.authorize.ProxyUsers;
|
|
||||||
import org.apache.hadoop.security.authorize.ProxyServers;
|
import org.apache.hadoop.security.authorize.ProxyServers;
|
||||||
|
import org.apache.hadoop.security.authorize.ProxyUsers;
|
||||||
import org.junit.Before;
|
import org.junit.Before;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
|
|
||||||
|
|
|
@ -17,11 +17,16 @@
|
||||||
*/
|
*/
|
||||||
package org.apache.hadoop.hdfs.server.namenode;
|
package org.apache.hadoop.hdfs.server.namenode;
|
||||||
|
|
||||||
|
import java.net.BindException;
|
||||||
|
import java.util.Random;
|
||||||
|
|
||||||
import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_NAMENODE_CHECKPOINT_PERIOD_KEY;
|
import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_NAMENODE_CHECKPOINT_PERIOD_KEY;
|
||||||
import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_NAMENODE_CHECKPOINT_TXNS_KEY;
|
import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_NAMENODE_CHECKPOINT_TXNS_KEY;
|
||||||
import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_NAMENODE_EDIT_LOG_AUTOROLL_CHECK_INTERVAL_MS;
|
import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_NAMENODE_EDIT_LOG_AUTOROLL_CHECK_INTERVAL_MS;
|
||||||
import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_NAMENODE_EDIT_LOG_AUTOROLL_MULTIPLIER_THRESHOLD;
|
import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_NAMENODE_EDIT_LOG_AUTOROLL_MULTIPLIER_THRESHOLD;
|
||||||
|
|
||||||
|
import org.apache.commons.logging.Log;
|
||||||
|
import org.apache.commons.logging.LogFactory;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
import org.apache.hadoop.fs.FileSystem;
|
import org.apache.hadoop.fs.FileSystem;
|
||||||
import org.apache.hadoop.fs.Path;
|
import org.apache.hadoop.fs.Path;
|
||||||
|
@ -43,6 +48,9 @@ public class TestEditLogAutoroll {
|
||||||
private NameNode nn0;
|
private NameNode nn0;
|
||||||
private FileSystem fs;
|
private FileSystem fs;
|
||||||
private FSEditLog editLog;
|
private FSEditLog editLog;
|
||||||
|
private final Random random = new Random();
|
||||||
|
|
||||||
|
private static final Log LOG = LogFactory.getLog(TestEditLog.class);
|
||||||
|
|
||||||
@Before
|
@Before
|
||||||
public void setUp() throws Exception {
|
public void setUp() throws Exception {
|
||||||
|
@ -54,10 +62,14 @@ public class TestEditLogAutoroll {
|
||||||
conf.setFloat(DFS_NAMENODE_EDIT_LOG_AUTOROLL_MULTIPLIER_THRESHOLD, 0.5f);
|
conf.setFloat(DFS_NAMENODE_EDIT_LOG_AUTOROLL_MULTIPLIER_THRESHOLD, 0.5f);
|
||||||
conf.setInt(DFS_NAMENODE_EDIT_LOG_AUTOROLL_CHECK_INTERVAL_MS, 100);
|
conf.setInt(DFS_NAMENODE_EDIT_LOG_AUTOROLL_CHECK_INTERVAL_MS, 100);
|
||||||
|
|
||||||
|
int retryCount = 0;
|
||||||
|
while (true) {
|
||||||
|
try {
|
||||||
|
int basePort = 10060 + random.nextInt(100) * 2;
|
||||||
MiniDFSNNTopology topology = new MiniDFSNNTopology()
|
MiniDFSNNTopology topology = new MiniDFSNNTopology()
|
||||||
.addNameservice(new MiniDFSNNTopology.NSConf("ns1")
|
.addNameservice(new MiniDFSNNTopology.NSConf("ns1")
|
||||||
.addNN(new MiniDFSNNTopology.NNConf("nn1").setHttpPort(10061))
|
.addNN(new MiniDFSNNTopology.NNConf("nn1").setHttpPort(basePort))
|
||||||
.addNN(new MiniDFSNNTopology.NNConf("nn2").setHttpPort(10062)));
|
.addNN(new MiniDFSNNTopology.NNConf("nn2").setHttpPort(basePort + 1)));
|
||||||
|
|
||||||
cluster = new MiniDFSCluster.Builder(conf)
|
cluster = new MiniDFSCluster.Builder(conf)
|
||||||
.nnTopology(topology)
|
.nnTopology(topology)
|
||||||
|
@ -72,6 +84,13 @@ public class TestEditLogAutoroll {
|
||||||
|
|
||||||
fs = cluster.getFileSystem(0);
|
fs = cluster.getFileSystem(0);
|
||||||
editLog = nn0.getNamesystem().getEditLog();
|
editLog = nn0.getNamesystem().getEditLog();
|
||||||
|
++retryCount;
|
||||||
|
break;
|
||||||
|
} catch (BindException e) {
|
||||||
|
LOG.info("Set up MiniDFSCluster failed due to port conflicts, retry "
|
||||||
|
+ retryCount + " times");
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@After
|
@After
|
||||||
|
|
|
@ -104,7 +104,7 @@ public class TestFailureToReadEdits {
|
||||||
HAUtil.setAllowStandbyReads(conf, true);
|
HAUtil.setAllowStandbyReads(conf, true);
|
||||||
|
|
||||||
if (clusterType == TestType.SHARED_DIR_HA) {
|
if (clusterType == TestType.SHARED_DIR_HA) {
|
||||||
MiniDFSNNTopology topology = MiniQJMHACluster.createDefaultTopology();
|
MiniDFSNNTopology topology = MiniQJMHACluster.createDefaultTopology(10000);
|
||||||
cluster = new MiniDFSCluster.Builder(conf)
|
cluster = new MiniDFSCluster.Builder(conf)
|
||||||
.nnTopology(topology)
|
.nnTopology(topology)
|
||||||
.numDataNodes(0)
|
.numDataNodes(0)
|
||||||
|
|
|
@ -25,12 +25,14 @@ import static org.junit.Assert.fail;
|
||||||
import java.io.File;
|
import java.io.File;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.OutputStream;
|
import java.io.OutputStream;
|
||||||
|
import java.net.BindException;
|
||||||
import java.lang.management.ManagementFactory;
|
import java.lang.management.ManagementFactory;
|
||||||
import java.lang.management.ThreadInfo;
|
import java.lang.management.ThreadInfo;
|
||||||
import java.lang.management.ThreadMXBean;
|
import java.lang.management.ThreadMXBean;
|
||||||
import java.net.URI;
|
import java.net.URI;
|
||||||
import java.net.URL;
|
import java.net.URL;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
import java.util.Random;
|
||||||
|
|
||||||
import org.apache.commons.logging.Log;
|
import org.apache.commons.logging.Log;
|
||||||
import org.apache.commons.logging.LogFactory;
|
import org.apache.commons.logging.LogFactory;
|
||||||
|
@ -73,6 +75,7 @@ public class TestStandbyCheckpoints {
|
||||||
protected MiniDFSCluster cluster;
|
protected MiniDFSCluster cluster;
|
||||||
protected NameNode nn0, nn1;
|
protected NameNode nn0, nn1;
|
||||||
protected FileSystem fs;
|
protected FileSystem fs;
|
||||||
|
private final Random random = new Random();
|
||||||
protected File tmpOivImgDir;
|
protected File tmpOivImgDir;
|
||||||
|
|
||||||
private static final Log LOG = LogFactory.getLog(TestStandbyCheckpoints.class);
|
private static final Log LOG = LogFactory.getLog(TestStandbyCheckpoints.class);
|
||||||
|
@ -87,10 +90,14 @@ public class TestStandbyCheckpoints {
|
||||||
conf.setInt(DFSConfigKeys.DFS_NAMENODE_NUM_CHECKPOINTS_RETAINED_KEY, 1);
|
conf.setInt(DFSConfigKeys.DFS_NAMENODE_NUM_CHECKPOINTS_RETAINED_KEY, 1);
|
||||||
conf.setInt(DFSConfigKeys.DFS_NAMENODE_NUM_EXTRA_EDITS_RETAINED_KEY, 0);
|
conf.setInt(DFSConfigKeys.DFS_NAMENODE_NUM_EXTRA_EDITS_RETAINED_KEY, 0);
|
||||||
|
|
||||||
|
int retryCount = 0;
|
||||||
|
while (true) {
|
||||||
|
try {
|
||||||
|
int basePort = 10060 + random.nextInt(100) * 2;
|
||||||
MiniDFSNNTopology topology = new MiniDFSNNTopology()
|
MiniDFSNNTopology topology = new MiniDFSNNTopology()
|
||||||
.addNameservice(new MiniDFSNNTopology.NSConf("ns1")
|
.addNameservice(new MiniDFSNNTopology.NSConf("ns1")
|
||||||
.addNN(new MiniDFSNNTopology.NNConf("nn1").setHttpPort(10061))
|
.addNN(new MiniDFSNNTopology.NNConf("nn1").setHttpPort(basePort))
|
||||||
.addNN(new MiniDFSNNTopology.NNConf("nn2").setHttpPort(10062)));
|
.addNN(new MiniDFSNNTopology.NNConf("nn2").setHttpPort(basePort + 1)));
|
||||||
|
|
||||||
cluster = new MiniDFSCluster.Builder(conf)
|
cluster = new MiniDFSCluster.Builder(conf)
|
||||||
.nnTopology(topology)
|
.nnTopology(topology)
|
||||||
|
@ -103,6 +110,13 @@ public class TestStandbyCheckpoints {
|
||||||
fs = HATestUtil.configureFailoverFs(cluster, conf);
|
fs = HATestUtil.configureFailoverFs(cluster, conf);
|
||||||
|
|
||||||
cluster.transitionToActive(0);
|
cluster.transitionToActive(0);
|
||||||
|
++retryCount;
|
||||||
|
break;
|
||||||
|
} catch (BindException e) {
|
||||||
|
LOG.info("Set up MiniDFSCluster failed due to port conflicts, retry "
|
||||||
|
+ retryCount + " times");
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
protected Configuration setupCommonConfig() {
|
protected Configuration setupCommonConfig() {
|
||||||
|
|
|
@ -41,6 +41,7 @@ import org.apache.hadoop.fs.Path;
|
||||||
import org.apache.hadoop.hdfs.MiniDFSCluster;
|
import org.apache.hadoop.hdfs.MiniDFSCluster;
|
||||||
import org.apache.hadoop.hdfs.tools.DFSAdmin;
|
import org.apache.hadoop.hdfs.tools.DFSAdmin;
|
||||||
import org.apache.hadoop.security.authorize.AuthorizationException;
|
import org.apache.hadoop.security.authorize.AuthorizationException;
|
||||||
|
import org.apache.hadoop.security.authorize.DefaultImpersonationProvider;
|
||||||
import org.apache.hadoop.security.authorize.ProxyUsers;
|
import org.apache.hadoop.security.authorize.ProxyUsers;
|
||||||
import org.junit.After;
|
import org.junit.After;
|
||||||
import org.junit.Before;
|
import org.junit.Before;
|
||||||
|
@ -150,8 +151,8 @@ public class TestRefreshUserMappings {
|
||||||
final String [] GROUP_NAMES2 = new String [] {"gr3" , "gr4"};
|
final String [] GROUP_NAMES2 = new String [] {"gr3" , "gr4"};
|
||||||
|
|
||||||
//keys in conf
|
//keys in conf
|
||||||
String userKeyGroups = ProxyUsers.getProxySuperuserGroupConfKey(SUPER_USER);
|
String userKeyGroups = DefaultImpersonationProvider.getProxySuperuserGroupConfKey(SUPER_USER);
|
||||||
String userKeyHosts = ProxyUsers.getProxySuperuserIpConfKey (SUPER_USER);
|
String userKeyHosts = DefaultImpersonationProvider.getProxySuperuserIpConfKey (SUPER_USER);
|
||||||
|
|
||||||
config.set(userKeyGroups, "gr3,gr4,gr5"); // superuser can proxy for this group
|
config.set(userKeyGroups, "gr3,gr4,gr5"); // superuser can proxy for this group
|
||||||
config.set(userKeyHosts,"127.0.0.1");
|
config.set(userKeyHosts,"127.0.0.1");
|
||||||
|
|
|
@ -202,6 +202,8 @@ Release 2.5.0 - UNRELEASED
|
||||||
|
|
||||||
MAPREDUCE-5809. Enhance distcp to support preserving HDFS ACLs. (cnauroth)
|
MAPREDUCE-5809. Enhance distcp to support preserving HDFS ACLs. (cnauroth)
|
||||||
|
|
||||||
|
MAPREDUCE-5899. Support incremental data copy in DistCp. (jing9)
|
||||||
|
|
||||||
OPTIMIZATIONS
|
OPTIMIZATIONS
|
||||||
|
|
||||||
BUG FIXES
|
BUG FIXES
|
||||||
|
@ -239,6 +241,12 @@ Release 2.5.0 - UNRELEASED
|
||||||
MAPREDUCE-5309. 2.0.4 JobHistoryParser can't parse certain failed job
|
MAPREDUCE-5309. 2.0.4 JobHistoryParser can't parse certain failed job
|
||||||
history files generated by 2.0.3 history server (Rushabh S Shah via jlowe)
|
history files generated by 2.0.3 history server (Rushabh S Shah via jlowe)
|
||||||
|
|
||||||
|
MAPREDUCE-5862. Line records longer than 2x split size aren't handled
|
||||||
|
correctly (bc Wong via jlowe)
|
||||||
|
|
||||||
|
MAPREDUCE-5895. Close streams properly to avoid leakage in TaskLog.
|
||||||
|
(Kousuke Saruta via devaraj)
|
||||||
|
|
||||||
Release 2.4.1 - UNRELEASED
|
Release 2.4.1 - UNRELEASED
|
||||||
|
|
||||||
INCOMPATIBLE CHANGES
|
INCOMPATIBLE CHANGES
|
||||||
|
|
|
@ -85,6 +85,15 @@
|
||||||
</execution>
|
</execution>
|
||||||
</executions>
|
</executions>
|
||||||
</plugin>
|
</plugin>
|
||||||
|
<plugin>
|
||||||
|
<groupId>org.apache.rat</groupId>
|
||||||
|
<artifactId>apache-rat-plugin</artifactId>
|
||||||
|
<configuration>
|
||||||
|
<excludes>
|
||||||
|
<exclude>src/test/resources/recordSpanningMultipleSplits.txt</exclude>
|
||||||
|
</excludes>
|
||||||
|
</configuration>
|
||||||
|
</plugin>
|
||||||
</plugins>
|
</plugins>
|
||||||
</build>
|
</build>
|
||||||
</project>
|
</project>
|
||||||
|
|
|
@ -184,7 +184,7 @@ public class LineRecordReader implements RecordReader<LongWritable, Text> {
|
||||||
private int maxBytesToConsume(long pos) {
|
private int maxBytesToConsume(long pos) {
|
||||||
return isCompressedInput()
|
return isCompressedInput()
|
||||||
? Integer.MAX_VALUE
|
? Integer.MAX_VALUE
|
||||||
: (int) Math.min(Integer.MAX_VALUE, end - pos);
|
: (int) Math.max(Math.min(Integer.MAX_VALUE, end - pos), maxLineLength);
|
||||||
}
|
}
|
||||||
|
|
||||||
private long getFilePosition() throws IOException {
|
private long getFilePosition() throws IOException {
|
||||||
|
@ -206,8 +206,7 @@ public class LineRecordReader implements RecordReader<LongWritable, Text> {
|
||||||
while (getFilePosition() <= end || in.needAdditionalRecordAfterSplit()) {
|
while (getFilePosition() <= end || in.needAdditionalRecordAfterSplit()) {
|
||||||
key.set(pos);
|
key.set(pos);
|
||||||
|
|
||||||
int newSize = in.readLine(value, maxLineLength,
|
int newSize = in.readLine(value, maxLineLength, maxBytesToConsume(pos));
|
||||||
Math.max(maxBytesToConsume(pos), maxLineLength));
|
|
||||||
if (newSize == 0) {
|
if (newSize == 0) {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
|
@ -199,16 +199,18 @@ public class TaskLog {
|
||||||
// file first and then rename.
|
// file first and then rename.
|
||||||
File tmpIndexFile = getTmpIndexFile(currentTaskid, isCleanup);
|
File tmpIndexFile = getTmpIndexFile(currentTaskid, isCleanup);
|
||||||
|
|
||||||
BufferedOutputStream bos =
|
BufferedOutputStream bos = null;
|
||||||
new BufferedOutputStream(
|
DataOutputStream dos = null;
|
||||||
|
try{
|
||||||
|
bos = new BufferedOutputStream(
|
||||||
SecureIOUtils.createForWrite(tmpIndexFile, 0644));
|
SecureIOUtils.createForWrite(tmpIndexFile, 0644));
|
||||||
DataOutputStream dos = new DataOutputStream(bos);
|
dos = new DataOutputStream(bos);
|
||||||
//the format of the index file is
|
//the format of the index file is
|
||||||
//LOG_DIR: <the dir where the task logs are really stored>
|
//LOG_DIR: <the dir where the task logs are really stored>
|
||||||
//STDOUT: <start-offset in the stdout file> <length>
|
//STDOUT: <start-offset in the stdout file> <length>
|
||||||
//STDERR: <start-offset in the stderr file> <length>
|
//STDERR: <start-offset in the stderr file> <length>
|
||||||
//SYSLOG: <start-offset in the syslog file> <length>
|
//SYSLOG: <start-offset in the syslog file> <length>
|
||||||
try{
|
|
||||||
dos.writeBytes(LogFileDetail.LOCATION + logLocation + "\n"
|
dos.writeBytes(LogFileDetail.LOCATION + logLocation + "\n"
|
||||||
+ LogName.STDOUT.toString() + ":");
|
+ LogName.STDOUT.toString() + ":");
|
||||||
dos.writeBytes(Long.toString(prevOutLength) + " ");
|
dos.writeBytes(Long.toString(prevOutLength) + " ");
|
||||||
|
@ -225,8 +227,10 @@ public class TaskLog {
|
||||||
+ "\n");
|
+ "\n");
|
||||||
dos.close();
|
dos.close();
|
||||||
dos = null;
|
dos = null;
|
||||||
|
bos.close();
|
||||||
|
bos = null;
|
||||||
} finally {
|
} finally {
|
||||||
IOUtils.cleanup(LOG, dos);
|
IOUtils.cleanup(LOG, dos, bos);
|
||||||
}
|
}
|
||||||
|
|
||||||
File indexFile = getIndexFile(currentTaskid, isCleanup);
|
File indexFile = getIndexFile(currentTaskid, isCleanup);
|
||||||
|
|
|
@ -121,7 +121,7 @@ public class LineRecordReader extends RecordReader<LongWritable, Text> {
|
||||||
private int maxBytesToConsume(long pos) {
|
private int maxBytesToConsume(long pos) {
|
||||||
return isCompressedInput
|
return isCompressedInput
|
||||||
? Integer.MAX_VALUE
|
? Integer.MAX_VALUE
|
||||||
: (int) Math.min(Integer.MAX_VALUE, end - pos);
|
: (int) Math.max(Math.min(Integer.MAX_VALUE, end - pos), maxLineLength);
|
||||||
}
|
}
|
||||||
|
|
||||||
private long getFilePosition() throws IOException {
|
private long getFilePosition() throws IOException {
|
||||||
|
@ -146,8 +146,7 @@ public class LineRecordReader extends RecordReader<LongWritable, Text> {
|
||||||
// We always read one extra line, which lies outside the upper
|
// We always read one extra line, which lies outside the upper
|
||||||
// split limit i.e. (end - 1)
|
// split limit i.e. (end - 1)
|
||||||
while (getFilePosition() <= end || in.needAdditionalRecordAfterSplit()) {
|
while (getFilePosition() <= end || in.needAdditionalRecordAfterSplit()) {
|
||||||
newSize = in.readLine(value, maxLineLength,
|
newSize = in.readLine(value, maxLineLength, maxBytesToConsume(pos));
|
||||||
Math.max(maxBytesToConsume(pos), maxLineLength));
|
|
||||||
pos += newSize;
|
pos += newSize;
|
||||||
if (newSize < maxLineLength) {
|
if (newSize < maxLineLength) {
|
||||||
break;
|
break;
|
||||||
|
|
|
@ -18,8 +18,6 @@
|
||||||
|
|
||||||
Hadoop MapReduce Next Generation - Distributed Cache Deploy
|
Hadoop MapReduce Next Generation - Distributed Cache Deploy
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
||||||
* Introduction
|
* Introduction
|
||||||
|
|
||||||
The MapReduce application framework has rudimentary support for deploying a
|
The MapReduce application framework has rudimentary support for deploying a
|
||||||
|
|
|
@ -18,8 +18,6 @@
|
||||||
|
|
||||||
Hadoop MapReduce Next Generation - Encrypted Shuffle
|
Hadoop MapReduce Next Generation - Encrypted Shuffle
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
||||||
* {Introduction}
|
* {Introduction}
|
||||||
|
|
||||||
The Encrypted Shuffle capability allows encryption of the MapReduce shuffle
|
The Encrypted Shuffle capability allows encryption of the MapReduce shuffle
|
||||||
|
|
|
@ -18,8 +18,6 @@
|
||||||
|
|
||||||
Apache Hadoop MapReduce - Migrating from Apache Hadoop 1.x to Apache Hadoop 2.x
|
Apache Hadoop MapReduce - Migrating from Apache Hadoop 1.x to Apache Hadoop 2.x
|
||||||
|
|
||||||
\[ {{{../../hadoop-yarn/hadoop-yarn-site/index.html}Go Back}} \]
|
|
||||||
|
|
||||||
* {Introduction}
|
* {Introduction}
|
||||||
|
|
||||||
This document provides information for users to migrate their Apache Hadoop
|
This document provides information for users to migrate their Apache Hadoop
|
||||||
|
|
|
@ -18,8 +18,6 @@
|
||||||
|
|
||||||
Hadoop MapReduce Next Generation - Pluggable Shuffle and Pluggable Sort
|
Hadoop MapReduce Next Generation - Pluggable Shuffle and Pluggable Sort
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
||||||
* Introduction
|
* Introduction
|
||||||
|
|
||||||
The pluggable shuffle and pluggable sort capabilities allow replacing the
|
The pluggable shuffle and pluggable sort capabilities allow replacing the
|
||||||
|
|
|
@ -23,9 +23,12 @@ import static org.junit.Assert.assertNotNull;
|
||||||
import static org.junit.Assert.assertTrue;
|
import static org.junit.Assert.assertTrue;
|
||||||
|
|
||||||
import java.io.File;
|
import java.io.File;
|
||||||
|
import java.io.FileInputStream;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.net.URL;
|
import java.net.URL;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
|
||||||
|
import org.apache.commons.compress.compressors.bzip2.BZip2CompressorInputStream;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
import org.apache.hadoop.fs.Path;
|
import org.apache.hadoop.fs.Path;
|
||||||
import org.apache.hadoop.io.LongWritable;
|
import org.apache.hadoop.io.LongWritable;
|
||||||
|
@ -97,4 +100,92 @@ public class TestLineRecordReader {
|
||||||
// character is a linefeed
|
// character is a linefeed
|
||||||
testSplitRecords("blockEndingInCRThenLF.txt.bz2", 136498);
|
testSplitRecords("blockEndingInCRThenLF.txt.bz2", 136498);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Use the LineRecordReader to read records from the file
|
||||||
|
public ArrayList<String> readRecords(URL testFileUrl, int splitSize)
|
||||||
|
throws IOException {
|
||||||
|
|
||||||
|
// Set up context
|
||||||
|
File testFile = new File(testFileUrl.getFile());
|
||||||
|
long testFileSize = testFile.length();
|
||||||
|
Path testFilePath = new Path(testFile.getAbsolutePath());
|
||||||
|
Configuration conf = new Configuration();
|
||||||
|
conf.setInt("io.file.buffer.size", 1);
|
||||||
|
|
||||||
|
// Gather the records returned by the record reader
|
||||||
|
ArrayList<String> records = new ArrayList<String>();
|
||||||
|
|
||||||
|
long offset = 0;
|
||||||
|
LongWritable key = new LongWritable();
|
||||||
|
Text value = new Text();
|
||||||
|
while (offset < testFileSize) {
|
||||||
|
FileSplit split =
|
||||||
|
new FileSplit(testFilePath, offset, splitSize, (String[]) null);
|
||||||
|
LineRecordReader reader = new LineRecordReader(conf, split);
|
||||||
|
|
||||||
|
while (reader.next(key, value)) {
|
||||||
|
records.add(value.toString());
|
||||||
|
}
|
||||||
|
offset += splitSize;
|
||||||
|
}
|
||||||
|
return records;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Gather the records by just splitting on new lines
|
||||||
|
public String[] readRecordsDirectly(URL testFileUrl, boolean bzip)
|
||||||
|
throws IOException {
|
||||||
|
int MAX_DATA_SIZE = 1024 * 1024;
|
||||||
|
byte[] data = new byte[MAX_DATA_SIZE];
|
||||||
|
FileInputStream fis = new FileInputStream(testFileUrl.getFile());
|
||||||
|
int count;
|
||||||
|
if (bzip) {
|
||||||
|
BZip2CompressorInputStream bzIn = new BZip2CompressorInputStream(fis);
|
||||||
|
count = bzIn.read(data);
|
||||||
|
bzIn.close();
|
||||||
|
} else {
|
||||||
|
count = fis.read(data);
|
||||||
|
}
|
||||||
|
fis.close();
|
||||||
|
assertTrue("Test file data too big for buffer", count < data.length);
|
||||||
|
return new String(data, 0, count, "UTF-8").split("\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
public void checkRecordSpanningMultipleSplits(String testFile,
|
||||||
|
int splitSize,
|
||||||
|
boolean bzip)
|
||||||
|
throws IOException {
|
||||||
|
URL testFileUrl = getClass().getClassLoader().getResource(testFile);
|
||||||
|
ArrayList<String> records = readRecords(testFileUrl, splitSize);
|
||||||
|
String[] actuals = readRecordsDirectly(testFileUrl, bzip);
|
||||||
|
|
||||||
|
assertEquals("Wrong number of records", actuals.length, records.size());
|
||||||
|
|
||||||
|
boolean hasLargeRecord = false;
|
||||||
|
for (int i = 0; i < actuals.length; ++i) {
|
||||||
|
assertEquals(actuals[i], records.get(i));
|
||||||
|
if (actuals[i].length() > 2 * splitSize) {
|
||||||
|
hasLargeRecord = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
assertTrue("Invalid test data. Doesn't have a large enough record",
|
||||||
|
hasLargeRecord);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRecordSpanningMultipleSplits()
|
||||||
|
throws IOException {
|
||||||
|
checkRecordSpanningMultipleSplits("recordSpanningMultipleSplits.txt",
|
||||||
|
10, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRecordSpanningMultipleSplitsCompressed()
|
||||||
|
throws IOException {
|
||||||
|
// The file is generated with bz2 block size of 100k. The split size
|
||||||
|
// needs to be larger than that for the CompressedSplitLineReader to
|
||||||
|
// work.
|
||||||
|
checkRecordSpanningMultipleSplits("recordSpanningMultipleSplits.txt.bz2",
|
||||||
|
200 * 1000, true);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -23,9 +23,12 @@ import static org.junit.Assert.assertNotNull;
|
||||||
import static org.junit.Assert.assertTrue;
|
import static org.junit.Assert.assertTrue;
|
||||||
|
|
||||||
import java.io.File;
|
import java.io.File;
|
||||||
|
import java.io.FileInputStream;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.net.URL;
|
import java.net.URL;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
|
||||||
|
import org.apache.commons.compress.compressors.bzip2.BZip2CompressorInputStream;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
import org.apache.hadoop.fs.Path;
|
import org.apache.hadoop.fs.Path;
|
||||||
import org.apache.hadoop.mapreduce.TaskAttemptContext;
|
import org.apache.hadoop.mapreduce.TaskAttemptContext;
|
||||||
|
@ -101,4 +104,93 @@ public class TestLineRecordReader {
|
||||||
// character is a linefeed
|
// character is a linefeed
|
||||||
testSplitRecords("blockEndingInCRThenLF.txt.bz2", 136498);
|
testSplitRecords("blockEndingInCRThenLF.txt.bz2", 136498);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Use the LineRecordReader to read records from the file
|
||||||
|
public ArrayList<String> readRecords(URL testFileUrl, int splitSize)
|
||||||
|
throws IOException {
|
||||||
|
|
||||||
|
// Set up context
|
||||||
|
File testFile = new File(testFileUrl.getFile());
|
||||||
|
long testFileSize = testFile.length();
|
||||||
|
Path testFilePath = new Path(testFile.getAbsolutePath());
|
||||||
|
Configuration conf = new Configuration();
|
||||||
|
conf.setInt("io.file.buffer.size", 1);
|
||||||
|
TaskAttemptContext context = new TaskAttemptContextImpl(conf, new TaskAttemptID());
|
||||||
|
|
||||||
|
// Gather the records returned by the record reader
|
||||||
|
ArrayList<String> records = new ArrayList<String>();
|
||||||
|
|
||||||
|
long offset = 0;
|
||||||
|
while (offset < testFileSize) {
|
||||||
|
FileSplit split = new FileSplit(testFilePath, offset, splitSize, null);
|
||||||
|
LineRecordReader reader = new LineRecordReader();
|
||||||
|
reader.initialize(split, context);
|
||||||
|
|
||||||
|
while (reader.nextKeyValue()) {
|
||||||
|
records.add(reader.getCurrentValue().toString());
|
||||||
|
}
|
||||||
|
offset += splitSize;
|
||||||
|
}
|
||||||
|
return records;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Gather the records by just splitting on new lines
|
||||||
|
public String[] readRecordsDirectly(URL testFileUrl, boolean bzip)
|
||||||
|
throws IOException {
|
||||||
|
int MAX_DATA_SIZE = 1024 * 1024;
|
||||||
|
byte[] data = new byte[MAX_DATA_SIZE];
|
||||||
|
FileInputStream fis = new FileInputStream(testFileUrl.getFile());
|
||||||
|
int count;
|
||||||
|
if (bzip) {
|
||||||
|
BZip2CompressorInputStream bzIn = new BZip2CompressorInputStream(fis);
|
||||||
|
count = bzIn.read(data);
|
||||||
|
bzIn.close();
|
||||||
|
} else {
|
||||||
|
count = fis.read(data);
|
||||||
|
}
|
||||||
|
fis.close();
|
||||||
|
assertTrue("Test file data too big for buffer", count < data.length);
|
||||||
|
return new String(data, 0, count, "UTF-8").split("\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
public void checkRecordSpanningMultipleSplits(String testFile,
|
||||||
|
int splitSize,
|
||||||
|
boolean bzip)
|
||||||
|
throws IOException {
|
||||||
|
URL testFileUrl = getClass().getClassLoader().getResource(testFile);
|
||||||
|
ArrayList<String> records = readRecords(testFileUrl, splitSize);
|
||||||
|
String[] actuals = readRecordsDirectly(testFileUrl, bzip);
|
||||||
|
|
||||||
|
assertEquals("Wrong number of records", actuals.length, records.size());
|
||||||
|
|
||||||
|
boolean hasLargeRecord = false;
|
||||||
|
for (int i = 0; i < actuals.length; ++i) {
|
||||||
|
assertEquals(actuals[i], records.get(i));
|
||||||
|
if (actuals[i].length() > 2 * splitSize) {
|
||||||
|
hasLargeRecord = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
assertTrue("Invalid test data. Doesn't have a large enough record",
|
||||||
|
hasLargeRecord);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRecordSpanningMultipleSplits()
|
||||||
|
throws IOException {
|
||||||
|
checkRecordSpanningMultipleSplits("recordSpanningMultipleSplits.txt",
|
||||||
|
10,
|
||||||
|
false);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRecordSpanningMultipleSplitsCompressed()
|
||||||
|
throws IOException {
|
||||||
|
// The file is generated with bz2 block size of 100k. The split size
|
||||||
|
// needs to be larger than that for the CompressedSplitLineReader to
|
||||||
|
// work.
|
||||||
|
checkRecordSpanningMultipleSplits("recordSpanningMultipleSplits.txt.bz2",
|
||||||
|
200 * 1000,
|
||||||
|
true);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,4 @@
|
||||||
|
Use with small split size,
|
||||||
|
like 32.
|
||||||
|
And then we give it a really really long line, which will surely span multiple splits,
|
||||||
|
to see how it handles.
|
Binary file not shown.
|
@ -50,6 +50,7 @@ public class DistCpConstants {
|
||||||
public static final String CONF_LABEL_COPY_STRATEGY = "distcp.copy.strategy";
|
public static final String CONF_LABEL_COPY_STRATEGY = "distcp.copy.strategy";
|
||||||
public static final String CONF_LABEL_SKIP_CRC = "distcp.skip.crc";
|
public static final String CONF_LABEL_SKIP_CRC = "distcp.skip.crc";
|
||||||
public static final String CONF_LABEL_OVERWRITE = "distcp.copy.overwrite";
|
public static final String CONF_LABEL_OVERWRITE = "distcp.copy.overwrite";
|
||||||
|
public static final String CONF_LABEL_APPEND = "distcp.copy.append";
|
||||||
public static final String CONF_LABEL_BANDWIDTH_MB = "distcp.map.bandwidth.mb";
|
public static final String CONF_LABEL_BANDWIDTH_MB = "distcp.map.bandwidth.mb";
|
||||||
|
|
||||||
public static final String CONF_LABEL_MAX_CHUNKS_TOLERABLE =
|
public static final String CONF_LABEL_MAX_CHUNKS_TOLERABLE =
|
||||||
|
|
|
@ -138,6 +138,10 @@ public enum DistCpOptionSwitch {
|
||||||
new Option("overwrite", false, "Choose to overwrite target files " +
|
new Option("overwrite", false, "Choose to overwrite target files " +
|
||||||
"unconditionally, even if they exist.")),
|
"unconditionally, even if they exist.")),
|
||||||
|
|
||||||
|
APPEND(DistCpConstants.CONF_LABEL_APPEND,
|
||||||
|
new Option("append", false,
|
||||||
|
"Reuse existing data in target files and append new data to them if possible")),
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Should DisctpExecution be blocking
|
* Should DisctpExecution be blocking
|
||||||
*/
|
*/
|
||||||
|
|
|
@ -39,6 +39,7 @@ public class DistCpOptions {
|
||||||
private boolean deleteMissing = false;
|
private boolean deleteMissing = false;
|
||||||
private boolean ignoreFailures = false;
|
private boolean ignoreFailures = false;
|
||||||
private boolean overwrite = false;
|
private boolean overwrite = false;
|
||||||
|
private boolean append = false;
|
||||||
private boolean skipCRC = false;
|
private boolean skipCRC = false;
|
||||||
private boolean blocking = true;
|
private boolean blocking = true;
|
||||||
|
|
||||||
|
@ -244,6 +245,22 @@ public class DistCpOptions {
|
||||||
this.overwrite = overwrite;
|
this.overwrite = overwrite;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @return whether we can append new data to target files
|
||||||
|
*/
|
||||||
|
public boolean shouldAppend() {
|
||||||
|
return append;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Set if we want to append new data to target files. This is valid only with
|
||||||
|
* update option and CRC is not skipped.
|
||||||
|
*/
|
||||||
|
public void setAppend(boolean append) {
|
||||||
|
validate(DistCpOptionSwitch.APPEND, append);
|
||||||
|
this.append = append;
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Should CRC/checksum check be skipped while checking files are identical
|
* Should CRC/checksum check be skipped while checking files are identical
|
||||||
*
|
*
|
||||||
|
@ -472,6 +489,7 @@ public class DistCpOptions {
|
||||||
value : this.atomicCommit);
|
value : this.atomicCommit);
|
||||||
boolean skipCRC = (option == DistCpOptionSwitch.SKIP_CRC ?
|
boolean skipCRC = (option == DistCpOptionSwitch.SKIP_CRC ?
|
||||||
value : this.skipCRC);
|
value : this.skipCRC);
|
||||||
|
boolean append = (option == DistCpOptionSwitch.APPEND ? value : this.append);
|
||||||
|
|
||||||
if (syncFolder && atomicCommit) {
|
if (syncFolder && atomicCommit) {
|
||||||
throw new IllegalArgumentException("Atomic commit can't be used with " +
|
throw new IllegalArgumentException("Atomic commit can't be used with " +
|
||||||
|
@ -492,6 +510,14 @@ public class DistCpOptions {
|
||||||
throw new IllegalArgumentException("Skip CRC is valid only with update options");
|
throw new IllegalArgumentException("Skip CRC is valid only with update options");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (!syncFolder && append) {
|
||||||
|
throw new IllegalArgumentException(
|
||||||
|
"Append is valid only with update options");
|
||||||
|
}
|
||||||
|
if (skipCRC && append) {
|
||||||
|
throw new IllegalArgumentException(
|
||||||
|
"Append is disallowed when skipping CRC");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -510,6 +536,8 @@ public class DistCpOptions {
|
||||||
String.valueOf(deleteMissing));
|
String.valueOf(deleteMissing));
|
||||||
DistCpOptionSwitch.addToConf(conf, DistCpOptionSwitch.OVERWRITE,
|
DistCpOptionSwitch.addToConf(conf, DistCpOptionSwitch.OVERWRITE,
|
||||||
String.valueOf(overwrite));
|
String.valueOf(overwrite));
|
||||||
|
DistCpOptionSwitch.addToConf(conf, DistCpOptionSwitch.APPEND,
|
||||||
|
String.valueOf(append));
|
||||||
DistCpOptionSwitch.addToConf(conf, DistCpOptionSwitch.SKIP_CRC,
|
DistCpOptionSwitch.addToConf(conf, DistCpOptionSwitch.SKIP_CRC,
|
||||||
String.valueOf(skipCRC));
|
String.valueOf(skipCRC));
|
||||||
DistCpOptionSwitch.addToConf(conf, DistCpOptionSwitch.BANDWIDTH,
|
DistCpOptionSwitch.addToConf(conf, DistCpOptionSwitch.BANDWIDTH,
|
||||||
|
|
|
@ -140,6 +140,10 @@ public class OptionsParser {
|
||||||
option.setOverwrite(true);
|
option.setOverwrite(true);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (command.hasOption(DistCpOptionSwitch.APPEND.getSwitch())) {
|
||||||
|
option.setAppend(true);
|
||||||
|
}
|
||||||
|
|
||||||
if (command.hasOption(DistCpOptionSwitch.DELETE_MISSING.getSwitch())) {
|
if (command.hasOption(DistCpOptionSwitch.DELETE_MISSING.getSwitch())) {
|
||||||
option.setDeleteMissing(true);
|
option.setDeleteMissing(true);
|
||||||
}
|
}
|
||||||
|
|
|
@ -18,13 +18,20 @@
|
||||||
|
|
||||||
package org.apache.hadoop.tools.mapred;
|
package org.apache.hadoop.tools.mapred;
|
||||||
|
|
||||||
|
import java.io.FileNotFoundException;
|
||||||
|
import java.io.FileOutputStream;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.OutputStream;
|
||||||
|
import java.util.Arrays;
|
||||||
|
import java.util.EnumSet;
|
||||||
|
|
||||||
import org.apache.commons.logging.Log;
|
import org.apache.commons.logging.Log;
|
||||||
import org.apache.commons.logging.LogFactory;
|
import org.apache.commons.logging.LogFactory;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
|
import org.apache.hadoop.fs.FileChecksum;
|
||||||
import org.apache.hadoop.fs.FileStatus;
|
import org.apache.hadoop.fs.FileStatus;
|
||||||
import org.apache.hadoop.fs.FileSystem;
|
import org.apache.hadoop.fs.FileSystem;
|
||||||
import org.apache.hadoop.fs.Path;
|
import org.apache.hadoop.fs.Path;
|
||||||
import org.apache.hadoop.fs.permission.AclEntry;
|
|
||||||
import org.apache.hadoop.io.Text;
|
import org.apache.hadoop.io.Text;
|
||||||
import org.apache.hadoop.mapreduce.JobContext;
|
import org.apache.hadoop.mapreduce.JobContext;
|
||||||
import org.apache.hadoop.mapreduce.Mapper;
|
import org.apache.hadoop.mapreduce.Mapper;
|
||||||
|
@ -36,11 +43,6 @@ import org.apache.hadoop.tools.DistCpOptions.FileAttribute;
|
||||||
import org.apache.hadoop.tools.util.DistCpUtils;
|
import org.apache.hadoop.tools.util.DistCpUtils;
|
||||||
import org.apache.hadoop.util.StringUtils;
|
import org.apache.hadoop.util.StringUtils;
|
||||||
|
|
||||||
import java.io.*;
|
|
||||||
import java.util.EnumSet;
|
|
||||||
import java.util.Arrays;
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Mapper class that executes the DistCp copy operation.
|
* Mapper class that executes the DistCp copy operation.
|
||||||
* Implements the o.a.h.mapreduce.Mapper<> interface.
|
* Implements the o.a.h.mapreduce.Mapper<> interface.
|
||||||
|
@ -62,6 +64,15 @@ public class CopyMapper extends Mapper<Text, CopyListingFileStatus, Text, Text>
|
||||||
BYTESSKIPPED, // Number of bytes that were skipped from copy.
|
BYTESSKIPPED, // Number of bytes that were skipped from copy.
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Indicate the action for each file
|
||||||
|
*/
|
||||||
|
static enum FileAction {
|
||||||
|
SKIP, // Skip copying the file since it's already in the target FS
|
||||||
|
APPEND, // Only need to append new data to the file in the target FS
|
||||||
|
OVERWRITE, // Overwrite the whole file
|
||||||
|
}
|
||||||
|
|
||||||
private static Log LOG = LogFactory.getLog(CopyMapper.class);
|
private static Log LOG = LogFactory.getLog(CopyMapper.class);
|
||||||
|
|
||||||
private Configuration conf;
|
private Configuration conf;
|
||||||
|
@ -70,6 +81,7 @@ public class CopyMapper extends Mapper<Text, CopyListingFileStatus, Text, Text>
|
||||||
private boolean ignoreFailures = false;
|
private boolean ignoreFailures = false;
|
||||||
private boolean skipCrc = false;
|
private boolean skipCrc = false;
|
||||||
private boolean overWrite = false;
|
private boolean overWrite = false;
|
||||||
|
private boolean append = false;
|
||||||
private EnumSet<FileAttribute> preserve = EnumSet.noneOf(FileAttribute.class);
|
private EnumSet<FileAttribute> preserve = EnumSet.noneOf(FileAttribute.class);
|
||||||
|
|
||||||
private FileSystem targetFS = null;
|
private FileSystem targetFS = null;
|
||||||
|
@ -90,6 +102,7 @@ public class CopyMapper extends Mapper<Text, CopyListingFileStatus, Text, Text>
|
||||||
ignoreFailures = conf.getBoolean(DistCpOptionSwitch.IGNORE_FAILURES.getConfigLabel(), false);
|
ignoreFailures = conf.getBoolean(DistCpOptionSwitch.IGNORE_FAILURES.getConfigLabel(), false);
|
||||||
skipCrc = conf.getBoolean(DistCpOptionSwitch.SKIP_CRC.getConfigLabel(), false);
|
skipCrc = conf.getBoolean(DistCpOptionSwitch.SKIP_CRC.getConfigLabel(), false);
|
||||||
overWrite = conf.getBoolean(DistCpOptionSwitch.OVERWRITE.getConfigLabel(), false);
|
overWrite = conf.getBoolean(DistCpOptionSwitch.OVERWRITE.getConfigLabel(), false);
|
||||||
|
append = conf.getBoolean(DistCpOptionSwitch.APPEND.getConfigLabel(), false);
|
||||||
preserve = DistCpUtils.unpackAttributes(conf.get(DistCpOptionSwitch.
|
preserve = DistCpUtils.unpackAttributes(conf.get(DistCpOptionSwitch.
|
||||||
PRESERVE_STATUS.getConfigLabel()));
|
PRESERVE_STATUS.getConfigLabel()));
|
||||||
|
|
||||||
|
@ -224,20 +237,19 @@ public class CopyMapper extends Mapper<Text, CopyListingFileStatus, Text, Text>
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (skipFile(sourceFS, sourceCurrStatus, target)) {
|
FileAction action = checkUpdate(sourceFS, sourceCurrStatus, target);
|
||||||
|
if (action == FileAction.SKIP) {
|
||||||
LOG.info("Skipping copy of " + sourceCurrStatus.getPath()
|
LOG.info("Skipping copy of " + sourceCurrStatus.getPath()
|
||||||
+ " to " + target);
|
+ " to " + target);
|
||||||
updateSkipCounters(context, sourceCurrStatus);
|
updateSkipCounters(context, sourceCurrStatus);
|
||||||
context.write(null, new Text("SKIP: " + sourceCurrStatus.getPath()));
|
context.write(null, new Text("SKIP: " + sourceCurrStatus.getPath()));
|
||||||
}
|
} else {
|
||||||
else {
|
|
||||||
copyFileWithRetry(description, sourceCurrStatus, target, context,
|
copyFileWithRetry(description, sourceCurrStatus, target, context,
|
||||||
fileAttributes);
|
action, fileAttributes);
|
||||||
}
|
}
|
||||||
|
|
||||||
DistCpUtils.preserve(target.getFileSystem(conf), target,
|
DistCpUtils.preserve(target.getFileSystem(conf), target,
|
||||||
sourceCurrStatus, fileAttributes);
|
sourceCurrStatus, fileAttributes);
|
||||||
|
|
||||||
} catch (IOException exception) {
|
} catch (IOException exception) {
|
||||||
handleFailures(exception, sourceFileStatus, target, context);
|
handleFailures(exception, sourceFileStatus, target, context);
|
||||||
}
|
}
|
||||||
|
@ -254,14 +266,14 @@ public class CopyMapper extends Mapper<Text, CopyListingFileStatus, Text, Text>
|
||||||
return DistCpUtils.unpackAttributes(attributeString);
|
return DistCpUtils.unpackAttributes(attributeString);
|
||||||
}
|
}
|
||||||
|
|
||||||
private void copyFileWithRetry(String description, FileStatus sourceFileStatus,
|
private void copyFileWithRetry(String description,
|
||||||
Path target, Context context,
|
FileStatus sourceFileStatus, Path target, Context context,
|
||||||
EnumSet<DistCpOptions.FileAttribute> fileAttributes) throws IOException {
|
FileAction action, EnumSet<DistCpOptions.FileAttribute> fileAttributes)
|
||||||
|
throws IOException {
|
||||||
long bytesCopied;
|
long bytesCopied;
|
||||||
try {
|
try {
|
||||||
bytesCopied = (Long)new RetriableFileCopyCommand(skipCrc, description)
|
bytesCopied = (Long) new RetriableFileCopyCommand(skipCrc, description,
|
||||||
.execute(sourceFileStatus, target, context, fileAttributes);
|
action).execute(sourceFileStatus, target, context, fileAttributes);
|
||||||
} catch (Exception e) {
|
} catch (Exception e) {
|
||||||
context.setStatus("Copy Failure: " + sourceFileStatus.getPath());
|
context.setStatus("Copy Failure: " + sourceFileStatus.getPath());
|
||||||
throw new IOException("File copy failed: " + sourceFileStatus.getPath() +
|
throw new IOException("File copy failed: " + sourceFileStatus.getPath() +
|
||||||
|
@ -311,25 +323,48 @@ public class CopyMapper extends Mapper<Text, CopyListingFileStatus, Text, Text>
|
||||||
context.getCounter(counter).increment(value);
|
context.getCounter(counter).increment(value);
|
||||||
}
|
}
|
||||||
|
|
||||||
private boolean skipFile(FileSystem sourceFS, FileStatus source, Path target)
|
private FileAction checkUpdate(FileSystem sourceFS, FileStatus source,
|
||||||
throws IOException {
|
Path target) throws IOException {
|
||||||
return targetFS.exists(target)
|
final FileStatus targetFileStatus;
|
||||||
&& !overWrite
|
try {
|
||||||
&& !mustUpdate(sourceFS, source, target);
|
targetFileStatus = targetFS.getFileStatus(target);
|
||||||
|
} catch (FileNotFoundException e) {
|
||||||
|
return FileAction.OVERWRITE;
|
||||||
|
}
|
||||||
|
if (targetFileStatus != null && !overWrite) {
|
||||||
|
if (canSkip(sourceFS, source, targetFileStatus)) {
|
||||||
|
return FileAction.SKIP;
|
||||||
|
} else if (append) {
|
||||||
|
long targetLen = targetFileStatus.getLen();
|
||||||
|
if (targetLen < source.getLen()) {
|
||||||
|
FileChecksum sourceChecksum = sourceFS.getFileChecksum(
|
||||||
|
source.getPath(), targetLen);
|
||||||
|
if (sourceChecksum != null
|
||||||
|
&& sourceChecksum.equals(targetFS.getFileChecksum(target))) {
|
||||||
|
// We require that the checksum is not null. Thus currently only
|
||||||
|
// DistributedFileSystem is supported
|
||||||
|
return FileAction.APPEND;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return FileAction.OVERWRITE;
|
||||||
}
|
}
|
||||||
|
|
||||||
private boolean mustUpdate(FileSystem sourceFS, FileStatus source, Path target)
|
private boolean canSkip(FileSystem sourceFS, FileStatus source,
|
||||||
throws IOException {
|
FileStatus target) throws IOException {
|
||||||
final FileStatus targetFileStatus = targetFS.getFileStatus(target);
|
if (!syncFolders) {
|
||||||
|
return true;
|
||||||
return syncFolders
|
}
|
||||||
&& (
|
boolean sameLength = target.getLen() == source.getLen();
|
||||||
targetFileStatus.getLen() != source.getLen()
|
boolean sameBlockSize = source.getBlockSize() == target.getBlockSize()
|
||||||
|| (!skipCrc &&
|
|| !preserve.contains(FileAttribute.BLOCKSIZE);
|
||||||
!DistCpUtils.checksumsAreEqual(sourceFS,
|
if (sameLength && sameBlockSize) {
|
||||||
source.getPath(), null, targetFS, target))
|
return skipCrc ||
|
||||||
|| (source.getBlockSize() != targetFileStatus.getBlockSize() &&
|
DistCpUtils.checksumsAreEqual(sourceFS, source.getPath(), null,
|
||||||
preserve.contains(FileAttribute.BLOCKSIZE))
|
targetFS, target.getPath());
|
||||||
);
|
} else {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -18,10 +18,8 @@
|
||||||
|
|
||||||
package org.apache.hadoop.tools.mapred;
|
package org.apache.hadoop.tools.mapred;
|
||||||
|
|
||||||
import java.io.BufferedInputStream;
|
|
||||||
import java.io.BufferedOutputStream;
|
import java.io.BufferedOutputStream;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.InputStream;
|
|
||||||
import java.io.OutputStream;
|
import java.io.OutputStream;
|
||||||
import java.util.EnumSet;
|
import java.util.EnumSet;
|
||||||
|
|
||||||
|
@ -29,6 +27,8 @@ import org.apache.commons.logging.Log;
|
||||||
import org.apache.commons.logging.LogFactory;
|
import org.apache.commons.logging.LogFactory;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
import org.apache.hadoop.fs.CreateFlag;
|
import org.apache.hadoop.fs.CreateFlag;
|
||||||
|
import org.apache.hadoop.fs.FSDataInputStream;
|
||||||
|
import org.apache.hadoop.fs.FSDataOutputStream;
|
||||||
import org.apache.hadoop.fs.FileChecksum;
|
import org.apache.hadoop.fs.FileChecksum;
|
||||||
import org.apache.hadoop.fs.FileStatus;
|
import org.apache.hadoop.fs.FileStatus;
|
||||||
import org.apache.hadoop.fs.FileSystem;
|
import org.apache.hadoop.fs.FileSystem;
|
||||||
|
@ -39,6 +39,7 @@ import org.apache.hadoop.io.IOUtils;
|
||||||
import org.apache.hadoop.mapreduce.Mapper;
|
import org.apache.hadoop.mapreduce.Mapper;
|
||||||
import org.apache.hadoop.tools.DistCpConstants;
|
import org.apache.hadoop.tools.DistCpConstants;
|
||||||
import org.apache.hadoop.tools.DistCpOptions.FileAttribute;
|
import org.apache.hadoop.tools.DistCpOptions.FileAttribute;
|
||||||
|
import org.apache.hadoop.tools.mapred.CopyMapper.FileAction;
|
||||||
import org.apache.hadoop.tools.util.DistCpUtils;
|
import org.apache.hadoop.tools.util.DistCpUtils;
|
||||||
import org.apache.hadoop.tools.util.RetriableCommand;
|
import org.apache.hadoop.tools.util.RetriableCommand;
|
||||||
import org.apache.hadoop.tools.util.ThrottledInputStream;
|
import org.apache.hadoop.tools.util.ThrottledInputStream;
|
||||||
|
@ -54,13 +55,15 @@ public class RetriableFileCopyCommand extends RetriableCommand {
|
||||||
private static Log LOG = LogFactory.getLog(RetriableFileCopyCommand.class);
|
private static Log LOG = LogFactory.getLog(RetriableFileCopyCommand.class);
|
||||||
private static int BUFFER_SIZE = 8 * 1024;
|
private static int BUFFER_SIZE = 8 * 1024;
|
||||||
private boolean skipCrc = false;
|
private boolean skipCrc = false;
|
||||||
|
private FileAction action;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Constructor, taking a description of the action.
|
* Constructor, taking a description of the action.
|
||||||
* @param description Verbose description of the copy operation.
|
* @param description Verbose description of the copy operation.
|
||||||
*/
|
*/
|
||||||
public RetriableFileCopyCommand(String description) {
|
public RetriableFileCopyCommand(String description, FileAction action) {
|
||||||
super(description);
|
super(description);
|
||||||
|
this.action = action;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -68,9 +71,11 @@ public class RetriableFileCopyCommand extends RetriableCommand {
|
||||||
*
|
*
|
||||||
* @param skipCrc Whether to skip the crc check.
|
* @param skipCrc Whether to skip the crc check.
|
||||||
* @param description A verbose description of the copy operation.
|
* @param description A verbose description of the copy operation.
|
||||||
|
* @param action We should overwrite the target file or append new data to it.
|
||||||
*/
|
*/
|
||||||
public RetriableFileCopyCommand(boolean skipCrc, String description) {
|
public RetriableFileCopyCommand(boolean skipCrc, String description,
|
||||||
this(description);
|
FileAction action) {
|
||||||
|
this(description, action);
|
||||||
this.skipCrc = skipCrc;
|
this.skipCrc = skipCrc;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -96,18 +101,17 @@ public class RetriableFileCopyCommand extends RetriableCommand {
|
||||||
}
|
}
|
||||||
|
|
||||||
private long doCopy(FileStatus sourceFileStatus, Path target,
|
private long doCopy(FileStatus sourceFileStatus, Path target,
|
||||||
Mapper.Context context,
|
Mapper.Context context, EnumSet<FileAttribute> fileAttributes)
|
||||||
EnumSet<FileAttribute> fileAttributes)
|
|
||||||
throws IOException {
|
throws IOException {
|
||||||
|
final boolean toAppend = action == FileAction.APPEND;
|
||||||
Path tmpTargetPath = getTmpFile(target, context);
|
Path targetPath = toAppend ? target : getTmpFile(target, context);
|
||||||
final Configuration configuration = context.getConfiguration();
|
final Configuration configuration = context.getConfiguration();
|
||||||
FileSystem targetFS = target.getFileSystem(configuration);
|
FileSystem targetFS = target.getFileSystem(configuration);
|
||||||
|
|
||||||
try {
|
try {
|
||||||
if (LOG.isDebugEnabled()) {
|
if (LOG.isDebugEnabled()) {
|
||||||
LOG.debug("Copying " + sourceFileStatus.getPath() + " to " + target);
|
LOG.debug("Copying " + sourceFileStatus.getPath() + " to " + target);
|
||||||
LOG.debug("Tmp-file path: " + tmpTargetPath);
|
LOG.debug("Target file path: " + targetPath);
|
||||||
}
|
}
|
||||||
final Path sourcePath = sourceFileStatus.getPath();
|
final Path sourcePath = sourceFileStatus.getPath();
|
||||||
final FileSystem sourceFS = sourcePath.getFileSystem(configuration);
|
final FileSystem sourceFS = sourcePath.getFileSystem(configuration);
|
||||||
|
@ -115,22 +119,31 @@ public class RetriableFileCopyCommand extends RetriableCommand {
|
||||||
.contains(FileAttribute.CHECKSUMTYPE) ? sourceFS
|
.contains(FileAttribute.CHECKSUMTYPE) ? sourceFS
|
||||||
.getFileChecksum(sourcePath) : null;
|
.getFileChecksum(sourcePath) : null;
|
||||||
|
|
||||||
long bytesRead = copyToTmpFile(tmpTargetPath, targetFS, sourceFileStatus,
|
final long offset = action == FileAction.APPEND ? targetFS.getFileStatus(
|
||||||
context, fileAttributes, sourceChecksum);
|
target).getLen() : 0;
|
||||||
|
long bytesRead = copyToFile(targetPath, targetFS, sourceFileStatus,
|
||||||
|
offset, context, fileAttributes, sourceChecksum);
|
||||||
|
|
||||||
compareFileLengths(sourceFileStatus, tmpTargetPath, configuration,
|
compareFileLengths(sourceFileStatus, targetPath, configuration, bytesRead
|
||||||
bytesRead);
|
+ offset);
|
||||||
//At this point, src&dest lengths are same. if length==0, we skip checksum
|
//At this point, src&dest lengths are same. if length==0, we skip checksum
|
||||||
if ((bytesRead != 0) && (!skipCrc)) {
|
if ((bytesRead != 0) && (!skipCrc)) {
|
||||||
compareCheckSums(sourceFS, sourceFileStatus.getPath(), sourceChecksum,
|
compareCheckSums(sourceFS, sourceFileStatus.getPath(), sourceChecksum,
|
||||||
targetFS, tmpTargetPath);
|
targetFS, targetPath);
|
||||||
|
}
|
||||||
|
// it's not append case, thus we first write to a temporary file, rename
|
||||||
|
// it to the target path.
|
||||||
|
if (!toAppend) {
|
||||||
|
promoteTmpToTarget(targetPath, target, targetFS);
|
||||||
}
|
}
|
||||||
promoteTmpToTarget(tmpTargetPath, target, targetFS);
|
|
||||||
return bytesRead;
|
return bytesRead;
|
||||||
|
|
||||||
} finally {
|
} finally {
|
||||||
if (targetFS.exists(tmpTargetPath))
|
// note that for append case, it is possible that we append partial data
|
||||||
targetFS.delete(tmpTargetPath, false);
|
// and then fail. In that case, for the next retry, we either reuse the
|
||||||
|
// partial appended data if it is good or we overwrite the whole file
|
||||||
|
if (!toAppend && targetFS.exists(targetPath)) {
|
||||||
|
targetFS.delete(targetPath, false);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -147,29 +160,37 @@ public class RetriableFileCopyCommand extends RetriableCommand {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
private long copyToTmpFile(Path tmpTargetPath, FileSystem targetFS,
|
private long copyToFile(Path targetPath, FileSystem targetFS,
|
||||||
FileStatus sourceFileStatus, Mapper.Context context,
|
FileStatus sourceFileStatus, long sourceOffset, Mapper.Context context,
|
||||||
EnumSet<FileAttribute> fileAttributes, final FileChecksum sourceChecksum)
|
EnumSet<FileAttribute> fileAttributes, final FileChecksum sourceChecksum)
|
||||||
throws IOException {
|
throws IOException {
|
||||||
FsPermission permission = FsPermission.getFileDefault().applyUMask(
|
FsPermission permission = FsPermission.getFileDefault().applyUMask(
|
||||||
FsPermission.getUMask(targetFS.getConf()));
|
FsPermission.getUMask(targetFS.getConf()));
|
||||||
OutputStream outStream = new BufferedOutputStream(
|
final OutputStream outStream;
|
||||||
targetFS.create(tmpTargetPath, permission,
|
if (action == FileAction.OVERWRITE) {
|
||||||
EnumSet.of(CreateFlag.CREATE, CreateFlag.OVERWRITE), BUFFER_SIZE,
|
final short repl = getReplicationFactor(fileAttributes, sourceFileStatus,
|
||||||
getReplicationFactor(fileAttributes, sourceFileStatus, targetFS,
|
targetFS, targetPath);
|
||||||
tmpTargetPath),
|
final long blockSize = getBlockSize(fileAttributes, sourceFileStatus,
|
||||||
getBlockSize(fileAttributes, sourceFileStatus, targetFS,
|
targetFS, targetPath);
|
||||||
tmpTargetPath),
|
FSDataOutputStream out = targetFS.create(targetPath, permission,
|
||||||
context, getChecksumOpt(fileAttributes, sourceChecksum)));
|
EnumSet.of(CreateFlag.CREATE, CreateFlag.OVERWRITE),
|
||||||
return copyBytes(sourceFileStatus, outStream, BUFFER_SIZE, context);
|
BUFFER_SIZE, repl, blockSize, context,
|
||||||
|
getChecksumOpt(fileAttributes, sourceChecksum));
|
||||||
|
outStream = new BufferedOutputStream(out);
|
||||||
|
} else {
|
||||||
|
outStream = new BufferedOutputStream(targetFS.append(targetPath,
|
||||||
|
BUFFER_SIZE));
|
||||||
|
}
|
||||||
|
return copyBytes(sourceFileStatus, sourceOffset, outStream, BUFFER_SIZE,
|
||||||
|
context);
|
||||||
}
|
}
|
||||||
|
|
||||||
private void compareFileLengths(FileStatus sourceFileStatus, Path target,
|
private void compareFileLengths(FileStatus sourceFileStatus, Path target,
|
||||||
Configuration configuration, long bytesRead)
|
Configuration configuration, long targetLen)
|
||||||
throws IOException {
|
throws IOException {
|
||||||
final Path sourcePath = sourceFileStatus.getPath();
|
final Path sourcePath = sourceFileStatus.getPath();
|
||||||
FileSystem fs = sourcePath.getFileSystem(configuration);
|
FileSystem fs = sourcePath.getFileSystem(configuration);
|
||||||
if (fs.getFileStatus(sourcePath).getLen() != bytesRead)
|
if (fs.getFileStatus(sourcePath).getLen() != targetLen)
|
||||||
throw new IOException("Mismatch in length of source:" + sourcePath
|
throw new IOException("Mismatch in length of source:" + sourcePath
|
||||||
+ " and target:" + target);
|
+ " and target:" + target);
|
||||||
}
|
}
|
||||||
|
@ -215,8 +236,8 @@ public class RetriableFileCopyCommand extends RetriableCommand {
|
||||||
}
|
}
|
||||||
|
|
||||||
@VisibleForTesting
|
@VisibleForTesting
|
||||||
long copyBytes(FileStatus sourceFileStatus, OutputStream outStream,
|
long copyBytes(FileStatus sourceFileStatus, long sourceOffset,
|
||||||
int bufferSize, Mapper.Context context)
|
OutputStream outStream, int bufferSize, Mapper.Context context)
|
||||||
throws IOException {
|
throws IOException {
|
||||||
Path source = sourceFileStatus.getPath();
|
Path source = sourceFileStatus.getPath();
|
||||||
byte buf[] = new byte[bufferSize];
|
byte buf[] = new byte[bufferSize];
|
||||||
|
@ -225,19 +246,21 @@ public class RetriableFileCopyCommand extends RetriableCommand {
|
||||||
|
|
||||||
try {
|
try {
|
||||||
inStream = getInputStream(source, context.getConfiguration());
|
inStream = getInputStream(source, context.getConfiguration());
|
||||||
int bytesRead = readBytes(inStream, buf);
|
int bytesRead = readBytes(inStream, buf, sourceOffset);
|
||||||
while (bytesRead >= 0) {
|
while (bytesRead >= 0) {
|
||||||
totalBytesRead += bytesRead;
|
totalBytesRead += bytesRead;
|
||||||
|
if (action == FileAction.APPEND) {
|
||||||
|
sourceOffset += bytesRead;
|
||||||
|
}
|
||||||
outStream.write(buf, 0, bytesRead);
|
outStream.write(buf, 0, bytesRead);
|
||||||
updateContextStatus(totalBytesRead, context, sourceFileStatus);
|
updateContextStatus(totalBytesRead, context, sourceFileStatus);
|
||||||
bytesRead = inStream.read(buf);
|
bytesRead = readBytes(inStream, buf, sourceOffset);
|
||||||
}
|
}
|
||||||
outStream.close();
|
outStream.close();
|
||||||
outStream = null;
|
outStream = null;
|
||||||
} finally {
|
} finally {
|
||||||
IOUtils.cleanup(LOG, outStream, inStream);
|
IOUtils.cleanup(LOG, outStream, inStream);
|
||||||
}
|
}
|
||||||
|
|
||||||
return totalBytesRead;
|
return totalBytesRead;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -254,24 +277,27 @@ public class RetriableFileCopyCommand extends RetriableCommand {
|
||||||
context.setStatus(message.toString());
|
context.setStatus(message.toString());
|
||||||
}
|
}
|
||||||
|
|
||||||
private static int readBytes(InputStream inStream, byte buf[])
|
private static int readBytes(ThrottledInputStream inStream, byte buf[],
|
||||||
throws IOException {
|
long position) throws IOException {
|
||||||
try {
|
try {
|
||||||
|
if (position == 0) {
|
||||||
return inStream.read(buf);
|
return inStream.read(buf);
|
||||||
|
} else {
|
||||||
|
return inStream.read(position, buf, 0, buf.length);
|
||||||
}
|
}
|
||||||
catch (IOException e) {
|
} catch (IOException e) {
|
||||||
throw new CopyReadException(e);
|
throw new CopyReadException(e);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private static ThrottledInputStream getInputStream(Path path, Configuration conf)
|
private static ThrottledInputStream getInputStream(Path path,
|
||||||
throws IOException {
|
Configuration conf) throws IOException {
|
||||||
try {
|
try {
|
||||||
FileSystem fs = path.getFileSystem(conf);
|
FileSystem fs = path.getFileSystem(conf);
|
||||||
long bandwidthMB = conf.getInt(DistCpConstants.CONF_LABEL_BANDWIDTH_MB,
|
long bandwidthMB = conf.getInt(DistCpConstants.CONF_LABEL_BANDWIDTH_MB,
|
||||||
DistCpConstants.DEFAULT_BANDWIDTH_MB);
|
DistCpConstants.DEFAULT_BANDWIDTH_MB);
|
||||||
return new ThrottledInputStream(new BufferedInputStream(fs.open(path)),
|
FSDataInputStream in = fs.open(path);
|
||||||
bandwidthMB * 1024 * 1024);
|
return new ThrottledInputStream(in, bandwidthMB * 1024 * 1024);
|
||||||
}
|
}
|
||||||
catch (IOException e) {
|
catch (IOException e) {
|
||||||
throw new CopyReadException(e);
|
throw new CopyReadException(e);
|
||||||
|
|
|
@ -21,6 +21,11 @@ package org.apache.hadoop.tools.util;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.io.InputStream;
|
import java.io.InputStream;
|
||||||
|
|
||||||
|
import org.apache.hadoop.fs.FSDataInputStream;
|
||||||
|
import org.apache.hadoop.fs.PositionedReadable;
|
||||||
|
|
||||||
|
import com.google.common.base.Preconditions;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* The ThrottleInputStream provides bandwidth throttling on a specified
|
* The ThrottleInputStream provides bandwidth throttling on a specified
|
||||||
* InputStream. It is implemented as a wrapper on top of another InputStream
|
* InputStream. It is implemented as a wrapper on top of another InputStream
|
||||||
|
@ -90,6 +95,25 @@ public class ThrottledInputStream extends InputStream {
|
||||||
return readLen;
|
return readLen;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Read bytes starting from the specified position. This requires rawStream is
|
||||||
|
* an instance of {@link PositionedReadable}.
|
||||||
|
*/
|
||||||
|
public int read(long position, byte[] buffer, int offset, int length)
|
||||||
|
throws IOException {
|
||||||
|
if (!(rawStream instanceof PositionedReadable)) {
|
||||||
|
throw new UnsupportedOperationException(
|
||||||
|
"positioned read is not supported by the internal stream");
|
||||||
|
}
|
||||||
|
throttle();
|
||||||
|
int readLen = ((PositionedReadable) rawStream).read(position, buffer,
|
||||||
|
offset, length);
|
||||||
|
if (readLen != -1) {
|
||||||
|
bytesRead += readLen;
|
||||||
|
}
|
||||||
|
return readLen;
|
||||||
|
}
|
||||||
|
|
||||||
private void throttle() throws IOException {
|
private void throttle() throws IOException {
|
||||||
if (getBytesPerSec() > maxBytesPerSec) {
|
if (getBytesPerSec() > maxBytesPerSec) {
|
||||||
try {
|
try {
|
||||||
|
|
|
@ -18,9 +18,12 @@
|
||||||
|
|
||||||
package org.apache.hadoop.tools;
|
package org.apache.hadoop.tools;
|
||||||
|
|
||||||
|
import static org.junit.Assert.fail;
|
||||||
|
|
||||||
import org.junit.Assert;
|
import org.junit.Assert;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
import org.apache.hadoop.fs.Path;
|
import org.apache.hadoop.fs.Path;
|
||||||
|
import org.apache.hadoop.test.GenericTestUtils;
|
||||||
import org.apache.hadoop.tools.DistCpOptions.*;
|
import org.apache.hadoop.tools.DistCpOptions.*;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
|
|
||||||
|
@ -554,4 +557,45 @@ public class TestOptionsParser {
|
||||||
Assert.assertEquals(conf.get(DistCpOptionSwitch.PRESERVE_STATUS.getConfigLabel()), "U");
|
Assert.assertEquals(conf.get(DistCpOptionSwitch.PRESERVE_STATUS.getConfigLabel()), "U");
|
||||||
Assert.assertEquals(conf.getInt(DistCpOptionSwitch.BANDWIDTH.getConfigLabel(), -1), 11);
|
Assert.assertEquals(conf.getInt(DistCpOptionSwitch.BANDWIDTH.getConfigLabel(), -1), 11);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testAppendOption() {
|
||||||
|
Configuration conf = new Configuration();
|
||||||
|
Assert.assertFalse(conf.getBoolean(
|
||||||
|
DistCpOptionSwitch.APPEND.getConfigLabel(), false));
|
||||||
|
Assert.assertFalse(conf.getBoolean(
|
||||||
|
DistCpOptionSwitch.SYNC_FOLDERS.getConfigLabel(), false));
|
||||||
|
|
||||||
|
DistCpOptions options = OptionsParser.parse(new String[] { "-update",
|
||||||
|
"-append", "hdfs://localhost:8020/source/first",
|
||||||
|
"hdfs://localhost:8020/target/" });
|
||||||
|
options.appendToConf(conf);
|
||||||
|
Assert.assertTrue(conf.getBoolean(
|
||||||
|
DistCpOptionSwitch.APPEND.getConfigLabel(), false));
|
||||||
|
Assert.assertTrue(conf.getBoolean(
|
||||||
|
DistCpOptionSwitch.SYNC_FOLDERS.getConfigLabel(), false));
|
||||||
|
|
||||||
|
// make sure -append is only valid when -update is specified
|
||||||
|
try {
|
||||||
|
options = OptionsParser.parse(new String[] { "-append",
|
||||||
|
"hdfs://localhost:8020/source/first",
|
||||||
|
"hdfs://localhost:8020/target/" });
|
||||||
|
fail("Append should fail if update option is not specified");
|
||||||
|
} catch (IllegalArgumentException e) {
|
||||||
|
GenericTestUtils.assertExceptionContains(
|
||||||
|
"Append is valid only with update options", e);
|
||||||
|
}
|
||||||
|
|
||||||
|
// make sure -append is invalid when skipCrc is specified
|
||||||
|
try {
|
||||||
|
options = OptionsParser.parse(new String[] {
|
||||||
|
"-append", "-update", "-skipcrccheck",
|
||||||
|
"hdfs://localhost:8020/source/first",
|
||||||
|
"hdfs://localhost:8020/target/" });
|
||||||
|
fail("Append should fail if skipCrc option is specified");
|
||||||
|
} catch (IllegalArgumentException e) {
|
||||||
|
GenericTestUtils.assertExceptionContains(
|
||||||
|
"Append is disallowed when skipping CRC", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -25,11 +25,13 @@ import java.security.PrivilegedAction;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.EnumSet;
|
import java.util.EnumSet;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
import java.util.Random;
|
||||||
|
|
||||||
import org.apache.commons.logging.Log;
|
import org.apache.commons.logging.Log;
|
||||||
import org.apache.commons.logging.LogFactory;
|
import org.apache.commons.logging.LogFactory;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
import org.apache.hadoop.fs.CreateFlag;
|
import org.apache.hadoop.fs.CreateFlag;
|
||||||
|
import org.apache.hadoop.fs.FSDataOutputStream;
|
||||||
import org.apache.hadoop.fs.FileStatus;
|
import org.apache.hadoop.fs.FileStatus;
|
||||||
import org.apache.hadoop.fs.FileSystem;
|
import org.apache.hadoop.fs.FileSystem;
|
||||||
import org.apache.hadoop.fs.Options.ChecksumOpt;
|
import org.apache.hadoop.fs.Options.ChecksumOpt;
|
||||||
|
@ -118,6 +120,16 @@ public class TestCopyMapper {
|
||||||
touchFile(SOURCE_PATH + "/7/8/9");
|
touchFile(SOURCE_PATH + "/7/8/9");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private static void appendSourceData() throws Exception {
|
||||||
|
FileSystem fs = cluster.getFileSystem();
|
||||||
|
for (Path source : pathList) {
|
||||||
|
if (fs.getFileStatus(source).isFile()) {
|
||||||
|
// append 2048 bytes per file
|
||||||
|
appendFile(source, DEFAULT_FILE_SIZE * 2);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
private static void createSourceDataWithDifferentBlockSize() throws Exception {
|
private static void createSourceDataWithDifferentBlockSize() throws Exception {
|
||||||
mkdirs(SOURCE_PATH + "/1");
|
mkdirs(SOURCE_PATH + "/1");
|
||||||
mkdirs(SOURCE_PATH + "/2");
|
mkdirs(SOURCE_PATH + "/2");
|
||||||
|
@ -201,18 +213,64 @@ public class TestCopyMapper {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Append specified length of bytes to a given file
|
||||||
|
*/
|
||||||
|
private static void appendFile(Path p, int length) throws IOException {
|
||||||
|
byte[] toAppend = new byte[length];
|
||||||
|
Random random = new Random();
|
||||||
|
random.nextBytes(toAppend);
|
||||||
|
FSDataOutputStream out = cluster.getFileSystem().append(p);
|
||||||
|
try {
|
||||||
|
out.write(toAppend);
|
||||||
|
} finally {
|
||||||
|
IOUtils.closeStream(out);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testCopyWithDifferentChecksumType() throws Exception {
|
public void testCopyWithDifferentChecksumType() throws Exception {
|
||||||
testCopy(true);
|
testCopy(true);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Test(timeout=40000)
|
@Test(timeout=40000)
|
||||||
public void testRun() {
|
public void testRun() throws Exception {
|
||||||
testCopy(false);
|
testCopy(false);
|
||||||
}
|
}
|
||||||
|
|
||||||
private void testCopy(boolean preserveChecksum) {
|
@Test
|
||||||
try {
|
public void testCopyWithAppend() throws Exception {
|
||||||
|
final FileSystem fs = cluster.getFileSystem();
|
||||||
|
// do the first distcp
|
||||||
|
testCopy(false);
|
||||||
|
// start appending data to source
|
||||||
|
appendSourceData();
|
||||||
|
|
||||||
|
// do the distcp again with -update and -append option
|
||||||
|
CopyMapper copyMapper = new CopyMapper();
|
||||||
|
StubContext stubContext = new StubContext(getConfiguration(), null, 0);
|
||||||
|
Mapper<Text, CopyListingFileStatus, Text, Text>.Context context =
|
||||||
|
stubContext.getContext();
|
||||||
|
// Enable append
|
||||||
|
context.getConfiguration().setBoolean(
|
||||||
|
DistCpOptionSwitch.APPEND.getConfigLabel(), true);
|
||||||
|
copyMapper.setup(context);
|
||||||
|
for (Path path: pathList) {
|
||||||
|
copyMapper.map(new Text(DistCpUtils.getRelativePath(new Path(SOURCE_PATH), path)),
|
||||||
|
new CopyListingFileStatus(cluster.getFileSystem().getFileStatus(
|
||||||
|
path)), context);
|
||||||
|
}
|
||||||
|
|
||||||
|
verifyCopy(fs, false);
|
||||||
|
// verify that we only copied new appended data
|
||||||
|
Assert.assertEquals(nFiles * DEFAULT_FILE_SIZE * 2, stubContext
|
||||||
|
.getReporter().getCounter(CopyMapper.Counter.BYTESCOPIED)
|
||||||
|
.getValue());
|
||||||
|
Assert.assertEquals(pathList.size(), stubContext.getReporter().
|
||||||
|
getCounter(CopyMapper.Counter.COPY).getValue());
|
||||||
|
}
|
||||||
|
|
||||||
|
private void testCopy(boolean preserveChecksum) throws Exception {
|
||||||
deleteState();
|
deleteState();
|
||||||
if (preserveChecksum) {
|
if (preserveChecksum) {
|
||||||
createSourceDataWithDifferentChecksumType();
|
createSourceDataWithDifferentChecksumType();
|
||||||
|
@ -238,30 +296,15 @@ public class TestCopyMapper {
|
||||||
copyMapper.setup(context);
|
copyMapper.setup(context);
|
||||||
|
|
||||||
for (Path path: pathList) {
|
for (Path path: pathList) {
|
||||||
copyMapper.map(new Text(DistCpUtils.getRelativePath(new Path(SOURCE_PATH), path)),
|
copyMapper.map(
|
||||||
|
new Text(DistCpUtils.getRelativePath(new Path(SOURCE_PATH), path)),
|
||||||
new CopyListingFileStatus(fs.getFileStatus(path)), context);
|
new CopyListingFileStatus(fs.getFileStatus(path)), context);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check that the maps worked.
|
// Check that the maps worked.
|
||||||
for (Path path : pathList) {
|
verifyCopy(fs, preserveChecksum);
|
||||||
final Path targetPath = new Path(path.toString()
|
Assert.assertEquals(pathList.size(), stubContext.getReporter()
|
||||||
.replaceAll(SOURCE_PATH, TARGET_PATH));
|
.getCounter(CopyMapper.Counter.COPY).getValue());
|
||||||
Assert.assertTrue(fs.exists(targetPath));
|
|
||||||
Assert.assertTrue(fs.isFile(targetPath) == fs.isFile(path));
|
|
||||||
FileStatus sourceStatus = fs.getFileStatus(path);
|
|
||||||
FileStatus targetStatus = fs.getFileStatus(targetPath);
|
|
||||||
Assert.assertEquals(sourceStatus.getReplication(),
|
|
||||||
targetStatus.getReplication());
|
|
||||||
if (preserveChecksum) {
|
|
||||||
Assert.assertEquals(sourceStatus.getBlockSize(),
|
|
||||||
targetStatus.getBlockSize());
|
|
||||||
}
|
|
||||||
Assert.assertTrue(!fs.isFile(targetPath)
|
|
||||||
|| fs.getFileChecksum(targetPath).equals(fs.getFileChecksum(path)));
|
|
||||||
}
|
|
||||||
|
|
||||||
Assert.assertEquals(pathList.size(),
|
|
||||||
stubContext.getReporter().getCounter(CopyMapper.Counter.COPY).getValue());
|
|
||||||
if (!preserveChecksum) {
|
if (!preserveChecksum) {
|
||||||
Assert.assertEquals(nFiles * DEFAULT_FILE_SIZE, stubContext
|
Assert.assertEquals(nFiles * DEFAULT_FILE_SIZE, stubContext
|
||||||
.getReporter().getCounter(CopyMapper.Counter.BYTESCOPIED)
|
.getReporter().getCounter(CopyMapper.Counter.BYTESCOPIED)
|
||||||
|
@ -274,12 +317,28 @@ public class TestCopyMapper {
|
||||||
|
|
||||||
testCopyingExistingFiles(fs, copyMapper, context);
|
testCopyingExistingFiles(fs, copyMapper, context);
|
||||||
for (Text value : stubContext.getWriter().values()) {
|
for (Text value : stubContext.getWriter().values()) {
|
||||||
Assert.assertTrue(value.toString() + " is not skipped", value.toString().startsWith("SKIP:"));
|
Assert.assertTrue(value.toString() + " is not skipped", value
|
||||||
|
.toString().startsWith("SKIP:"));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
catch (Exception e) {
|
|
||||||
LOG.error("Unexpected exception: ", e);
|
private void verifyCopy(FileSystem fs, boolean preserveChecksum)
|
||||||
Assert.assertTrue(false);
|
throws Exception {
|
||||||
|
for (Path path : pathList) {
|
||||||
|
final Path targetPath = new Path(path.toString().replaceAll(SOURCE_PATH,
|
||||||
|
TARGET_PATH));
|
||||||
|
Assert.assertTrue(fs.exists(targetPath));
|
||||||
|
Assert.assertTrue(fs.isFile(targetPath) == fs.isFile(path));
|
||||||
|
FileStatus sourceStatus = fs.getFileStatus(path);
|
||||||
|
FileStatus targetStatus = fs.getFileStatus(targetPath);
|
||||||
|
Assert.assertEquals(sourceStatus.getReplication(),
|
||||||
|
targetStatus.getReplication());
|
||||||
|
if (preserveChecksum) {
|
||||||
|
Assert.assertEquals(sourceStatus.getBlockSize(),
|
||||||
|
targetStatus.getBlockSize());
|
||||||
|
}
|
||||||
|
Assert.assertTrue(!fs.isFile(targetPath)
|
||||||
|
|| fs.getFileChecksum(targetPath).equals(fs.getFileChecksum(path)));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -22,6 +22,7 @@ import org.apache.hadoop.conf.Configuration;
|
||||||
import org.apache.hadoop.fs.FileStatus;
|
import org.apache.hadoop.fs.FileStatus;
|
||||||
import org.apache.hadoop.fs.Path;
|
import org.apache.hadoop.fs.Path;
|
||||||
import org.apache.hadoop.mapreduce.*;
|
import org.apache.hadoop.mapreduce.*;
|
||||||
|
import org.apache.hadoop.tools.mapred.CopyMapper.FileAction;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
import static org.junit.Assert.*;
|
import static org.junit.Assert.*;
|
||||||
import static org.mockito.Mockito.*;
|
import static org.mockito.Mockito.*;
|
||||||
|
@ -48,8 +49,8 @@ public class TestRetriableFileCopyCommand {
|
||||||
|
|
||||||
Exception actualEx = null;
|
Exception actualEx = null;
|
||||||
try {
|
try {
|
||||||
new RetriableFileCopyCommand("testFailOnCloseError")
|
new RetriableFileCopyCommand("testFailOnCloseError", FileAction.OVERWRITE)
|
||||||
.copyBytes(stat, out, 512, context);
|
.copyBytes(stat, 0, out, 512, context);
|
||||||
} catch (Exception e) {
|
} catch (Exception e) {
|
||||||
actualEx = e;
|
actualEx = e;
|
||||||
}
|
}
|
||||||
|
|
|
@ -18,8 +18,6 @@
|
||||||
|
|
||||||
Yarn Scheduler Load Simulator (SLS)
|
Yarn Scheduler Load Simulator (SLS)
|
||||||
|
|
||||||
\[ {{{./index.html}Go Back}} \]
|
|
||||||
|
|
||||||
%{toc|section=1|fromDepth=0}
|
%{toc|section=1|fromDepth=0}
|
||||||
|
|
||||||
* Overview
|
* Overview
|
||||||
|
|
|
@ -90,6 +90,33 @@ Release 2.5.0 - UNRELEASED
|
||||||
|
|
||||||
YARN-2017. Merged some of the common scheduler code. (Jian He via vinodkv)
|
YARN-2017. Merged some of the common scheduler code. (Jian He via vinodkv)
|
||||||
|
|
||||||
|
YARN-2049. Added delegation-token support for the Timeline Server. (Zhijie
|
||||||
|
Shen via vinodkv)
|
||||||
|
|
||||||
|
YARN-1936. Added security support for the Timeline Client. (Zhijie Shen via
|
||||||
|
vinodkv)
|
||||||
|
|
||||||
|
YARN-1937. Added owner-only ACLs support for Timeline Client and server.
|
||||||
|
(Zhijie Shen via vinodkv)
|
||||||
|
|
||||||
|
YARN-2012. Fair Scheduler: allow default queue placement rule to take an
|
||||||
|
arbitrary queue (Ashwin Shankar via Sandy Ryza)
|
||||||
|
|
||||||
|
YARN-2059. Added admin ACLs support to Timeline Server. (Zhijie Shen via
|
||||||
|
vinodkv)
|
||||||
|
|
||||||
|
YARN-2073. Fair Scheduler: Add a utilization threshold to prevent preempting
|
||||||
|
resources when cluster is free (Karthik Kambatla via Sandy Ryza)
|
||||||
|
|
||||||
|
YARN-2071. Modified levelDB store permissions to be readable only by the
|
||||||
|
server user. (Zhijie Shen via vinodkv)
|
||||||
|
|
||||||
|
YARN-2107. Refactored timeline classes into o.a.h.y.s.timeline package. (Vinod
|
||||||
|
Kumar Vavilapalli via zjshen)
|
||||||
|
|
||||||
|
YARN-596. Use scheduling policies throughout the queue hierarchy to decide
|
||||||
|
which containers to preempt (Wei Yan via Sandy Ryza)
|
||||||
|
|
||||||
OPTIMIZATIONS
|
OPTIMIZATIONS
|
||||||
|
|
||||||
BUG FIXES
|
BUG FIXES
|
||||||
|
@ -125,6 +152,16 @@ Release 2.5.0 - UNRELEASED
|
||||||
YARN-2089. FairScheduler: QueuePlacementPolicy and QueuePlacementRule
|
YARN-2089. FairScheduler: QueuePlacementPolicy and QueuePlacementRule
|
||||||
are missing audience annotations. (Zhihai Xu via kasha)
|
are missing audience annotations. (Zhihai Xu via kasha)
|
||||||
|
|
||||||
|
YARN-2096. Race in TestRMRestart#testQueueMetricsOnRMRestart.
|
||||||
|
(Anubhav Dhoot via kasha)
|
||||||
|
|
||||||
|
YARN-2105. Fix TestFairScheduler after YARN-2012. (Ashwin Shankar via
|
||||||
|
Sandy Ryza)
|
||||||
|
|
||||||
|
YARN-2112. Fixed yarn-common's pom.xml to include jackson dependencies so
|
||||||
|
that both Timeline Server and client can access them. (Zhijie Shen via
|
||||||
|
vinodkv)
|
||||||
|
|
||||||
Release 2.4.1 - UNRELEASED
|
Release 2.4.1 - UNRELEASED
|
||||||
|
|
||||||
INCOMPATIBLE CHANGES
|
INCOMPATIBLE CHANGES
|
||||||
|
|
|
@ -0,0 +1,63 @@
|
||||||
|
/**
|
||||||
|
* 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.hadoop.yarn.api.records.timeline;
|
||||||
|
|
||||||
|
import javax.xml.bind.annotation.XmlAccessType;
|
||||||
|
import javax.xml.bind.annotation.XmlAccessorType;
|
||||||
|
import javax.xml.bind.annotation.XmlElement;
|
||||||
|
import javax.xml.bind.annotation.XmlRootElement;
|
||||||
|
|
||||||
|
import org.apache.hadoop.classification.InterfaceAudience.Public;
|
||||||
|
import org.apache.hadoop.classification.InterfaceStability.Unstable;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The response of delegation token related request
|
||||||
|
*/
|
||||||
|
@XmlRootElement(name = "delegationtoken")
|
||||||
|
@XmlAccessorType(XmlAccessType.NONE)
|
||||||
|
@Public
|
||||||
|
@Unstable
|
||||||
|
public class TimelineDelegationTokenResponse {
|
||||||
|
|
||||||
|
private String type;
|
||||||
|
private Object content;
|
||||||
|
|
||||||
|
public TimelineDelegationTokenResponse() {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
@XmlElement(name = "type")
|
||||||
|
public String getType() {
|
||||||
|
return type;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setType(String type) {
|
||||||
|
this.type = type;
|
||||||
|
}
|
||||||
|
|
||||||
|
@XmlElement(name = "content")
|
||||||
|
public Object getContent() {
|
||||||
|
return content;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setContent(Object content) {
|
||||||
|
this.content = content;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -107,6 +107,17 @@ public class TimelinePutResponse {
|
||||||
*/
|
*/
|
||||||
public static final int IO_EXCEPTION = 2;
|
public static final int IO_EXCEPTION = 2;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Error code returned if the user specifies the timeline system reserved
|
||||||
|
* filter key
|
||||||
|
*/
|
||||||
|
public static final int SYSTEM_FILTER_CONFLICT = 3;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Error code returned if the user is denied to access the timeline data
|
||||||
|
*/
|
||||||
|
public static final int ACCESS_DENIED = 4;
|
||||||
|
|
||||||
private String entityId;
|
private String entityId;
|
||||||
private String entityType;
|
private String entityType;
|
||||||
private int errorCode;
|
private int errorCode;
|
||||||
|
|
|
@ -1136,14 +1136,6 @@ public class YarnConfiguration extends Configuration {
|
||||||
public static final String DEFAULT_TIMELINE_SERVICE_WEBAPP_HTTPS_ADDRESS =
|
public static final String DEFAULT_TIMELINE_SERVICE_WEBAPP_HTTPS_ADDRESS =
|
||||||
"0.0.0.0:" + DEFAULT_TIMELINE_SERVICE_WEBAPP_HTTPS_PORT;
|
"0.0.0.0:" + DEFAULT_TIMELINE_SERVICE_WEBAPP_HTTPS_PORT;
|
||||||
|
|
||||||
/**The kerberos principal to be used for spnego filter for timeline service.*/
|
|
||||||
public static final String TIMELINE_SERVICE_WEBAPP_SPNEGO_USER_NAME_KEY =
|
|
||||||
TIMELINE_SERVICE_PREFIX + "webapp.spnego-principal";
|
|
||||||
|
|
||||||
/**The kerberos keytab to be used for spnego filter for timeline service.*/
|
|
||||||
public static final String TIMELINE_SERVICE_WEBAPP_SPNEGO_KEYTAB_FILE_KEY =
|
|
||||||
TIMELINE_SERVICE_PREFIX + "webapp.spnego-keytab-file";
|
|
||||||
|
|
||||||
/** Timeline service store class */
|
/** Timeline service store class */
|
||||||
public static final String TIMELINE_SERVICE_STORE =
|
public static final String TIMELINE_SERVICE_STORE =
|
||||||
TIMELINE_SERVICE_PREFIX + "store-class";
|
TIMELINE_SERVICE_PREFIX + "store-class";
|
||||||
|
|
|
@ -23,17 +23,17 @@ import java.io.IOException;
|
||||||
import org.apache.hadoop.classification.InterfaceAudience.Private;
|
import org.apache.hadoop.classification.InterfaceAudience.Private;
|
||||||
import org.apache.hadoop.classification.InterfaceAudience.Public;
|
import org.apache.hadoop.classification.InterfaceAudience.Public;
|
||||||
import org.apache.hadoop.classification.InterfaceStability.Unstable;
|
import org.apache.hadoop.classification.InterfaceStability.Unstable;
|
||||||
|
import org.apache.hadoop.security.token.Token;
|
||||||
import org.apache.hadoop.service.AbstractService;
|
import org.apache.hadoop.service.AbstractService;
|
||||||
import org.apache.hadoop.yarn.api.records.timeline.TimelineEntity;
|
import org.apache.hadoop.yarn.api.records.timeline.TimelineEntity;
|
||||||
import org.apache.hadoop.yarn.api.records.timeline.TimelinePutResponse;
|
import org.apache.hadoop.yarn.api.records.timeline.TimelinePutResponse;
|
||||||
import org.apache.hadoop.yarn.client.api.impl.TimelineClientImpl;
|
import org.apache.hadoop.yarn.client.api.impl.TimelineClientImpl;
|
||||||
import org.apache.hadoop.yarn.exceptions.YarnException;
|
import org.apache.hadoop.yarn.exceptions.YarnException;
|
||||||
|
import org.apache.hadoop.yarn.security.client.TimelineDelegationTokenIdentifier;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* A client library that can be used to post some information in terms of a
|
* A client library that can be used to post some information in terms of a
|
||||||
* number of conceptual entities.
|
* number of conceptual entities.
|
||||||
*
|
|
||||||
* @See Entity
|
|
||||||
*/
|
*/
|
||||||
@Public
|
@Public
|
||||||
@Unstable
|
@Unstable
|
||||||
|
@ -67,4 +67,22 @@ public abstract class TimelineClient extends AbstractService {
|
||||||
public abstract TimelinePutResponse putEntities(
|
public abstract TimelinePutResponse putEntities(
|
||||||
TimelineEntity... entities) throws IOException, YarnException;
|
TimelineEntity... entities) throws IOException, YarnException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* <p>
|
||||||
|
* Get a delegation token so as to be able to talk to the timeline server in a
|
||||||
|
* secure way.
|
||||||
|
* </p>
|
||||||
|
*
|
||||||
|
* @param renewer
|
||||||
|
* Address of the renewer who can renew these tokens when needed by
|
||||||
|
* securely talking to the timeline server
|
||||||
|
* @return a delegation token ({@link Token}) that can be used to talk to the
|
||||||
|
* timeline server
|
||||||
|
* @throws IOException
|
||||||
|
* @throws YarnException
|
||||||
|
*/
|
||||||
|
@Public
|
||||||
|
public abstract Token<TimelineDelegationTokenIdentifier> getDelegationToken(
|
||||||
|
String renewer) throws IOException, YarnException;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,252 @@
|
||||||
|
/**
|
||||||
|
* 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.hadoop.yarn.client.api.impl;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.lang.reflect.Constructor;
|
||||||
|
import java.net.HttpURLConnection;
|
||||||
|
import java.net.URL;
|
||||||
|
import java.net.URLEncoder;
|
||||||
|
import java.text.MessageFormat;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
import org.apache.hadoop.classification.InterfaceAudience.Private;
|
||||||
|
import org.apache.hadoop.classification.InterfaceStability.Unstable;
|
||||||
|
import org.apache.hadoop.security.authentication.client.AuthenticatedURL;
|
||||||
|
import org.apache.hadoop.security.authentication.client.AuthenticationException;
|
||||||
|
import org.apache.hadoop.security.authentication.client.Authenticator;
|
||||||
|
import org.apache.hadoop.security.authentication.client.KerberosAuthenticator;
|
||||||
|
import org.apache.hadoop.security.token.Token;
|
||||||
|
import org.apache.hadoop.yarn.api.records.timeline.TimelineDelegationTokenResponse;
|
||||||
|
import org.apache.hadoop.yarn.security.client.TimelineDelegationTokenIdentifier;
|
||||||
|
import org.apache.hadoop.yarn.security.client.TimelineDelegationTokenOperation;
|
||||||
|
import org.apache.hadoop.yarn.security.client.TimelineAuthenticationConsts;
|
||||||
|
import org.apache.hadoop.yarn.webapp.YarnJacksonJaxbJsonProvider;
|
||||||
|
import org.codehaus.jackson.JsonNode;
|
||||||
|
import org.codehaus.jackson.map.ObjectMapper;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A <code>KerberosAuthenticator</code> subclass that fallback to
|
||||||
|
* {@link TimelineAuthenticationConsts}.
|
||||||
|
*/
|
||||||
|
@Private
|
||||||
|
@Unstable
|
||||||
|
public class TimelineAuthenticator extends KerberosAuthenticator {
|
||||||
|
|
||||||
|
private static ObjectMapper mapper;
|
||||||
|
|
||||||
|
static {
|
||||||
|
mapper = new ObjectMapper();
|
||||||
|
YarnJacksonJaxbJsonProvider.configObjectMapper(mapper);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns the fallback authenticator if the server does not use Kerberos
|
||||||
|
* SPNEGO HTTP authentication.
|
||||||
|
*
|
||||||
|
* @return a {@link TimelineAuthenticationConsts} instance.
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
protected Authenticator getFallBackAuthenticator() {
|
||||||
|
return new TimelineAuthenticator();
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void injectDelegationToken(Map<String, String> params,
|
||||||
|
Token<?> dtToken)
|
||||||
|
throws IOException {
|
||||||
|
if (dtToken != null) {
|
||||||
|
params.put(TimelineAuthenticationConsts.DELEGATION_PARAM,
|
||||||
|
dtToken.encodeToUrlString());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private boolean hasDelegationToken(URL url) {
|
||||||
|
return url.getQuery().contains(
|
||||||
|
TimelineAuthenticationConsts.DELEGATION_PARAM + "=");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void authenticate(URL url, AuthenticatedURL.Token token)
|
||||||
|
throws IOException, AuthenticationException {
|
||||||
|
if (!hasDelegationToken(url)) {
|
||||||
|
super.authenticate(url, token);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static Token<TimelineDelegationTokenIdentifier> getDelegationToken(
|
||||||
|
URL url, AuthenticatedURL.Token token, String renewer) throws IOException {
|
||||||
|
TimelineDelegationTokenOperation op =
|
||||||
|
TimelineDelegationTokenOperation.GETDELEGATIONTOKEN;
|
||||||
|
Map<String, String> params = new HashMap<String, String>();
|
||||||
|
params.put(TimelineAuthenticationConsts.OP_PARAM, op.toString());
|
||||||
|
params.put(TimelineAuthenticationConsts.RENEWER_PARAM, renewer);
|
||||||
|
url = appendParams(url, params);
|
||||||
|
AuthenticatedURL aUrl =
|
||||||
|
new AuthenticatedURL(new TimelineAuthenticator());
|
||||||
|
try {
|
||||||
|
HttpURLConnection conn = aUrl.openConnection(url, token);
|
||||||
|
conn.setRequestMethod(op.getHttpMethod());
|
||||||
|
TimelineDelegationTokenResponse dtRes = validateAndParseResponse(conn);
|
||||||
|
if (!dtRes.getType().equals(
|
||||||
|
TimelineAuthenticationConsts.DELEGATION_TOKEN_URL)) {
|
||||||
|
throw new IOException("The response content is not expected: "
|
||||||
|
+ dtRes.getContent());
|
||||||
|
}
|
||||||
|
String tokenStr = dtRes.getContent().toString();
|
||||||
|
Token<TimelineDelegationTokenIdentifier> dToken =
|
||||||
|
new Token<TimelineDelegationTokenIdentifier>();
|
||||||
|
dToken.decodeFromUrlString(tokenStr);
|
||||||
|
return dToken;
|
||||||
|
} catch (AuthenticationException ex) {
|
||||||
|
throw new IOException(ex.toString(), ex);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static long renewDelegationToken(URL url,
|
||||||
|
AuthenticatedURL.Token token,
|
||||||
|
Token<TimelineDelegationTokenIdentifier> dToken) throws IOException {
|
||||||
|
Map<String, String> params = new HashMap<String, String>();
|
||||||
|
params.put(TimelineAuthenticationConsts.OP_PARAM,
|
||||||
|
TimelineDelegationTokenOperation.RENEWDELEGATIONTOKEN.toString());
|
||||||
|
params.put(TimelineAuthenticationConsts.TOKEN_PARAM,
|
||||||
|
dToken.encodeToUrlString());
|
||||||
|
url = appendParams(url, params);
|
||||||
|
AuthenticatedURL aUrl =
|
||||||
|
new AuthenticatedURL(new TimelineAuthenticator());
|
||||||
|
try {
|
||||||
|
HttpURLConnection conn = aUrl.openConnection(url, token);
|
||||||
|
conn.setRequestMethod(
|
||||||
|
TimelineDelegationTokenOperation.RENEWDELEGATIONTOKEN.getHttpMethod());
|
||||||
|
TimelineDelegationTokenResponse dtRes = validateAndParseResponse(conn);
|
||||||
|
if (!dtRes.getType().equals(
|
||||||
|
TimelineAuthenticationConsts.DELEGATION_TOKEN_EXPIRATION_TIME)) {
|
||||||
|
throw new IOException("The response content is not expected: "
|
||||||
|
+ dtRes.getContent());
|
||||||
|
}
|
||||||
|
return Long.valueOf(dtRes.getContent().toString());
|
||||||
|
} catch (AuthenticationException ex) {
|
||||||
|
throw new IOException(ex.toString(), ex);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void cancelDelegationToken(URL url,
|
||||||
|
AuthenticatedURL.Token token,
|
||||||
|
Token<TimelineDelegationTokenIdentifier> dToken) throws IOException {
|
||||||
|
Map<String, String> params = new HashMap<String, String>();
|
||||||
|
params.put(TimelineAuthenticationConsts.OP_PARAM,
|
||||||
|
TimelineDelegationTokenOperation.CANCELDELEGATIONTOKEN.toString());
|
||||||
|
params.put(TimelineAuthenticationConsts.TOKEN_PARAM,
|
||||||
|
dToken.encodeToUrlString());
|
||||||
|
url = appendParams(url, params);
|
||||||
|
AuthenticatedURL aUrl =
|
||||||
|
new AuthenticatedURL(new TimelineAuthenticator());
|
||||||
|
try {
|
||||||
|
HttpURLConnection conn = aUrl.openConnection(url, token);
|
||||||
|
conn.setRequestMethod(TimelineDelegationTokenOperation.CANCELDELEGATIONTOKEN
|
||||||
|
.getHttpMethod());
|
||||||
|
validateAndParseResponse(conn);
|
||||||
|
} catch (AuthenticationException ex) {
|
||||||
|
throw new IOException(ex.toString(), ex);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Convenience method that appends parameters an HTTP <code>URL</code>.
|
||||||
|
*
|
||||||
|
* @param url
|
||||||
|
* the url.
|
||||||
|
* @param params
|
||||||
|
* the query string parameters.
|
||||||
|
*
|
||||||
|
* @return a <code>URL</code>
|
||||||
|
*
|
||||||
|
* @throws IOException
|
||||||
|
* thrown if an IO error occurs.
|
||||||
|
*/
|
||||||
|
public static URL appendParams(URL url, Map<String, String> params)
|
||||||
|
throws IOException {
|
||||||
|
StringBuilder sb = new StringBuilder();
|
||||||
|
sb.append(url);
|
||||||
|
String separator = url.toString().contains("?") ? "&" : "?";
|
||||||
|
for (Map.Entry<String, String> entry : params.entrySet()) {
|
||||||
|
sb.append(separator).append(entry.getKey()).append("=").
|
||||||
|
append(URLEncoder.encode(entry.getValue(), "UTF8"));
|
||||||
|
separator = "&";
|
||||||
|
}
|
||||||
|
return new URL(sb.toString());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Validates the response of an <code>HttpURLConnection</code>. If the current
|
||||||
|
* status code is not 200, it will throw an exception with a detail message
|
||||||
|
* using Server side error messages if available. Otherwise,
|
||||||
|
* {@link TimelineDelegationTokenResponse} will be parsed and returned.
|
||||||
|
*
|
||||||
|
* @param conn
|
||||||
|
* the <code>HttpURLConnection</code>.
|
||||||
|
* @return
|
||||||
|
* @throws IOException
|
||||||
|
* thrown if the current status code is not 200 or the JSON response
|
||||||
|
* cannot be parsed correctly
|
||||||
|
*/
|
||||||
|
private static TimelineDelegationTokenResponse validateAndParseResponse(
|
||||||
|
HttpURLConnection conn) throws IOException {
|
||||||
|
int status = conn.getResponseCode();
|
||||||
|
JsonNode json = mapper.readTree(conn.getInputStream());
|
||||||
|
if (status == HttpURLConnection.HTTP_OK) {
|
||||||
|
return mapper.readValue(json, TimelineDelegationTokenResponse.class);
|
||||||
|
} else {
|
||||||
|
// If the status code is not 200, some thing wrong should happen at the
|
||||||
|
// server side, the JSON content is going to contain exception details.
|
||||||
|
// We can use the JSON content to reconstruct the exception object.
|
||||||
|
try {
|
||||||
|
String message =
|
||||||
|
json.get(TimelineAuthenticationConsts.ERROR_MESSAGE_JSON)
|
||||||
|
.getTextValue();
|
||||||
|
String exception =
|
||||||
|
json.get(TimelineAuthenticationConsts.ERROR_EXCEPTION_JSON)
|
||||||
|
.getTextValue();
|
||||||
|
String className =
|
||||||
|
json.get(TimelineAuthenticationConsts.ERROR_CLASSNAME_JSON)
|
||||||
|
.getTextValue();
|
||||||
|
|
||||||
|
try {
|
||||||
|
ClassLoader cl = TimelineAuthenticator.class.getClassLoader();
|
||||||
|
Class<?> klass = cl.loadClass(className);
|
||||||
|
Constructor<?> constr = klass.getConstructor(String.class);
|
||||||
|
throw (IOException) constr.newInstance(message);
|
||||||
|
} catch (IOException ex) {
|
||||||
|
throw ex;
|
||||||
|
} catch (Exception ex) {
|
||||||
|
throw new IOException(MessageFormat.format("{0} - {1}", exception,
|
||||||
|
message));
|
||||||
|
}
|
||||||
|
} catch (IOException ex) {
|
||||||
|
if (ex.getCause() instanceof IOException) {
|
||||||
|
throw (IOException) ex.getCause();
|
||||||
|
}
|
||||||
|
throw new IOException(
|
||||||
|
MessageFormat.format("HTTP status [{0}], {1}",
|
||||||
|
status, conn.getResponseMessage()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -18,24 +18,43 @@
|
||||||
|
|
||||||
package org.apache.hadoop.yarn.client.api.impl;
|
package org.apache.hadoop.yarn.client.api.impl;
|
||||||
|
|
||||||
|
import java.io.File;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
|
import java.net.HttpURLConnection;
|
||||||
import java.net.URI;
|
import java.net.URI;
|
||||||
|
import java.net.URL;
|
||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
import javax.ws.rs.core.MediaType;
|
import javax.ws.rs.core.MediaType;
|
||||||
|
|
||||||
|
import org.apache.commons.cli.CommandLine;
|
||||||
|
import org.apache.commons.cli.GnuParser;
|
||||||
|
import org.apache.commons.cli.HelpFormatter;
|
||||||
|
import org.apache.commons.cli.Options;
|
||||||
import org.apache.commons.logging.Log;
|
import org.apache.commons.logging.Log;
|
||||||
import org.apache.commons.logging.LogFactory;
|
import org.apache.commons.logging.LogFactory;
|
||||||
import org.apache.hadoop.classification.InterfaceAudience.Private;
|
import org.apache.hadoop.classification.InterfaceAudience.Private;
|
||||||
import org.apache.hadoop.classification.InterfaceStability.Unstable;
|
import org.apache.hadoop.classification.InterfaceStability.Unstable;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
|
import org.apache.hadoop.io.Text;
|
||||||
|
import org.apache.hadoop.security.UserGroupInformation;
|
||||||
|
import org.apache.hadoop.security.authentication.client.AuthenticatedURL;
|
||||||
|
import org.apache.hadoop.security.authentication.client.AuthenticationException;
|
||||||
|
import org.apache.hadoop.security.token.Token;
|
||||||
import org.apache.hadoop.yarn.api.records.timeline.TimelineEntities;
|
import org.apache.hadoop.yarn.api.records.timeline.TimelineEntities;
|
||||||
import org.apache.hadoop.yarn.api.records.timeline.TimelineEntity;
|
import org.apache.hadoop.yarn.api.records.timeline.TimelineEntity;
|
||||||
import org.apache.hadoop.yarn.api.records.timeline.TimelinePutResponse;
|
import org.apache.hadoop.yarn.api.records.timeline.TimelinePutResponse;
|
||||||
import org.apache.hadoop.yarn.client.api.TimelineClient;
|
import org.apache.hadoop.yarn.client.api.TimelineClient;
|
||||||
import org.apache.hadoop.yarn.conf.YarnConfiguration;
|
import org.apache.hadoop.yarn.conf.YarnConfiguration;
|
||||||
import org.apache.hadoop.yarn.exceptions.YarnException;
|
import org.apache.hadoop.yarn.exceptions.YarnException;
|
||||||
|
import org.apache.hadoop.yarn.exceptions.YarnRuntimeException;
|
||||||
|
import org.apache.hadoop.yarn.security.client.TimelineDelegationTokenIdentifier;
|
||||||
|
import org.apache.hadoop.yarn.security.client.TimelineDelegationTokenSelector;
|
||||||
|
import org.apache.hadoop.yarn.util.timeline.TimelineUtils;
|
||||||
import org.apache.hadoop.yarn.webapp.YarnJacksonJaxbJsonProvider;
|
import org.apache.hadoop.yarn.webapp.YarnJacksonJaxbJsonProvider;
|
||||||
|
import org.codehaus.jackson.map.ObjectMapper;
|
||||||
|
|
||||||
import com.google.common.annotations.VisibleForTesting;
|
import com.google.common.annotations.VisibleForTesting;
|
||||||
import com.google.common.base.Joiner;
|
import com.google.common.base.Joiner;
|
||||||
|
@ -44,6 +63,8 @@ import com.sun.jersey.api.client.ClientResponse;
|
||||||
import com.sun.jersey.api.client.WebResource;
|
import com.sun.jersey.api.client.WebResource;
|
||||||
import com.sun.jersey.api.client.config.ClientConfig;
|
import com.sun.jersey.api.client.config.ClientConfig;
|
||||||
import com.sun.jersey.api.client.config.DefaultClientConfig;
|
import com.sun.jersey.api.client.config.DefaultClientConfig;
|
||||||
|
import com.sun.jersey.client.urlconnection.HttpURLConnectionFactory;
|
||||||
|
import com.sun.jersey.client.urlconnection.URLConnectionClientHandler;
|
||||||
|
|
||||||
@Private
|
@Private
|
||||||
@Unstable
|
@Unstable
|
||||||
|
@ -52,17 +73,30 @@ public class TimelineClientImpl extends TimelineClient {
|
||||||
private static final Log LOG = LogFactory.getLog(TimelineClientImpl.class);
|
private static final Log LOG = LogFactory.getLog(TimelineClientImpl.class);
|
||||||
private static final String RESOURCE_URI_STR = "/ws/v1/timeline/";
|
private static final String RESOURCE_URI_STR = "/ws/v1/timeline/";
|
||||||
private static final Joiner JOINER = Joiner.on("");
|
private static final Joiner JOINER = Joiner.on("");
|
||||||
|
private static Options opts;
|
||||||
|
static {
|
||||||
|
opts = new Options();
|
||||||
|
opts.addOption("put", true, "Put the TimelineEntities in a JSON file");
|
||||||
|
opts.getOption("put").setArgName("Path to the JSON file");
|
||||||
|
opts.addOption("help", false, "Print usage");
|
||||||
|
}
|
||||||
|
|
||||||
private Client client;
|
private Client client;
|
||||||
private URI resURI;
|
private URI resURI;
|
||||||
private boolean isEnabled;
|
private boolean isEnabled;
|
||||||
|
private TimelineAuthenticatedURLConnectionFactory urlFactory;
|
||||||
|
|
||||||
public TimelineClientImpl() {
|
public TimelineClientImpl() {
|
||||||
super(TimelineClientImpl.class.getName());
|
super(TimelineClientImpl.class.getName());
|
||||||
ClientConfig cc = new DefaultClientConfig();
|
ClientConfig cc = new DefaultClientConfig();
|
||||||
cc.getClasses().add(YarnJacksonJaxbJsonProvider.class);
|
cc.getClasses().add(YarnJacksonJaxbJsonProvider.class);
|
||||||
|
if (UserGroupInformation.isSecurityEnabled()) {
|
||||||
|
urlFactory = new TimelineAuthenticatedURLConnectionFactory();
|
||||||
|
client = new Client(new URLConnectionClientHandler(urlFactory), cc);
|
||||||
|
} else {
|
||||||
client = Client.create(cc);
|
client = Client.create(cc);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
protected void serviceInit(Configuration conf) throws Exception {
|
protected void serviceInit(Configuration conf) throws Exception {
|
||||||
isEnabled = conf.getBoolean(
|
isEnabled = conf.getBoolean(
|
||||||
|
@ -83,6 +117,9 @@ public class TimelineClientImpl extends TimelineClient {
|
||||||
YarnConfiguration.DEFAULT_TIMELINE_SERVICE_WEBAPP_ADDRESS),
|
YarnConfiguration.DEFAULT_TIMELINE_SERVICE_WEBAPP_ADDRESS),
|
||||||
RESOURCE_URI_STR));
|
RESOURCE_URI_STR));
|
||||||
}
|
}
|
||||||
|
if (UserGroupInformation.isSecurityEnabled()) {
|
||||||
|
urlFactory.setService(TimelineUtils.buildTimelineTokenService(conf));
|
||||||
|
}
|
||||||
LOG.info("Timeline service address: " + resURI);
|
LOG.info("Timeline service address: " + resURI);
|
||||||
}
|
}
|
||||||
super.serviceInit(conf);
|
super.serviceInit(conf);
|
||||||
|
@ -124,6 +161,13 @@ public class TimelineClientImpl extends TimelineClient {
|
||||||
return resp.getEntity(TimelinePutResponse.class);
|
return resp.getEntity(TimelinePutResponse.class);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Token<TimelineDelegationTokenIdentifier> getDelegationToken(
|
||||||
|
String renewer) throws IOException, YarnException {
|
||||||
|
return TimelineAuthenticator.getDelegationToken(resURI.toURL(),
|
||||||
|
urlFactory.token, renewer);
|
||||||
|
}
|
||||||
|
|
||||||
@Private
|
@Private
|
||||||
@VisibleForTesting
|
@VisibleForTesting
|
||||||
public ClientResponse doPostingEntities(TimelineEntities entities) {
|
public ClientResponse doPostingEntities(TimelineEntities entities) {
|
||||||
|
@ -133,4 +177,138 @@ public class TimelineClientImpl extends TimelineClient {
|
||||||
.post(ClientResponse.class, entities);
|
.post(ClientResponse.class, entities);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private static class TimelineAuthenticatedURLConnectionFactory
|
||||||
|
implements HttpURLConnectionFactory {
|
||||||
|
|
||||||
|
private AuthenticatedURL.Token token;
|
||||||
|
private TimelineAuthenticator authenticator;
|
||||||
|
private Token<TimelineDelegationTokenIdentifier> dToken;
|
||||||
|
private Text service;
|
||||||
|
|
||||||
|
public TimelineAuthenticatedURLConnectionFactory() {
|
||||||
|
token = new AuthenticatedURL.Token();
|
||||||
|
authenticator = new TimelineAuthenticator();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public HttpURLConnection getHttpURLConnection(URL url) throws IOException {
|
||||||
|
try {
|
||||||
|
if (dToken == null) {
|
||||||
|
//TODO: need to take care of the renew case
|
||||||
|
dToken = selectToken();
|
||||||
|
if (LOG.isDebugEnabled()) {
|
||||||
|
LOG.debug("Timeline delegation token: " + dToken.toString());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (dToken != null) {
|
||||||
|
Map<String, String> params = new HashMap<String, String>();
|
||||||
|
TimelineAuthenticator.injectDelegationToken(params, dToken);
|
||||||
|
url = TimelineAuthenticator.appendParams(url, params);
|
||||||
|
if (LOG.isDebugEnabled()) {
|
||||||
|
LOG.debug("URL with delegation token: " + url);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return new AuthenticatedURL(authenticator).openConnection(url, token);
|
||||||
|
} catch (AuthenticationException e) {
|
||||||
|
LOG.error("Authentication failed when openning connection [" + url
|
||||||
|
+ "] with token [" + token + "].", e);
|
||||||
|
throw new IOException(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private Token<TimelineDelegationTokenIdentifier> selectToken() {
|
||||||
|
UserGroupInformation ugi;
|
||||||
|
try {
|
||||||
|
ugi = UserGroupInformation.getCurrentUser();
|
||||||
|
} catch (IOException e) {
|
||||||
|
String msg = "Error when getting the current user";
|
||||||
|
LOG.error(msg, e);
|
||||||
|
throw new YarnRuntimeException(msg, e);
|
||||||
|
}
|
||||||
|
TimelineDelegationTokenSelector tokenSelector =
|
||||||
|
new TimelineDelegationTokenSelector();
|
||||||
|
return tokenSelector.selectToken(
|
||||||
|
service, ugi.getCredentials().getAllTokens());
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setService(Text service) {
|
||||||
|
this.service = service;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void main(String[] argv) throws Exception {
|
||||||
|
CommandLine cliParser = new GnuParser().parse(opts, argv);
|
||||||
|
if (cliParser.hasOption("put")) {
|
||||||
|
String path = cliParser.getOptionValue("put");
|
||||||
|
if (path != null && path.length() > 0) {
|
||||||
|
putTimelineEntitiesInJSONFile(path);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
printUsage();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Put timeline data in a JSON file via command line.
|
||||||
|
*
|
||||||
|
* @param path
|
||||||
|
* path to the {@link TimelineEntities} JSON file
|
||||||
|
*/
|
||||||
|
private static void putTimelineEntitiesInJSONFile(String path) {
|
||||||
|
File jsonFile = new File(path);
|
||||||
|
if (!jsonFile.exists()) {
|
||||||
|
System.out.println("Error: File [" + jsonFile.getAbsolutePath()
|
||||||
|
+ "] doesn't exist");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
ObjectMapper mapper = new ObjectMapper();
|
||||||
|
YarnJacksonJaxbJsonProvider.configObjectMapper(mapper);
|
||||||
|
TimelineEntities entities = null;
|
||||||
|
try {
|
||||||
|
entities = mapper.readValue(jsonFile, TimelineEntities.class);
|
||||||
|
} catch (Exception e) {
|
||||||
|
System.err.println("Error: " + e.getMessage());
|
||||||
|
e.printStackTrace(System.err);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
Configuration conf = new YarnConfiguration();
|
||||||
|
TimelineClient client = TimelineClient.createTimelineClient();
|
||||||
|
client.init(conf);
|
||||||
|
client.start();
|
||||||
|
try {
|
||||||
|
if (UserGroupInformation.isSecurityEnabled()
|
||||||
|
&& conf.getBoolean(YarnConfiguration.TIMELINE_SERVICE_ENABLED, false)) {
|
||||||
|
Token<TimelineDelegationTokenIdentifier> token =
|
||||||
|
client.getDelegationToken(
|
||||||
|
UserGroupInformation.getCurrentUser().getUserName());
|
||||||
|
UserGroupInformation.getCurrentUser().addToken(token);
|
||||||
|
}
|
||||||
|
TimelinePutResponse response = client.putEntities(
|
||||||
|
entities.getEntities().toArray(
|
||||||
|
new TimelineEntity[entities.getEntities().size()]));
|
||||||
|
if (response.getErrors().size() == 0) {
|
||||||
|
System.out.println("Timeline data is successfully put");
|
||||||
|
} else {
|
||||||
|
for (TimelinePutResponse.TimelinePutError error : response.getErrors()) {
|
||||||
|
System.out.println("TimelineEntity [" + error.getEntityType() + ":" +
|
||||||
|
error.getEntityId() + "] is not successfully put. Error code: " +
|
||||||
|
error.getErrorCode());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (Exception e) {
|
||||||
|
System.err.println("Error: " + e.getMessage());
|
||||||
|
e.printStackTrace(System.err);
|
||||||
|
} finally {
|
||||||
|
client.stop();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Helper function to print out usage
|
||||||
|
*/
|
||||||
|
private static void printUsage() {
|
||||||
|
new HelpFormatter().printHelp("TimelineClient", opts);
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -19,6 +19,7 @@
|
||||||
package org.apache.hadoop.yarn.client.api.impl;
|
package org.apache.hadoop.yarn.client.api.impl;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
|
import java.nio.ByteBuffer;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.EnumSet;
|
import java.util.EnumSet;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
@ -29,8 +30,13 @@ import org.apache.commons.logging.LogFactory;
|
||||||
import org.apache.hadoop.classification.InterfaceAudience.Private;
|
import org.apache.hadoop.classification.InterfaceAudience.Private;
|
||||||
import org.apache.hadoop.classification.InterfaceStability.Unstable;
|
import org.apache.hadoop.classification.InterfaceStability.Unstable;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
|
import org.apache.hadoop.io.DataInputByteBuffer;
|
||||||
|
import org.apache.hadoop.io.DataOutputBuffer;
|
||||||
import org.apache.hadoop.io.Text;
|
import org.apache.hadoop.io.Text;
|
||||||
import org.apache.hadoop.ipc.RPC;
|
import org.apache.hadoop.ipc.RPC;
|
||||||
|
import org.apache.hadoop.security.Credentials;
|
||||||
|
import org.apache.hadoop.security.UserGroupInformation;
|
||||||
|
import org.apache.hadoop.security.token.TokenIdentifier;
|
||||||
import org.apache.hadoop.yarn.api.ApplicationClientProtocol;
|
import org.apache.hadoop.yarn.api.ApplicationClientProtocol;
|
||||||
import org.apache.hadoop.yarn.api.protocolrecords.GetApplicationAttemptReportRequest;
|
import org.apache.hadoop.yarn.api.protocolrecords.GetApplicationAttemptReportRequest;
|
||||||
import org.apache.hadoop.yarn.api.protocolrecords.GetApplicationAttemptReportResponse;
|
import org.apache.hadoop.yarn.api.protocolrecords.GetApplicationAttemptReportResponse;
|
||||||
|
@ -64,6 +70,7 @@ import org.apache.hadoop.yarn.api.records.ApplicationId;
|
||||||
import org.apache.hadoop.yarn.api.records.ApplicationReport;
|
import org.apache.hadoop.yarn.api.records.ApplicationReport;
|
||||||
import org.apache.hadoop.yarn.api.records.ApplicationSubmissionContext;
|
import org.apache.hadoop.yarn.api.records.ApplicationSubmissionContext;
|
||||||
import org.apache.hadoop.yarn.api.records.ContainerId;
|
import org.apache.hadoop.yarn.api.records.ContainerId;
|
||||||
|
import org.apache.hadoop.yarn.api.records.ContainerLaunchContext;
|
||||||
import org.apache.hadoop.yarn.api.records.ContainerReport;
|
import org.apache.hadoop.yarn.api.records.ContainerReport;
|
||||||
import org.apache.hadoop.yarn.api.records.NodeReport;
|
import org.apache.hadoop.yarn.api.records.NodeReport;
|
||||||
import org.apache.hadoop.yarn.api.records.NodeState;
|
import org.apache.hadoop.yarn.api.records.NodeState;
|
||||||
|
@ -74,6 +81,7 @@ import org.apache.hadoop.yarn.api.records.YarnApplicationState;
|
||||||
import org.apache.hadoop.yarn.api.records.YarnClusterMetrics;
|
import org.apache.hadoop.yarn.api.records.YarnClusterMetrics;
|
||||||
import org.apache.hadoop.yarn.client.ClientRMProxy;
|
import org.apache.hadoop.yarn.client.ClientRMProxy;
|
||||||
import org.apache.hadoop.yarn.client.api.AHSClient;
|
import org.apache.hadoop.yarn.client.api.AHSClient;
|
||||||
|
import org.apache.hadoop.yarn.client.api.TimelineClient;
|
||||||
import org.apache.hadoop.yarn.client.api.YarnClient;
|
import org.apache.hadoop.yarn.client.api.YarnClient;
|
||||||
import org.apache.hadoop.yarn.client.api.YarnClientApplication;
|
import org.apache.hadoop.yarn.client.api.YarnClientApplication;
|
||||||
import org.apache.hadoop.yarn.conf.YarnConfiguration;
|
import org.apache.hadoop.yarn.conf.YarnConfiguration;
|
||||||
|
@ -82,8 +90,10 @@ import org.apache.hadoop.yarn.exceptions.ApplicationNotFoundException;
|
||||||
import org.apache.hadoop.yarn.exceptions.YarnException;
|
import org.apache.hadoop.yarn.exceptions.YarnException;
|
||||||
import org.apache.hadoop.yarn.exceptions.YarnRuntimeException;
|
import org.apache.hadoop.yarn.exceptions.YarnRuntimeException;
|
||||||
import org.apache.hadoop.yarn.security.AMRMTokenIdentifier;
|
import org.apache.hadoop.yarn.security.AMRMTokenIdentifier;
|
||||||
|
import org.apache.hadoop.yarn.security.client.TimelineDelegationTokenIdentifier;
|
||||||
import org.apache.hadoop.yarn.util.ConverterUtils;
|
import org.apache.hadoop.yarn.util.ConverterUtils;
|
||||||
import org.apache.hadoop.yarn.util.Records;
|
import org.apache.hadoop.yarn.util.Records;
|
||||||
|
import org.apache.hadoop.yarn.util.timeline.TimelineUtils;
|
||||||
|
|
||||||
import com.google.common.annotations.VisibleForTesting;
|
import com.google.common.annotations.VisibleForTesting;
|
||||||
|
|
||||||
|
@ -97,8 +107,11 @@ public class YarnClientImpl extends YarnClient {
|
||||||
protected long submitPollIntervalMillis;
|
protected long submitPollIntervalMillis;
|
||||||
private long asyncApiPollIntervalMillis;
|
private long asyncApiPollIntervalMillis;
|
||||||
private long asyncApiPollTimeoutMillis;
|
private long asyncApiPollTimeoutMillis;
|
||||||
protected AHSClient historyClient;
|
private AHSClient historyClient;
|
||||||
private boolean historyServiceEnabled;
|
private boolean historyServiceEnabled;
|
||||||
|
protected TimelineClient timelineClient;
|
||||||
|
protected Text timelineService;
|
||||||
|
protected boolean timelineServiceEnabled;
|
||||||
|
|
||||||
private static final String ROOT = "root";
|
private static final String ROOT = "root";
|
||||||
|
|
||||||
|
@ -126,10 +139,17 @@ public class YarnClientImpl extends YarnClient {
|
||||||
if (conf.getBoolean(YarnConfiguration.APPLICATION_HISTORY_ENABLED,
|
if (conf.getBoolean(YarnConfiguration.APPLICATION_HISTORY_ENABLED,
|
||||||
YarnConfiguration.DEFAULT_APPLICATION_HISTORY_ENABLED)) {
|
YarnConfiguration.DEFAULT_APPLICATION_HISTORY_ENABLED)) {
|
||||||
historyServiceEnabled = true;
|
historyServiceEnabled = true;
|
||||||
historyClient = AHSClientImpl.createAHSClient();
|
historyClient = AHSClient.createAHSClient();
|
||||||
historyClient.init(getConfig());
|
historyClient.init(conf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (conf.getBoolean(YarnConfiguration.TIMELINE_SERVICE_ENABLED,
|
||||||
|
YarnConfiguration.DEFAULT_TIMELINE_SERVICE_ENABLED)) {
|
||||||
|
timelineServiceEnabled = true;
|
||||||
|
timelineClient = TimelineClient.createTimelineClient();
|
||||||
|
timelineClient.init(conf);
|
||||||
|
timelineService = TimelineUtils.buildTimelineTokenService(conf);
|
||||||
|
}
|
||||||
super.serviceInit(conf);
|
super.serviceInit(conf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -141,6 +161,9 @@ public class YarnClientImpl extends YarnClient {
|
||||||
if (historyServiceEnabled) {
|
if (historyServiceEnabled) {
|
||||||
historyClient.start();
|
historyClient.start();
|
||||||
}
|
}
|
||||||
|
if (timelineServiceEnabled) {
|
||||||
|
timelineClient.start();
|
||||||
|
}
|
||||||
} catch (IOException e) {
|
} catch (IOException e) {
|
||||||
throw new YarnRuntimeException(e);
|
throw new YarnRuntimeException(e);
|
||||||
}
|
}
|
||||||
|
@ -155,6 +178,9 @@ public class YarnClientImpl extends YarnClient {
|
||||||
if (historyServiceEnabled) {
|
if (historyServiceEnabled) {
|
||||||
historyClient.stop();
|
historyClient.stop();
|
||||||
}
|
}
|
||||||
|
if (timelineServiceEnabled) {
|
||||||
|
timelineClient.stop();
|
||||||
|
}
|
||||||
super.serviceStop();
|
super.serviceStop();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -189,6 +215,12 @@ public class YarnClientImpl extends YarnClient {
|
||||||
Records.newRecord(SubmitApplicationRequest.class);
|
Records.newRecord(SubmitApplicationRequest.class);
|
||||||
request.setApplicationSubmissionContext(appContext);
|
request.setApplicationSubmissionContext(appContext);
|
||||||
|
|
||||||
|
// Automatically add the timeline DT into the CLC
|
||||||
|
// Only when the security and the timeline service are both enabled
|
||||||
|
if (isSecurityEnabled() && timelineServiceEnabled) {
|
||||||
|
addTimelineDelegationToken(appContext.getAMContainerSpec());
|
||||||
|
}
|
||||||
|
|
||||||
//TODO: YARN-1763:Handle RM failovers during the submitApplication call.
|
//TODO: YARN-1763:Handle RM failovers during the submitApplication call.
|
||||||
rmClient.submitApplication(request);
|
rmClient.submitApplication(request);
|
||||||
|
|
||||||
|
@ -238,6 +270,48 @@ public class YarnClientImpl extends YarnClient {
|
||||||
return applicationId;
|
return applicationId;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private void addTimelineDelegationToken(
|
||||||
|
ContainerLaunchContext clc) throws YarnException, IOException {
|
||||||
|
org.apache.hadoop.security.token.Token<TimelineDelegationTokenIdentifier> timelineDelegationToken =
|
||||||
|
timelineClient.getDelegationToken(
|
||||||
|
UserGroupInformation.getCurrentUser().getUserName());
|
||||||
|
if (timelineDelegationToken == null) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
Credentials credentials = new Credentials();
|
||||||
|
DataInputByteBuffer dibb = new DataInputByteBuffer();
|
||||||
|
ByteBuffer tokens = clc.getTokens();
|
||||||
|
if (tokens != null) {
|
||||||
|
dibb.reset(tokens);
|
||||||
|
credentials.readTokenStorageStream(dibb);
|
||||||
|
tokens.rewind();
|
||||||
|
}
|
||||||
|
// If the timeline delegation token is already in the CLC, no need to add
|
||||||
|
// one more
|
||||||
|
for (org.apache.hadoop.security.token.Token<? extends TokenIdentifier> token : credentials
|
||||||
|
.getAllTokens()) {
|
||||||
|
TokenIdentifier tokenIdentifier = token.decodeIdentifier();
|
||||||
|
if (tokenIdentifier instanceof TimelineDelegationTokenIdentifier) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
credentials.addToken(timelineService, timelineDelegationToken);
|
||||||
|
if (LOG.isDebugEnabled()) {
|
||||||
|
LOG.debug("Add timline delegation token into credentials: "
|
||||||
|
+ timelineDelegationToken);
|
||||||
|
}
|
||||||
|
DataOutputBuffer dob = new DataOutputBuffer();
|
||||||
|
credentials.writeTokenStorageToStream(dob);
|
||||||
|
tokens = ByteBuffer.wrap(dob.getData(), 0, dob.getLength());
|
||||||
|
clc.setTokens(tokens);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Private
|
||||||
|
@VisibleForTesting
|
||||||
|
protected boolean isSecurityEnabled() {
|
||||||
|
return UserGroupInformation.isSecurityEnabled();
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public void killApplication(ApplicationId applicationId)
|
public void killApplication(ApplicationId applicationId)
|
||||||
throws YarnException, IOException {
|
throws YarnException, IOException {
|
||||||
|
|
|
@ -25,19 +25,26 @@ import static org.mockito.Mockito.verify;
|
||||||
import static org.mockito.Mockito.when;
|
import static org.mockito.Mockito.when;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
|
import java.nio.ByteBuffer;
|
||||||
import java.security.PrivilegedExceptionAction;
|
import java.security.PrivilegedExceptionAction;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
|
import java.util.Collection;
|
||||||
import java.util.EnumSet;
|
import java.util.EnumSet;
|
||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
import java.util.HashSet;
|
import java.util.HashSet;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Set;
|
import java.util.Set;
|
||||||
|
|
||||||
import org.junit.Assert;
|
|
||||||
|
|
||||||
import org.apache.commons.io.IOUtils;
|
import org.apache.commons.io.IOUtils;
|
||||||
import org.apache.hadoop.conf.Configuration;
|
import org.apache.hadoop.conf.Configuration;
|
||||||
|
import org.apache.hadoop.io.DataInputByteBuffer;
|
||||||
|
import org.apache.hadoop.io.DataOutputBuffer;
|
||||||
|
import org.apache.hadoop.security.Credentials;
|
||||||
|
import org.apache.hadoop.security.SecurityUtil;
|
||||||
import org.apache.hadoop.security.UserGroupInformation;
|
import org.apache.hadoop.security.UserGroupInformation;
|
||||||
|
import org.apache.hadoop.security.UserGroupInformation.AuthenticationMethod;
|
||||||
|
import org.apache.hadoop.security.token.Token;
|
||||||
|
import org.apache.hadoop.security.token.TokenIdentifier;
|
||||||
import org.apache.hadoop.yarn.api.ApplicationClientProtocol;
|
import org.apache.hadoop.yarn.api.ApplicationClientProtocol;
|
||||||
import org.apache.hadoop.yarn.api.protocolrecords.GetApplicationAttemptReportRequest;
|
import org.apache.hadoop.yarn.api.protocolrecords.GetApplicationAttemptReportRequest;
|
||||||
import org.apache.hadoop.yarn.api.protocolrecords.GetApplicationAttemptReportResponse;
|
import org.apache.hadoop.yarn.api.protocolrecords.GetApplicationAttemptReportResponse;
|
||||||
|
@ -69,19 +76,23 @@ import org.apache.hadoop.yarn.api.records.Priority;
|
||||||
import org.apache.hadoop.yarn.api.records.Resource;
|
import org.apache.hadoop.yarn.api.records.Resource;
|
||||||
import org.apache.hadoop.yarn.api.records.YarnApplicationAttemptState;
|
import org.apache.hadoop.yarn.api.records.YarnApplicationAttemptState;
|
||||||
import org.apache.hadoop.yarn.api.records.YarnApplicationState;
|
import org.apache.hadoop.yarn.api.records.YarnApplicationState;
|
||||||
|
import org.apache.hadoop.yarn.client.api.TimelineClient;
|
||||||
import org.apache.hadoop.yarn.client.api.YarnClient;
|
import org.apache.hadoop.yarn.client.api.YarnClient;
|
||||||
import org.apache.hadoop.yarn.client.api.YarnClientApplication;
|
import org.apache.hadoop.yarn.client.api.YarnClientApplication;
|
||||||
import org.apache.hadoop.yarn.conf.YarnConfiguration;
|
import org.apache.hadoop.yarn.conf.YarnConfiguration;
|
||||||
import org.apache.hadoop.yarn.exceptions.ApplicationIdNotProvidedException;
|
import org.apache.hadoop.yarn.exceptions.ApplicationIdNotProvidedException;
|
||||||
import org.apache.hadoop.yarn.exceptions.YarnException;
|
import org.apache.hadoop.yarn.exceptions.YarnException;
|
||||||
|
import org.apache.hadoop.yarn.security.client.TimelineDelegationTokenIdentifier;
|
||||||
import org.apache.hadoop.yarn.server.MiniYARNCluster;
|
import org.apache.hadoop.yarn.server.MiniYARNCluster;
|
||||||
import org.apache.hadoop.yarn.server.resourcemanager.MockRM;
|
import org.apache.hadoop.yarn.server.resourcemanager.MockRM;
|
||||||
import org.apache.hadoop.yarn.server.resourcemanager.ResourceManager;
|
import org.apache.hadoop.yarn.server.resourcemanager.ResourceManager;
|
||||||
import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMApp;
|
import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMApp;
|
||||||
import org.apache.hadoop.yarn.util.Records;
|
import org.apache.hadoop.yarn.util.Records;
|
||||||
|
import org.apache.hadoop.yarn.util.timeline.TimelineUtils;
|
||||||
import org.apache.log4j.Level;
|
import org.apache.log4j.Level;
|
||||||
import org.apache.log4j.LogManager;
|
import org.apache.log4j.LogManager;
|
||||||
import org.apache.log4j.Logger;
|
import org.apache.log4j.Logger;
|
||||||
|
import org.junit.Assert;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
|
|
||||||
public class TestYarnClient {
|
public class TestYarnClient {
|
||||||
|
@ -725,4 +736,80 @@ public class TestYarnClient {
|
||||||
IOUtils.closeQuietly(client);
|
IOUtils.closeQuietly(client);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testAutomaticTimelineDelegationTokenLoading()
|
||||||
|
throws Exception {
|
||||||
|
Configuration conf = new YarnConfiguration();
|
||||||
|
conf.setBoolean(YarnConfiguration.TIMELINE_SERVICE_ENABLED, true);
|
||||||
|
SecurityUtil.setAuthenticationMethod(AuthenticationMethod.KERBEROS, conf);
|
||||||
|
final Token<TimelineDelegationTokenIdentifier> dToken =
|
||||||
|
new Token<TimelineDelegationTokenIdentifier>();
|
||||||
|
// crate a mock client
|
||||||
|
YarnClientImpl client = new YarnClientImpl() {
|
||||||
|
@Override
|
||||||
|
protected void serviceInit(Configuration conf) throws Exception {
|
||||||
|
if (getConfig().getBoolean(YarnConfiguration.TIMELINE_SERVICE_ENABLED,
|
||||||
|
YarnConfiguration.DEFAULT_TIMELINE_SERVICE_ENABLED)) {
|
||||||
|
timelineServiceEnabled = true;
|
||||||
|
timelineClient = mock(TimelineClient.class);
|
||||||
|
when(timelineClient.getDelegationToken(any(String.class)))
|
||||||
|
.thenReturn(dToken);
|
||||||
|
timelineClient.init(getConfig());
|
||||||
|
timelineService = TimelineUtils.buildTimelineTokenService(getConfig());
|
||||||
|
}
|
||||||
|
this.setConfig(conf);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void serviceStart() throws Exception {
|
||||||
|
rmClient = mock(ApplicationClientProtocol.class);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void serviceStop() throws Exception {
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ApplicationReport getApplicationReport(ApplicationId appId) {
|
||||||
|
ApplicationReport report = mock(ApplicationReport.class);
|
||||||
|
when(report.getYarnApplicationState())
|
||||||
|
.thenReturn(YarnApplicationState.SUBMITTED);
|
||||||
|
return report;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean isSecurityEnabled() {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
client.init(conf);
|
||||||
|
client.start();
|
||||||
|
ApplicationSubmissionContext context =
|
||||||
|
mock(ApplicationSubmissionContext.class);
|
||||||
|
ApplicationId applicationId = ApplicationId.newInstance(0, 1);
|
||||||
|
when(context.getApplicationId()).thenReturn(applicationId);
|
||||||
|
DataOutputBuffer dob = new DataOutputBuffer();
|
||||||
|
Credentials credentials = new Credentials();
|
||||||
|
credentials.writeTokenStorageToStream(dob);
|
||||||
|
ByteBuffer tokens = ByteBuffer.wrap(dob.getData(), 0, dob.getLength());
|
||||||
|
ContainerLaunchContext clc = ContainerLaunchContext.newInstance(
|
||||||
|
null, null, null, null, tokens, null);
|
||||||
|
when(context.getAMContainerSpec()).thenReturn(clc);
|
||||||
|
client.submitApplication(context);
|
||||||
|
// Check whether token is added or not
|
||||||
|
credentials = new Credentials();
|
||||||
|
DataInputByteBuffer dibb = new DataInputByteBuffer();
|
||||||
|
tokens = clc.getTokens();
|
||||||
|
if (tokens != null) {
|
||||||
|
dibb.reset(tokens);
|
||||||
|
credentials.readTokenStorageStream(dibb);
|
||||||
|
tokens.rewind();
|
||||||
|
}
|
||||||
|
Collection<Token<? extends TokenIdentifier>> dTokens =
|
||||||
|
credentials.getAllTokens();
|
||||||
|
Assert.assertEquals(1, dTokens.size());
|
||||||
|
Assert.assertEquals(dToken, dTokens.iterator().next());
|
||||||
|
client.stop();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -70,10 +70,22 @@
|
||||||
<groupId>com.sun.jersey</groupId>
|
<groupId>com.sun.jersey</groupId>
|
||||||
<artifactId>jersey-core</artifactId>
|
<artifactId>jersey-core</artifactId>
|
||||||
</dependency>
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.codehaus.jackson</groupId>
|
||||||
|
<artifactId>jackson-core-asl</artifactId>
|
||||||
|
</dependency>
|
||||||
<dependency>
|
<dependency>
|
||||||
<groupId>org.codehaus.jackson</groupId>
|
<groupId>org.codehaus.jackson</groupId>
|
||||||
<artifactId>jackson-mapper-asl</artifactId>
|
<artifactId>jackson-mapper-asl</artifactId>
|
||||||
</dependency>
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.codehaus.jackson</groupId>
|
||||||
|
<artifactId>jackson-jaxrs</artifactId>
|
||||||
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.codehaus.jackson</groupId>
|
||||||
|
<artifactId>jackson-xc</artifactId>
|
||||||
|
</dependency>
|
||||||
<dependency>
|
<dependency>
|
||||||
<groupId>com.google.guava</groupId>
|
<groupId>com.google.guava</groupId>
|
||||||
<artifactId>guava</artifactId>
|
<artifactId>guava</artifactId>
|
||||||
|
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue