Java Code Examples for io.netty.util.AsciiString#of()

The following examples show how to use io.netty.util.AsciiString#of() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example 1

Source File: InboundHeadersBenchmark.java From grpc-nebula-java with Apache License 2.0

6 votes

private static void setupRequestHeaders() {
  requestHeaders = new AsciiString[18];
  int i = 0;
  requestHeaders[i++] = AsciiString.of(":method");
  requestHeaders[i++] = AsciiString.of("POST");
  requestHeaders[i++] = AsciiString.of(":scheme");
  requestHeaders[i++] = AsciiString.of("http");
  requestHeaders[i++] = AsciiString.of(":path");
  requestHeaders[i++] = AsciiString.of("/google.pubsub.v2.PublisherService/CreateTopic");
  requestHeaders[i++] = AsciiString.of(":authority");
  requestHeaders[i++] = AsciiString.of("pubsub.googleapis.com");
  requestHeaders[i++] = AsciiString.of("te");
  requestHeaders[i++] = AsciiString.of("trailers");
  requestHeaders[i++] = AsciiString.of("grpc-timeout");
  requestHeaders[i++] = AsciiString.of("1S");
  requestHeaders[i++] = AsciiString.of("content-type");
  requestHeaders[i++] = AsciiString.of("application/grpc+proto");
  requestHeaders[i++] = AsciiString.of("grpc-encoding");
  requestHeaders[i++] = AsciiString.of("gzip");
  requestHeaders[i++] = AsciiString.of("authorization");
  requestHeaders[i] = AsciiString.of("Bearer y235.wef315yfh138vh31hv93hv8h3v");
}

Example 2

Source File: NettyClientStreamTest.java From grpc-java with Apache License 2.0

6 votes

@Override
protected NettyClientStream createStream() {
  when(handler.getWriteQueue()).thenReturn(writeQueue);
  NettyClientStream stream = new NettyClientStream(
      new TransportStateImpl(handler, DEFAULT_MAX_MESSAGE_SIZE),
      methodDescriptor,
      new Metadata(),
      channel,
      AsciiString.of("localhost"),
      AsciiString.of("http"),
      AsciiString.of("agent"),
      StatsTraceContext.NOOP,
      transportTracer,
      CallOptions.DEFAULT,
      false);
  stream.start(listener);
  stream.transportState().setHttp2Stream(http2Stream);
  reset(listener);
  return stream;
}

Example 3

Source File: NettyClientStreamTest.java From grpc-java with Apache License 2.0

6 votes

@Test
public void setHttp2StreamShouldNotifyReady() {
  listener = mock(ClientStreamListener.class);

  stream = new NettyClientStream(new TransportStateImpl(handler, DEFAULT_MAX_MESSAGE_SIZE),
      methodDescriptor,
      new Metadata(),
      channel,
      AsciiString.of("localhost"),
      AsciiString.of("http"),
      AsciiString.of("agent"),
      StatsTraceContext.NOOP,
      transportTracer,
      CallOptions.DEFAULT,
      false);
  stream.start(listener);
  stream().transportState().setId(STREAM_ID);
  verify(listener, never()).onReady();
  assertFalse(stream.isReady());
  stream().transportState().setHttp2Stream(http2Stream);
  verify(listener).onReady();
  assertTrue(stream.isReady());
}

Example 4

Source File: InboundHeadersBenchmark.java From grpc-java with Apache License 2.0

6 votes

private static void setupRequestHeaders() {
  requestHeaders = new AsciiString[18];
  int i = 0;
  requestHeaders[i++] = AsciiString.of(":method");
  requestHeaders[i++] = AsciiString.of("POST");
  requestHeaders[i++] = AsciiString.of(":scheme");
  requestHeaders[i++] = AsciiString.of("http");
  requestHeaders[i++] = AsciiString.of(":path");
  requestHeaders[i++] = AsciiString.of("/google.pubsub.v2.PublisherService/CreateTopic");
  requestHeaders[i++] = AsciiString.of(":authority");
  requestHeaders[i++] = AsciiString.of("pubsub.googleapis.com");
  requestHeaders[i++] = AsciiString.of("te");
  requestHeaders[i++] = AsciiString.of("trailers");
  requestHeaders[i++] = AsciiString.of("grpc-timeout");
  requestHeaders[i++] = AsciiString.of("1S");
  requestHeaders[i++] = AsciiString.of("content-type");
  requestHeaders[i++] = AsciiString.of("application/grpc+proto");
  requestHeaders[i++] = AsciiString.of("grpc-encoding");
  requestHeaders[i++] = AsciiString.of("gzip");
  requestHeaders[i++] = AsciiString.of("authorization");
  requestHeaders[i] = AsciiString.of("Bearer y235.wef315yfh138vh31hv93hv8h3v");
}

Example 5

Source File: ServerHandshakePacket.java From spring-boot-protocol with Apache License 2.0

6 votes

private ServerHandshakePacket(Builder builder) {
	super(builder.authPluginData);
	if (builder.authPluginData.readableBytes() < Constants.AUTH_PLUGIN_DATA_PART1_LEN) {
		throw new IllegalArgumentException("authPluginData can not contain less than " + Constants.AUTH_PLUGIN_DATA_PART1_LEN + " bytes.");
	}
	protocolVersion = builder.protocolVersion;
	if (builder.serverVersion == null) {
		throw new NullPointerException("serverVersion can not be null");
	}
	sequenceId = builder.sequenceId;
	serverVersion = AsciiString.of(builder.serverVersion);
	connectionId = builder.connectionId;
	capabilities = Collections.unmodifiableSet(builder.capabilities);
	characterSet = builder.characterSet;
	serverStatus = Collections.unmodifiableSet(builder.serverStatus);
	authPluginName = builder.authPluginName;
}

Example 6

Source File: NettyClientStreamTest.java From grpc-nebula-java with Apache License 2.0

6 votes

@Override
protected NettyClientStream createStream() {
  when(handler.getWriteQueue()).thenReturn(writeQueue);
  NettyClientStream stream = new NettyClientStream(
      new TransportStateImpl(handler, DEFAULT_MAX_MESSAGE_SIZE),
      methodDescriptor,
      new Metadata(),
      channel,
      AsciiString.of("localhost"),
      AsciiString.of("http"),
      AsciiString.of("agent"),
      StatsTraceContext.NOOP,
      transportTracer,
      CallOptions.DEFAULT);
  stream.start(listener);
  stream.transportState().setId(STREAM_ID);
  stream.transportState().setHttp2Stream(http2Stream);
  reset(listener);
  return stream;
}

Example 7

Source File: NettyClientStreamTest.java From grpc-nebula-java with Apache License 2.0

6 votes

@Test
public void setHttp2StreamShouldNotifyReady() {
  listener = mock(ClientStreamListener.class);

  stream = new NettyClientStream(new TransportStateImpl(handler, DEFAULT_MAX_MESSAGE_SIZE),
      methodDescriptor,
      new Metadata(),
      channel,
      AsciiString.of("localhost"),
      AsciiString.of("http"),
      AsciiString.of("agent"),
      StatsTraceContext.NOOP,
      transportTracer,
      CallOptions.DEFAULT);
  stream.start(listener);
  stream().transportState().setId(STREAM_ID);
  verify(listener, never()).onReady();
  assertFalse(stream.isReady());
  stream().transportState().setHttp2Stream(http2Stream);
  verify(listener).onReady();
  assertTrue(stream.isReady());
}

Example 8

Source File: NettyClientStreamTest.java From grpc-nebula-java with Apache License 2.0

5 votes

@Test
public void getRequestSentThroughHeader() {
  // Creating a GET method
  MethodDescriptor<?, ?> descriptor = MethodDescriptor.<Void, Void>newBuilder()
      .setType(MethodDescriptor.MethodType.UNARY)
      .setFullMethodName("testService/test")
      .setRequestMarshaller(marshaller)
      .setResponseMarshaller(marshaller)
      .setIdempotent(true)
      .setSafe(true)
      .build();
  NettyClientStream stream = new NettyClientStream(
      new TransportStateImpl(handler, DEFAULT_MAX_MESSAGE_SIZE),
      descriptor,
      new Metadata(),
      channel,
      AsciiString.of("localhost"),
      AsciiString.of("http"),
      AsciiString.of("agent"),
      StatsTraceContext.NOOP,
      transportTracer,
      CallOptions.DEFAULT);
  stream.start(listener);
  stream.transportState().setId(STREAM_ID);
  stream.transportState().setHttp2Stream(http2Stream);

  byte[] msg = smallMessage();
  stream.writeMessage(new ByteArrayInputStream(msg));
  stream.flush();
  stream.halfClose();
  ArgumentCaptor<CreateStreamCommand> cmdCap = ArgumentCaptor.forClass(CreateStreamCommand.class);
  verify(writeQueue).enqueue(cmdCap.capture(), eq(true));
  ImmutableListMultimap<CharSequence, CharSequence> headers =
      ImmutableListMultimap.copyOf(cmdCap.getValue().headers());
  assertThat(headers).containsEntry(AsciiString.of(":method"), Utils.HTTP_GET_METHOD);
  assertThat(headers)
      .containsEntry(
          AsciiString.of(":path"),
          AsciiString.of("/testService/test?" + BaseEncoding.base64().encode(msg)));
}

Example 9

Source File: NettyClientStreamTest.java From grpc-nebula-java with Apache License 2.0

5 votes

@Test
public void removeUserAgentFromApplicationHeaders() {
  Metadata metadata = new Metadata();
  metadata.put(GrpcUtil.USER_AGENT_KEY, "bad agent");
  listener = mock(ClientStreamListener.class);
  Mockito.reset(writeQueue);
  ChannelPromise completedPromise = new DefaultChannelPromise(channel)
      .setSuccess();
  when(writeQueue.enqueue(any(QueuedCommand.class), any(boolean.class)))
      .thenReturn(completedPromise);

  stream = new NettyClientStream(
      new TransportStateImpl(handler, DEFAULT_MAX_MESSAGE_SIZE),
      methodDescriptor,
      new Metadata(),
      channel,
      AsciiString.of("localhost"),
      AsciiString.of("http"),
      AsciiString.of("good agent"),
      StatsTraceContext.NOOP,
      transportTracer,
      CallOptions.DEFAULT);
  stream.start(listener);

  ArgumentCaptor<CreateStreamCommand> cmdCap = ArgumentCaptor.forClass(CreateStreamCommand.class);
  verify(writeQueue).enqueue(cmdCap.capture(), eq(false));
  assertThat(ImmutableListMultimap.copyOf(cmdCap.getValue().headers()))
      .containsEntry(Utils.USER_AGENT, AsciiString.of("good agent"));
}

Example 10

Source File: HttpHeaderNamesTest.java From armeria with Apache License 2.0

5 votes

@Test
void testOfAsciiString() {
    // Should produce a lower-cased AsciiString.
    final AsciiString mixedCased = AsciiString.of("Foo");
    assertThat((Object) HttpHeaderNames.of(mixedCased)).isNotSameAs(mixedCased);
    assertThat(HttpHeaderNames.of(mixedCased).toString()).isEqualTo("foo");

    // Should not produce a new instance for an AsciiString that's already lower-cased.
    final AsciiString lowerCased = AsciiString.of("foo");
    assertThat((Object) HttpHeaderNames.of(lowerCased)).isSameAs(lowerCased);

    // Should reuse known header name instances.
    assertThat((Object) HttpHeaderNames.of(AsciiString.of("date"))).isSameAs(HttpHeaderNames.DATE);
}

Example 11

Source File: InboundHeadersBenchmark.java From grpc-java with Apache License 2.0

5 votes

private static void setupResponseHeaders() {
  responseHeaders = new AsciiString[4];
  int i = 0;
  responseHeaders[i++] = AsciiString.of(":status");
  responseHeaders[i++] = AsciiString.of("200");
  responseHeaders[i++] = AsciiString.of("grpc-encoding");
  responseHeaders[i] = AsciiString.of("gzip");
}

Example 12

Source File: InboundHeadersBenchmark.java From grpc-nebula-java with Apache License 2.0

5 votes

private static void setupResponseHeaders() {
  responseHeaders = new AsciiString[4];
  int i = 0;
  responseHeaders[i++] = AsciiString.of(":status");
  responseHeaders[i++] = AsciiString.of("200");
  responseHeaders[i++] = AsciiString.of("grpc-encoding");
  responseHeaders[i] = AsciiString.of("gzip");
}

Example 13

Source File: NettyClientStreamTest.java From grpc-java with Apache License 2.0

5 votes

@Test
public void removeUserAgentFromApplicationHeaders() {
  Metadata metadata = new Metadata();
  metadata.put(GrpcUtil.USER_AGENT_KEY, "bad agent");
  listener = mock(ClientStreamListener.class);
  Mockito.reset(writeQueue);
  ChannelPromise completedPromise = new DefaultChannelPromise(channel)
      .setSuccess();
  when(writeQueue.enqueue(any(QueuedCommand.class), anyBoolean())).thenReturn(completedPromise);

  stream = new NettyClientStream(
      new TransportStateImpl(handler, DEFAULT_MAX_MESSAGE_SIZE),
      methodDescriptor,
      new Metadata(),
      channel,
      AsciiString.of("localhost"),
      AsciiString.of("http"),
      AsciiString.of("good agent"),
      StatsTraceContext.NOOP,
      transportTracer,
      CallOptions.DEFAULT,
      false);
  stream.start(listener);

  ArgumentCaptor<CreateStreamCommand> cmdCap = ArgumentCaptor.forClass(CreateStreamCommand.class);
  verify(writeQueue).enqueue(cmdCap.capture(), eq(false));
  assertThat(ImmutableListMultimap.copyOf(cmdCap.getValue().headers()))
      .containsEntry(Utils.USER_AGENT, AsciiString.of("good agent"));
}

Example 14

Source File: HttpConversionUtil.java From netty-4.1.22 with Apache License 2.0

4 votes

public static void toHttp2Headers(HttpHeaders inHeaders, Http2Headers out) {
    Iterator<Entry<CharSequence, CharSequence>> iter = inHeaders.iteratorCharSequence();
    // Choose 8 as a default size because it is unlikely we will see more than 4 Connection headers values, but
    // still allowing for "enough" space in the map to reduce the chance of hash code collision.
    CharSequenceMap<AsciiString> connectionBlacklist =
        toLowercaseMap(inHeaders.valueCharSequenceIterator(CONNECTION), 8);
    while (iter.hasNext()) {
        Entry<CharSequence, CharSequence> entry = iter.next();
        final AsciiString aName = AsciiString.of(entry.getKey()).toLowerCase();
        if (!HTTP_TO_HTTP2_HEADER_BLACKLIST.contains(aName) && !connectionBlacklist.contains(aName)) {
            // https://tools.ietf.org/html/rfc7540#section-8.1.2.2 makes a special exception for TE
            if (aName.contentEqualsIgnoreCase(TE)) {
                toHttp2HeadersFilterTE(entry, out);
            } else if (aName.contentEqualsIgnoreCase(COOKIE)) {
                AsciiString value = AsciiString.of(entry.getValue());
                // split up cookies to allow for better compression
                // https://tools.ietf.org/html/rfc7540#section-8.1.2.5
                try {
                    int index = value.forEachByte(FIND_SEMI_COLON);
                    if (index != -1) {
                        int start = 0;
                        do {
                            out.add(COOKIE, value.subSequence(start, index, false));
                            // skip 2 characters "; " (see https://tools.ietf.org/html/rfc6265#section-4.2.1)
                            start = index + 2;
                        } while (start < value.length() &&
                                (index = value.forEachByte(start, value.length() - start, FIND_SEMI_COLON)) != -1);
                        if (start >= value.length()) {
                            throw new IllegalArgumentException("cookie value is of unexpected format: " + value);
                        }
                        out.add(COOKIE, value.subSequence(start, value.length(), false));
                    } else {
                        out.add(COOKIE, value);
                    }
                } catch (Exception e) {
                    // This is not expect to happen because FIND_SEMI_COLON never throws but must be caught
                    // because of the ByteProcessor interface.
                    throw new IllegalStateException(e);
                }
            } else {
                out.add(aName, entry.getValue());
            }
        }
    }
}

Example 15

Source File: SmtpCommand.java From netty-4.1.22 with Apache License 2.0

4 votes

/**
 * Returns the {@link SmtpCommand} for the given command name.
 */
public static SmtpCommand valueOf(CharSequence commandName) {
    ObjectUtil.checkNotNull(commandName, "commandName");
    SmtpCommand command = COMMANDS.get(commandName.toString());
    return command != null ? command : new SmtpCommand(AsciiString.of(commandName));
}

Example 16

Source File: ViaHeaderAppendingInterceptor.java From styx with Apache License 2.0

4 votes

public ViaHeaderAppendingInterceptor(final String via) {
    final String value = isBlank(via) ? DEFAULT_VIA : via;
    via10 = AsciiString.of("1.0 " + value);
    via11 = AsciiString.of("1.1 " + value);
}

Example 17

Source File: GrpcServiceServerTest.java From armeria with Apache License 2.0

4 votes

@BeforeAll
static void createLargePayload() {
    LARGE_PAYLOAD = AsciiString.of(Strings.repeat("a", MAX_MESSAGE_SIZE + 1));
}

Example 18

Source File: NettyClientStream.java From grpc-java with Apache License 2.0

4 votes

@Override
public void setAuthority(String authority) {
  this.authority = AsciiString.of(checkNotNull(authority, "authority"));
}

Example 19

Source File: NettyClientStreamTest.java From grpc-java with Apache License 2.0

4 votes

@Test
public void getRequestSentThroughHeader() {
  // Creating a GET method
  MethodDescriptor<?, ?> descriptor = MethodDescriptor.<Void, Void>newBuilder()
      .setType(MethodDescriptor.MethodType.UNARY)
      .setFullMethodName("testService/test")
      .setRequestMarshaller(marshaller)
      .setResponseMarshaller(marshaller)
      .setIdempotent(true)
      .setSafe(true)
      .build();
  NettyClientStream stream = new NettyClientStream(
      new TransportStateImpl(handler, DEFAULT_MAX_MESSAGE_SIZE),
      descriptor,
      new Metadata(),
      channel,
      AsciiString.of("localhost"),
      AsciiString.of("http"),
      AsciiString.of("agent"),
      StatsTraceContext.NOOP,
      transportTracer,
      CallOptions.DEFAULT,
      true);
  stream.start(listener);
  stream.transportState().setId(STREAM_ID);
  stream.transportState().setHttp2Stream(http2Stream);

  byte[] msg = smallMessage();
  stream.writeMessage(new ByteArrayInputStream(msg));
  stream.flush();
  stream.halfClose();
  ArgumentCaptor<CreateStreamCommand> cmdCap = ArgumentCaptor.forClass(CreateStreamCommand.class);
  verify(writeQueue).enqueue(cmdCap.capture(), eq(true));
  ImmutableListMultimap<CharSequence, CharSequence> headers =
      ImmutableListMultimap.copyOf(cmdCap.getValue().headers());
  assertThat(headers).containsEntry(AsciiString.of(":method"), Utils.HTTP_GET_METHOD);
  assertThat(headers)
      .containsEntry(
          AsciiString.of(":path"),
          AsciiString.of("/testService/test?" + BaseEncoding.base64().encode(msg)));
}

Example 20

Source File: NettyClientStream.java From grpc-nebula-java with Apache License 2.0

4 votes

@Override
public void setAuthority(String authority) {
  this.authority = AsciiString.of(checkNotNull(authority, "authority"));
}