@Test public void handleMessageReceived() { Type type = Type.RECEIVED; long uncompressed = 456L; HttpRequestContext context = new HttpRequestContext(fakeSpan, tagContext); handler.handleMessageReceived(context, uncompressed); verify(fakeSpan).addMessageEvent(captor.capture()); MessageEvent messageEvent = captor.getValue(); assertThat(messageEvent.getType()).isEqualTo(type); assertThat(messageEvent.getMessageId()).isEqualTo(1L); assertThat(messageEvent.getUncompressedMessageSize()).isEqualTo(uncompressed); assertThat(messageEvent.getCompressedMessageSize()).isEqualTo(0); }
@Test public void handleMessageSent() { Type type = Type.SENT; long uncompressed = 456L; HttpRequestContext context = new HttpRequestContext(fakeSpan, tagContext); handler.handleMessageSent(context, uncompressed); verify(fakeSpan).addMessageEvent(captor.capture()); MessageEvent messageEvent = captor.getValue(); assertThat(messageEvent.getType()).isEqualTo(type); assertThat(messageEvent.getMessageId()).isEqualTo(1L); assertThat(messageEvent.getUncompressedMessageSize()).isEqualTo(uncompressed); assertThat(messageEvent.getCompressedMessageSize()).isEqualTo(0); }
@Test public void buildMessageEvent_WithCompressedMessageSize() { MessageEvent messageEvent = MessageEvent.builder(MessageEvent.Type.SENT, 1L).setCompressedMessageSize(123L).build(); assertThat(messageEvent.getType()).isEqualTo(MessageEvent.Type.SENT); assertThat(messageEvent.getMessageId()).isEqualTo(1L); assertThat(messageEvent.getCompressedMessageSize()).isEqualTo(123L); }
@Test public void buildMessageEvent_WithAllValues() { MessageEvent messageEvent = MessageEvent.builder(MessageEvent.Type.RECEIVED, 1L) .setUncompressedMessageSize(123L) .setCompressedMessageSize(63L) .build(); assertThat(messageEvent.getType()).isEqualTo(MessageEvent.Type.RECEIVED); assertThat(messageEvent.getMessageId()).isEqualTo(1L); assertThat(messageEvent.getUncompressedMessageSize()).isEqualTo(123L); assertThat(messageEvent.getCompressedMessageSize()).isEqualTo(63L); }
/** * Cast or convert a {@link io.opencensus.trace.BaseMessageEvent} to {@link * io.opencensus.trace.NetworkEvent}. * * @param event the {@code BaseMessageEvent} that is being cast or converted. * @return a {@code io.opencensus.trace.NetworkEvent} representation of the input. */ public static io.opencensus.trace.NetworkEvent asNetworkEvent( io.opencensus.trace.BaseMessageEvent event) { Utils.checkNotNull(event, "event"); if (event instanceof io.opencensus.trace.NetworkEvent) { return (io.opencensus.trace.NetworkEvent) event; } io.opencensus.trace.MessageEvent messageEvent = (io.opencensus.trace.MessageEvent) event; io.opencensus.trace.NetworkEvent.Type type = (messageEvent.getType() == io.opencensus.trace.MessageEvent.Type.RECEIVED) ? io.opencensus.trace.NetworkEvent.Type.RECV : io.opencensus.trace.NetworkEvent.Type.SENT; return io.opencensus.trace.NetworkEvent.builder(type, messageEvent.getMessageId()) .setUncompressedMessageSize(messageEvent.getUncompressedMessageSize()) .setCompressedMessageSize(messageEvent.getCompressedMessageSize()) .build(); }
final Tag tagCompressedSize = new Tag(MESSAGE_EVENT_COMPRESSED_SIZE, TagType.LONG) .setVLong(event.getEvent().getCompressedMessageSize()); final Tag tagUncompressedSize = new Tag(MESSAGE_EVENT_UNCOMPRESSED_SIZE, TagType.LONG)
private static TimeEvent toTimeMessageEventProto( TimedEvent<io.opencensus.trace.MessageEvent> timedEvent) { TimeEvent.Builder timeEventBuilder = TimeEvent.newBuilder().setTime(toTimestampProto(timedEvent.getTimestamp())); io.opencensus.trace.MessageEvent messageEvent = timedEvent.getEvent(); timeEventBuilder.setMessageEvent( TimeEvent.MessageEvent.newBuilder() .setId(messageEvent.getMessageId()) .setCompressedSize(messageEvent.getCompressedMessageSize()) .setUncompressedSize(messageEvent.getUncompressedMessageSize()) .setType(toMessageEventTypeProto(messageEvent)) .build()); return timeEventBuilder.build(); }
private static TimeEvent toTimeMessageEventProto( TimedEvent<io.opencensus.trace.MessageEvent> timedEvent) { TimeEvent.Builder timeEventBuilder = TimeEvent.newBuilder().setTime(toTimestampProto(timedEvent.getTimestamp())); io.opencensus.trace.MessageEvent messageEvent = timedEvent.getEvent(); timeEventBuilder.setMessageEvent( TimeEvent.MessageEvent.newBuilder() .setId(messageEvent.getMessageId()) .setCompressedSizeBytes(messageEvent.getCompressedMessageSize()) .setUncompressedSizeBytes(messageEvent.getUncompressedMessageSize()) .setType(toMessageEventTypeProto(messageEvent)) .build()); return timeEventBuilder.build(); }
private static TimeEvent toTimeMessageEventProto( TimedEvent<io.opencensus.trace.MessageEvent> timedEvent) { TimeEvent.Builder timeEventBuilder = TimeEvent.newBuilder().setTime(toTimestampProto(timedEvent.getTimestamp())); io.opencensus.trace.MessageEvent messageEvent = timedEvent.getEvent(); timeEventBuilder.setMessageEvent( TimeEvent.MessageEvent.newBuilder() .setId(messageEvent.getMessageId()) .setCompressedSizeBytes(messageEvent.getCompressedMessageSize()) .setUncompressedSizeBytes(messageEvent.getUncompressedMessageSize()) .setType(toMessageEventTypeProto(messageEvent)) .build()); return timeEventBuilder.build(); }