honeymoose
/
OpenSearch
mirror of https://github.com/honeymoose/OpenSearch.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add helper methods to TransportActionProxy to identify proxy actions and requests (#25124) 

Downstream users of out network intercept infrastructure need this information which is
hidden due to member and class visibility.
This commit is contained in:
Simon Willnauer 2017-06-08 09:07:22 +02:00 committed by GitHub
parent d187fa78fd
commit ce24331d1f
2 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
core/src/main/java/org/elasticsearch/transport/TransportActionProxy.java
View File

@ -158,4 +158,18 @@ public final class TransportActionProxy {
}
return request;
}
/**
* Returns <code>true</code> iff the given action is a proxy action
*/
public static boolean isProxyAction(String action) {
return action.startsWith(PROXY_ACTION_PREFIX);
}
/**
* Returns <code>true</code> iff the given request is a proxy request
*/
public static boolean isProxyRequest(TransportRequest request) {
return request instanceof ProxyRequest;
}
}

12
core/src/test/java/org/elasticsearch/transport/TransportActionProxyTests.java
View File

@ -258,4 +258,16 @@ public class TransportActionProxyTests extends ESTestCase {
assertTrue(transportRequest instanceof TransportActionProxy.ProxyRequest);
assertSame(TransportService.HandshakeRequest.INSTANCE, TransportActionProxy.unwrapRequest(transportRequest));
}
public void testIsProxyAction() {
String action = "foo/bar";
String proxyAction = TransportActionProxy.getProxyAction(action);
assertTrue(TransportActionProxy.isProxyAction(proxyAction));
assertFalse(TransportActionProxy.isProxyAction(action));
}
public void testIsProxyRequest() {
assertTrue(TransportActionProxy.isProxyRequest(new TransportActionProxy.ProxyRequest<>(() -> null)));
assertFalse(TransportActionProxy.isProxyRequest(TransportRequest.Empty.INSTANCE));
}
}