org.elasticsearch.transport.CloseableConnection Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of elasticsearch Show documentation
Show all versions of elasticsearch Show documentation
Elasticsearch subproject :server
/*
* Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
* or more contributor license agreements. Licensed under the Elastic License
* 2.0 and the Server Side Public License, v 1; you may not use this file except
* in compliance with, at your election, the Elastic License 2.0 or the Server
* Side Public License, v 1.
*/
package org.elasticsearch.transport;
import org.elasticsearch.action.ActionListener;
import org.elasticsearch.core.AbstractRefCounted;
import org.elasticsearch.core.CompletableContext;
/**
* Abstract Transport.Connection that provides common close logic.
*/
public abstract class CloseableConnection extends AbstractRefCounted implements Transport.Connection {
private final CompletableContext closeContext = new CompletableContext<>();
private final CompletableContext removeContext = new CompletableContext<>();
@Override
public void addCloseListener(ActionListener listener) {
closeContext.addListener(ActionListener.toBiConsumer(listener));
}
@Override
public void addRemovedListener(ActionListener listener) {
removeContext.addListener(ActionListener.toBiConsumer(listener));
}
@Override
public boolean isClosed() {
return closeContext.isDone();
}
@Override
public void close() {
// This method is safe to call multiple times as the close context will provide concurrency
// protection and only be completed once. The attached listeners will only be notified once.
closeContext.complete(null);
}
@Override
public void onRemoved() {
removeContext.complete(null);
}
@Override
protected void closeInternal() {
close();
}
}