[ONOS-2592] BGP peer connection initiation on peer configuration.
Change-Id: I1892c9b9f702a4a233790a69f4289e5f4edb0ec3
Showing
1 changed file
with
133 additions
and
0 deletions
1 | +/* | ||
2 | + * Copyright 2015 Open Networking Laboratory | ||
3 | + * | ||
4 | + * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with | ||
5 | + * the License. You may obtain a copy of the License at | ||
6 | + * | ||
7 | + * http://www.apache.org/licenses/LICENSE-2.0 | ||
8 | + * | ||
9 | + * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on | ||
10 | + * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the | ||
11 | + * specific language governing permissions and limitations under the License. | ||
12 | + */ | ||
13 | +package org.onosproject.bgp.controller.impl; | ||
14 | + | ||
15 | +import java.net.InetSocketAddress; | ||
16 | +import java.util.concurrent.Executors; | ||
17 | +import java.util.concurrent.ScheduledExecutorService; | ||
18 | +import java.util.concurrent.TimeUnit; | ||
19 | + | ||
20 | +import org.jboss.netty.bootstrap.ClientBootstrap; | ||
21 | +import org.jboss.netty.channel.ChannelFuture; | ||
22 | +import org.jboss.netty.channel.ChannelFutureListener; | ||
23 | +import org.jboss.netty.channel.ChannelPipelineFactory; | ||
24 | +import org.onosproject.bgp.controller.BGPCfg; | ||
25 | +import org.onosproject.bgp.controller.BGPController; | ||
26 | +import org.onosproject.bgp.controller.BGPPeerCfg; | ||
27 | +import org.onosproject.bgp.controller.BgpConnectPeer; | ||
28 | +import org.slf4j.Logger; | ||
29 | +import org.slf4j.LoggerFactory; | ||
30 | + | ||
31 | +/** | ||
32 | + * Implements connection initiation to peer on peer configuration and manage channel using netty channel handler. | ||
33 | + */ | ||
34 | +public class BgpConnectPeerImpl implements BgpConnectPeer { | ||
35 | + private static final Logger log = LoggerFactory.getLogger(BgpConnectPeerImpl.class); | ||
36 | + | ||
37 | + private ScheduledExecutorService connectExecutor = null; | ||
38 | + private final String peerHost; | ||
39 | + private static final int RETRY_INTERVAL = 4; | ||
40 | + private final int peerPort; | ||
41 | + private int connectRetryCounter = 0; | ||
42 | + private int connectRetryTime; | ||
43 | + private ChannelPipelineFactory pfact; | ||
44 | + private ClientBootstrap peerBootstrap; | ||
45 | + private BGPCfg bgpconfig; | ||
46 | + | ||
47 | + /** | ||
48 | + * Initialize timer and initiate pipeline factory. | ||
49 | + * | ||
50 | + * @param bgpController parent BGP controller | ||
51 | + * @param remoteHost remote host to connect | ||
52 | + * @param remotePort remote port to connect | ||
53 | + */ | ||
54 | + public BgpConnectPeerImpl(BGPController bgpController, String remoteHost, int remotePort) { | ||
55 | + | ||
56 | + this.bgpconfig = bgpController.getConfig(); | ||
57 | + this.pfact = new BGPPipelineFactory(bgpController, false); | ||
58 | + this.peerBootstrap = Controller.peerBootstrap(); | ||
59 | + this.peerBootstrap.setPipelineFactory(pfact); | ||
60 | + this.peerHost = remoteHost; | ||
61 | + this.peerPort = remotePort; | ||
62 | + this.connectRetryTime = 0; | ||
63 | + } | ||
64 | + | ||
65 | + @Override | ||
66 | + public void disconnectPeer() { | ||
67 | + if (connectExecutor != null) { | ||
68 | + connectExecutor.shutdown(); | ||
69 | + connectExecutor = null; | ||
70 | + } | ||
71 | + } | ||
72 | + | ||
73 | + @Override | ||
74 | + public void connectPeer() { | ||
75 | + scheduleConnectionRetry(this.connectRetryTime); | ||
76 | + } | ||
77 | + | ||
78 | + /** | ||
79 | + * Retry connection with exponential back-off mechanism. | ||
80 | + * | ||
81 | + * @param retryDelay retry delay | ||
82 | + */ | ||
83 | + private void scheduleConnectionRetry(long retryDelay) { | ||
84 | + if (this.connectExecutor == null) { | ||
85 | + this.connectExecutor = Executors.newSingleThreadScheduledExecutor(); | ||
86 | + } | ||
87 | + this.connectExecutor.schedule(new ConnectionRetry(), retryDelay, TimeUnit.MINUTES); | ||
88 | + } | ||
89 | + | ||
90 | + /** | ||
91 | + * Implements BGP connection and manages connection to peer with back-off mechanism in case of failure. | ||
92 | + */ | ||
93 | + class ConnectionRetry implements Runnable { | ||
94 | + @Override | ||
95 | + public void run() { | ||
96 | + log.debug("Connect to peer {}", peerHost); | ||
97 | + | ||
98 | + InetSocketAddress connectToSocket = new InetSocketAddress(peerHost, peerPort); | ||
99 | + | ||
100 | + try { | ||
101 | + bgpconfig.setPeerConnState(peerHost, BGPPeerCfg.State.CONNECT); | ||
102 | + peerBootstrap.connect(connectToSocket).addListener(new ChannelFutureListener() { | ||
103 | + @Override | ||
104 | + public void operationComplete(ChannelFuture future) throws Exception { | ||
105 | + if (!future.isSuccess()) { | ||
106 | + bgpconfig.setPeerConnState(peerHost, BGPPeerCfg.State.ACTIVE); | ||
107 | + connectRetryCounter++; | ||
108 | + log.error("Connection failed, ConnectRetryCounter {} remote host {}", connectRetryCounter, | ||
109 | + peerHost); | ||
110 | + /* | ||
111 | + * Reconnect to peer on failure is exponential till 4 mins, later on retry after every 4 | ||
112 | + * mins. | ||
113 | + */ | ||
114 | + if (connectRetryTime < RETRY_INTERVAL) { | ||
115 | + connectRetryTime = (connectRetryTime != 0) ? connectRetryTime * 2 : 1; | ||
116 | + } | ||
117 | + scheduleConnectionRetry(connectRetryTime); | ||
118 | + } else { | ||
119 | + | ||
120 | + connectRetryCounter++; | ||
121 | + log.info("Connected to remote host {}, Connect Counter {}", peerHost, connectRetryCounter); | ||
122 | + disconnectPeer(); | ||
123 | + return; | ||
124 | + } | ||
125 | + } | ||
126 | + }); | ||
127 | + } catch (Exception e) { | ||
128 | + log.info("Connect peer exception : " + e.toString()); | ||
129 | + disconnectPeer(); | ||
130 | + } | ||
131 | + } | ||
132 | + } | ||
133 | +} |
-
Please register or login to post a comment