io.vertx.ext.eventbus.bridge.tcp.impl.protocol.FrameHelper Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2015 Red Hat, Inc.
 *
 *  All rights reserved. This program and the accompanying materials
 *  are made available under the terms of the Eclipse Public License v1.0
 *  and Apache License v2.0 which accompanies this distribution.
 *
 *  The Eclipse Public License is available at
 *  http://www.eclipse.org/legal/epl-v10.html
 *
 *  The Apache License v2.0 is available at
 *  http://www.opensource.org/licenses/apache2.0.php
 *
 *  You may elect to redistribute this code under either of these licenses.
 */
package io.vertx.ext.eventbus.bridge.tcp.impl.protocol;

import io.vertx.core.buffer.Buffer;
import io.vertx.core.eventbus.ReplyException;
import io.vertx.core.json.JsonObject;
import io.vertx.core.streams.WriteStream;

import java.nio.charset.Charset;

/**
 * Helper class to format and send frames over a socket
 * @author Paulo Lopes
 */
public class FrameHelper {

  private static final Charset UTF8 = Charset.forName("UTF-8");

  private FrameHelper() {}

  public static void sendFrame(String type, String address, String replyAddress, JsonObject headers, Boolean send, JsonObject body, WriteStream handler) {
    final JsonObject payload = new JsonObject().put("type", type);

    if (address != null) {
      payload.put("address", address);
    }

    if (replyAddress != null) {
      payload.put("replyAddress", replyAddress);
    }

    if (headers != null) {
      payload.put("headers", headers);
    }

    if (body != null) {
      payload.put("body", body);
    }

    if (send != null) {
      payload.put("send", send);
    }

    writeFrame(payload, handler);
  }

  public static void sendFrame(String type, String address, String replyAddress, JsonObject body, WriteStream handler) {
    sendFrame(type, address, replyAddress, null, null, body, handler);
  }

  public static void sendFrame(String type, String address, JsonObject body, WriteStream handler) {
    sendFrame(type, address, null, null, null, body, handler);
  }

  public static void sendFrame(String type, WriteStream handler) {
    sendFrame(type, null, null, null, null, null, handler);
  }

  public static void sendErrFrame(String address, String replyAddress, ReplyException failure, WriteStream handler) {
    final JsonObject payload = new JsonObject()
        .put("type", "err")
        .put("address", replyAddress)
        .put("sourceAddress", address)
        .put("failureCode", failure.failureCode())
        .put("failureType", failure.failureType().name())
        .put("message", failure.getMessage());

    writeFrame(payload, handler);
  }

  public static void sendErrFrame(String message, WriteStream handler) {
    final JsonObject payload = new JsonObject()
        .put("type", "err")
        .put("message", message);

    writeFrame(payload, handler);
  }

  public static void writeFrame(JsonObject payload, WriteStream handler) {
    // encode
    byte[] data = payload.encode().getBytes(UTF8);

    handler.write(Buffer.buffer().appendInt(data.length).appendBytes(data));
  }
}