Beispiel #1
0
  static void sendError(HttpServerExchange exchange, boolean isGet, boolean encode, String msg) {

    if (encode) {

      try {
        ModelNode response = new ModelNode();
        response.set(msg);

        ByteArrayOutputStream bout = new ByteArrayOutputStream();
        response.writeBase64(bout);
        byte[] bytes = bout.toByteArray();

        exchange
            .getResponseHeaders()
            .put(Headers.CONTENT_TYPE, APPLICATION_DMR_ENCODED + ";" + UTF_8);
        exchange.getResponseHeaders().put(Headers.CONTENT_LENGTH, String.valueOf(bytes.length));
        exchange.setResponseCode(500);

        exchange.getResponseSender().send(new String(bytes), IoCallback.END_EXCHANGE);

      } catch (IOException e) {
        // fallback, should not happen
        sendError(exchange, isGet, false, msg);
      }

    } else {
      byte[] bytes = msg.getBytes();
      exchange.getResponseHeaders().put(Headers.CONTENT_TYPE, TEXT_PLAIN + ";" + UTF_8);
      exchange.getResponseHeaders().put(Headers.CONTENT_LENGTH, String.valueOf(bytes.length));
      exchange.setResponseCode(500);

      exchange.getResponseSender().send(msg, IoCallback.END_EXCHANGE);
    }
  }
  /**
   * Writes the HTTP response to the output stream.
   *
   * @param http The HttpExchange object that allows access to the request and response.
   * @param isGet Flag indicating whether or not the request was a GET request or POST request.
   * @param pretty Flag indicating whether or not the output, if JSON, should be pretty printed or
   *     not.
   * @param response The DMR response from the operation.
   * @param status The HTTP status code to be included in the response.
   * @param encode Flag indicating whether or not to Base64 encode the response payload.
   * @throws IOException if an error occurs while attempting to generate the HTTP response.
   */
  private void writeResponse(
      final HttpExchange http,
      boolean isGet,
      boolean pretty,
      ModelNode response,
      int status,
      boolean encode,
      String contentType)
      throws IOException {
    final Headers responseHeaders = http.getResponseHeaders();
    responseHeaders.add(CONTENT_TYPE, contentType);
    http.sendResponseHeaders(status, 0);

    final OutputStream out = http.getResponseBody();
    final PrintWriter print = new PrintWriter(out);

    // GET (read) operations will never have a compensating update, and the status is already
    // available via the http response status code, so unwrap them.
    if (isGet && status == OK) response = response.get("result");

    try {
      if (encode) {
        response.writeBase64(out);
      } else {
        response.writeJSONString(print, !pretty);
      }
    } finally {
      print.flush();
      out.flush();
      safeClose(print);
      safeClose(out);
    }
  }
 @Override
 public long getContentLength() {
   // Needed for the http client
   final ByteArrayOutputStream os = new ByteArrayOutputStream();
   try {
     model.writeBase64(os);
     return os.size();
   } catch (Exception e) {
     return -1;
   }
 }
 @Override
 public void writeTo(OutputStream out) throws IOException {
   model.writeBase64(out);
 }