コード例 #1
0
  @Test
  public void should_set_primaryKey_to_true_when_embedded_id() throws Exception {
    // Given
    EntityParsingContext entityContext = mock(EntityParsingContext.class, RETURNS_DEEP_STUBS);
    when(entityContext.getCurrentEntityClass()).thenReturn((Class) CompleteBean.class);
    Field field = CompleteBean.class.getDeclaredField("age");

    PropertyParsingContext context = new PropertyParsingContext(entityContext, field);

    // When
    context.setEmbeddedId(true);

    // Then
    assertThat(context.isEmbeddedId()).isTrue();
    assertThat(context.isPrimaryKey()).isTrue();
    assertThat(context.<CompleteBean>getCurrentEntityClass()).isEqualTo(CompleteBean.class);
    assertThat(context.getCurrentPropertyName()).isEqualTo("age_in_years");
  }
コード例 #2
0
ファイル: PropertyParser.java プロジェクト: relgames/Achilles
  public PropertyMeta parse(PropertyParsingContext context) {
    log.debug(
        "Parsing property {} of entity class {}",
        context.getCurrentPropertyName(),
        context.getCurrentEntityClass().getCanonicalName());

    Field field = context.getCurrentField();
    inferPropertyName(context);
    context.setCustomConsistencyLevels(hasConsistencyAnnotation(context.getCurrentField()));

    validator.validateNoDuplicate(context);
    validator.validateIndexIfSet(context);

    Class<?> fieldType = field.getType();
    PropertyMeta propertyMeta;

    if (List.class.isAssignableFrom(fieldType)) {
      propertyMeta = parseListProperty(context);
    } else if (Set.class.isAssignableFrom(fieldType)) {
      propertyMeta = parseSetProperty(context);
    } else if (Map.class.isAssignableFrom(fieldType)) {
      propertyMeta = parseMapProperty(context);
    } else if (Counter.class.isAssignableFrom(fieldType)) {
      propertyMeta = parseCounterProperty(context);
    } else if (context.isEmbeddedId()) {
      propertyMeta = parseEmbeddedId(context);
    } else if (context.isPrimaryKey()) {
      propertyMeta = parseId(context);
    } else {
      propertyMeta = parseSimpleProperty(context);
      String indexName = getIndexName(field);
      if (indexName != null) {
        propertyMeta.setIndexProperties(new IndexProperties(indexName));
      }
    }
    context.getPropertyMetas().put(context.getCurrentPropertyName(), propertyMeta);
    return propertyMeta;
  }