def test_portable_class_def(self): builder_inner = ClassDefinitionBuilder(FACTORY_ID, InnerPortable.CLASS_ID) builder_inner.add_utf_field("param_str") builder_inner.add_int_field("param_int") class_def_inner = builder_inner.build() builder = ClassDefinitionBuilder(FACTORY_ID, SerializationV1Portable.CLASS_ID) builder.add_byte_field("1") builder.add_boolean_field("2") builder.add_char_field("3") builder.add_short_field("4") builder.add_int_field("5") builder.add_long_field("6") builder.add_float_field("7") builder.add_double_field("8") builder.add_utf_field("9") builder.add_byte_array_field("a1") builder.add_boolean_array_field("a2") builder.add_char_array_field("a3") builder.add_short_array_field("a4") builder.add_int_array_field("a5") builder.add_long_array_field("a6") builder.add_float_array_field("a7") builder.add_double_array_field("a8") builder.add_utf_array_field("a9") builder.add_portable_field("p", class_def_inner) builder.add_portable_array_field("ap", class_def_inner) class_def = builder.build() config = hazelcast.ClientConfig() config.serialization_config.portable_factories[ FACTORY_ID] = the_factory config.serialization_config.class_definitions.add(class_def) config.serialization_config.class_definitions.add(class_def_inner) service = SerializationServiceV1(config.serialization_config) service2 = SerializationServiceV1(config.serialization_config) obj = SerializationV1Portable() data = service.to_data(obj) obj2 = service2.to_object(data) self.assertTrue(obj == obj2)
def test_portable_class_def(self): builder_inner = ClassDefinitionBuilder(FACTORY_ID, InnerPortable.CLASS_ID) builder_inner.add_utf_field("param_str") builder_inner.add_int_field("param_int") class_def_inner = builder_inner.build() builder = ClassDefinitionBuilder(FACTORY_ID, SerializationV1Portable.CLASS_ID) builder.add_byte_field("1") builder.add_boolean_field("2") builder.add_char_field("3") builder.add_short_field("4") builder.add_int_field("5") builder.add_long_field("6") builder.add_float_field("7") builder.add_double_field("8") builder.add_utf_field("9") builder.add_byte_array_field("a1") builder.add_boolean_array_field("a2") builder.add_char_array_field("a3") builder.add_short_array_field("a4") builder.add_int_array_field("a5") builder.add_long_array_field("a6") builder.add_float_array_field("a7") builder.add_double_array_field("a8") builder.add_utf_array_field("a9") builder.add_portable_field("p", class_def_inner) builder.add_portable_array_field("ap", class_def_inner) class_def = builder.build() config = hazelcast.ClientConfig() config.serialization_config.portable_factories[FACTORY_ID] = the_factory config.serialization_config.class_definitions.add(class_def) config.serialization_config.class_definitions.add(class_def_inner) service = SerializationServiceV1(config.serialization_config) service2 = SerializationServiceV1(config.serialization_config) obj = SerializationV1Portable() data = service.to_data(obj) obj2 = service2.to_object(data) self.assertTrue(obj == obj2)
class ClassDefinitionWriter(PortableWriter): def __init__(self, portable_context, factory_id=None, class_id=None, version=None, class_def_builder=None): self.portable_context = portable_context if class_def_builder is None: self._builder = ClassDefinitionBuilder(factory_id, class_id, version) else: self._builder = class_def_builder def write_byte(self, field_name, value): self._builder.add_byte_field(field_name) def write_boolean(self, field_name, value): self._builder.add_boolean_field(field_name) def write_char(self, field_name, value): self._builder.add_char_field(field_name) def write_short(self, field_name, value): self._builder.add_short_field(field_name) def write_int(self, field_name, value): self._builder.add_int_field(field_name) def write_long(self, field_name, value): self._builder.add_long_field(field_name) def write_float(self, field_name, value): self._builder.add_float_field(field_name) def write_double(self, field_name, value): self._builder.add_double_field(field_name) def write_utf(self, field_name, value): self._builder.add_utf_field(field_name) def write_byte_array(self, field_name, values): self._builder.add_byte_array_field(field_name) def write_boolean_array(self, field_name, values): self._builder.add_boolean_array_field(field_name) def write_char_array(self, field_name, values): self._builder.add_char_array_field(field_name) def write_short_array(self, field_name, values): self._builder.add_short_array_field(field_name) def write_int_array(self, field_name, values): self._builder.add_int_array_field(field_name) def write_long_array(self, field_name, values): self._builder.add_long_array_field(field_name) def write_float_array(self, field_name, values): self._builder.add_float_array_field(field_name) def write_double_array(self, field_name, values): self._builder.add_double_array_field(field_name) def write_utf_array(self, field_name, values): self._builder.add_utf_array_field(field_name) def write_portable(self, field_name, portable): if portable is None: raise HazelcastSerializationError( "Cannot write None portable without explicitly registering class definition!" ) version = util.get_portable_version( portable, self.portable_context.portable_version) factory_id = portable.get_factory_id() class_id = portable.get_class_id() nested_builder = ClassDefinitionBuilder(factory_id, class_id, version) nested_class_def = self._create_nested_class_def( portable, nested_builder) self._builder.add_portable_field(field_name, nested_class_def) def write_portable_array(self, field_name, values): if values is None or len(values) == 0: raise HazelcastSerializationError( "Cannot write None portable without explicitly registering class definition!" ) portable = values[0] _class_id = portable.get_class_id() for i in xrange(1, len(values)): if values[i].get_class_id() != _class_id: raise ValueError( "Detected different class-ids in portable array!") version = util.get_portable_version( portable, self.portable_context.portable_version) nested_builder = ClassDefinitionBuilder(portable.get_factory_id(), portable.get_class_id(), version) nested_class_def = self._create_nested_class_def( portable, nested_builder) self._builder.add_portable_array_field(field_name, nested_class_def) def write_null_portable(self, field_name, factory_id, class_id): nested_class_def = self.portable_context.lookup_class_definition( factory_id, class_id, self.portable_context.portable_version) if nested_class_def is None: raise HazelcastSerializationError( "Cannot write None portable without explicitly registering class definition!" ) self._builder.add_portable_field(field_name, nested_class_def) def get_raw_data_output(self): return EmptyObjectDataOutput() def _create_nested_class_def(self, portable, nested_builder): _writer = ClassDefinitionWriter(self.portable_context, class_def_builder=nested_builder) portable.write_portable(_writer) return self.portable_context.register_class_definition( nested_builder.build()) def register_and_get(self): cd = self._builder.build() return self.portable_context.register_class_definition(cd)
class ClassDefinitionWriter(PortableWriter): def __init__(self, portable_context, factory_id, class_id, version, class_def_builder=None): self.portable_context = portable_context if class_def_builder is None: self._builder = ClassDefinitionBuilder(factory_id, class_id, version) else: self._builder = class_def_builder def write_utf_array(self, field_name, values): self._builder.add_utf_array_field(field_name) def write_utf(self, field_name, value): self._builder.add_utf_array_field(field_name) def write_short_array(self, field_name, values): self._builder.add_utf_field(field_name) def write_short(self, field_name, value): self._builder.add_short_field(field_name) def write_portable_array(self, field_name, values): if values is None or len(values) == 0: raise HazelcastSerializationError("Cannot write None portable without explicitly registering class definition!") portable = values[0] _class_id = portable.get_class_id() for i in xrange(1, len(values)): if values[i].get_class_id() != _class_id: raise ValueError("Detected different class-ids in portable array!") version = util.get_portable_version(portable, self.portable_context.get_version()) nested_builder = ClassDefinitionBuilder(portable.get_factory_id(), portable.get_class_id(), version) nested_class_def = self._create_nested_class_def(portable, nested_builder) self._builder.add_portable_array_field(field_name, nested_class_def) def write_portable(self, field_name, portable): if portable is None: raise HazelcastSerializationError("Cannot write None portable without explicitly registering class definition!") version = util.get_portable_version(portable, self.portable_context.get_version()) nested_builder = ClassDefinitionBuilder(portable.get_factory_id(), portable.get_class_id(), version) nested_class_def = self._create_nested_class_def(portable, nested_builder) self._builder.add_portable_field(field_name, nested_class_def) def write_null_portable(self, field_name, factory_id, class_id): nested_class_def = self.portable_context.lookup_class_definition(factory_id,class_id, self.portable_context.get_version()) if nested_class_def is None: raise HazelcastSerializationError("Cannot write None portable without explicitly registering class definition!") self._builder.add_portable_field(field_name, nested_class_def) def write_long_array(self, field_name, values): self._builder.add_long_array_field(field_name) def write_long(self, field_name, value): self._builder.add_long_field(field_name) def write_int_array(self, field_name, values): self._builder.add_int_array_field(field_name) def write_int(self, field_name, value): self._builder.add_int_field(field_name) def write_float_array(self, field_name, values): self._builder.add_float_array_field(field_name) def write_float(self, field_name, value): self._builder.add_float_field(field_name) def write_double_array(self, field_name, values): self._builder.add_double_array_field(field_name) def write_double(self, field_name, value): self._builder.add_double_field(field_name) def write_char_array(self, field_name, values): self._builder.add_char_array_field(field_name) def write_char(self, field_name, value): self._builder.add_char_field(field_name) def write_byte_array(self, field_name, values): self._builder.add_byte_array_field(field_name) def write_byte(self, field_name, value): self._builder.add_byte_field(field_name) def write_boolean_array(self, field_name, values): self._builder.add_boolean_array_field(field_name) def write_boolean(self, field_name, value): self._builder.add_boolean_field(field_name) def get_raw_data_output(self): return EmptyObjectDataOutput() def _create_nested_class_def(self, portable, nested_builder): _writer = ClassDefinitionWriter(self.portable_context, nested_builder) portable.write_portable(_writer) return self.portable_context.register_class_definition(nested_builder.build())