Skip to content

Commit

Permalink
Added PR feedback, including lazy remote host lookup
Browse files Browse the repository at this point in the history
  • Loading branch information
fickludd committed Sep 16, 2016
1 parent f107748 commit f70712e
Show file tree
Hide file tree
Showing 8 changed files with 14 additions and 44 deletions.
Expand Up @@ -123,6 +123,6 @@ public String toString()
return format( return format(
"ExecutingQuery{queryId=%d, querySource='%s', username='%s', queryText='%s', queryParameters=%s, " + "ExecutingQuery{queryId=%d, querySource='%s', username='%s', queryText='%s', queryParameters=%s, " +
"startTime=%d}", "startTime=%d}",
queryId, querySource.toString( ':' ), usernameAsString(), queryText, queryParameters, startTime ); queryId, querySource.toString( ":" ), usernameAsString(), queryText, queryParameters, startTime );
} }
} }
Expand Up @@ -267,9 +267,4 @@ public AccessMode accessMode()
{ {
return mode; return mode;
} }

public static Function<TransactionalContext, String> describe()
{
return transactionalContext -> "We didn't really care to describe this for you in this test";
}
} }
Expand Up @@ -41,27 +41,11 @@ public QuerySource append( String newPart )
@Override @Override
public String toString() public String toString()
{ {
return toString( '\t' ); return toString( "\t" );
} }


public String toString( Character sep ) public String toString( String sep )
{ {
StringBuilder builder = new StringBuilder( ); return String.join( sep, parts );

boolean isFirst = true;
for ( String part : parts )
{
if ( isFirst )
{
isFirst = false;
}
else
{
builder.append( sep );
}
builder.append( part );
}

return builder.toString();
} }
} }
Expand Up @@ -69,18 +69,12 @@ public void checkKernelStatementOnCheck() throws Exception
KernelTransaction.Type transactionType = null; KernelTransaction.Type transactionType = null;
AccessMode transactionMode = null; AccessMode transactionMode = null;
ExecutingQuery executingQuery = null; ExecutingQuery executingQuery = null;
PropertyContainerLocker locker = null;
DbmsOperations.Factory dbmsOperationsFactory = null; DbmsOperations.Factory dbmsOperationsFactory = null;
ThreadToStatementContextBridge txBridge = null; ThreadToStatementContextBridge txBridge = null;
Neo4jTransactionalContext transactionalContext = Neo4jTransactionalContext transactionalContext =
//<<<<<<< HEAD
// new Neo4jTransactionalContext(
// databaseQueryService, transaction, statement, "", Collections.emptyMap(), propertyContainerLocker );
//=======
new Neo4jTransactionalContext( new Neo4jTransactionalContext(
databaseQueryService, transaction, transactionType, transactionMode, statement, executingQuery, databaseQueryService, transaction, transactionType, transactionMode, statement, executingQuery,
propertyContainerLocker, txBridge, dbmsOperationsFactory, guard ); propertyContainerLocker, txBridge, dbmsOperationsFactory, guard );
//>>>>>>> b556d13... Moved functionality from QuerySession to TransactionContext


transactionalContext.check(); transactionalContext.check();


Expand Down
Expand Up @@ -35,6 +35,7 @@
import java.util.Enumeration; import java.util.Enumeration;
import java.util.HashMap; import java.util.HashMap;
import java.util.Map; import java.util.Map;
import java.util.function.Supplier;
import javax.servlet.DispatcherType; import javax.servlet.DispatcherType;
import javax.servlet.ReadListener; import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream; import javax.servlet.ServletInputStream;
Expand All @@ -43,6 +44,7 @@
import javax.servlet.http.HttpServletResponse; import javax.servlet.http.HttpServletResponse;
import javax.ws.rs.core.MediaType; import javax.ws.rs.core.MediaType;


import org.neo4j.function.Suppliers;
import org.neo4j.string.UTF8; import org.neo4j.string.UTF8;


public class InternalJettyServletRequest extends Request public class InternalJettyServletRequest extends Request
Expand Down Expand Up @@ -216,7 +218,7 @@ public String getRemoteAddr()
@Override @Override
public String getRemoteHost() public String getRemoteHost()
{ {
return requestData.remoteHost; return requestData.remoteHost.get();
} }


@Override @Override
Expand Down Expand Up @@ -362,7 +364,7 @@ public HttpChannelState getHttpChannelState()
public static class RequestData public static class RequestData
{ {
public final String remoteAddress; public final String remoteAddress;
public final String remoteHost; public final Supplier<String> remoteHost;
public final boolean isSecure; public final boolean isSecure;
public final int remotePort; public final int remotePort;
public final String localName; public final String localName;
Expand All @@ -372,7 +374,7 @@ public static class RequestData


public RequestData( public RequestData(
String remoteAddress, String remoteAddress,
String remoteHost, Supplier<String> remoteHost,
boolean isSecure, boolean isSecure,
int remotePort, int remotePort,
String localName, String localName,
Expand All @@ -394,7 +396,7 @@ public static RequestData from( HttpServletRequest req )
{ {
return new RequestData( return new RequestData(
req.getRemoteAddr(), req.getRemoteAddr(),
req.getRemoteHost(), Suppliers.lazySingleton( req::getRemoteHost ),
req.isSecure(), req.isSecure(),
req.getRemotePort(), req.getRemotePort(),
req.getLocalName(), req.getLocalName(),
Expand Down
Expand Up @@ -29,6 +29,6 @@ public static QuerySource describe( HttpServletRequest request )
{ {
return request == null ? return request == null ?
new QuerySource( "server-session" ) : new QuerySource( "server-session" ) :
new QuerySource( "server-session", "http", request.getRemoteAddr(), request.getRequestURI() ); new QuerySource( "server-session", request.getScheme(), request.getRemoteAddr(), request.getRequestURI() );
} }
} }
Expand Up @@ -34,6 +34,7 @@
import org.neo4j.server.rest.web.InternalJettyServletResponse; import org.neo4j.server.rest.web.InternalJettyServletResponse;


import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.mock; import static org.mockito.Mockito.mock;


public class BatchOperationsTest { public class BatchOperationsTest {
Expand Down Expand Up @@ -83,7 +84,7 @@ public void shouldForwardMetadataFromRequestData() throws Exception
{ {
// Given // Given
RequestData mock = new RequestData( RequestData mock = new RequestData(
"127.0.0.1", "localhost", true, 1, "127.0.0.1", () -> "localhost", true, 1,
"TheLocalName", "129.0.0.1", 2, "authorization/auth" ); "TheLocalName", "129.0.0.1", 2, "authorization/auth" );


InternalJettyServletRequest req = new InternalJettyServletRequest( "POST", InternalJettyServletRequest req = new InternalJettyServletRequest( "POST",
Expand All @@ -93,7 +94,7 @@ public void shouldForwardMetadataFromRequestData() throws Exception
// When & then // When & then
assertEquals( "127.0.0.1", req.getRemoteAddr()); assertEquals( "127.0.0.1", req.getRemoteAddr());
assertEquals( "localhost", req.getRemoteHost()); assertEquals( "localhost", req.getRemoteHost());
assertEquals( true, req.isSecure() ); assertTrue( req.isSecure() );
assertEquals( 1, req.getRemotePort()); assertEquals( 1, req.getRemotePort());
assertEquals( "TheLocalName", req.getLocalName()); assertEquals( "TheLocalName", req.getLocalName());
assertEquals( "129.0.0.1", req.getLocalAddr()); assertEquals( "129.0.0.1", req.getLocalAddr());
Expand Down
Expand Up @@ -225,11 +225,5 @@ private static String mapAsString( Map<String,Object> params )


return builder.toString(); return builder.toString();
} }

@SuppressWarnings( "unchecked" )
private static Class<Map<String,Object>> paramsClass()
{
return (Class<Map<String,Object>>) (Object) Map.class;
}
} }
} }

0 comments on commit f70712e

Please sign in to comment.