/**
* Copyright 2013 Lennart Koopmann <lennart@torch.sh>
*
* This file is part of Graylog2.
*
* Graylog2 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.
*
* Graylog2 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 Graylog2. If not, see <http://www.gnu.org/licenses/>.
*
*/
package controllers.api;
import com.google.common.collect.Maps;
import controllers.AuthenticatedController;
import org.graylog2.restclient.lib.APIException;
import org.graylog2.restclient.lib.ApiClient;
import org.graylog2.restclient.lib.Tools;
import org.graylog2.restclient.models.ClusterService;
import org.graylog2.restclient.models.Input;
import org.graylog2.restclient.models.Node;
import org.graylog2.restclient.models.NodeService;
import org.graylog2.restclient.models.api.results.MessageResult;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import play.libs.Json;
import play.mvc.Result;
import javax.inject.Inject;
import java.io.IOException;
import java.util.Map;
public class InputsApiController extends AuthenticatedController {
private static final Logger log = LoggerFactory.getLogger(InputsApiController.class);
private final NodeService nodeService;
private final ClusterService clusterService;
@Inject
public InputsApiController(NodeService nodeService, ClusterService clusterService) {
this.nodeService = nodeService;
this.clusterService = clusterService;
}
public Result io(String nodeId, String inputId) {
try {
Map<String, Object> result = Maps.newHashMap();
final Node node = nodeService.loadNode(nodeId);
final Input input = node.getInput(inputId);
final Input.IoStats ioStats = input.getIoStats();
result.put("total_rx", Tools.byteToHuman(ioStats.readBytesTotal));
result.put("total_tx", Tools.byteToHuman(ioStats.writtenBytesTotal));
result.put("rx", Tools.byteToHuman(ioStats.readBytes));
result.put("tx", Tools.byteToHuman(ioStats.writtenBytes));
return ok(Json.toJson(result));
} catch (IOException e) {
return internalServerError("io exception");
} catch (APIException e) {
return internalServerError("api exception " + e);
} catch (NodeService.NodeNotFoundException e) {
return status(404, views.html.errors.error.render(ApiClient.ERROR_MSG_NODE_NOT_FOUND, e, request()));
}
}
public Result globaIO(String inputId) {
Map<String, Object> result = Maps.newHashMap();
final Input globalInput;
try {
globalInput = nodeService.loadMasterNode().getInput(inputId);
if (!globalInput.getGlobal()) {
return badRequest();
}
} catch (IOException e) {
log.error("Could not load input.");
return internalServerError();
} catch (APIException e) {
log.error("Could not load input.");
return internalServerError();
}
final Input.IoStats ioStats = clusterService.getGlobalInputIo(globalInput);
result.put("total_rx", Tools.byteToHuman(ioStats.readBytesTotal));
result.put("total_tx", Tools.byteToHuman(ioStats.writtenBytesTotal));
result.put("rx", Tools.byteToHuman(ioStats.readBytes));
result.put("tx", Tools.byteToHuman(ioStats.writtenBytes));
return ok(Json.toJson(result));
}
public Result connections(String nodeId, String inputId) {
try {
Map<String, Object> result = Maps.newHashMap();
final Node node = nodeService.loadNode(nodeId);
final Input input = node.getInput(inputId);
result.put("active", input.getConnections());
result.put("total", input.getTotalConnections());
return ok(Json.toJson(result));
} catch (IOException e) {
return internalServerError("io exception");
} catch (APIException e) {
return internalServerError("api exception " + e);
} catch (NodeService.NodeNotFoundException e) {
return status(404, views.html.errors.error.render(ApiClient.ERROR_MSG_NODE_NOT_FOUND, e, request()));
}
}
public Result recentMessage(String nodeId, String inputId) {
return recentMessage(nodeId, inputId, true);
}
public Result recentMessage(String nodeId, String inputId, Boolean filtered) {
try {
Node node = nodeService.loadNode(nodeId);
MessageResult recentlyReceivedMessage = node.getInput(inputId).getRecentlyReceivedMessage(nodeId);
if (recentlyReceivedMessage == null) {
return notFound();
}
Map<String, Object> result = Maps.newHashMap();
result.put("id", recentlyReceivedMessage.getId());
result.put("index", recentlyReceivedMessage.getIndex());
if (filtered)
result.put("fields", recentlyReceivedMessage.getFilteredFields());
else
result.put("fields", recentlyReceivedMessage.getFields());
return ok(Json.toJson(result));
} catch (IOException e) {
return status(500);
} catch (APIException e) {
return status(e.getHttpCode());
} catch (NodeService.NodeNotFoundException e) {
return status(404, views.html.errors.error.render(ApiClient.ERROR_MSG_NODE_NOT_FOUND, e, request()));
}
}
}