io.grpc.CompressorRegistry源码实例Demo-ICode联盟

类io.grpc.CompressorRegistry源码实例Demo

下面列出了怎么用io.grpc.CompressorRegistry的API类实例代码及写法，或者点击链接到github查看源代码。

源代码1 项目： grpc-nebula-java 文件： ServerCallImplTest.java

private void sendMessage_serverSendsOne_closeOnSecondCall(
    MethodDescriptor<Long, Long> method) {
  ServerCallImpl<Long, Long> serverCall = new ServerCallImpl<Long, Long>(
      stream,
      method,
      requestHeaders,
      context,
      DecompressorRegistry.getDefaultInstance(),
      CompressorRegistry.getDefaultInstance(),
      serverCallTracer);
  serverCall.sendHeaders(new Metadata());
  serverCall.sendMessage(1L);
  verify(stream, times(1)).writeMessage(any(InputStream.class));
  verify(stream, never()).close(any(Status.class), any(Metadata.class));

  // trying to send a second message causes gRPC to close the underlying stream
  serverCall.sendMessage(1L);
  verify(stream, times(1)).writeMessage(any(InputStream.class));
  ArgumentCaptor<Status> statusCaptor = ArgumentCaptor.forClass(Status.class);
  verify(stream, times(1)).cancel(statusCaptor.capture());
  assertEquals(Status.Code.INTERNAL, statusCaptor.getValue().getCode());
  assertEquals(ServerCallImpl.TOO_MANY_RESPONSES, statusCaptor.getValue().getDescription());
}

源代码2 项目： grpc-nebula-java 文件： ServerCallImplTest.java

private void sendMessage_serverSendsOne_closeOnSecondCall_appRunToCompletion(
    MethodDescriptor<Long, Long> method) {
  ServerCallImpl<Long, Long> serverCall = new ServerCallImpl<Long, Long>(
      stream,
      method,
      requestHeaders,
      context,
      DecompressorRegistry.getDefaultInstance(),
      CompressorRegistry.getDefaultInstance(),
      serverCallTracer);
  serverCall.sendHeaders(new Metadata());
  serverCall.sendMessage(1L);
  serverCall.sendMessage(1L);
  verify(stream, times(1)).writeMessage(any(InputStream.class));
  verify(stream, times(1)).cancel(any(Status.class));

  // App runs to completion but everything is ignored
  serverCall.sendMessage(1L);
  serverCall.close(Status.OK, new Metadata());
  try {
    serverCall.close(Status.OK, new Metadata());
    fail("calling a second time should still cause an error");
  } catch (IllegalStateException expected) {
    // noop
  }
}

源代码3 项目： grpc-nebula-java 文件： ServerCallImplTest.java

private void serverSendsOne_okFailsOnMissingResponse(
    MethodDescriptor<Long, Long> method) {
  ServerCallImpl<Long, Long> serverCall = new ServerCallImpl<Long, Long>(
      stream,
      method,
      requestHeaders,
      context,
      DecompressorRegistry.getDefaultInstance(),
      CompressorRegistry.getDefaultInstance(),
      serverCallTracer);
  serverCall.close(Status.OK, new Metadata());
  ArgumentCaptor<Status> statusCaptor = ArgumentCaptor.forClass(Status.class);
  verify(stream, times(1)).cancel(statusCaptor.capture());
  assertEquals(Status.Code.INTERNAL, statusCaptor.getValue().getCode());
  assertEquals(ServerCallImpl.MISSING_RESPONSE, statusCaptor.getValue().getDescription());
}

源代码4 项目： armeria 文件： GrpcServiceBuilder.java

/**
 * Constructs a new {@link GrpcService} that can be bound to
 * {@link ServerBuilder}. It is recommended to bind the service to a server using
 * {@linkplain ServerBuilder#service(HttpServiceWithRoutes, Function[])
 * ServerBuilder.service(HttpServiceWithRoutes)} to mount all service paths
 * without interfering with other services.
 */
public GrpcService build() {
    final HandlerRegistry handlerRegistry = registryBuilder.build();

    final GrpcService grpcService = new FramedGrpcService(
            handlerRegistry,
            handlerRegistry
                    .methods()
                    .keySet()
                    .stream()
                    .map(path -> Route.builder().exact('/' + path).build())
                    .collect(ImmutableSet.toImmutableSet()),
            firstNonNull(decompressorRegistry, DecompressorRegistry.getDefaultInstance()),
            firstNonNull(compressorRegistry, CompressorRegistry.getDefaultInstance()),
            supportedSerializationFormats,
            jsonMarshallerFactory,
            protoReflectionServiceInterceptor,
            maxOutboundMessageSizeBytes,
            useBlockingTaskExecutor,
            unsafeWrapRequestBuffers,
            useClientTimeoutHeader,
            maxInboundMessageSizeBytes);
    return enableUnframedRequests ? new UnframedGrpcService(grpcService) : grpcService;
}

源代码5 项目： armeria 文件： ArmeriaServerCallTest.java

@Test
public void messageRead_wrappedByteBuf() {
    tearDown();

    call = new ArmeriaServerCall<>(
            HttpHeaders.of(),
            TestServiceGrpc.getUnaryCallMethod(),
            CompressorRegistry.getDefaultInstance(),
            DecompressorRegistry.getDefaultInstance(),
            res,
            MAX_MESSAGE_BYTES,
            MAX_MESSAGE_BYTES,
            ctx,
            GrpcSerializationFormats.PROTO,
            new DefaultJsonMarshaller(MessageMarshaller.builder().build()),
            true,
            false,
            ResponseHeaders.builder(HttpStatus.OK)
                           .contentType(GrpcSerializationFormats.PROTO.mediaType())
                           .build());

    final ByteBuf buf = GrpcTestUtil.requestByteBuf();
    call.messageRead(new DeframedMessage(buf, 0));

    verify(buffersAttr).put(any(), same(buf));
}

源代码6 项目： grpc-java 文件： ServerCallImplTest.java

private void sendMessage_serverSendsOne_closeOnSecondCall(
    MethodDescriptor<Long, Long> method) {
  ServerCallImpl<Long, Long> serverCall = new ServerCallImpl<>(
      stream,
      method,
      requestHeaders,
      context,
      DecompressorRegistry.getDefaultInstance(),
      CompressorRegistry.getDefaultInstance(),
      serverCallTracer,
      PerfMark.createTag());
  serverCall.sendHeaders(new Metadata());
  serverCall.sendMessage(1L);
  verify(stream, times(1)).writeMessage(any(InputStream.class));
  verify(stream, never()).close(any(Status.class), any(Metadata.class));

  // trying to send a second message causes gRPC to close the underlying stream
  serverCall.sendMessage(1L);
  verify(stream, times(1)).writeMessage(any(InputStream.class));
  ArgumentCaptor<Status> statusCaptor = ArgumentCaptor.forClass(Status.class);
  verify(stream, times(1)).cancel(statusCaptor.capture());
  assertEquals(Status.Code.INTERNAL, statusCaptor.getValue().getCode());
  assertEquals(ServerCallImpl.TOO_MANY_RESPONSES, statusCaptor.getValue().getDescription());
}

源代码7 项目： grpc-java 文件： ServerCallImplTest.java

private void serverSendsOne_okFailsOnMissingResponse(
    MethodDescriptor<Long, Long> method) {
  ServerCallImpl<Long, Long> serverCall = new ServerCallImpl<>(
      stream,
      method,
      requestHeaders,
      context,
      DecompressorRegistry.getDefaultInstance(),
      CompressorRegistry.getDefaultInstance(),
      serverCallTracer,
      PerfMark.createTag());
  serverCall.close(Status.OK, new Metadata());
  ArgumentCaptor<Status> statusCaptor = ArgumentCaptor.forClass(Status.class);
  verify(stream, times(1)).cancel(statusCaptor.capture());
  assertEquals(Status.Code.INTERNAL, statusCaptor.getValue().getCode());
  assertEquals(ServerCallImpl.MISSING_RESPONSE, statusCaptor.getValue().getDescription());
}

源代码8 项目： grpc-nebula-java 文件： AbstractManagedChannelImplBuilder.java

@Override
public final T compressorRegistry(CompressorRegistry registry) {
  if (registry != null) {
    this.compressorRegistry = registry;
  } else {
    this.compressorRegistry = DEFAULT_COMPRESSOR_REGISTRY;
  }
  return thisT();
}

源代码9 项目： grpc-nebula-java 文件： ServerCallImpl.java

ServerCallImpl(ServerStream stream, MethodDescriptor<ReqT, RespT> method,
    Metadata inboundHeaders, Context.CancellableContext context,
    DecompressorRegistry decompressorRegistry, CompressorRegistry compressorRegistry,
    CallTracer serverCallTracer) {
  this.stream = stream;
  this.method = method;
  this.context = context;
  this.messageAcceptEncoding = inboundHeaders.get(MESSAGE_ACCEPT_ENCODING_KEY);
  this.decompressorRegistry = decompressorRegistry;
  this.compressorRegistry = compressorRegistry;
  this.serverCallTracer = serverCallTracer;
  this.serverCallTracer.reportCallStarted();
}

源代码10 项目： grpc-nebula-java 文件： AbstractServerImplBuilder.java

@Override
public final T compressorRegistry(CompressorRegistry registry) {
  if (registry != null) {
    compressorRegistry = registry;
  } else {
    compressorRegistry = DEFAULT_COMPRESSOR_REGISTRY;
  }
  return thisT();
}

源代码11 项目： grpc-nebula-java 文件： ServerCallImplTest.java

@Before
public void setUp() {
  MockitoAnnotations.initMocks(this);
  context = Context.ROOT.withCancellation();
  call = new ServerCallImpl<Long, Long>(stream, UNARY_METHOD, requestHeaders, context,
      DecompressorRegistry.getDefaultInstance(), CompressorRegistry.getDefaultInstance(),
      serverCallTracer);
}

源代码12 项目： grpc-nebula-java 文件： ServerCallImplTest.java

private void callTracer0(Status status) {
  CallTracer tracer = CallTracer.getDefaultFactory().create();
  ServerStats.Builder beforeBuilder = new ServerStats.Builder();
  tracer.updateBuilder(beforeBuilder);
  ServerStats before = beforeBuilder.build();
  assertEquals(0, before.callsStarted);
  assertEquals(0, before.lastCallStartedNanos);

  call = new ServerCallImpl<Long, Long>(stream, UNARY_METHOD, requestHeaders, context,
      DecompressorRegistry.getDefaultInstance(), CompressorRegistry.getDefaultInstance(),
      tracer);

  // required boilerplate
  call.sendHeaders(new Metadata());
  call.sendMessage(123L);
  // end: required boilerplate

  call.close(status, new Metadata());
  ServerStats.Builder afterBuilder = new ServerStats.Builder();
  tracer.updateBuilder(afterBuilder);
  ServerStats after = afterBuilder.build();
  assertEquals(1, after.callsStarted);
  if (status.isOk()) {
    assertEquals(1, after.callsSucceeded);
  } else {
    assertEquals(1, after.callsFailed);
  }
}

源代码13 项目： grpc-nebula-java 文件： AbstractManagedChannelImplBuilderTest.java

@Test
public void compressorRegistry_normal() {
  CompressorRegistry compressorRegistry = CompressorRegistry.newEmptyInstance();
  assertNotEquals(compressorRegistry, builder.compressorRegistry);
  assertEquals(builder, builder.compressorRegistry(compressorRegistry));
  assertEquals(compressorRegistry, builder.compressorRegistry);
}

源代码14 项目： grpc-nebula-java 文件： AbstractManagedChannelImplBuilderTest.java

@Test
public void compressorRegistry_null() {
  CompressorRegistry defaultValue = builder.compressorRegistry;
  builder.compressorRegistry(CompressorRegistry.newEmptyInstance());
  assertNotEquals(defaultValue, builder.compressorRegistry);
  assertEquals(builder, builder.compressorRegistry(null));
  assertEquals(defaultValue, builder.compressorRegistry);
}

源代码15 项目： grpc-spring-boot-starter 文件： GrpcCommonCodecAutoConfiguration.java

@ConditionalOnBean(GrpcCodecDiscoverer.class)
@ConditionalOnMissingBean
@Bean
public CompressorRegistry defaultCompressorRegistry(final GrpcCodecDiscoverer codecDiscoverer) {
    log.debug("Found GrpcCodecDiscoverer -> Creating custom CompressorRegistry");
    final CompressorRegistry registry = CompressorRegistry.getDefaultInstance();
    for (final GrpcCodecDefinition definition : codecDiscoverer.findGrpcCodecs()) {
        if (definition.getCodecType().isForCompression()) {
            final Codec codec = definition.getCodec();
            log.debug("Registering compressor: '{}' ({})", codec.getMessageEncoding(), codec.getClass().getName());
            registry.register(codec);
        }
    }
    return registry;
}

源代码16 项目： grpc-spring-boot-starter 文件： GrpcCommonCodecAutoConfiguration.java

@ConditionalOnBean(GrpcCodecDiscoverer.class)
@ConditionalOnMissingBean
@Bean
public CompressorRegistry defaultCompressorRegistry(final GrpcCodecDiscoverer codecDiscoverer) {
    log.debug("Found GrpcCodecDiscoverer -> Creating custom CompressorRegistry");
    final CompressorRegistry registry = CompressorRegistry.getDefaultInstance();
    for (final GrpcCodecDefinition definition : codecDiscoverer.findGrpcCodecs()) {
        if (definition.getCodecType().isForCompression()) {
            final Codec codec = definition.getCodec();
            log.debug("Registering compressor: '{}' ({})", codec.getMessageEncoding(), codec.getClass().getName());
            registry.register(codec);
        }
    }
    return registry;
}

源代码17 项目： armeria 文件： ArmeriaServerCallTest.java

@Before
public void setUp() {
    completionFuture = new CompletableFuture<>();
    when(res.whenComplete()).thenReturn(completionFuture);

    ctx = ServiceRequestContext.builder(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,
            new DefaultJsonMarshaller(MessageMarshaller.builder().build()),
            false,
            false,
            ResponseHeaders.builder(HttpStatus.OK)
                           .contentType(GrpcSerializationFormats.PROTO.mediaType())
                           .build());
    call.setListener(listener);
    call.messageReader().onSubscribe(subscription);

    ctx.setAttr(GrpcUnsafeBufferUtil.BUFFERS, buffersAttr);
}

源代码18 项目： grpc-java 文件： AbstractManagedChannelImplBuilder.java

@Override
public final T compressorRegistry(CompressorRegistry registry) {
  if (registry != null) {
    this.compressorRegistry = registry;
  } else {
    this.compressorRegistry = DEFAULT_COMPRESSOR_REGISTRY;
  }
  return thisT();
}

源代码19 项目： grpc-java 文件： ServerCallImpl.java

ServerCallImpl(ServerStream stream, MethodDescriptor<ReqT, RespT> method,
    Metadata inboundHeaders, Context.CancellableContext context,
    DecompressorRegistry decompressorRegistry, CompressorRegistry compressorRegistry,
    CallTracer serverCallTracer, Tag tag) {
  this.stream = stream;
  this.method = method;
  this.context = context;
  this.messageAcceptEncoding = inboundHeaders.get(MESSAGE_ACCEPT_ENCODING_KEY);
  this.decompressorRegistry = decompressorRegistry;
  this.compressorRegistry = compressorRegistry;
  this.serverCallTracer = serverCallTracer;
  this.serverCallTracer.reportCallStarted();
  this.tag = tag;
}

源代码20 项目： grpc-java 文件： ServerCallImplTest.java

@Before
public void setUp() {
  MockitoAnnotations.initMocks(this);
  context = Context.ROOT.withCancellation();
  call = new ServerCallImpl<>(stream, UNARY_METHOD, requestHeaders, context,
      DecompressorRegistry.getDefaultInstance(), CompressorRegistry.getDefaultInstance(),
      serverCallTracer, PerfMark.createTag());
}

源代码21 项目： grpc-java 文件： ServerCallImplTest.java

private void callTracer0(Status status) {
  CallTracer tracer = CallTracer.getDefaultFactory().create();
  ServerStats.Builder beforeBuilder = new ServerStats.Builder();
  tracer.updateBuilder(beforeBuilder);
  ServerStats before = beforeBuilder.build();
  assertEquals(0, before.callsStarted);
  assertEquals(0, before.lastCallStartedNanos);

  call = new ServerCallImpl<>(stream, UNARY_METHOD, requestHeaders, context,
      DecompressorRegistry.getDefaultInstance(), CompressorRegistry.getDefaultInstance(),
      tracer, PerfMark.createTag());

  // required boilerplate
  call.sendHeaders(new Metadata());
  call.sendMessage(123L);
  // end: required boilerplate

  call.close(status, new Metadata());
  ServerStats.Builder afterBuilder = new ServerStats.Builder();
  tracer.updateBuilder(afterBuilder);
  ServerStats after = afterBuilder.build();
  assertEquals(1, after.callsStarted);
  if (status.isOk()) {
    assertEquals(1, after.callsSucceeded);
  } else {
    assertEquals(1, after.callsFailed);
  }
}

源代码22 项目： grpc-java 文件： ServerCallImplTest.java

private void sendMessage_serverSendsOne_closeOnSecondCall_appRunToCompletion(
    MethodDescriptor<Long, Long> method) {
  ServerCallImpl<Long, Long> serverCall = new ServerCallImpl<>(
      stream,
      method,
      requestHeaders,
      context,
      DecompressorRegistry.getDefaultInstance(),
      CompressorRegistry.getDefaultInstance(),
      serverCallTracer,
      PerfMark.createTag());
  serverCall.sendHeaders(new Metadata());
  serverCall.sendMessage(1L);
  serverCall.sendMessage(1L);
  verify(stream, times(1)).writeMessage(any(InputStream.class));
  verify(stream, times(1)).cancel(any(Status.class));

  // App runs to completion but everything is ignored
  serverCall.sendMessage(1L);
  serverCall.close(Status.OK, new Metadata());
  try {
    serverCall.close(Status.OK, new Metadata());
    fail("calling a second time should still cause an error");
  } catch (IllegalStateException expected) {
    // noop
  }
}

源代码23 项目： grpc-java 文件： AbstractManagedChannelImplBuilderTest.java

@Test
public void compressorRegistry_normal() {
  CompressorRegistry compressorRegistry = CompressorRegistry.newEmptyInstance();
  assertNotEquals(compressorRegistry, builder.compressorRegistry);
  assertEquals(builder, builder.compressorRegistry(compressorRegistry));
  assertEquals(compressorRegistry, builder.compressorRegistry);
}

源代码24 项目： grpc-java 文件： AbstractManagedChannelImplBuilderTest.java

@Test
public void compressorRegistry_null() {
  CompressorRegistry defaultValue = builder.compressorRegistry;
  builder.compressorRegistry(CompressorRegistry.newEmptyInstance());
  assertNotEquals(defaultValue, builder.compressorRegistry);
  assertEquals(builder, builder.compressorRegistry(null));
  assertEquals(defaultValue, builder.compressorRegistry);
}

源代码25 项目： grpc-nebula-java 文件： AltsServerBuilder.java

/** {@inheritDoc} */
@Override
public AltsServerBuilder compressorRegistry(CompressorRegistry registry) {
  delegate.compressorRegistry(registry);
  return this;
}

源代码26 项目： grpc-nebula-java 文件： ClientCallImpl.java

ClientCallImpl<ReqT, RespT> setCompressorRegistry(CompressorRegistry compressorRegistry) {
  this.compressorRegistry = compressorRegistry;
  return this;
}

源代码27 项目： grpc-spring-boot-starter 文件： GrpcServerAutoConfiguration.java

@ConditionalOnBean(CompressorRegistry.class)
@Bean
public GrpcServerConfigurer compressionServerConfigurer(final CompressorRegistry registry) {
    return builder -> builder.compressorRegistry(registry);
}

源代码28 项目： grpc-spring-boot-starter 文件： GrpcClientAutoConfiguration.java

@ConditionalOnBean(CompressorRegistry.class)
@Bean
GrpcChannelConfigurer compressionChannelConfigurer(final CompressorRegistry registry) {
    return (builder, name) -> builder.compressorRegistry(registry);
}

源代码29 项目： grpc-spring-boot-starter 文件： GrpcServerAutoConfiguration.java

@ConditionalOnBean(CompressorRegistry.class)
@Bean
public GrpcServerConfigurer compressionServerConfigurer(final CompressorRegistry registry) {
    return builder -> builder.compressorRegistry(registry);
}

源代码30 项目： grpc-spring-boot-starter 文件： GrpcClientAutoConfiguration.java

@ConditionalOnBean(CompressorRegistry.class)
@Bean
GrpcChannelConfigurer compressionChannelConfigurer(final CompressorRegistry registry) {
    return (builder, name) -> builder.compressorRegistry(registry);
}

类所在包

io.grpc

类方法

getDefaultInstance ( )

同包方法

io.grpc.stub.StreamObserver

io.grpc.ManagedChannel

io.grpc.ManagedChannelBuilder

io.grpc.Server

io.grpc.StatusRuntimeException

io.grpc.ServerBuilder

io.grpc.Status