/* * 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<Buffer> 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<Buffer> handler) { sendFrame(type, address, replyAddress, null, null, body, handler); } public static void sendFrame(String type, String address, JsonObject body, WriteStream<Buffer> handler) { sendFrame(type, address, null, null, null, body, handler); } public static void sendFrame(String type, WriteStream<Buffer> handler) { sendFrame(type, null, null, null, null, null, handler); } public static void sendErrFrame(String address, String replyAddress, ReplyException failure, WriteStream<Buffer> 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<Buffer> handler) { final JsonObject payload = new JsonObject() .put("type", "err") .put("message", message); writeFrame(payload, handler); } public static void writeFrame(JsonObject payload, WriteStream<Buffer> handler) { // encode byte[] data = payload.encode().getBytes(UTF8); handler.write(Buffer.buffer().appendInt(data.length).appendBytes(data)); } }