View Javadoc
1   package com.ericsson.research.transport;
2   
3   /*
4    * ##_BEGIN_LICENSE_##
5    * Transport Abstraction Package (trap)
6    * ----------
7    * Copyright (C) 2014 Ericsson AB
8    * ----------
9    * Redistribution and use in source and binary forms, with or without modification,
10   * are permitted provided that the following conditions are met:
11   * 
12   * 1. Redistributions of source code must retain the above copyright notice, this
13   *    list of conditions and the following disclaimer.
14   * 
15   * 2. Redistributions in binary form must reproduce the above copyright notice,
16   *    this list of conditions and the following disclaimer in the documentation
17   *    and/or other materials provided with the distribution.
18   * 
19   * 3. Neither the name of the Ericsson AB nor the names of its contributors
20   *    may be used to endorse or promote products derived from this software without
21   *    specific prior written permission.
22   * 
23   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
24   * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
25   * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
26   * IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
27   * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
28   * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
29   * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
30   * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
31   * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
32   * OF THE POSSIBILITY OF SUCH DAMAGE.
33   * ##_END_LICENSE_##
34   */
35  
36  import java.io.IOException;
37  import java.net.SocketAddress;
38  import java.nio.channels.SelectableChannel;
39  import java.nio.channels.ServerSocketChannel;
40  import java.nio.channels.SocketChannel;
41  
42  public class NioWaitingSocket {
43  	
44  	private final NioEndpoint	socket;
45  	private final int			ops;
46  	private boolean				done		= false;
47  	private long				timeoutTime	= System.currentTimeMillis() + 30000;
48  	private int					retries		= 5;
49  	private final boolean		server;
50  	private final SocketAddress	addr;
51  	private SelectableChannel	channel;
52  	
53  	public NioWaitingSocket(NioEndpoint socket, SocketAddress addr, boolean server, int ops) {
54  		this.socket = socket;
55  		this.addr = addr;
56  		this.server = server;
57  		this.ops = ops;
58  	}
59  	
60  	public SelectableChannel getChannel() {
61  		return this.channel;
62  	}
63  
64  	public SelectableChannel createChannel() throws IOException {
65  		if (this.server) {
66  			this.channel = ServerSocketChannel.open();
67  			this.channel.configureBlocking(false);
68  			((ServerSocketChannel) this.channel).socket().setReuseAddress(true);
69  			((ServerSocketChannel) this.channel).socket().bind(this.addr);
70  		} else {
71  			this.channel = SocketChannel.open();
72  			this.channel.configureBlocking(false);
73  			((SocketChannel) this.channel).connect(this.addr);
74  		}
75  		return this.channel;
76  	}
77  
78  	public int getOps() {
79  		return this.ops;
80  	}
81  
82  	public NioEndpoint getSocket() {
83  		return this.socket;
84  	}
85  
86  	public void setDone(boolean done) {
87  		this.done = done;
88  	}
89  
90  	public boolean isDone() {
91  		return this.done;
92  	}
93  
94  	public void setTimeoutTime(long timeoutTime) {
95  		this.timeoutTime = timeoutTime;
96  	}
97  
98  	public long getTimeoutTime() {
99  		return this.timeoutTime;
100 	}
101 
102 	public void setRetries(int retries) {
103 		this.retries = retries;
104 	}
105 
106 	public int getRetries() {
107 		return this.retries;
108 	}
109 	
110 }