Package com.linkedin.r2.message.rest

Examples of com.linkedin.r2.message.rest.RestResponseBuilder


      restRequest = request;
      restRequestContext = requestContext;
      restWireAttrs = wireAttrs;
      restCallback = callback;

      callback.onResponse(TransportResponseImpl.<RestResponse> success(new RestResponseBuilder().build(),
                                                                       wireAttrs));
    }
View Full Code Here


    final String msg = request.getEntity().asString(CHARSET);
    _echoService.echo(msg, new CallbackAdapter<RestResponse, String>(callback) {
      @Override
      protected RestResponse convertResponse(String responseMsg)
      {
        return new RestResponseBuilder()
                .setEntity(ByteString.copyString(responseMsg, CHARSET))
                .setHeader("Content-Type", "text/plain")
                .build();
      }
    });
View Full Code Here

   * @return
   */
  public RestResponse buildResponse(final RoutingResult routingResult,
                                     PartialRestResponse partialResponse)
  {
    RestResponseBuilder builder =
        new RestResponseBuilder().setHeaders(partialResponse.getHeaders()).setStatus(partialResponse.getStatus()
                                                                                                    .getCode());
    if (partialResponse.hasData())
    {
      DataMap dataMap = partialResponse.getDataMap();
      String mimeType = ((ServerResourceContext) routingResult.getContext()).getResponseMimeType();
      builder = encodeResult(mimeType, builder, dataMap);
    }
    return builder.build();
  }
View Full Code Here

    return _errorResponseBuilder.buildRestLiResponseData(request, routingResult, object, headers);
  }

  public RestException buildRestException(final Throwable e, PartialRestResponse partialResponse)
  {
    RestResponseBuilder builder =
        new RestResponseBuilder().setHeaders(partialResponse.getHeaders()).setStatus(partialResponse.getStatus()
                                                                                                    .getCode());
    if (partialResponse.hasData())
    {
      DataMap dataMap = partialResponse.getDataMap();
      ByteArrayOutputStream baos = new ByteArrayOutputStream(4096);
      DataMapUtils.write(dataMap, null, baos, true); // partialResponse.getSchema()
      builder.setEntity(baos.toByteArray());
    }
    RestResponse restResponse = builder.build();
    RestException restException = new RestException(restResponse, e);
    return restException;
  }
View Full Code Here

      LOG.debug("{}: handling a response", e.getChannel().getRemoteAddress());
      final Map<String, String> headers = new HashMap<String, String>(response.getHeaders());
      final Map<String, String> wireAttrs =
            new HashMap<String, String>(WireAttributeHelper.removeWireAttributes(headers));

      final RestResponse newResponse = new RestResponseBuilder(response)
              .unsafeSetHeaders(headers)
              .build();

      callback.onResponse(TransportResponseImpl.success(newResponse, wireAttrs));
    }
View Full Code Here

    protected Object decode(ChannelHandlerContext ctx, Channel channel, Object msg)
            throws Exception
    {
      HttpResponse nettyResponse = (HttpResponse) msg;

      RestResponseBuilder builder = new RestResponseBuilder();

      HttpResponseStatus status = nettyResponse.getStatus();
      builder.setStatus(status.getCode());

      for (Map.Entry<String, String> e : nettyResponse.getHeaders())
      {
        builder.unsafeAddHeaderValue(e.getKey(), e.getValue());
      }

      ChannelBuffer buf = nettyResponse.getContent();
      byte[] array = new byte[buf.readableBytes()];
      buf.readBytes(array);
      builder.setEntity(array);

      return builder.build();
    }
View Full Code Here

{
  @Test
  public void testReplayWithRestException()
  {
    final Request req = request();
    final RestResponse res = new RestResponseBuilder().setStatus(RestStatus.NOT_FOUND).build();

    final CaptureLastCallFilter captureFilter = new CaptureLastCallFilter();
    final FilterChain fc = getFilterChain().addFirst(captureFilter);

    // Record a response for the request we will fire
View Full Code Here

  }

  @Test
  public void testSimpleRestRes() throws IOException
  {
    final RestResponse expected = new RestResponseBuilder()
            .build();
    assertMsgEquals(expected, _serializer.readRestResponse(getResource("simple-rest-res.txt")));
  }
View Full Code Here

    return new RestRequestBuilder(URI.create("http://linkedin.com")).build();
  }

  private RestResponse createRestResponse()
  {
    return new RestResponseBuilder().build();
  }
View Full Code Here

    );
  }

  private void fireRestResponse(FilterChain fc)
  {
    fc.onRestResponse(new RestResponseBuilder().build(),
                      createRequestContext(), createWireAttributes()
    );
  }
View Full Code Here

TOP

Related Classes of com.linkedin.r2.message.rest.RestResponseBuilder

Copyright © 2018 www.massapicom. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.