Examples with FieldDef com.linkedin.data.template.FieldDef used on opensource projects

Example 16 with FieldDef

use of com.linkedin.data.template.FieldDef in project rest.li by linkedin.

the class TestAllPartitionsRequestBuilder method testSendAllPartitionsRequests.

@Test(dataProvider = com.linkedin.restli.internal.common.TestConstants.RESTLI_PROTOCOL_1_2_PREFIX + "restliRequestOptions")
public void testSendAllPartitionsRequests(RestliRequestOptions options, RingFactory<URI> ringFactory) throws ServiceUnavailableException, URISyntaxException, RestException, InterruptedException {
    final int PARTITION_NUM = 5;
    List<URI> expectedUris = new ArrayList<URI>();
    ConsistentHashKeyMapper mapper = getKeyToHostMapper(PARTITION_NUM, expectedUris, ringFactory);
    AllPartitionsRequestBuilder<Greeting> searchRB = new AllPartitionsRequestBuilder<Greeting>(mapper);
    ActionRequestBuilder<Long, Greeting> builder = new ActionRequestBuilder<Long, Greeting>(TEST_URI, Greeting.class, _COLL_SPEC, options);
    ActionRequest<Greeting> request = builder.name("updateTone").id(1L).setParam(new FieldDef<Tone>("newTone", Tone.class, DataTemplateUtil.getSchema(Tone.class)), Tone.FRIENDLY).build();
    final Map<String, Greeting> results = new ConcurrentHashMap<String, Greeting>();
    final CountDownLatch latch = new CountDownLatch(PARTITION_NUM);
    final List<Throwable> errors = new ArrayList<Throwable>();
    final List<Greeting> responses = new ArrayList<Greeting>();
    Callback<Response<Greeting>> cb = new Callback<Response<Greeting>>() {

        @Override
        public void onError(Throwable e) {
            synchronized (errors) {
                errors.add(e);
            }
            latch.countDown();
        }

        @Override
        public void onSuccess(Response<Greeting> response) {
            results.put(response.getEntity().toString(), response.getEntity());
            synchronized (responses) {
                responses.add(response.getEntity());
            }
            latch.countDown();
        }
    };
    HostSet hostsResult = searchRB.sendRequests(getClient(), request, new RequestContext(), cb);
    List<URI> uris = hostsResult.getAllHosts();
    Assert.assertTrue(uris.containsAll(expectedUris));
    Assert.assertTrue(expectedUris.containsAll(uris));
    latch.await();
    if (!errors.isEmpty()) {
        Assert.fail("I knew it: " + errors.toString());
    }
    Assert.assertEquals(PARTITION_NUM, responses.size());
}

Example 17 with FieldDef

use of com.linkedin.data.template.FieldDef in project rest.li by linkedin.

the class ResourceSchemaToResourceSpecTranslator method toActionMetadata.

// this is dynamic, don't have concrete classes for the FieldDef
@SuppressWarnings({ "unchecked", "rawtypes" })
private ActionMetadata toActionMetadata(ActionSchema action) {
    ArrayList<FieldDef<?>> fieldDefs = new ArrayList<FieldDef<?>>();
    if (action.hasParameters()) {
        for (ParameterSchema parameterSchema : action.getParameters()) {
            DataSchema dataSchema = RestSpecCodec.textToSchema(parameterSchema.getType(), _schemaResolver);
            Class<?> paramClass = toType(dataSchema);
            FieldDef<?> fieldDef = new FieldDef(parameterSchema.getName(), paramClass, dataSchema);
            fieldDefs.add(fieldDef);
        }
    }
    Collection<FieldDef<?>> response;
    if (action.hasReturns()) {
        DataSchema returnType = RestSpecCodec.textToSchema(action.getReturns(), _schemaResolver);
        Class<?> returnClass = toType(returnType);
        response = Collections.<FieldDef<?>>singletonList(new FieldDef("value", returnClass, returnType));
    } else {
        response = Collections.emptyList();
    }
    return new ActionMetadata(action.getName(), new DynamicRecordMetadata(action.getName(), fieldDefs), new DynamicRecordMetadata(action.getName(), response));
}

Example 18 with FieldDef

use of com.linkedin.data.template.FieldDef in project rest.li by linkedin.

the class ExampleRequestResponseGenerator method addParams.

private void addParams(ActionRequestBuilder<?, ?> request, DynamicRecordMetadata requestMetadata, ParameterSchemaArray parameters) {
    if (parameters != null) {
        for (ParameterSchema parameter : parameters) {
            FieldDef<?> fieldDef = requestMetadata.getFieldDef(parameter.getName());
            Object value = generateFieldDefValue(fieldDef);
            // to the dereferenced type so the example values can be set on the request without coercing.
            if (fieldDef.getDataSchema().getType() == Type.TYPEREF) {
                FieldDef<?> deRefFieldDef = new FieldDef<>(fieldDef.getName(), fieldDef.getDataClass(), fieldDef.getDataSchema().getDereferencedDataSchema());
                deRefFieldDef.getField().setRecord(fieldDef.getField().getRecord());
                fieldDef = deRefFieldDef;
            }
            request.setParam(fieldDef, value);
        }
    }
}