Examples with RequestContext org.hypertrace.core.grpcutils.context.RequestContext used on opensource projects

Example 16 with RequestContext

use of org.hypertrace.core.grpcutils.context.RequestContext in project entity-service by hypertrace.

the class EntityTypeCachingClientTest method supportsMultipleConcurrentCacheKeys.

@Test
void supportsMultipleConcurrentCacheKeys() throws Exception {
    EntityType defaultRetrieved = this.grpcTestContext.call(() -> this.typeClient.get("first").blockingGet());
    assertSame(this.type1, defaultRetrieved);
    verify(this.mockTypeService, times(1)).queryEntityTypes(any(), any());
    RequestContext otherMockContext = mock(RequestContext.class);
    when(otherMockContext.getTenantId()).thenReturn(Optional.of("other tenant"));
    Context otherGrpcContext = Context.current().withValue(RequestContext.CURRENT, otherMockContext);
    EntityType otherContextType = EntityType.newBuilder(this.type1).build();
    this.responseTypes = List.of(otherContextType);
    EntityType otherRetrieved = otherGrpcContext.call(() -> this.typeClient.get("first").blockingGet());
    assertSame(otherContextType, otherRetrieved);
    assertNotSame(defaultRetrieved, otherRetrieved);
    verify(this.mockTypeService, times(2)).queryEntityTypes(any(), any());
    verifyNoMoreInteractions(this.mockTypeService);
    assertSame(defaultRetrieved, this.grpcTestContext.call(() -> this.typeClient.get("first").blockingGet()));
    assertSame(otherRetrieved, otherGrpcContext.call(() -> this.typeClient.get("first").blockingGet()));
}

Example 17 with RequestContext

use of org.hypertrace.core.grpcutils.context.RequestContext in project entity-service by hypertrace.

the class EntityTypeCachingClientTest method hasConfigurableCacheSize.

@Test
void hasConfigurableCacheSize() throws Exception {
    this.typeClient = EntityTypeClient.builder(this.grpcChannel).withMaximumCacheContexts(1).build();
    RequestContext otherMockContext = mock(RequestContext.class);
    when(otherMockContext.getTenantId()).thenReturn(Optional.of("other tenant"));
    this.grpcTestContext.call(() -> this.typeClient.get("first").blockingGet());
    // This call should evict the original call
    Context.current().withValue(RequestContext.CURRENT, otherMockContext).call(() -> this.typeClient.get("first").blockingGet());
    // Rerunning this call now fire again, a third server call
    this.grpcTestContext.call(() -> this.typeClient.get("first").blockingGet());
    verify(this.mockTypeService, times(3)).queryEntityTypes(any(), any());
}