Examples with Struct org.apache.kafka.common.protocol.types.Struct used on opensource projects

Example 61 with Struct

use of org.apache.kafka.common.protocol.types.Struct in project kafka by apache.

the class ConnectProtocol method deserializeMetadata.

public static WorkerState deserializeMetadata(ByteBuffer buffer) {
    Struct header = CONNECT_PROTOCOL_HEADER_SCHEMA.read(buffer);
    Short version = header.getShort(VERSION_KEY_NAME);
    checkVersionCompatibility(version);
    Struct struct = CONFIG_STATE_V0.read(buffer);
    long configOffset = struct.getLong(CONFIG_OFFSET_KEY_NAME);
    String url = struct.getString(URL_KEY_NAME);
    return new WorkerState(url, configOffset);
}

Example 62 with Struct

use of org.apache.kafka.common.protocol.types.Struct in project kafka by apache.

the class ConnectProtocol method deserializeAssignment.

public static Assignment deserializeAssignment(ByteBuffer buffer) {
    Struct header = CONNECT_PROTOCOL_HEADER_SCHEMA.read(buffer);
    Short version = header.getShort(VERSION_KEY_NAME);
    checkVersionCompatibility(version);
    Struct struct = ASSIGNMENT_V0.read(buffer);
    short error = struct.getShort(ERROR_KEY_NAME);
    String leader = struct.getString(LEADER_KEY_NAME);
    String leaderUrl = struct.getString(LEADER_URL_KEY_NAME);
    long offset = struct.getLong(CONFIG_OFFSET_KEY_NAME);
    List<String> connectorIds = new ArrayList<>();
    List<ConnectorTaskId> taskIds = new ArrayList<>();
    for (Object structObj : struct.getArray(ASSIGNMENT_KEY_NAME)) {
        Struct assignment = (Struct) structObj;
        String connector = assignment.getString(CONNECTOR_KEY_NAME);
        for (Object taskIdObj : assignment.getArray(TASKS_KEY_NAME)) {
            Integer taskId = (Integer) taskIdObj;
            if (taskId == CONNECTOR_TASK)
                connectorIds.add(connector);
            else
                taskIds.add(new ConnectorTaskId(connector, taskId));
        }
    }
    return new Assignment(error, leader, leaderUrl, offset, connectorIds, taskIds);
}

Example 63 with Struct

use of org.apache.kafka.common.protocol.types.Struct in project kafka by apache.

the class ConsumerProtocol method serializeSubscription.

public static ByteBuffer serializeSubscription(PartitionAssignor.Subscription subscription) {
    Struct struct = new Struct(SUBSCRIPTION_V0);
    struct.set(USER_DATA_KEY_NAME, subscription.userData());
    struct.set(TOPICS_KEY_NAME, subscription.topics().toArray());
    ByteBuffer buffer = ByteBuffer.allocate(CONSUMER_PROTOCOL_HEADER_V0.sizeOf() + SUBSCRIPTION_V0.sizeOf(struct));
    CONSUMER_PROTOCOL_HEADER_V0.writeTo(buffer);
    SUBSCRIPTION_V0.write(buffer, struct);
    buffer.flip();
    return buffer;
}