/*
 * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */
package org.opendaylight.protocol.bgp.rib.impl;

import java.io.Closeable;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.util.concurrent.ExecutionException;

import org.opendaylight.protocol.bgp.parser.BGPSession;
import org.opendaylight.protocol.bgp.parser.BGPSessionListener;
import org.opendaylight.protocol.bgp.rib.impl.spi.BGPDispatcher;
import org.opendaylight.protocol.bgp.rib.impl.spi.BGPSessionProposal;
import org.opendaylight.protocol.concepts.ListenerRegistration;
import org.opendaylight.protocol.framework.ReconnectStrategy;

import com.google.common.base.Preconditions;

/**
 * Implementation of {@link BGP}.
 */
public class BGPImpl implements BGP, Closeable {
	/**
	 * Wrapper class to give listener a close method.
	 */
	public class BGPListenerRegistration extends ListenerRegistration<BGPSessionListener> {
		private final BGPSession session;

		public BGPListenerRegistration(final BGPSessionListener l, final BGPSession session) {
			super(l);
			this.session = session;
		}

		@Override
		public void removeRegistration() {
			this.session.close();
		}
	}

	private final BGPDispatcher dispatcher;

	private final InetSocketAddress address;

	private final BGPSessionProposal proposal;

	public BGPImpl(final BGPDispatcher dispatcher, final InetSocketAddress address, final BGPSessionProposal proposal) {
		this.dispatcher = Preconditions.checkNotNull(dispatcher);
		this.address = Preconditions.checkNotNull(address);
		this.proposal = Preconditions.checkNotNull(proposal);
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public BGPListenerRegistration registerUpdateListener(final BGPSessionListener listener, final ReconnectStrategy strategy) throws IOException {
		final BGPSession session;
		try {
			session = this.dispatcher.createClient(this.address, this.proposal.getProposal(), listener, strategy).get();
		} catch (InterruptedException | ExecutionException e) {
			throw new IOException("Failed to connect to peer", e);
		}
		return new BGPListenerRegistration(listener, session);
	}

	@Override
	public void close() {

	}
}