1 /* 2 * Licensed to the Apache Software Foundation (ASF) under one or more 3 * contributor license agreements. See the NOTICE file distributed with 4 * this work for additional information regarding copyright ownership. 5 * The ASF licenses this file to You under the Apache License, Version 2.0 6 * (the "License"); you may not use this file except in compliance with 7 * the License. You may obtain a copy of the License at 8 * 9 * http://www.apache.org/licenses/LICENSE-2.0 10 * 11 * Unless required by applicable law or agreed to in writing, software 12 * distributed under the License is distributed on an "AS IS" BASIS, 13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14 * See the License for the specific language governing permissions and 15 * limitations under the License. 16 */ 17 package nginx.unit.websocket; 18 19 import java.util.ArrayList; 20 import java.util.List; 21 import java.util.Map; 22 import java.util.Map.Entry; 23 24 import javax.websocket.HandshakeResponse; 25 26 import org.apache.tomcat.util.collections.CaseInsensitiveKeyMap; 27 28 /** 29 * Represents the response to a WebSocket handshake. 30 */ 31 public class WsHandshakeResponse implements HandshakeResponse { 32 33 private final Map<String,List<String>> headers = new CaseInsensitiveKeyMap<>(); 34 35 WsHandshakeResponse()36 public WsHandshakeResponse() { 37 } 38 39 WsHandshakeResponse(Map<String,List<String>> headers)40 public WsHandshakeResponse(Map<String,List<String>> headers) { 41 for (Entry<String,List<String>> entry : headers.entrySet()) { 42 if (this.headers.containsKey(entry.getKey())) { 43 this.headers.get(entry.getKey()).addAll(entry.getValue()); 44 } else { 45 List<String> values = new ArrayList<>(entry.getValue()); 46 this.headers.put(entry.getKey(), values); 47 } 48 } 49 } 50 51 52 @Override getHeaders()53 public Map<String,List<String>> getHeaders() { 54 return headers; 55 } 56 } 57