Commit a84cb315 authored by pengys5's avatar pengys5
Browse files

Provide the service that use for get last heart beat time. 

#351
parent b671d634

apm-collector/apm-collector-ui/src/main/java/org/skywalking/apm/collector/ui/dao/IInstanceDAO.java

0 → 100644
+10 −0
Original line number Diff line number Diff line 
package org.skywalking.apm.collector.ui.dao;



/**

 * @author pengys5

 */

public interface IInstanceDAO {

    Long lastHeartBeatTime();



    Long instanceLastHeartBeatTime(long applicationInstanceId);

}

apm-collector/apm-collector-ui/src/main/java/org/skywalking/apm/collector/ui/dao/InstanceEsDAO.java

0 → 100644
+63 −0
Original line number Diff line number Diff line 
package org.skywalking.apm.collector.ui.dao;



import org.elasticsearch.action.search.SearchRequestBuilder;

import org.elasticsearch.action.search.SearchResponse;

import org.elasticsearch.action.search.SearchType;

import org.elasticsearch.index.query.AbstractQueryBuilder;

import org.elasticsearch.index.query.BoolQueryBuilder;

import org.elasticsearch.index.query.MatchQueryBuilder;

import org.elasticsearch.index.query.QueryBuilders;

import org.elasticsearch.index.query.RangeQueryBuilder;

import org.elasticsearch.search.SearchHit;

import org.elasticsearch.search.sort.SortBuilders;

import org.elasticsearch.search.sort.SortMode;

import org.skywalking.apm.collector.storage.elasticsearch.dao.EsDAO;

import org.skywalking.apm.collector.storage.table.node.NodeComponentTable;

import org.skywalking.apm.collector.storage.table.register.InstanceTable;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;



/**

 * @author pengys5

 */

public class InstanceEsDAO extends EsDAO implements IInstanceDAO {



    private final Logger logger = LoggerFactory.getLogger(InstanceEsDAO.class);



    @Override public Long lastHeartBeatTime() {

        long fiveMinuteBefore = System.currentTimeMillis() - 5 * 60 * 1000;

        RangeQueryBuilder rangeQueryBuilder = QueryBuilders.rangeQuery(InstanceTable.COLUMN_HEARTBEAT_TIME).gt(fiveMinuteBefore);

        return heartBeatTime(rangeQueryBuilder);

    }



    @Override public Long instanceLastHeartBeatTime(long applicationInstanceId) {

        long fiveMinuteBefore = System.currentTimeMillis() - 5 * 60 * 1000;

        BoolQueryBuilder boolQueryBuilder = new BoolQueryBuilder();

        RangeQueryBuilder rangeQueryBuilder = QueryBuilders.rangeQuery(InstanceTable.COLUMN_HEARTBEAT_TIME).gt(fiveMinuteBefore);

        MatchQueryBuilder matchQueryBuilder = QueryBuilders.matchQuery(InstanceTable.COLUMN_INSTANCE_ID, applicationInstanceId);

        boolQueryBuilder.must(rangeQueryBuilder);

        boolQueryBuilder.must(matchQueryBuilder);

        return heartBeatTime(boolQueryBuilder);

    }



    private Long heartBeatTime(AbstractQueryBuilder queryBuilder) {

        SearchRequestBuilder searchRequestBuilder = getClient().prepareSearch(InstanceTable.TABLE);

        searchRequestBuilder.setTypes(NodeComponentTable.TABLE_TYPE);

        searchRequestBuilder.setSearchType(SearchType.DFS_QUERY_THEN_FETCH);

        searchRequestBuilder.setQuery(queryBuilder);

        searchRequestBuilder.setSize(1);

        searchRequestBuilder.setFetchSource(InstanceTable.COLUMN_HEARTBEAT_TIME, null);

        searchRequestBuilder.addSort(SortBuilders.fieldSort(InstanceTable.COLUMN_HEARTBEAT_TIME).sortMode(SortMode.MAX));



        SearchResponse searchResponse = searchRequestBuilder.execute().actionGet();

        SearchHit[] searchHits = searchResponse.getHits().getHits();



        Long heartBeatTime = 0L;

        for (SearchHit searchHit : searchHits) {

            heartBeatTime = (Long)searchHit.getSource().get(InstanceTable.COLUMN_HEARTBEAT_TIME);

            logger.debug("heartBeatTime: {}", heartBeatTime);

            heartBeatTime = heartBeatTime - 5;

        }

        return heartBeatTime;

    }

}

apm-collector/apm-collector-ui/src/main/java/org/skywalking/apm/collector/ui/dao/InstanceH2DAO.java

0 → 100644
+16 −0
Original line number Diff line number Diff line 
package org.skywalking.apm.collector.ui.dao;



import org.skywalking.apm.collector.storage.h2.dao.H2DAO;



/**

 * @author pengys5

 */

public class InstanceH2DAO extends H2DAO implements IInstanceDAO {

    @Override public Long lastHeartBeatTime() {

        return null;

    }



    @Override public Long instanceLastHeartBeatTime(long applicationInstanceId) {

        return null;

    }

}

apm-collector/apm-collector-ui/src/main/java/org/skywalking/apm/collector/ui/jetty/handler/time/AllInstanceLastTimeGetHandler.java

0 → 100644
+36 −0
Original line number Diff line number Diff line 
package org.skywalking.apm.collector.ui.jetty.handler.time;



import com.google.gson.JsonElement;

import com.google.gson.JsonObject;

import javax.servlet.http.HttpServletRequest;

import org.skywalking.apm.collector.server.jetty.ArgumentsParseException;

import org.skywalking.apm.collector.server.jetty.JettyHandler;

import org.skywalking.apm.collector.ui.service.TimeSynchronousService;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;



/**

 * @author pengys5

 */

public class AllInstanceLastTimeGetHandler extends JettyHandler {



    private final Logger logger = LoggerFactory.getLogger(AllInstanceLastTimeGetHandler.class);



    @Override public String pathSpec() {

        return "/time/allInstance";

    }



    private TimeSynchronousService service = new TimeSynchronousService();



    @Override protected JsonElement doGet(HttpServletRequest req) throws ArgumentsParseException {

        Long time = service.allInstanceLastTime();

        logger.debug("all instance last time: {}", time);

        JsonObject timeJson = new JsonObject();

        timeJson.addProperty("time", time);

        return timeJson;

    }



    @Override protected JsonElement doPost(HttpServletRequest req) throws ArgumentsParseException {

        throw new UnsupportedOperationException();

    }

}

apm-collector/apm-collector-ui/src/main/java/org/skywalking/apm/collector/ui/jetty/handler/time/InstanceLastTimeGetHandler.java

0 → 100644
+46 −0
Original line number Diff line number Diff line 
package org.skywalking.apm.collector.ui.jetty.handler.time;



import com.google.gson.JsonElement;

import com.google.gson.JsonObject;

import javax.servlet.http.HttpServletRequest;

import org.skywalking.apm.collector.server.jetty.ArgumentsParseException;

import org.skywalking.apm.collector.server.jetty.JettyHandler;

import org.skywalking.apm.collector.ui.service.TimeSynchronousService;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;



/**

 * @author pengys5

 */

public class InstanceLastTimeGetHandler extends JettyHandler {



    private final Logger logger = LoggerFactory.getLogger(InstanceLastTimeGetHandler.class);



    @Override public String pathSpec() {

        return "/time/instanceId";

    }



    private TimeSynchronousService service = new TimeSynchronousService();



    @Override protected JsonElement doGet(HttpServletRequest req) throws ArgumentsParseException {

        String applicationInstanceIdStr = req.getParameter("applicationInstanceId");

        logger.debug("applicationInstanceId: {}", applicationInstanceIdStr);



        int applicationInstanceId;

        try {

            applicationInstanceId = Integer.parseInt(applicationInstanceIdStr);

        } catch (NumberFormatException e) {

            throw new ArgumentsParseException("application instance id must be integer");

        }



        Long time = service.instanceLastTime(applicationInstanceId);

        logger.debug("application instance id: {}, instance last time: {}", applicationInstanceId, time);

        JsonObject timeJson = new JsonObject();

        timeJson.addProperty("time", time);

        return timeJson;

    }



    @Override protected JsonElement doPost(HttpServletRequest req) throws ArgumentsParseException {

        throw new UnsupportedOperationException();

    }

}