public class MyServiceFactory implements Factory<MyService> { @Context private HttpHeaders headers; @Override public MyService provide() { return new MyService(headers.getHeaderString("X-Header")); } @Override public void dispose(MyService service) { /* noop */ } } register(new AbstractBinder() { @Override public void configure() { bindFactory(MyServiceFactory.class).to(MyService.class) .in(RequestScoped.class); } });
private void checkCRC(HttpHeaders headers, String offlineTableName, String segmentName, long existingCrc) { String expectedCrcStr = headers.getHeaderString(HttpHeaders.IF_MATCH); if (expectedCrcStr != null) { long expectedCrc; try { expectedCrc = Long.parseLong(expectedCrcStr); } catch (NumberFormatException e) { throw new ControllerApplicationException(LOGGER, "Caught exception for segment: " + segmentName + " of table: " + offlineTableName + " while parsing IF-MATCH CRC: \"" + expectedCrcStr + "\"", Response.Status.PRECONDITION_FAILED); } if (expectedCrc != existingCrc) { throw new ControllerApplicationException(LOGGER, "For segment: " + segmentName + " of table: " + offlineTableName + ", expected CRC: " + expectedCrc + " does not match existing CRC: " + existingCrc, Response.Status.PRECONDITION_FAILED); } } }
String rpcTraceContext = headers.getHeaderString(RemotingConstants.NEW_RPC_TRACE_NAME); if (StringUtils.isNotBlank(rpcTraceContext)) { String traceIdKey = headers.getHeaderString(RemotingConstants.HTTP_HEADER_TRACE_ID_KEY); String rpcIdKey = headers.getHeaderString(RemotingConstants.HTTP_HEADER_RPC_ID_KEY); if (StringUtils.isEmpty(rpcIdKey)) { rpcIdKey = request.getUri().getQueryParameters().getFirst(RemotingConstants.RPC_ID_KEY); map.put(RemotingConstants.TRACE_ID_KEY, traceIdKey); map.put(RemotingConstants.RPC_ID_KEY, rpcIdKey); String penAttrs = headers.getHeaderString(RemotingConstants.PEN_ATTRS_KEY); map.put(RemotingConstants.PEN_ATTRS_KEY, penAttrs); sofaRequest.addRequestProp(RemotingConstants.RPC_TRACE_NAME, map);
String rpcTraceContext = headers.getHeaderString(RemotingConstants.NEW_RPC_TRACE_NAME); if (StringUtils.isNotBlank(rpcTraceContext)) { String traceIdKey = headers.getHeaderString(RemotingConstants.HTTP_HEADER_TRACE_ID_KEY); String rpcIdKey = headers.getHeaderString(RemotingConstants.HTTP_HEADER_RPC_ID_KEY); if (StringUtils.isEmpty(rpcIdKey)) { rpcIdKey = request.getUri().getQueryParameters().getFirst(RemotingConstants.RPC_ID_KEY); map.put(RemotingConstants.TRACE_ID_KEY, traceIdKey); map.put(RemotingConstants.RPC_ID_KEY, rpcIdKey); String penAttrs = headers.getHeaderString(RemotingConstants.PEN_ATTRS_KEY); map.put(RemotingConstants.PEN_ATTRS_KEY, penAttrs); sofaRequest.addRequestProp(RemotingConstants.RPC_TRACE_NAME, map);
public void completeSegmentOperations(SegmentMetadata segmentMetadata, URI finalSegmentLocationURI, File currentSegmentLocation, boolean enableParallelPushProtection, HttpHeaders headers, String zkDownloadURI, boolean moveSegmentToFinalLocation) throws Exception { String rawTableName = segmentMetadata.getTableName(); String offlineTableName = TableNameBuilder.OFFLINE.tableNameWithType(rawTableName); String segmentName = segmentMetadata.getName(); ZNRecord znRecord = _pinotHelixResourceManager.getSegmentMetadataZnRecord(offlineTableName, segmentName); // Brand new segment, not refresh, directly add the segment if (znRecord == null) { LOGGER.info("Adding new segment {} from table {}", segmentName, rawTableName); String crypter = headers.getHeaderString(FileUploadDownloadClient.CustomHeaders.CRYPTER); processNewSegment(segmentMetadata, finalSegmentLocationURI, currentSegmentLocation, zkDownloadURI, crypter, rawTableName, segmentName, moveSegmentToFinalLocation); return; } LOGGER.info("Segment {} from table {} already exists, refreshing if necessary", segmentName, rawTableName); processExistingSegment(segmentMetadata, finalSegmentLocationURI, currentSegmentLocation, enableParallelPushProtection, headers, zkDownloadURI, offlineTableName, segmentName, znRecord, moveSegmentToFinalLocation); }
@ApiOperation(value = "List all collectors") public Response listCollectors(@Context HttpHeaders httpHeaders) { String ifNoneMatch = httpHeaders.getHeaderString("If-None-Match"); Boolean etagCached = false; Response.ResponseBuilder builder = Response.noContent();
@ApiParam(name = "configurationId", required = true) @PathParam("configurationId") String configurationId) throws RenderTemplateException { String ifNoneMatch = httpHeaders.getHeaderString("If-None-Match"); Boolean etagCached = false; Response.ResponseBuilder builder = Response.noContent();
String uploadTypeStr = headers.getHeaderString(FileUploadDownloadClient.CustomHeaders.UPLOAD_TYPE); FileUploadDownloadClient.FileUploadType uploadType = getUploadType(uploadTypeStr); String crypterClassHeader = headers.getHeaderString(FileUploadDownloadClient.CustomHeaders.CRYPTER); String currentSegmentLocationURI = headers.getHeaderString(FileUploadDownloadClient.CustomHeaders.DOWNLOAD_URI);
String crypter = headers.getHeaderString(FileUploadDownloadClient.CustomHeaders.CRYPTER); existingSegmentZKMetadata.setCrypterName(crypter); headers.getHeaderString(FileUploadDownloadClient.CustomHeaders.SEGMENT_ZK_METADATA_CUSTOM_MAP_MODIFIER); SegmentZKMetadataCustomMapModifier segmentZKMetadataCustomMapModifier; if (segmentZKMetadataCustomMapModifierStr != null) {
String remoteIP = httpHeaders.getHeaderString("X-Forwarded-For"); if (remoteIP == null) { remoteIP = httpHeaders.getHeaderString("X-Real-IP");
String remoteIP = httpHeaders.getHeaderString("X-Forwarded-For"); if (remoteIP == null) { remoteIP = httpHeaders.getHeaderString("X-Real-IP");
@Override public String getHeaderString(String name) { return httpRequest.getHttpHeaders().getHeaderString(name); }
@Override public InputStream getInputSteam(InputStream is, HttpHeaders httpHeaders) throws IOException { String fileUrl = httpHeaders.getHeaderString("fileUrl"); if(fileUrl != null && !"".equals(fileUrl)){ return TikaInputStream.get(new URL(fileUrl)); } return is; } }
@GET @Timed @Compress(buffer=(1024 * 40)) @Produces(APPLICATION_JSON_WITH_CHARSET) public Response get(@Context HugeConfig conf, @Context HttpHeaders headers, @Context UriInfo uriInfo) { String location = conf.get(ServerOptions.GREMLIN_SERVER_URL); String auth = headers.getHeaderString(HttpHeaders.AUTHORIZATION); String query = uriInfo.getRequestUri().getRawQuery(); return doGetRequest(location, auth, query); } }
static String getHeaderString(HttpRequest request, String header) { if (request == null) { return null; } if (request.getHttpHeaders() != null) { return request.getHttpHeaders().getHeaderString(header); } return null; }
@POST @Timed @Compress @Consumes(APPLICATION_JSON) @Produces(APPLICATION_JSON_WITH_CHARSET) public Response post(@Context HugeConfig conf, @Context HttpHeaders headers, String request) { /* The following code is reserved for forwarding request */ // context.getRequestDispatcher(location).forward(request, response); // return Response.seeOther(UriBuilder.fromUri(location).build()) // .build(); // Response.temporaryRedirect(UriBuilder.fromUri(location).build()) // .build(); String location = conf.get(ServerOptions.GREMLIN_SERVER_URL); String auth = headers.getHeaderString(HttpHeaders.AUTHORIZATION); return doPostRequest(location, auth, request); }
MyService service = Mockito.mock(MyService.class); when(service.getObject()).thenReturn(new Object()); HttpHeaders headers = Mockito.mock(HttpHeaders.class); when(headers.getHeaderString("X-Header")).thenReturn("blah"); UriInfo uriInfo = Mockito.mock(UriInfo.class); when(uriInfo.getRequestUri()).thenReturn(URI.create("http://localhost"));
@Override public String getHeaderString(String name) { return httpRequest.getHttpHeaders().getHeaderString(name); }
public Book getBook(Long id) throws BookNotFoundFault { if (hs == null) { throw new WebApplicationException(Response.serverError().build()); } boolean springProxy = PropertyUtils.isTrue(hs.getHeaderString("SpringProxy")); if (!springProxy && ui == null) { throw new WebApplicationException(Response.serverError().build()); } return books.get(id); }
private ResponseBuilder getBookByHeaderSimpleBuilder(@HeaderParam("BOOK") String headerBook, @HeaderParam("Simple") String headerSimple) throws Exception { ResponseBuilder builder = Response.ok(doGetBook(headerBook)); if (headerSimple != null) { builder.header("Simple", headerSimple); } String aHeaderValue = httpHeaders.getHeaderString("a"); if (aHeaderValue != null) { builder.header("a", aHeaderValue); } return builder; }