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 * Extends a {@link SharedBlockingCallback} to a WebSocket {@link WriteCallback}
29 */
30 public class BlockingWriteCallback extends SharedBlockingCallback
31 {
32 public BlockingWriteCallback()
33 {
34 }
35
36 public WriteBlocker acquireWriteBlocker() throws IOException
37 {
38 return new WriteBlocker(acquire());
39 }
40
41 public static class WriteBlocker implements WriteCallback, Callback.NonBlocking, AutoCloseable
42 {
43 private final Blocker blocker;
44
45 protected WriteBlocker(Blocker blocker)
46 {
47 this.blocker=blocker;
48 }
49
50 @Override
51 public void writeFailed(Throwable x)
52 {
53 blocker.failed(x);
54 }
55
56 @Override
57 public void writeSuccess()
58 {
59 blocker.succeeded();
60 }
61
62 @Override
63 public void succeeded()
64 {
65 blocker.succeeded();
66 }
67
68 @Override
69 public void failed(Throwable x)
70 {
71 blocker.failed(x);
72 }
73
74 @Override
75 public void close()
76 {
77 blocker.close();
78 }
79
80 public void block() throws IOException
81 {
82 blocker.block();
83 }
84 }
85 }