Examples with ContainerHeartbeatResponse org.apache.samza.container.ContainerHeartbeatResponse used on opensource projects

Example 1 with ContainerHeartbeatResponse

use of org.apache.samza.container.ContainerHeartbeatResponse in project samza by apache.

the class YarnContainerHeartbeatServlet method doGet.

@Override
protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
    ContainerId yarnContainerId;
    PrintWriter printWriter = resp.getWriter();
    String containerIdParam = req.getParameter(CoordinationConstants.YARN_EXECUTION_ENVIRONMENT_CONTAINER_ID);
    ContainerHeartbeatResponse response;
    resp.setContentType(APPLICATION_JSON);
    boolean alive = false;
    try {
        yarnContainerId = ContainerId.fromString(containerIdParam);
        for (YarnContainer yarnContainer : yarnAppState.runningProcessors.values()) {
            if (yarnContainer.id().compareTo(yarnContainerId) == 0) {
                alive = true;
                break;
            }
        }
        if (!alive) {
            heartbeatsExpiredCount.inc();
        }
        response = new ContainerHeartbeatResponse(alive);
        printWriter.write(mapper.writeValueAsString(response));
    } catch (IllegalArgumentException e) {
        LOG.error("Container ID {} passed is invalid", containerIdParam);
        resp.sendError(HttpServletResponse.SC_BAD_REQUEST, e.getMessage());
    }
}