Examples with BatchEntityResponse com.linkedin.restli.internal.client.response.BatchEntityResponse used on opensource projects

Example 6 with BatchEntityResponse

use of com.linkedin.restli.internal.client.response.BatchEntityResponse in project rest.li by linkedin.

the class MockBatchEntityResponseFactory method createWithComplexKey.

/**
 * Create a {@link BatchKVResponse} where the key is a {@link ComplexResourceKey}
 *
 * @param valueClass the value class
 * @param keyKeyClass the class of the key part of the {@link ComplexResourceKey}
 * @param keyParamsClass the class of the params part of the {@link ComplexResourceKey}
 * @param recordTemplates the data that will be returned for a call to {@link com.linkedin.restli.client.response.BatchKVResponse#getResults()}
 *                        NOTE: the params part of the {@link ComplexResourceKey} is removed in this map. A new
 *                        instance of the params class is created with no data in it.
 * @param statuses The HTTP status codes that will be returned as part of {@link EntityResponse}s
 *                 returned in {@link com.linkedin.restli.client.response.BatchKVResponse#getResults()}
 * @param errorResponses the data that will be returned for a call to {@link com.linkedin.restli.client.response.BatchKVResponse#getErrors()}
 *                       NOTE: the params part of the {@link ComplexResourceKey} is removed in this map. A new
 *                       instance of the params class is created with no data in it.
 * @param <V>
 */
@SuppressWarnings("rawtypes")
public static <KK extends RecordTemplate, KP extends RecordTemplate, V extends RecordTemplate> BatchKVResponse<ComplexResourceKey<KK, KP>, EntityResponse<V>> createWithComplexKey(Class<V> valueClass, Class<KK> keyKeyClass, Class<KP> keyParamsClass, Map<ComplexResourceKey<KK, KP>, V> recordTemplates, Map<ComplexResourceKey<KK, KP>, HttpStatus> statuses, Map<ComplexResourceKey<KK, KP>, ErrorResponse> errorResponses) {
    ProtocolVersion version = AllProtocolVersions.BASELINE_PROTOCOL_VERSION;
    DataMap batchResponseDataMap = buildDataMap(recordTemplates, statuses, errorResponses, version);
    @SuppressWarnings("unchecked") BatchKVResponse<ComplexResourceKey<KK, KP>, EntityResponse<V>> response = (BatchKVResponse<ComplexResourceKey<KK, KP>, EntityResponse<V>>) (Object) new BatchEntityResponse<>(batchResponseDataMap, new TypeSpec<>(ComplexResourceKey.class), TypeSpec.forClassMaybeNull(valueClass), null, ComplexKeySpec.forClassesMaybeNull(keyKeyClass, keyParamsClass), version);
    return response;
}