Java Code Examples for org.apache.pulsar.client.api.Schema#KeyValue

The following examples show how to use org.apache.pulsar.client.api.Schema#KeyValue . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example 1

Source File: TypedMessageBuilderImplTest.java From pulsar with Apache License 2.0

6 votes

@Test
public void testSeparatedValue() {
    producerBase = mock(ProducerBase.class);

    AvroSchema<SchemaTestUtils.Foo> fooSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Foo>builder().withPojo(SchemaTestUtils.Foo.class).build());
    AvroSchema<SchemaTestUtils.Bar> barSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Bar>builder().withPojo(SchemaTestUtils.Bar.class).build());

    Schema<KeyValue<SchemaTestUtils.Foo, SchemaTestUtils.Bar>> keyValueSchema = Schema.KeyValue(fooSchema, barSchema, KeyValueEncodingType.SEPARATED);
    TypedMessageBuilderImpl typedMessageBuilderImpl = new TypedMessageBuilderImpl(producerBase, keyValueSchema);

    SchemaTestUtils.Foo foo = new SchemaTestUtils.Foo();
    foo.setField1("field1");
    foo.setField2("field2");
    SchemaTestUtils.Bar bar = new SchemaTestUtils.Bar();
    bar.setField1(true);
    KeyValue<SchemaTestUtils.Foo, SchemaTestUtils.Bar> keyValue = new KeyValue<>(foo, bar);

    // Check kv.encoding.type SEPARATED
    TypedMessageBuilderImpl typedMessageBuilder = (TypedMessageBuilderImpl)typedMessageBuilderImpl.value(keyValue);
    ByteBuffer content = typedMessageBuilder.getContent();
    byte[] contentByte = new byte[content.remaining()];
    content.get(contentByte);
    assertTrue(typedMessageBuilderImpl.hasKey());
    assertEquals(typedMessageBuilderImpl.getKey(), Base64.getEncoder().encodeToString(fooSchema.encode(foo)));
    assertEquals(barSchema.decode(contentByte), bar);
}

Example 2

Source File: KeyValueSchemaTest.java From pulsar with Apache License 2.0

6 votes

@Test
public void testNotAllowNullAvroSchemaCreate() {
    AvroSchema<Foo> fooSchema = AvroSchema.of(SchemaDefinition.<Foo>builder().withPojo(Foo.class).withAlwaysAllowNull(false).build());
    AvroSchema<Bar> barSchema = AvroSchema.of(SchemaDefinition.<Bar>builder().withPojo(Bar.class).withAlwaysAllowNull(false).build());

    Schema<KeyValue<Foo, Bar>> keyValueSchema1 = Schema.KeyValue(fooSchema, barSchema);
    Schema<KeyValue<Foo, Bar>> keyValueSchema2 = Schema.KeyValue(AvroSchema.of(SchemaDefinition.<Foo>builder().withPojo(Foo.class).withAlwaysAllowNull(false).build()),
            AvroSchema.of(SchemaDefinition.<Bar>builder().withPojo(Bar.class).withAlwaysAllowNull(false).build()));

    assertEquals(keyValueSchema1.getSchemaInfo().getType(), SchemaType.KEY_VALUE);
    assertEquals(keyValueSchema2.getSchemaInfo().getType(), SchemaType.KEY_VALUE);

    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema1).getKeySchema().getSchemaInfo().getType(),
        SchemaType.AVRO);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema1).getValueSchema().getSchemaInfo().getType(),
        SchemaType.AVRO);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema2).getKeySchema().getSchemaInfo().getType(),
        SchemaType.AVRO);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema2).getValueSchema().getSchemaInfo().getType(),
        SchemaType.AVRO);

    String schemaInfo1 = new String(keyValueSchema1.getSchemaInfo().getSchema());
    String schemaInfo2 = new String(keyValueSchema2.getSchemaInfo().getSchema());
    assertEquals(schemaInfo1, schemaInfo2);
}

Example 3

Source File: KeyValueSchemaTest.java From pulsar with Apache License 2.0

6 votes

@Test
public void testFillParametersToSchemainfo() {
    AvroSchema<Foo> fooSchema = AvroSchema.of(SchemaDefinition.<Foo>builder().withPojo(Foo.class).build());
    AvroSchema<Bar> barSchema = AvroSchema.of(SchemaDefinition.<Bar>builder().withPojo(Bar.class).build());

    fooSchema.getSchemaInfo().setName("foo");
    fooSchema.getSchemaInfo().setType(SchemaType.AVRO);
    Map<String, String> keyProperties = Maps.newTreeMap();
    keyProperties.put("foo.key1", "value");
    keyProperties.put("foo.key2", "value");
    fooSchema.getSchemaInfo().setProperties(keyProperties);
    barSchema.getSchemaInfo().setName("bar");
    barSchema.getSchemaInfo().setType(SchemaType.AVRO);
    Map<String, String> valueProperties = Maps.newTreeMap();
    valueProperties.put("bar.key", "key");
    barSchema.getSchemaInfo().setProperties(valueProperties);
    Schema<KeyValue<Foo, Bar>> keyValueSchema1 = Schema.KeyValue(fooSchema, barSchema);

    assertEquals(keyValueSchema1.getSchemaInfo().getProperties().get("key.schema.name"), "foo");
    assertEquals(keyValueSchema1.getSchemaInfo().getProperties().get("key.schema.type"), String.valueOf(SchemaType.AVRO));
    assertEquals(keyValueSchema1.getSchemaInfo().getProperties().get("key.schema.properties"), "{\"foo.key1\":\"value\",\"foo.key2\":\"value\"}");
    assertEquals(keyValueSchema1.getSchemaInfo().getProperties().get("value.schema.name"), "bar");
    assertEquals(keyValueSchema1.getSchemaInfo().getProperties().get("value.schema.type"), String.valueOf(SchemaType.AVRO));
    assertEquals(keyValueSchema1.getSchemaInfo().getProperties().get("value.schema.properties"), "{\"bar.key\":\"key\"}");
}

Example 4

Source File: TypedMessageBuilderImplTest.java From pulsar with Apache License 2.0

6 votes

@Test
public void testSetKeyEncodingTypeSeparated() {
    producerBase = mock(ProducerBase.class);

    AvroSchema<SchemaTestUtils.Foo> fooSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Foo>builder().withPojo(SchemaTestUtils.Foo.class).build());
    AvroSchema<SchemaTestUtils.Bar> barSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Bar>builder().withPojo(SchemaTestUtils.Bar.class).build());

    Schema<KeyValue<SchemaTestUtils.Foo, SchemaTestUtils.Bar>> keyValueSchema = Schema.KeyValue(fooSchema, barSchema, KeyValueEncodingType.SEPARATED);
    TypedMessageBuilderImpl typedMessageBuilderImpl = new TypedMessageBuilderImpl(producerBase, keyValueSchema);


    try {
        TypedMessageBuilderImpl typedMessageBuilder = (TypedMessageBuilderImpl)typedMessageBuilderImpl.key("separated");
        fail("This should fail");
    } catch (IllegalArgumentException e) {
        assertTrue(e.getMessage().contains("This method is not allowed to set keys when in encoding type is SEPARATED"));
    }
}

Example 5

Source File: KeyValueSchemaTest.java From pulsar with Apache License 2.0

6 votes

@Test
public void testAllowNullAvroSchemaCreate() {
    AvroSchema<Foo> fooSchema = AvroSchema.of(SchemaDefinition.<Foo>builder().withPojo(Foo.class).build());
    AvroSchema<Bar> barSchema = AvroSchema.of(SchemaDefinition.<Bar>builder().withPojo(Bar.class).build());

    Schema<KeyValue<Foo, Bar>> keyValueSchema1 = Schema.KeyValue(fooSchema, barSchema);
    Schema<KeyValue<Foo, Bar>> keyValueSchema2 = Schema.KeyValue(Foo.class, Bar.class, SchemaType.AVRO);

    assertEquals(keyValueSchema1.getSchemaInfo().getType(), SchemaType.KEY_VALUE);
    assertEquals(keyValueSchema2.getSchemaInfo().getType(), SchemaType.KEY_VALUE);

    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema1).getKeySchema().getSchemaInfo().getType(),
            SchemaType.AVRO);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema1).getValueSchema().getSchemaInfo().getType(),
            SchemaType.AVRO);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema2).getKeySchema().getSchemaInfo().getType(),
            SchemaType.AVRO);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema2).getValueSchema().getSchemaInfo().getType(),
            SchemaType.AVRO);

    String schemaInfo1 = new String(keyValueSchema1.getSchemaInfo().getSchema());
    String schemaInfo2 = new String(keyValueSchema2.getSchemaInfo().getSchema());
    assertEquals(schemaInfo1, schemaInfo2);
}

Example 6

Source File: KeyValueSchemaTest.java From pulsar with Apache License 2.0

6 votes

@Test
public void testNotAllowNullSchemaEncodeAndDecode() {
    Schema keyValueSchema = Schema.KeyValue(JSONSchema.of(SchemaDefinition.<Foo>builder().withPojo(Foo.class).withAlwaysAllowNull(false).build()),
            JSONSchema.of(SchemaDefinition.<Bar>builder().withPojo(Bar.class).withAlwaysAllowNull(false).build()));

    Bar bar = new Bar();
    bar.setField1(true);

    Foo foo = new Foo();
    foo.setField1("field1");
    foo.setField2("field2");
    foo.setField3(3);
    foo.setField4(bar);
    foo.setColor(Color.RED);

    byte[] encodeBytes = keyValueSchema.encode(new KeyValue(foo, bar));
    Assert.assertTrue(encodeBytes.length > 0);

    KeyValue<Foo, Bar> keyValue = (KeyValue<Foo, Bar>) keyValueSchema.decode(encodeBytes);
    Foo fooBack = keyValue.getKey();
    Bar barBack = keyValue.getValue();

    assertEquals(foo, fooBack);
    assertEquals(bar, barBack);
}

Example 7

Source File: TypedMessageBuilderImplTest.java From pulsar with Apache License 2.0

6 votes

@Test
public void testSetKeyBytesEncodingTypeSeparated() {
    producerBase = mock(ProducerBase.class);

    AvroSchema<SchemaTestUtils.Foo> fooSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Foo>builder().withPojo(SchemaTestUtils.Foo.class).build());
    AvroSchema<SchemaTestUtils.Bar> barSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Bar>builder().withPojo(SchemaTestUtils.Bar.class).build());

    Schema<KeyValue<SchemaTestUtils.Foo, SchemaTestUtils.Bar>> keyValueSchema = Schema.KeyValue(fooSchema, barSchema, KeyValueEncodingType.SEPARATED);
    TypedMessageBuilderImpl typedMessageBuilderImpl = new TypedMessageBuilderImpl(producerBase, keyValueSchema);


    try {
        TypedMessageBuilderImpl typedMessageBuilder = (TypedMessageBuilderImpl)typedMessageBuilderImpl.keyBytes("separated".getBytes());
        fail("This should fail");
    } catch (IllegalArgumentException e) {
        assertTrue(e.getMessage().contains("This method is not allowed to set keys when in encoding type is SEPARATED"));
    }
}

Example 8

Source File: KeyValueSchemaInfoTest.java From pulsar with Apache License 2.0

6 votes

@Test(dataProvider = "encodingTypes")
public void encodeDecodeKeyValueSchemaInfo(KeyValueEncodingType encodingType) {
    Schema<KeyValue<Foo, Bar>> kvSchema = Schema.KeyValue(
        FOO_SCHEMA,
        BAR_SCHEMA,
        encodingType
    );
    SchemaInfo kvSchemaInfo = kvSchema.getSchemaInfo();
    assertEquals(
        DefaultImplementation.decodeKeyValueEncodingType(kvSchemaInfo),
        encodingType);

    SchemaInfo encodedSchemaInfo =
        DefaultImplementation.encodeKeyValueSchemaInfo(FOO_SCHEMA, BAR_SCHEMA, encodingType);
    assertEquals(encodedSchemaInfo, kvSchemaInfo);
    assertEquals(
        DefaultImplementation.decodeKeyValueEncodingType(encodedSchemaInfo),
        encodingType);

    KeyValue<SchemaInfo, SchemaInfo> schemaInfoKeyValue =
        DefaultImplementation.decodeKeyValueSchemaInfo(kvSchemaInfo);

    assertEquals(schemaInfoKeyValue.getKey(), FOO_SCHEMA.getSchemaInfo());
    assertEquals(schemaInfoKeyValue.getValue(), BAR_SCHEMA.getSchemaInfo());
}

Example 9

Source File: KeyValueSchemaTest.java From pulsar with Apache License 2.0

5 votes

@Test
public void testInlineKeyValueEncodingTypeSchemaEncodeAndDecode() {

    AvroSchema<Foo> fooSchema = AvroSchema.of(SchemaDefinition.<Foo>builder().withPojo(Foo.class).build());
    AvroSchema<Bar> barSchema = AvroSchema.of(SchemaDefinition.<Bar>builder().withPojo(Bar.class).build());

    Schema<KeyValue<Foo, Bar>> keyValueSchema = Schema.KeyValue(fooSchema, barSchema, KeyValueEncodingType.INLINE);


    Bar bar = new Bar();
    bar.setField1(true);

    Foo foo = new Foo();
    foo.setField1("field1");
    foo.setField2("field2");
    foo.setField3(3);
    foo.setField4(bar);
    foo.setColor(Color.RED);

    // Check kv.encoding.type INLINE
    byte[] encodeBytes = keyValueSchema.encode(new KeyValue(foo, bar));
    Assert.assertTrue(encodeBytes.length > 0);
    KeyValue<Foo, Bar>  keyValue = (KeyValue<Foo, Bar>) keyValueSchema.decode(encodeBytes);
    Foo fooBack = keyValue.getKey();
    Bar barBack = keyValue.getValue();
    assertEquals(foo, fooBack);
    assertEquals(bar, barBack);

}

Example 10

Source File: AdminApiSchemaTest.java From pulsar with Apache License 2.0

5 votes

@DataProvider(name = "schemas")
public Object[][] schemas() {
    return new Object[][] {
        { Schema.BOOL },
        { Schema.INT8 },
        { Schema.INT16 },
        { Schema.INT32 },
        { Schema.INT64 },
        { StringSchema.utf8() },
        { new StringSchema(US_ASCII) },
        { Schema.FLOAT },
        { Schema.DOUBLE },
        { Schema.DATE },
        { Schema.TIME },
        { Schema.TIMESTAMP },
        { Schema.AVRO(
            SchemaDefinition.builder()
                .withPojo(Foo.class)
                .withProperties(PROPS)
                .build()
        ) },
        { Schema.JSON(
            SchemaDefinition.builder()
                .withPojo(Foo.class)
                .withProperties(PROPS)
                .build()
        )},
        { Schema.KeyValue(
            StringSchema.utf8(),
            new StringSchema(US_ASCII)
        )}
    };
}

Example 11

Source File: KeyValueSchemaTest.java From pulsar with Apache License 2.0

5 votes

@Test
public void testNotAllowNullJsonSchemaCreate() {
    JSONSchema<Foo> fooSchema = JSONSchema.of(
            SchemaDefinition.<Foo>builder().withPojo(Foo.class).withAlwaysAllowNull(false).build());
    JSONSchema<Bar> barSchema = JSONSchema.of(
            SchemaDefinition.<Bar>builder().withPojo(Bar.class).withAlwaysAllowNull(false).build());

    Schema<KeyValue<Foo, Bar>> keyValueSchema1 = Schema.KeyValue(fooSchema, barSchema);
    Schema<KeyValue<Foo, Bar>> keyValueSchema2 = Schema.KeyValue(JSONSchema.of(SchemaDefinition.<Foo>builder().withPojo(Foo.class).withAlwaysAllowNull(false).build()),
            JSONSchema.of(SchemaDefinition.<Bar>builder().withPojo(Bar.class).withAlwaysAllowNull(false).build()));

    Schema<KeyValue<Foo, Bar>> keyValueSchema3 = Schema.KeyValue(JSONSchema.of(SchemaDefinition.<Foo>builder().withPojo(Foo.class).withAlwaysAllowNull(false).build()),
            JSONSchema.of(SchemaDefinition.<Bar>builder().withPojo(Bar.class).withAlwaysAllowNull(false).build()));

    assertEquals(keyValueSchema1.getSchemaInfo().getType(), SchemaType.KEY_VALUE);
    assertEquals(keyValueSchema2.getSchemaInfo().getType(), SchemaType.KEY_VALUE);
    assertEquals(keyValueSchema3.getSchemaInfo().getType(), SchemaType.KEY_VALUE);

    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema1).getKeySchema().getSchemaInfo().getType(),
        SchemaType.JSON);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema1).getValueSchema().getSchemaInfo().getType(),
        SchemaType.JSON);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema2).getKeySchema().getSchemaInfo().getType(),
        SchemaType.JSON);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema2).getValueSchema().getSchemaInfo().getType(),
        SchemaType.JSON);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema3).getKeySchema().getSchemaInfo().getType(),
        SchemaType.JSON);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema3).getValueSchema().getSchemaInfo().getType(),
        SchemaType.JSON);

    String schemaInfo1 = new String(keyValueSchema1.getSchemaInfo().getSchema());
    String schemaInfo2 = new String(keyValueSchema2.getSchemaInfo().getSchema());
    String schemaInfo3 = new String(keyValueSchema3.getSchemaInfo().getSchema());
    assertEquals(schemaInfo1, schemaInfo2);
    assertEquals(schemaInfo1, schemaInfo3);
}

Example 12

Source File: KeyValueSchemaTest.java From pulsar with Apache License 2.0

5 votes

@Test
public void testAllowNullJsonSchemaCreate() {
    JSONSchema<Foo> fooSchema = JSONSchema.of(SchemaDefinition.<Foo>builder().withPojo(Foo.class).build());
    JSONSchema<Bar> barSchema = JSONSchema.of(SchemaDefinition.<Bar>builder().withPojo(Bar.class).build());

    Schema<KeyValue<Foo, Bar>> keyValueSchema1 = Schema.KeyValue(fooSchema, barSchema);
    Schema<KeyValue<Foo, Bar>> keyValueSchema2 = Schema.KeyValue(Foo.class, Bar.class, SchemaType.JSON);
    Schema<KeyValue<Foo, Bar>> keyValueSchema3 = Schema.KeyValue(Foo.class, Bar.class);

    assertEquals(keyValueSchema1.getSchemaInfo().getType(), SchemaType.KEY_VALUE);
    assertEquals(keyValueSchema2.getSchemaInfo().getType(), SchemaType.KEY_VALUE);
    assertEquals(keyValueSchema3.getSchemaInfo().getType(), SchemaType.KEY_VALUE);

    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema1).getKeySchema().getSchemaInfo().getType(),
            SchemaType.JSON);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema1).getValueSchema().getSchemaInfo().getType(),
            SchemaType.JSON);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema2).getKeySchema().getSchemaInfo().getType(),
            SchemaType.JSON);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema2).getValueSchema().getSchemaInfo().getType(),
            SchemaType.JSON);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema3).getKeySchema().getSchemaInfo().getType(),
            SchemaType.JSON);
    assertEquals(((KeyValueSchema<Foo, Bar>) keyValueSchema3).getValueSchema().getSchemaInfo().getType(),
            SchemaType.JSON);

    String schemaInfo1 = new String(keyValueSchema1.getSchemaInfo().getSchema());
    String schemaInfo2 = new String(keyValueSchema2.getSchemaInfo().getSchema());
    String schemaInfo3 = new String(keyValueSchema3.getSchemaInfo().getSchema());
    assertEquals(schemaInfo1, schemaInfo2);
    assertEquals(schemaInfo1, schemaInfo3);
}

Example 13

Source File: KeyValueSchemaInfoTest.java From pulsar with Apache License 2.0

5 votes

@Test(dataProvider = "encodingTypes")
public void encodeDecodeNestedKeyValueSchemaInfo(KeyValueEncodingType encodingType) {
    Schema<KeyValue<String, Bar>> nestedSchema =
        Schema.KeyValue(Schema.STRING, BAR_SCHEMA, KeyValueEncodingType.INLINE);
    Schema<KeyValue<Foo, KeyValue<String, Bar>>> kvSchema = Schema.KeyValue(
        FOO_SCHEMA,
        nestedSchema,
        encodingType
    );
    SchemaInfo kvSchemaInfo = kvSchema.getSchemaInfo();
    assertEquals(
        DefaultImplementation.decodeKeyValueEncodingType(kvSchemaInfo),
        encodingType);

    SchemaInfo encodedSchemaInfo =
        DefaultImplementation.encodeKeyValueSchemaInfo(
            FOO_SCHEMA,
            nestedSchema,
            encodingType);
    assertEquals(encodedSchemaInfo, kvSchemaInfo);
    assertEquals(
        DefaultImplementation.decodeKeyValueEncodingType(encodedSchemaInfo),
        encodingType);

    KeyValue<SchemaInfo, SchemaInfo> schemaInfoKeyValue =
        DefaultImplementation.decodeKeyValueSchemaInfo(kvSchemaInfo);

    assertEquals(schemaInfoKeyValue.getKey(), FOO_SCHEMA.getSchemaInfo());
    assertEquals(schemaInfoKeyValue.getValue().getType(), SchemaType.KEY_VALUE);
    KeyValue<SchemaInfo, SchemaInfo> nestedSchemaInfoKeyValue =
        DefaultImplementation.decodeKeyValueSchemaInfo(schemaInfoKeyValue.getValue());

    assertEquals(nestedSchemaInfoKeyValue.getKey(), Schema.STRING.getSchemaInfo());
    assertEquals(nestedSchemaInfoKeyValue.getValue(), BAR_SCHEMA.getSchemaInfo());
}

Example 14

Source File: KeyValueSchemaTest.java From pulsar with Apache License 2.0

5 votes

@Test
public void testDefaultKeyValueEncodingTypeSchemaEncodeAndDecode() {
    AvroSchema<Foo> fooSchema = AvroSchema.of(SchemaDefinition.<Foo>builder().withPojo(Foo.class).build());
    AvroSchema<Bar> barSchema = AvroSchema.of(SchemaDefinition.<Bar>builder().withPojo(Bar.class).build());

    Schema<KeyValue<Foo, Bar>> keyValueSchema = Schema.KeyValue(fooSchema, barSchema);

    Bar bar = new Bar();
    bar.setField1(true);

    Foo foo = new Foo();
    foo.setField1("field1");
    foo.setField2("field2");
    foo.setField3(3);
    foo.setField4(bar);
    foo.setColor(Color.RED);

    // Check kv.encoding.type default not set value
    byte[] encodeBytes = keyValueSchema.encode(new KeyValue(foo, bar));
    Assert.assertTrue(encodeBytes.length > 0);

    KeyValue<Foo, Bar> keyValue = (KeyValue<Foo, Bar>) keyValueSchema.decode(encodeBytes);
    Foo fooBack = keyValue.getKey();
    Bar barBack = keyValue.getValue();

    assertEquals(foo, fooBack);
    assertEquals(bar, barBack);
}

Example 15

Source File: TypedMessageBuilderImplTest.java From pulsar with Apache License 2.0

5 votes

@Test
public void testSetKeyBytesEncodingTypeDefault() {
    producerBase = mock(ProducerBase.class);

    AvroSchema<SchemaTestUtils.Foo> fooSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Foo>builder().withPojo(SchemaTestUtils.Foo.class).build());
    AvroSchema<SchemaTestUtils.Bar> barSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Bar>builder().withPojo(SchemaTestUtils.Bar.class).build());

    Schema<KeyValue<SchemaTestUtils.Foo, SchemaTestUtils.Bar>> keyValueSchema = Schema.KeyValue(fooSchema, barSchema);
    TypedMessageBuilderImpl typedMessageBuilderImpl = new TypedMessageBuilderImpl(producerBase, keyValueSchema);

    TypedMessageBuilderImpl typedMessageBuilder = (TypedMessageBuilderImpl)typedMessageBuilderImpl.keyBytes("default".getBytes());
    assertEquals(typedMessageBuilder.getKey(), Base64.getEncoder().encodeToString("default".getBytes()));
    assertTrue(typedMessageBuilder.getMetadataBuilder().getPartitionKeyB64Encoded());
}

Example 16

Source File: TypedMessageBuilderImplTest.java From pulsar with Apache License 2.0

5 votes

@Test
public void testSetKeyEncodingTypeInline() {
    producerBase = mock(ProducerBase.class);

    AvroSchema<SchemaTestUtils.Foo> fooSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Foo>builder().withPojo(SchemaTestUtils.Foo.class).build());
    AvroSchema<SchemaTestUtils.Bar> barSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Bar>builder().withPojo(SchemaTestUtils.Bar.class).build());

    Schema<KeyValue<SchemaTestUtils.Foo, SchemaTestUtils.Bar>> keyValueSchema = Schema.KeyValue(fooSchema, barSchema, KeyValueEncodingType.INLINE);
    TypedMessageBuilderImpl typedMessageBuilderImpl = new TypedMessageBuilderImpl(producerBase, keyValueSchema);

    TypedMessageBuilderImpl typedMessageBuilder = (TypedMessageBuilderImpl)typedMessageBuilderImpl.key("inline");
    assertEquals(typedMessageBuilder.getKey(), "inline");
    assertFalse(typedMessageBuilder.getMetadataBuilder().getPartitionKeyB64Encoded());
}

Example 17

Source File: TypedMessageBuilderImplTest.java From pulsar with Apache License 2.0

5 votes

@Test
public void testSetKeyEncodingTypeDefault() {
    producerBase = mock(ProducerBase.class);

    AvroSchema<SchemaTestUtils.Foo> fooSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Foo>builder().withPojo(SchemaTestUtils.Foo.class).build());
    AvroSchema<SchemaTestUtils.Bar> barSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Bar>builder().withPojo(SchemaTestUtils.Bar.class).build());

    Schema<KeyValue<SchemaTestUtils.Foo, SchemaTestUtils.Bar>> keyValueSchema = Schema.KeyValue(fooSchema, barSchema);
    TypedMessageBuilderImpl typedMessageBuilderImpl = new TypedMessageBuilderImpl(producerBase, keyValueSchema);

    TypedMessageBuilderImpl typedMessageBuilder = (TypedMessageBuilderImpl)typedMessageBuilderImpl.key("default");
    assertEquals(typedMessageBuilder.getKey(), "default");
    assertFalse(typedMessageBuilder.getMetadataBuilder().getPartitionKeyB64Encoded());
}

Example 18

Source File: TypedMessageBuilderImplTest.java From pulsar with Apache License 2.0

5 votes

@Test
public void testInlineValue() {
    producerBase = mock(ProducerBase.class);

    AvroSchema<SchemaTestUtils.Foo> fooSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Foo>builder().withPojo(SchemaTestUtils.Foo.class).build());
    AvroSchema<SchemaTestUtils.Bar> barSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Bar>builder().withPojo(SchemaTestUtils.Bar.class).build());

    Schema<KeyValue<SchemaTestUtils.Foo, SchemaTestUtils.Bar>> keyValueSchema = Schema.KeyValue(fooSchema, barSchema, KeyValueEncodingType.INLINE);
    TypedMessageBuilderImpl typedMessageBuilderImpl = new TypedMessageBuilderImpl(producerBase, keyValueSchema);

    SchemaTestUtils.Foo foo = new SchemaTestUtils.Foo();
    foo.setField1("field1");
    foo.setField2("field2");
    SchemaTestUtils.Bar bar = new SchemaTestUtils.Bar();
    bar.setField1(true);
    KeyValue<SchemaTestUtils.Foo, SchemaTestUtils.Bar> keyValue = new KeyValue<>(foo, bar);

    // Check kv.encoding.type INLINE
    TypedMessageBuilderImpl<KeyValue> typedMessageBuilder = (TypedMessageBuilderImpl)typedMessageBuilderImpl.value(keyValue);
    ByteBuffer content = typedMessageBuilder.getContent();
    byte[] contentByte = new byte[content.remaining()];
    content.get(contentByte);
    KeyValue<SchemaTestUtils.Foo, SchemaTestUtils.Bar> decodeKeyValue = keyValueSchema.decode(contentByte);
    assertEquals(decodeKeyValue.getKey(), foo);
    assertEquals(decodeKeyValue.getValue(), bar);
    assertFalse(typedMessageBuilderImpl.hasKey());
}

Example 19

Source File: TypedMessageBuilderImplTest.java From pulsar with Apache License 2.0

5 votes

@Test
public void testDefaultValue() {
    producerBase = mock(ProducerBase.class);

    AvroSchema<SchemaTestUtils.Foo> fooSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Foo>builder().withPojo(SchemaTestUtils.Foo.class).build());
    AvroSchema<SchemaTestUtils.Bar> barSchema = AvroSchema.of(SchemaDefinition.<SchemaTestUtils.Bar>builder().withPojo(SchemaTestUtils.Bar.class).build());

    Schema<KeyValue<SchemaTestUtils.Foo, SchemaTestUtils.Bar>> keyValueSchema = Schema.KeyValue(fooSchema, barSchema);
    TypedMessageBuilderImpl typedMessageBuilderImpl = new TypedMessageBuilderImpl(producerBase, keyValueSchema);

    SchemaTestUtils.Foo foo = new SchemaTestUtils.Foo();
    foo.setField1("field1");
    foo.setField2("field2");
    SchemaTestUtils.Bar bar = new SchemaTestUtils.Bar();
    bar.setField1(true);
    KeyValue<SchemaTestUtils.Foo, SchemaTestUtils.Bar> keyValue = new KeyValue<>(foo, bar);

    // Check kv.encoding.type default, not set value
    TypedMessageBuilderImpl<KeyValue>  typedMessageBuilder = (TypedMessageBuilderImpl)typedMessageBuilderImpl.value(keyValue);
    ByteBuffer content = typedMessageBuilder.getContent();
    byte[] contentByte = new byte[content.remaining()];
    content.get(contentByte);
    KeyValue<SchemaTestUtils.Foo, SchemaTestUtils.Bar>  decodeKeyValue = keyValueSchema.decode(contentByte);
    assertEquals(decodeKeyValue.getKey(), foo);
    assertEquals(decodeKeyValue.getValue(), bar);
    assertFalse(typedMessageBuilderImpl.hasKey());
}

Example 20

Source File: AdminApiSchemaTest.java From pulsar with Apache License 2.0

5 votes

@Test
public void createKeyValueSchema() throws Exception {
    String topicName = "schematest/test/test-key-value-schema";
    Schema keyValueSchema = Schema.KeyValue(Schema.AVRO(Foo.class), Schema.AVRO(Foo.class));
    admin.schemas().createSchema(topicName,
            keyValueSchema.getSchemaInfo());
    SchemaInfo schemaInfo = admin.schemas().getSchemaInfo(topicName);

    assertEquals(schemaInfo, keyValueSchema.getSchemaInfo());
}