org.argeo.cms.ee/src/org/argeo/cms/websocket/server/WebSocketHandshakeResponse.java

   1 package org.argeo.cms.websocket.server;
   2
   3 import java.util.ArrayList;
   4 import java.util.Collections;
   5 import java.util.List;
   6
   7 import javax.websocket.HandshakeResponse;
   8
   9 import org.argeo.cms.auth.RemoteAuthResponse;
  10
  11 public class WebSocketHandshakeResponse implements RemoteAuthResponse {
  12         private final HandshakeResponse handshakeResponse;
  13
  14         public WebSocketHandshakeResponse(HandshakeResponse handshakeResponse) {
  15                 this.handshakeResponse = handshakeResponse;
  16         }
  17
  18         @Override
  19         public void setHeader(String headerName, String value) {
  20                 handshakeResponse.getHeaders().put(headerName, Collections.singletonList(value));
  21         }
  22
  23         @Override
  24         public void addHeader(String headerName, String value) {
  25                 List<String> values = handshakeResponse.getHeaders().getOrDefault(headerName, new ArrayList<>());
  26                 values.add(value);
  27         }
  28
  29 }