big-data/hadoop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

YARN-9072. Send exit command to terminate docker exec on connection close. Contributed by Eric Yang

Browse Source
This commit is contained in:
Billie Rinaldi 2018-12-18 08:43:09 -08:00
parent 99b5061265
commit c5c73182e5
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-nodemanager/src/main/java/org/apache/hadoop/yarn/server/nodemanager/webapp/ContainerShellWebSocket.java
View File

@ -113,6 +113,7 @@ public void onConnect(Session session) {
.fromString(cId));
if (!checkAuthorization(session, container)) {
session.close(1008, "Forbidden");
return;
}
LOG.info(session.getRemoteAddress().getHostString() + " connected!");
LOG.info(
@ -135,6 +136,9 @@ public void onConnect(Session session) {
public void onClose(Session session, int status, String reason) {
try {
LOG.info(session.getRemoteAddress().getHostString() + " closed!");
String exit = "exit\r\n";
pair.out.write(exit.getBytes(Charset.forName("UTF-8")));
pair.out.flush();
pair.in.close();
pair.out.close();
} catch (IOException e) {

2
hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-nodemanager/src/test/java/org/apache/hadoop/yarn/server/nodemanager/webapp/TestNMWebTerminal.java
View File

@ -48,7 +48,7 @@
public class TestNMWebTerminal {
private static final File TESTROOTDIR = new File("target",
TestNMWebServer.class.getSimpleName());
private static File TESTLOGDIR = new File("target",
private static final File TESTLOGDIR = new File("target",
TestNMWebServer.class.getSimpleName() + "LogDir");
private NodeHealthCheckerService healthChecker;
private WebServer server;