Examples with TBase org.apache.thrift.TBase used on opensource projects

Example 11 with TBase

use of org.apache.thrift.TBase in project pinpoint by naver.

the class ChunkHeaderTBaseDeserializerTest method deserialize.

@Test
public void deserialize() throws Exception {
    final ChunkHeaderTBaseDeserializer deserializer = new ChunkHeaderTBaseDeserializer(DEFAULT_PROTOCOL_FACTORY, DEFAULT_TBASE_LOCATOR);
    UnsafeByteArrayOutputStream out = new UnsafeByteArrayOutputStream();
    ChunkHeaderBufferedTBaseSerializer serializer = new ChunkHeaderBufferedTBaseSerializer(out, DEFAULT_PROTOCOL_FACTORY, DEFAULT_TBASE_LOCATOR);
    TSpanChunk chunk = new TSpanMockBuilder().buildChunk(3, 10);
    serializer.add(chunk);
    List<TBase<?, ?>> list = deserializer.deserialize(serializer.getTransport().getBuffer(), 0, serializer.getTransport().getBufferPosition());
    assertEquals(1, list.size());
    TSpanChunk result = (TSpanChunk) list.get(0);
    assertEquals(3, result.getSpanEventList().size());
}

Example 12 with TBase

use of org.apache.thrift.TBase in project pinpoint by naver.

the class TCommandRegistryTest method registryTest6.

@Test
public void registryTest6() throws TException {
    TCommandRegistry registry = new TCommandRegistry(TCommandTypeVersion.V_1_0_2_SNAPSHOT);
    TBase tBase = registry.tBaseLookup(TCommandType.RESULT.getCode());
    Assert.assertEquals(tBase.getClass(), TResult.class);
    tBase = registry.tBaseLookup(TCommandType.THREAD_DUMP.getCode());
    Assert.assertEquals(tBase.getClass(), TCommandThreadDump.class);
}

Example 13 with TBase

use of org.apache.thrift.TBase in project pinpoint by naver.

the class SpanChunkStreamSendDataPlanerTest method deserialize.

private List<TSpanEvent> deserialize(byte[] data) throws TException {
    ByteBuffer bb = ByteBuffer.wrap(data);
    bb.get();
    bb.get();
    int chunkSize = bb.get();
    List<TSpanEvent> eventList = new ArrayList<TSpanEvent>();
    for (int i = 0; i < chunkSize; i++) {
        short componentSize = bb.getShort();
        byte[] component = new byte[componentSize];
        bb.get(component);
        HeaderTBaseDeserializer deserialize = new HeaderTBaseDeserializerFactory().createDeserializer();
        List<TBase<?, ?>> value = deserialize.deserializeList(component);
        for (int j = 0; j < value.size(); j++) {
            TBase tbase = value.get(j);
            if (tbase instanceof TSpanEvent) {
                eventList.add((TSpanEvent) tbase);
            } else {
            }
        }
    }
    return eventList;
}

Example 14 with TBase

use of org.apache.thrift.TBase in project pinpoint by naver.

the class SpanStreamSendDataPlanerTest method deserialize.

private List<TSpanEvent> deserialize(byte[] data) throws TException {
    ByteBuffer bb = ByteBuffer.wrap(data);
    bb.get();
    bb.get();
    int chunkSize = bb.get();
    List<TSpanEvent> eventList = new ArrayList<TSpanEvent>();
    for (int i = 0; i < chunkSize; i++) {
        short componentSize = bb.getShort();
        byte[] component = new byte[componentSize];
        bb.get(component);
        HeaderTBaseDeserializer deserialize = new HeaderTBaseDeserializerFactory().createDeserializer();
        List<TBase<?, ?>> value = deserialize.deserializeList(component);
        for (int j = 0; j < value.size(); j++) {
            TBase tbase = value.get(j);
            if (tbase instanceof TSpanEvent) {
                eventList.add((TSpanEvent) tbase);
            } else {
            }
        }
    }
    return eventList;
}

Example 15 with TBase

use of org.apache.thrift.TBase in project brisk by riptano.

the class CassandraHiveMetaStore method getTables.

/**
     * Retrieve the tables for the given database and pattern.
     * 
     * @param dbName
     * @param tableNamePattern the pattern passed as is to {@link String#matches(String)} of
     * {@link Table#getTableName()}
     */
public List<String> getTables(String dbName, String tableNamePattern) throws MetaException {
    log.info("in getTables with dbName: {} and tableNamePattern: {}", dbName, tableNamePattern);
    if (schemaManagerService.getAutoCreateSchema()) {
        KsDef ksDef = schemaManagerService.getKeyspaceForDatabaseName(dbName);
        if (ksDef != null) {
            log.debug("Checking for new column families to add");
            schemaManagerService.createNewColumnFamilyTables(ksDef);
        }
    }
    List<TBase> tables = metaStorePersister.find(new Table(), dbName);
    List<String> results = new ArrayList<String>(tables.size());
    for (TBase tBase : tables) {
        Table table = (Table) tBase;
        if (StringUtils.isEmpty(tableNamePattern) || table.getTableName().matches(tableNamePattern))
            results.add(table.getTableName());
    }
    return results;
}