@SuppressWarnings("deprecation") public GetRequestGroup(Request<?> request, int maxBatchSize) { _baseUriTemplate = request.getBaseUriTemplate(); _headers = request.getHeaders(); _queryParams = getQueryParamsForBatchingKey(request); _resourceSpec = request.getResourceSpec(); _requestOptions = request.getRequestOptions(); _pathKeys = request.getPathKeys(); _maxBatchSize = maxBatchSize; }
private static IndividualRequest toIndividualRequest(Request<?> request, IndividualRequestMap dependantRequests) throws RestLiEncodingException { //TODO: Hardcoding RESTLI_PROTOCOL_2_0_0 for now. We need to refactor this code later to get protocol version using the mechanism similar to // RestClient.getProtocolVersionForService() ProtocolVersion protocolVersion = AllProtocolVersions.RESTLI_PROTOCOL_2_0_0.getProtocolVersion(); String relativeUrl = getRelativeUrl(request, protocolVersion); IndividualRequest individualRequest = new IndividualRequest(); individualRequest.setRelativeUrl(relativeUrl); individualRequest.setMethod(request.getMethod().getHttpMethod().name()); individualRequest.setHeaders(new StringMap(request.getHeaders())); List<HttpCookie> cookies = request.getCookies(); if (cookies != null && !cookies.isEmpty()) { throw new IllegalArgumentException(String.format("Cookies for individual request '%s' MUST be added at the envelope request level", relativeUrl)); } individualRequest.setBody(getBody(request, protocolVersion), SetMode.IGNORE_NULL); individualRequest.setDependentRequests(dependantRequests); return individualRequest; }
@Override public void onSuccess(ProtocolVersion protocolVersion) { URI requestUri = RestliUriBuilderUtil.createUriBuilder(request, _uriPrefix, protocolVersion).build(); final ResourceMethod method = request.getMethod(); final String methodName = request.getMethodName(); addDisruptContext(request.getBaseUriTemplate(), method, methodName, requestContext); sendRestRequestImpl(requestContext, requestUri, method, input != null ? RequestBodyTransformer.transform(request, protocolVersion) : null, request.getHeaders(), CookieUtil.encodeCookies(request.getCookies()), methodName, protocolVersion, request.getRequestOptions(), callback); } });
@Override public void onSuccess(ProtocolVersion protocolVersion) { URI requestUri = RestliUriBuilderUtil.createUriBuilder(request, _uriPrefix, protocolVersion).build(); final ResourceMethod method = request.getMethod(); final String methodName = request.getMethodName(); addDisruptContext(request.getBaseUriTemplate(), method, methodName, requestContext); sendStreamRequestImpl(requestContext, requestUri, method, input != null ? RequestBodyTransformer.transform(request, protocolVersion) : null, request.getHeaders(), CookieUtil.encodeCookies(request.getCookies()), methodName, protocolVersion, request.getRequestOptions(), request.getStreamingAttachments(), callback); } });