io.servicefabric.transport.ProtostuffProtocol Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of servicefabric-transport Show documentation

The newest version!

package io.servicefabric.transport;

import io.protostuff.runtime.RuntimeSchema;

/**
 * Protostuff implementation for protocol.
 * @author Anton Kharenko
 */
public class ProtostuffProtocol implements Protocol {

  private final ProtostuffFrameHandlerFactory frameHandlerFactory;
  private final ProtostuffMessageDeserializer messageDeserializer;
  private final ProtostuffMessageSerializer messageSerializer;

  /**
   * Creates new protostuff protocol, registers protostuff runtime schema for {@link io.servicefabric.transport.Message}
   */
  public ProtostuffProtocol() {
    frameHandlerFactory = new ProtostuffFrameHandlerFactory();
    messageDeserializer = new ProtostuffMessageDeserializer();
    messageSerializer = new ProtostuffMessageSerializer();

    // Register message schema
    if (!RuntimeSchema.isRegistered(Message.class)) {
      RuntimeSchema.register(Message.class, new MessageSchema());
    }
  }

  @Override
  public FrameHandlerFactory getFrameHandlerFactory() {
    return frameHandlerFactory;
  }

  @Override
  public MessageDeserializer getMessageDeserializer() {
    return messageDeserializer;
  }

  @Override
  public MessageSerializer getMessageSerializer() {
    return messageSerializer;
  }
}