org.graylog2.restclient.models.Radio Maven / Gradle / Ivy
The newest version!
/**
* This file is part of Graylog.
*
* Graylog is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Graylog 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Graylog. If not, see getAllInputTypeInformation() throws IOException, APIException {
Map types = Maps.newHashMap();
for (String type : getInputTypes().keySet()) {
InputTypeSummaryResponse itr = getInputTypeInformation(type);
types.put(itr.type, itr);
}
return types;
}
public Map getInputTypes() throws IOException, APIException {
return api.path(routes.radio().InputTypesResource().types(), InputTypesResponse.class).radio(this).execute().types;
}
public InputTypeSummaryResponse getInputTypeInformation(String type) throws IOException, APIException {
return api.path(routes.radio().InputTypesResource().info(type), InputTypeSummaryResponse.class).radio(this).execute();
}
public List getInputs() {
List inputs = Lists.newArrayList();
for (InputStateSummaryResponse input : inputs().inputs) {
inputs.add(inputFactory.fromSummaryResponse(input.messageinput, this));
}
return inputs;
}
public Input getInput(String inputId) throws IOException, APIException {
final InputSummaryResponse inputSummaryResponse = api
.path(routes.radio().InputsResource().single(inputId), InputSummaryResponse.class).radio(this).execute();
return inputFactory.fromSummaryResponse(inputSummaryResponse, this);
}
public int numberOfInputs() {
return inputs().total;
}
private InputsResponse inputs() {
try {
return api.path(routes.radio().InputsResource().list(), InputsResponse.class).radio(this).execute();
} catch (Exception e) {
LOG.error("Could not get inputs.", e);
throw new RuntimeException("Could not get inputs.", e);
}
}
@Override
public InputLaunchResponse launchInput(String title, String type, Boolean global, Map configuration, boolean isExclusive, String nodeId) throws ExclusiveInputException {
if (isExclusive) {
for (Input input : getInputs()) {
if (input.getType().equals(type)) {
throw new ExclusiveInputException();
}
}
}
final InputLaunchRequest request = InputLaunchRequest.create(title, type, global, configuration, nodeId);
try {
return api.path(routes.radio().InputsResource().create(), InputLaunchResponse.class)
.radio(this)
.body(request)
.expect(Http.Status.ACCEPTED)
.execute();
} catch (Exception e) {
LOG.error("Could not launch input " + title, e);
return null;
}
}
public BufferInfo getBuffers() {
loadBufferInformation();
if (bufferInfo == null) {
return BufferInfo.buildEmpty();
} else {
return bufferInfo;
}
}
public String getThreadDump() throws IOException, APIException {
return api.path(routes.radio().SystemResource().threaddump(), String.class)
.radio(this)
.accept(MediaType.ANY_TEXT_TYPE)
.execute();
}
public int getThroughput() {
try {
return api.path(routes.radio().ThroughputResource().total(), NodeThroughputResponse.class).radio(this).execute().throughput;
} catch (Exception e) {
LOG.error("Could not load throughput for radio " + this, e);
}
return 0;
}
public Map getMetrics(String namespace) throws APIException, IOException {
MetricsListResponse response = api.path(routes.radio().MetricsResource().byNamespace(namespace), MetricsListResponse.class)
.radio(this)
.expect(200, 404)
.execute();
return response.getMetrics();
}
@Override
public String toString() {
final StringBuilder b = new StringBuilder();
if (id == null) {
b.append("UnresolvedNode {'").append(transportAddress).append("'}");
return b.toString();
}
b.append("Node {");
b.append("'").append(clusterId).append("'");
b.append("'").append(id).append("'");
b.append(", ").append(transportAddress);
b.append("}");
return b.toString();
}
@Override
public void stopInput(String inputId) throws IOException, APIException {
api.path(routes.radio().InputsResource().stop(inputId))
.radio(this)
.expect(Http.Status.ACCEPTED)
.execute();
}
@Override
public void startInput(String inputId) throws IOException, APIException {
api.path(routes.radio().InputsResource().launchExisting(inputId))
.radio(this)
.expect(Http.Status.ACCEPTED)
.execute();
}
@Override
public void restartInput(String inputId) throws IOException, APIException {
api.path(routes.InputsResource().restart(inputId))
.radio(this)
.expect(Http.Status.ACCEPTED)
.execute();
}
}