View Javadoc

1   //
2   //  ========================================================================
3   //  Copyright (c) 1995-2016 Mort Bay Consulting Pty. Ltd.
4   //  ------------------------------------------------------------------------
5   //  All rights reserved. This program and the accompanying materials
6   //  are made available under the terms of the Eclipse Public License v1.0
7   //  and Apache License v2.0 which accompanies this distribution.
8   //
9   //      The Eclipse Public License is available at
10  //      http://www.eclipse.org/legal/epl-v10.html
11  //
12  //      The Apache License v2.0 is available at
13  //      http://www.opensource.org/licenses/apache2.0.php
14  //
15  //  You may elect to redistribute this code under either of these licenses.
16  //  ========================================================================
17  //
18  
19  package org.eclipse.jetty.websocket.jsr356.messages;
20  
21  import javax.websocket.DecodeException;
22  import javax.websocket.Decoder;
23  import javax.websocket.Decoder.Binary;
24  import javax.websocket.MessageHandler;
25  import javax.websocket.MessageHandler.Whole;
26  
27  import org.eclipse.jetty.util.BufferUtil;
28  import org.eclipse.jetty.websocket.api.WebSocketException;
29  import org.eclipse.jetty.websocket.common.events.EventDriver;
30  import org.eclipse.jetty.websocket.common.message.SimpleBinaryMessage;
31  import org.eclipse.jetty.websocket.jsr356.DecoderFactory;
32  import org.eclipse.jetty.websocket.jsr356.MessageHandlerWrapper;
33  
34  public class BinaryWholeMessage extends SimpleBinaryMessage
35  {
36      private final MessageHandlerWrapper msgWrapper;
37      private final MessageHandler.Whole<Object> wholeHandler;
38  
39      @SuppressWarnings("unchecked")
40      public BinaryWholeMessage(EventDriver onEvent, MessageHandlerWrapper wrapper)
41      {
42          super(onEvent);
43          this.msgWrapper = wrapper;
44          this.wholeHandler = (Whole<Object>)wrapper.getHandler();
45      }
46  
47      @SuppressWarnings("unchecked")
48      @Override
49      public void messageComplete()
50      {
51          super.finished = true;
52  
53          byte data[] = out.toByteArray();
54  
55          DecoderFactory.Wrapper decoder = msgWrapper.getDecoder();
56          Decoder.Binary<Object> binaryDecoder = (Binary<Object>)decoder.getDecoder();
57          try
58          {
59              Object obj = binaryDecoder.decode(BufferUtil.toBuffer(data));
60              wholeHandler.onMessage(obj);
61          }
62          catch (DecodeException e)
63          {
64              throw new WebSocketException("Unable to decode binary data",e);
65          }
66      }
67  }