Merge "Fixed imports."
[bgpcep.git] / bgp / rib-impl / src / main / java / org / opendaylight / protocol / bgp / rib / impl / BGPImpl.java
1 /*
2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.protocol.bgp.rib.impl;
9
10 import java.io.Closeable;
11 import java.io.IOException;
12 import java.net.InetSocketAddress;
13 import java.util.concurrent.ExecutionException;
14
15 import org.opendaylight.protocol.bgp.parser.BGPSession;
16 import org.opendaylight.protocol.bgp.parser.BGPSessionListener;
17 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPDispatcher;
18 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPSessionProposal;
19 import org.opendaylight.protocol.concepts.ListenerRegistration;
20 import org.opendaylight.protocol.framework.ReconnectStrategy;
21
22 import com.google.common.base.Preconditions;
23
24 /**
25  * Implementation of {@link BGP}.
26  */
27 public class BGPImpl implements BGP, Closeable {
28         /**
29          * Wrapper class to give listener a close method.
30          */
31         public class BGPListenerRegistration extends ListenerRegistration<BGPSessionListener> {
32                 private final BGPSession session;
33
34                 public BGPListenerRegistration(final BGPSessionListener l, final BGPSession session) {
35                         super(l);
36                         this.session = session;
37                 }
38
39                 @Override
40                 public void removeRegistration() {
41                         this.session.close();
42                 }
43         }
44
45         private final BGPDispatcher dispatcher;
46
47         private final InetSocketAddress address;
48
49         private final BGPSessionProposal proposal;
50
51         public BGPImpl(final BGPDispatcher dispatcher, final InetSocketAddress address, final BGPSessionProposal proposal) {
52                 this.dispatcher = Preconditions.checkNotNull(dispatcher);
53                 this.address = Preconditions.checkNotNull(address);
54                 this.proposal = Preconditions.checkNotNull(proposal);
55         }
56
57         /**
58          * {@inheritDoc}
59          */
60         @Override
61         public BGPListenerRegistration registerUpdateListener(final BGPSessionListener listener, final ReconnectStrategy strategy) throws IOException {
62                 final BGPSession session;
63                 try {
64                         session = this.dispatcher.createClient(this.address, this.proposal.getProposal(), listener, strategy).get();
65                 } catch (InterruptedException | ExecutionException e) {
66                         throw new IOException("Failed to connect to peer", e);
67                 }
68                 return new BGPListenerRegistration(listener, session);
69         }
70
71         @Override
72         public void close() {
73
74         }
75 }