1 /* 2 * Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have any 23 * questions. 24 */ 25 26 package jdk.internal.net.http.common; 27 28 import java.net.http.HttpHeaders; 29 import java.util.ArrayList; 30 import java.util.List; 31 import java.util.Map; 32 import java.util.TreeMap; 33 import static jdk.internal.net.http.common.Utils.ACCEPT_ALL; 34 35 /** A mutable builder for collecting and building HTTP headers. */ 36 public class HttpHeadersBuilder { 37 38 private final TreeMap<String, List<String>> headersMap; 39 HttpHeadersBuilder()40 public HttpHeadersBuilder() { 41 headersMap = new TreeMap<>(String.CASE_INSENSITIVE_ORDER); 42 } 43 structuralCopy()44 public HttpHeadersBuilder structuralCopy() { 45 HttpHeadersBuilder builder = new HttpHeadersBuilder(); 46 for (Map.Entry<String, List<String>> entry : headersMap.entrySet()) { 47 List<String> valuesCopy = new ArrayList<>(entry.getValue()); 48 builder.headersMap.put(entry.getKey(), valuesCopy); 49 } 50 return builder; 51 } 52 addHeader(String name, String value)53 public void addHeader(String name, String value) { 54 headersMap.computeIfAbsent(name, k -> new ArrayList<>(1)) 55 .add(value); 56 } 57 setHeader(String name, String value)58 public void setHeader(String name, String value) { 59 // headers typically have one value 60 List<String> values = new ArrayList<>(1); 61 values.add(value); 62 headersMap.put(name, values); 63 } 64 clear()65 public void clear() { 66 headersMap.clear(); 67 } 68 map()69 public Map<String, List<String>> map() { 70 return headersMap; 71 } 72 build()73 public HttpHeaders build() { 74 return HttpHeaders.of(headersMap, ACCEPT_ALL); 75 } 76 77 @Override toString()78 public String toString() { 79 StringBuilder sb = new StringBuilder(); 80 sb.append(super.toString()).append(" { "); 81 sb.append(map()); 82 sb.append(" }"); 83 return sb.toString(); 84 } 85 } 86