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.common;
20
21 import java.io.IOException;
22
23 import org.eclipse.jetty.util.Callback;
24 import org.eclipse.jetty.util.SharedBlockingCallback;
25 import org.eclipse.jetty.websocket.api.WriteCallback;
26
27
28 /* ------------------------------------------------------------ */
29 /** extend a SharedlBlockingCallback to an websocket WriteCallback
30 */
31 public class BlockingWriteCallback extends SharedBlockingCallback
32 {
33 public BlockingWriteCallback()
34 {
35 }
36
37 public WriteBlocker acquireWriteBlocker() throws IOException
38 {
39 return new WriteBlocker(acquire());
40 }
41
42 public static class WriteBlocker implements WriteCallback, Callback, AutoCloseable
43 {
44 Blocker blocker;
45
46 WriteBlocker(Blocker blocker)
47 {
48 this.blocker=blocker;
49 }
50
51 @Override
52 public void writeFailed(Throwable x)
53 {
54 blocker.failed(x);
55 }
56
57 @Override
58 public void writeSuccess()
59 {
60 blocker.succeeded();
61 }
62
63 @Override
64 public void succeeded()
65 {
66 blocker.succeeded();
67 }
68
69 @Override
70 public void failed(Throwable x)
71 {
72 blocker.failed(x);
73 }
74
75 @Override
76 public void close() throws IOException
77 {
78 blocker.close();
79 }
80
81 public void block() throws IOException
82 {
83 blocker.block();
84 }
85 }
86 }