Search in sources :

Example 11 with Header

use of com.navercorp.pinpoint.io.header.Header in project pinpoint by naver.

the class ChunkHeaderBufferedTBaseSerializer method write.

// write chunk header + header + body
private void write(final TBase<?, ?> base, final String fieldName, final List<ByteArrayOutput> list) throws TException {
    final TReplaceListProtocol protocol = new TReplaceListProtocol(protocolFactory.getProtocol(transport));
    // write chunk header
    writeChunkHeader(protocol);
    // write header
    final Header header = locator.headerLookup(base);
    if (header == null) {
        throw new TException("header must not be null base:" + base);
    }
    HeaderUtils.writeHeader(protocol, header);
    if (CollectionUtils.hasLength(list)) {
        protocol.addReplaceField(fieldName, list);
    }
    base.write(protocol);
    if (isNeedFlush()) {
        flush();
    }
}
Also used : TException(org.apache.thrift.TException) Header(com.navercorp.pinpoint.io.header.Header)

Example 12 with Header

use of com.navercorp.pinpoint.io.header.Header in project pinpoint by naver.

the class HeaderTest method testGetSignature.

@Test
public void testGetSignature() {
    Header header = new HeaderV1((short) 1);
    byte signature = header.getSignature();
    short type = header.getType();
    byte version = header.getVersion();
}
Also used : Header(com.navercorp.pinpoint.io.header.Header) HeaderV1(com.navercorp.pinpoint.io.header.v1.HeaderV1) Test(org.junit.Test)

Example 13 with Header

use of com.navercorp.pinpoint.io.header.Header in project pinpoint by naver.

the class HeaderUtilsTest method validateSignature_error.

@Test(expected = InvalidHeaderException.class)
public void validateSignature_error() throws TException {
    Header error = new HeaderV1((byte) 0x11, (byte) 0x20, (short) 1);
    Assert.assertNotEquals(HeaderUtils.validateSignature(error.getSignature()), HeaderUtils.OK);
    logger.debug(error.toString());
}
Also used : Header(com.navercorp.pinpoint.io.header.Header) HeaderV1(com.navercorp.pinpoint.io.header.v1.HeaderV1) Test(org.junit.Test)

Example 14 with Header

use of com.navercorp.pinpoint.io.header.Header in project pinpoint by naver.

the class TCommandRegistryTest method registryTest5.

@Test
public void registryTest5() throws TException {
    TypeLocator<TBase<?, ?>> registry = TCommandRegistry.build(TCommandTypeVersion.V_1_0_2_SNAPSHOT);
    Header header = registry.headerLookup(new TResult());
    Assert.assertNotNull(header);
}
Also used : Header(com.navercorp.pinpoint.io.header.Header) TBase(org.apache.thrift.TBase) TResult(com.navercorp.pinpoint.thrift.dto.TResult) Test(org.junit.Test)

Example 15 with Header

use of com.navercorp.pinpoint.io.header.Header in project pinpoint by naver.

the class HeaderTBaseSerializer method writeHeader.

private void writeHeader(TBase<?, ?> base, HeaderEntity headerEntity) {
    try {
        final Header header = locator.headerLookup(base);
        if (header == null) {
            throw new TException("header must not be null base:" + base);
        }
        HeaderWriter headerWriter = new ByteArrayHeaderWriter(header, headerEntity);
        byte[] headerBytes = headerWriter.writeHeader();
        baos.write(headerBytes);
    } catch (Exception e) {
        throw new InvalidHeaderException("can not write header.", e);
    }
}
Also used : TException(org.apache.thrift.TException) Header(com.navercorp.pinpoint.io.header.Header) ByteArrayHeaderWriter(com.navercorp.pinpoint.io.header.ByteArrayHeaderWriter) HeaderWriter(com.navercorp.pinpoint.io.header.HeaderWriter) ByteArrayHeaderWriter(com.navercorp.pinpoint.io.header.ByteArrayHeaderWriter) InvalidHeaderException(com.navercorp.pinpoint.io.header.InvalidHeaderException) TException(org.apache.thrift.TException) InvalidHeaderException(com.navercorp.pinpoint.io.header.InvalidHeaderException) UnsupportedEncodingException(java.io.UnsupportedEncodingException)

Aggregations

Header (com.navercorp.pinpoint.io.header.Header)18 DefaultMessage (com.navercorp.pinpoint.io.request.DefaultMessage)6 HeaderEntity (com.navercorp.pinpoint.io.header.HeaderEntity)5 TException (org.apache.thrift.TException)5 HeaderV2 (com.navercorp.pinpoint.io.header.v2.HeaderV2)4 Test (org.junit.Test)4 HeaderV1 (com.navercorp.pinpoint.io.header.v1.HeaderV1)3 ByteArrayHeaderReader (com.navercorp.pinpoint.io.header.ByteArrayHeaderReader)2 HeaderReader (com.navercorp.pinpoint.io.header.HeaderReader)2 TBase (org.apache.thrift.TBase)2 ByteArrayHeaderWriter (com.navercorp.pinpoint.io.header.ByteArrayHeaderWriter)1 HeaderWriter (com.navercorp.pinpoint.io.header.HeaderWriter)1 InvalidHeaderException (com.navercorp.pinpoint.io.header.InvalidHeaderException)1 Message (com.navercorp.pinpoint.io.request.Message)1 TResult (com.navercorp.pinpoint.thrift.dto.TResult)1 UnsupportedEncodingException (java.io.UnsupportedEncodingException)1 ArrayList (java.util.ArrayList)1 TProtocol (org.apache.thrift.protocol.TProtocol)1