Example usage for org.apache.hadoop.yarn.client.api NMTokenCache getNMToken

Introduction

In this page you can find the example usage for org.apache.hadoop.yarn.client.api NMTokenCache getNMToken.

Prototype

@Public
public static Token getNMToken(String nodeAddr) 

Source Link

Document

Returns NMToken, null if absent.

Usage

From source file:org.springframework.yarn.am.AppmasterCmTemplate.java

License:Apache License

@Override
protected UserGroupInformation getUser() {
    InetSocketAddress rpcAddress = getRpcAddress(getConfiguration());

    // TODO: at some point remove static cache
    Token token = NMTokenCache.getNMToken(container.getNodeId().toString());

    // this is what node manager requires for auth
    UserGroupInformation user = UserGroupInformation
            .createRemoteUser(container.getId().getApplicationAttemptId().toString());
    org.apache.hadoop.security.token.Token<NMTokenIdentifier> nmToken = ConverterUtils.convertFromYarn(token,
            rpcAddress);/*ww  w . ja v  a  2  s. c o m*/
    user.addToken(nmToken);

    return user;
}

---

From source file:org.springframework.yarn.am.NMTokenCacheTests.java

License:Apache License

@Test
public void testTokenCacheGenericUsage() {
    // cdh//from  www. j  a  v  a2  s. c  o  m
    // static org.apache.hadoop.yarn.client.api.NMTokenCache.getSingleton()
    // void org.apache.hadoop.yarn.client.api.NMTokenCache.setNMToken(String, Token)

    // vanilla
    // static void org.apache.hadoop.yarn.client.api.NMTokenCache.setNMToken(String, Token)

    NMTokenCache cache1 = NMTokenCacheCompat.getNMTokenCache();
    NMTokenCache cache2 = NMTokenCacheCompat.getNMTokenCache();
    cache1.setNMToken("nodeAddr1", new TestToken("kind1"));
    cache2.setNMToken("nodeAddr2", new TestToken("kind2"));

    assertThat(cache1.getNMToken("nodeAddr1").getKind(), is("kind1"));
    assertThat(cache2.getNMToken("nodeAddr1").getKind(), is("kind1"));
    assertThat(cache1.getNMToken("nodeAddr2").getKind(), is("kind2"));
    assertThat(cache2.getNMToken("nodeAddr2").getKind(), is("kind2"));

    assertThat(cache1, sameInstance(cache2));
    assertThat(NMTokenCacheCompat.getNMTokenCache(), sameInstance(NMTokenCacheCompat.getNMTokenCache()));
}

---