Package org.kitesdk.data

Examples of org.kitesdk.data.ColumnMapping$Builder


    );
  }

  @Test
  public void testAddEmbeddedColumnMapping() {
    ColumnMapping mapping = new ColumnMapping.Builder()
        .key("id")
        .column("username", "u", "username")
        .column("real_name", "u", "name")
        .build();
    Schema original = new Schema.Parser().parse("{" +
View Full Code Here


    junit.framework.Assert.assertEquals(mapping, ColumnMappingParser.parseFromSchema(embedded));
  }

  @Test
  public void testReplaceEmbeddedPartitionStrategy() {
    ColumnMapping mapping = new ColumnMapping.Builder()
        .key("id")
        .column("username", "u", "username")
        .column("real_name", "u", "name")
        .build();
    Schema original = new Schema.Parser().parse("{" +
View Full Code Here

        .size());
  }

  @Test
  public void testOverrideColumnMapping() {
    ColumnMapping desc = new ColumnMapping.Builder()
        .column("field1", "override", "field1")
        .counter("version", "override", "version").build();
    AvroEntitySchema avroEntitySchema = parser.parseEntitySchema(entitySchema,
        desc);
    desc = avroEntitySchema.getColumnMappingDescriptor();

    assertEquals(2, desc.getFieldMappings().size());
    assertEquals(FieldMapping.MappingType.COLUMN, desc.getFieldMapping("field1")
        .getMappingType());
    assertEquals(FieldMapping.MappingType.COUNTER, desc.getFieldMapping("version")
        .getMappingType());
  }
View Full Code Here

      List<FieldMapping> fieldMappings = new ArrayList<FieldMapping>();
      for (EntityMapper<?> entityMapper : entityMappers) {
        fieldMappings.addAll(entityMapper.getEntitySchema()
            .getColumnMappingDescriptor().getFieldMappings());
      }
      ColumnMapping mappingDescriptor = new ColumnMapping.Builder()
          .fieldMappings(fieldMappings).build();
      return new EntitySchema(null, null, mappingDescriptor);
    }
View Full Code Here

          .getResourceAsStream("/TestRecord.avsc"));

  @Test
  public void testGoodSchema() {
    AvroEntitySchema avroEntitySchema = parser.parseEntitySchema(entitySchema);
    ColumnMapping descriptor = avroEntitySchema
        .getColumnMappingDescriptor();
    assertEquals(9, descriptor.getFieldMappings().size());
    assertEquals(FieldMapping.MappingType.COLUMN, descriptor.getFieldMapping("field1")
        .getMappingType());
    assertEquals(FieldMapping.MappingType.KEY_AS_COLUMN, descriptor
        .getFieldMapping("field4").getMappingType());
    assertEquals(FieldMapping.MappingType.OCC_VERSION, descriptor.getFieldMapping("version")
        .getMappingType());

    AvroKeySchema avroKeySchema = parser.parseKeySchema(entitySchema);
    assertEquals(Type.STRING, avroKeySchema.getAvroSchema()
        .getField("keyPart1_copy").schema().getType());
View Full Code Here

TOP

Related Classes of org.kitesdk.data.ColumnMapping$Builder

Copyright © 2018 www.massapicom. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.