private static ServiceRequestContext newServiceContext( String path, @Nullable String query) throws Exception { final InetSocketAddress remoteAddress = new InetSocketAddress( InetAddress.getByAddress("client.com", new byte[] { 1, 2, 3, 4 }), 5678); final InetSocketAddress localAddress = new InetSocketAddress( InetAddress.getByAddress("server.com", new byte[] { 5, 6, 7, 8 }), 8080); final String pathAndQuery = path + (query != null ? '?' + query : ""); final HttpRequest req = HttpRequest.of(HttpHeaders.of(HttpMethod.GET, pathAndQuery) .authority("server.com:8080") .set(HttpHeaderNames.USER_AGENT, "some-client")); final ServiceRequestContext ctx = ServiceRequestContextBuilder.of(req) .sslSession(newSslSession()) .remoteAddress(remoteAddress) .localAddress(localAddress) .clientAddress(InetAddress.getByName("9.10.11.12")) .build(); ctx.attr(MY_ATTR).set(new CustomValue("some-attr")); return ctx; }
@Before public void setUp() { completionFuture = new CompletableFuture<>(); when(res.completionFuture()).thenReturn(completionFuture); ctx = ServiceRequestContextBuilder.of(HttpRequest.of(HttpMethod.POST, "/")) .eventLoop(eventLoop.get()) .build(); call = new ArmeriaServerCall<>( HttpHeaders.of(), TestServiceGrpc.getUnaryCallMethod(), CompressorRegistry.getDefaultInstance(), DecompressorRegistry.getDefaultInstance(), res, MAX_MESSAGE_BYTES, MAX_MESSAGE_BYTES, ctx, GrpcSerializationFormats.PROTO, MessageMarshaller.builder().build(), false, false, "gzip"); call.setListener(listener); call.messageReader().onSubscribe(subscription); ctx.attr(GrpcUnsafeBufferUtil.BUFFERS).set(buffersAttr); }
public static User currentUser(ServiceRequestContext ctx) { return ctx.attr(CURRENT_USER_KEY).get(); }
public static User currentUser(ServiceRequestContext ctx) { return ctx.attr(CURRENT_USER_KEY).get(); }
public static User currentUser(ServiceRequestContext ctx) { return ctx.attr(CURRENT_USER_KEY).get(); }
/** * Returns the {@link MetadataService} instance from the specified {@link ServiceRequestContext}. */ public static MetadataService getMetadataService(ServiceRequestContext ctx) { final MetadataService mds = ctx.attr(METADATA_SERVICE_ATTRIBUTE_KEY).get(); if (mds != null) { return mds; } throw new IllegalStateException("No metadata service instance exists."); } }
/** * Returns the {@link MetadataService} instance from the specified {@link ServiceRequestContext}. */ public static MetadataService getMetadataService(ServiceRequestContext ctx) { final MetadataService mds = ctx.attr(METADATA_SERVICE_ATTRIBUTE_KEY).get(); if (mds != null) { return mds; } throw new IllegalStateException("No metadata service instance exists."); } }
/** * Returns the {@link MetadataService} instance from the specified {@link ServiceRequestContext}. */ public static MetadataService getMetadataService(ServiceRequestContext ctx) { final MetadataService mds = ctx.attr(METADATA_SERVICE_ATTRIBUTE_KEY).get(); if (mds != null) { return mds; } throw new IllegalStateException("No metadata service instance exists."); } }
public static void setCurrentUser(ServiceRequestContext ctx, User currentUser) { requireNonNull(ctx, "ctx"); requireNonNull(currentUser, "currentUser"); ctx.attr(CURRENT_USER_KEY).set(currentUser); }
public static void setCurrentUser(ServiceRequestContext ctx, User currentUser) { requireNonNull(ctx, "ctx"); requireNonNull(currentUser, "currentUser"); ctx.attr(CURRENT_USER_KEY).set(currentUser); }
public static void setCurrentUser(ServiceRequestContext ctx, User currentUser) { requireNonNull(ctx, "ctx"); requireNonNull(currentUser, "currentUser"); ctx.attr(CURRENT_USER_KEY).set(currentUser); }
@Override public HttpResponse serve(ServiceRequestContext ctx, HttpRequest req) throws Exception { ctx.attr(METADATA_SERVICE_ATTRIBUTE_KEY).set(mds); return delegate().serve(ctx, req); }
@Override public HttpResponse serve(ServiceRequestContext ctx, HttpRequest req) throws Exception { ctx.attr(METADATA_SERVICE_ATTRIBUTE_KEY).set(mds); return delegate().serve(ctx, req); }
@Override public HttpResponse serve(ServiceRequestContext ctx, HttpRequest req) throws Exception { ctx.attr(METADATA_SERVICE_ATTRIBUTE_KEY).set(mds); return delegate().serve(ctx, req); }
public static Author currentAuthor(ServiceRequestContext ctx) { final User user = ctx.attr(CURRENT_USER_KEY).get(); assert user != null; return user == User.DEFAULT ? Author.DEFAULT : new Author(user.name(), user.email()); }
public static Author currentAuthor(ServiceRequestContext ctx) { final User user = ctx.attr(CURRENT_USER_KEY).get(); assert user != null; return user == User.DEFAULT ? Author.DEFAULT : new Author(user.name(), user.email()); }
public static Author currentAuthor(ServiceRequestContext ctx) { final User user = ctx.attr(CURRENT_USER_KEY).get(); assert user != null; return user == User.DEFAULT ? Author.DEFAULT : new Author(user.name(), user.email()); }