Commit 4d415a8a authored by peng-yongsheng's avatar peng-yongsheng
Browse files

Add the time bucket setting in segment DAO class.

parent 90bd26c8

apm-collector/apm-collector-storage/collector-storage-es-provider/src/main/java/org/skywalking/apm/collector/storage/es/dao/SegmentEsPersistenceDAO.java

+1 −0
Original line number Diff line number Diff line
@@ -53,6 +53,7 @@ public class SegmentEsPersistenceDAO extends EsDAO implements ISegmentPersistenc 
    @Override public IndexRequestBuilder prepareBatchInsert(Segment data) {

        Map<String, Object> source = new HashMap<>();

        source.put(SegmentTable.COLUMN_DATA_BINARY, new String(Base64.getEncoder().encode(data.getDataBinary())));

        source.put(SegmentTable.COLUMN_TIME_BUCKET, data.getTimeBucket());

        logger.debug("segment source: {}", source.toString());

        return getClient().prepareIndex(SegmentTable.TABLE, data.getId()).setSource(source);

    }

apm-collector/apm-collector-storage/collector-storage-h2-provider/src/main/java/org/skywalking/apm/collector/storage/h2/dao/SegmentH2PersistenceDAO.java

+1 −0
Original line number Diff line number Diff line
@@ -50,6 +50,7 @@ public class SegmentH2PersistenceDAO extends H2DAO implements ISegmentPersistenc 
        H2SqlEntity entity = new H2SqlEntity();

        source.put(SegmentTable.COLUMN_ID, data.getId());

        source.put(SegmentTable.COLUMN_DATA_BINARY, data.getDataBinary());

        source.put(SegmentTable.COLUMN_TIME_BUCKET, data.getTimeBucket());

        logger.debug("segment source: {}", source.toString());



        String sql = SqlBuilder.buildBatchInsertSql(SegmentTable.TABLE, source.keySet());