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 java.io.IOException;
22 import java.nio.ByteBuffer;
23
24 import javax.websocket.OnMessage;
25
26 import org.eclipse.jetty.util.BufferUtil;
27 import org.eclipse.jetty.websocket.common.message.MessageAppender;
28 import org.eclipse.jetty.websocket.jsr356.endpoints.JsrAnnotatedEventDriver;
29
30 /**
31 * Partial BINARY MessageAppender for @{@link OnMessage} annotated methods
32 */
33 public class BinaryPartialOnMessage implements MessageAppender
34 {
35 private final JsrAnnotatedEventDriver driver;
36 private boolean finished;
37
38 public BinaryPartialOnMessage(JsrAnnotatedEventDriver driver)
39 {
40 this.driver = driver;
41 this.finished = false;
42 }
43
44 @Override
45 public void appendFrame(ByteBuffer payload, boolean isLast) throws IOException
46 {
47 if (finished)
48 {
49 throw new IOException("Cannot append to finished buffer");
50 }
51 if (payload == null)
52 {
53 driver.onPartialBinaryMessage(BufferUtil.EMPTY_BUFFER,isLast);
54 }
55 else
56 {
57 driver.onPartialBinaryMessage(payload,isLast);
58 }
59 }
60
61 @Override
62 public void messageComplete()
63 {
64 finished = true;
65 }
66 }