org.jdiameter.client.impl.StackImpl Maven / Gradle / Ivy
The newest version!
/*
* TeleStax, Open Source Cloud Communications
* Copyright 2011-2016, TeleStax Inc. and individual contributors
* by the @authors tag.
*
* This program is free software: you can redistribute it and/or modify
* under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation; either version 3 of
* the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see peerTable = peerManager.getPeerTable();
// considering only "to connect" peers are on the table at this time...
final CountDownLatch barrier = new CountDownLatch(Mode.ANY_PEER.equals(mode) ? Math.min(peerTable.size(), 1) : peerTable.size());
StateChangeListener listener = new AbstractStateChangeListener() {
@Override
public void stateChanged(Enum oldState, Enum newState) {
if (PeerState.OKAY.equals(newState)) {
barrier.countDown();
}
}
};
for (Peer p : peerTable) {
((IPeer) p).addStateChangeListener(listener);
}
startPeerManager();
try {
barrier.await(timeOut, timeUnit);
if (barrier.getCount() != 0) {
throw new InternalException("TimeOut");
}
state = StackState.STARTED;
}
catch (InterruptedException e) {
throw new InternalException("TimeOut");
}
finally {
for (Peer p : peerTable) {
((IPeer) p).remStateChangeListener(listener);
}
}
}
finally {
lock.unlock();
}
}
private void startPeerManager() throws InternalException {
try {
if (peerManager != null) {
peerManager.start();
}
getMetaData().unwrap(IMetaData.class).updateLocalHostStateId();
}
catch (Exception e) {
throw new InternalException(e);
}
}
@Override
@SuppressWarnings("unchecked")
public void stop(long timeOut, TimeUnit timeUnit, int disconnectCause) throws IllegalDiameterStateException, InternalException {
lock.lock();
try {
if (state == StackState.STARTED || state == StackState.CONFIGURED) {
if (log.isInfoEnabled()) {
log.info("(-)(-)(-)(-)(-) Stopping " + VersionProperties.instance.getProperty("vendor") + " DIAMETER Stack v" +
VersionProperties.instance.getProperty("version") + " (-)(-)(-)(-)(-)");
}
List peerTable = peerManager.getPeerTable();
final CountDownLatch barrier = new CountDownLatch(peerTable.size());
StateChangeListener listener = new AbstractStateChangeListener() {
@Override
public void stateChanged(Enum oldState, Enum newState) {
if (PeerState.DOWN.equals(newState)) {
barrier.countDown();
}
}
};
for (Peer p : peerTable) {
if (p.getState(PeerState.class).equals(PeerState.DOWN)) {
barrier.countDown();
}
else {
((IPeer) p).addStateChangeListener(listener);
}
}
if (peerManager != null) {
try {
peerManager.stopping(disconnectCause);
}
catch (Exception e) {
log.warn("Stopping error", e);
}
}
try {
barrier.await(timeOut, timeUnit);
if (barrier.getCount() != 0) {
throw new InternalException("TimeOut");
}
}
catch (InterruptedException e) {
throw new InternalException("TimeOut");
}
finally {
state = StackState.STOPPED;
for (Peer p : peerTable) {
((IPeer) p).remStateChangeListener(listener);
}
}
assembler.getComponentInstance(ISessionDatasource.class).stop();
assembler.getComponentInstance(IStatisticProcessor.class).stop();
try {
if (peerManager != null) {
peerManager.stopped();
}
// Clear all timeout tasks
if (scheduledFacility != null) {
concurrentFactory.shutdownNow(scheduledFacility);
}
}
catch (Exception e) {
log.warn("Stopped error", e);
}
state = StackState.STOPPED;
if (log.isInfoEnabled()) {
log.info("(-)(-)(-)(-)(-) Stopped " + VersionProperties.instance.getProperty("vendor") + " DIAMETER Stack v" +
VersionProperties.instance.getProperty("version") + " (-)(-)(-)(-)(-)");
}
}
}
finally {
lock.unlock();
}
}
@Override
public void destroy() {
// Be friendly
if (state == StackState.STARTED) {
log.warn("Calling destroy() with Stack in STARTED state. Calling stop(REBOOTING) before, please do it yourself with the proper cause.");
stop(DisconnectCause.REBOOTING);
}
lock.lock();
try {
if (peerManager != null) {
peerManager.destroy();
}
if (assembler != null) {
assembler.destroy();
}
if (scheduledFacility != null) {
concurrentFactory.shutdownNow(scheduledFacility);
}
}
catch (Exception e) {
log.warn("Destroy error", e);
}
finally {
state = StackState.IDLE;
lock.unlock();
}
}
@Override
public boolean isActive() {
return state == StackState.STARTED;
}
@Override
public java.util.logging.Logger getLogger() {
return java.util.logging.Logger.getAnonymousLogger();
}
@Override
public MetaData getMetaData() {
if (state == StackState.IDLE) {
throw new IllegalStateException("Meta data not defined");
}
return assembler.getComponentInstance(IMetaData.class);
}
@Override
@SuppressWarnings("unchecked")
public T getSession(String sessionId, Class clazz) throws InternalException {
if (getState() == StackState.IDLE) {
throw new InternalException("Illegal state of stack");
}
BaseSession bs = assembler.getComponentInstance(ISessionDatasource.class).getSession(sessionId);
return bs != null ? (T) bs : null;
}
@Override
public boolean isWrapperFor(Class aClass) throws InternalException {
boolean isWrap = aClass == PeerTable.class;
if (!isWrap) {
isWrap = assembler.getChilds()[StackLayer.id()].getComponentInstance(aClass) != null;
}
if (!isWrap) {
isWrap = assembler.getChilds()[ControllerLayer.id()].getComponentInstance(aClass) != null;
}
if (!isWrap) {
isWrap = assembler.getChilds()[TransportLayer.id()].getComponentInstance(aClass) != null;
}
return isWrap;
}
@Override
@SuppressWarnings("unchecked")
public T unwrap(Class aClass) throws InternalException {
Object unwrapObject = null;
if (aClass == PeerTable.class) {
unwrapObject = assembler.getComponentInstance(aClass);
}
// TODO: "layers" should be removed....
if (unwrapObject == null) {
unwrapObject = assembler.getChilds()[StackLayer.id()].getComponentInstance(aClass);
}
if (unwrapObject == null) {
unwrapObject = assembler.getChilds()[ControllerLayer.id()].getComponentInstance(aClass);
}
if (unwrapObject == null) {
unwrapObject = assembler.getChilds()[TransportLayer.id()].getComponentInstance(aClass);
}
return (T) unwrapObject;
}
// Extended methods
@Override
public StackState getState() {
return state;
}
@Override
public Configuration getConfiguration() {
return config;
}
@Override
public IAssembler getAssemblerFacility() {
return assembler;
}
@Override
public void sendMessage(IMessage message) throws RouteException, AvpDataException, IllegalDiameterStateException, IOException {
peerManager.sendMessage(message);
}
@Override
public void addSessionListener(String sessionId, NetworkReqListener listener) {
peerManager.addSessionReqListener(sessionId, listener);
}
@Override
public void removeSessionListener(String sessionId) {
peerManager.removeSessionListener(sessionId);
}
@Override
public ScheduledExecutorService getScheduledFacility() {
return scheduledFacility;
}
@Override
public IConcurrentFactory getConcurrentFactory() {
return this.concurrentFactory;
}
@Override
public String configuration() {
return config != null ? config.toString() : "not set";
}
@Override
public String metaData() {
try {
return getMetaData().toString();
}
catch (Exception exc) {
return "not set";
}
}
@Override
public String peerDescription(String name) {
try {
for (Peer p : unwrap(PeerTable.class).getPeerTable()) {
if (p.getUri().getFQDN().equals(name)) {
return p.toString();
}
}
}
catch (InternalException e) {
log.debug("InternalException", e);
}
return "not set";
}
@Override
public String peerList() {
try {
return unwrap(PeerTable.class).getPeerTable().toString();
}
catch (InternalException e) {
return "not set";
}
}
@Override
public void stop(int disconnectCause) {
try {
stop(10, TimeUnit.SECONDS, disconnectCause);
}
catch (Exception e) {
log.debug("Exception", e);
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy