Java tutorial
/* * Copyright 2015 NAVER Corp. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.jxt.web.dao.hbase; import java.util.ArrayList; import java.util.Collection; import java.util.List; import com.jxt.web.dao.AgentLifeCycleDao; import com.jxt.web.vo.AgentInfo; import com.jxt.web.vo.AgentStatus; import com.navercorp.pinpoint.common.server.util.AgentLifeCycleState; import org.apache.commons.collections.CollectionUtils; import org.apache.hadoop.hbase.client.Result; import org.apache.hadoop.hbase.client.ResultScanner; import org.apache.hadoop.hbase.client.Scan; import org.apache.hadoop.hbase.util.Bytes; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Qualifier; import org.springframework.stereotype.Repository; import com.navercorp.pinpoint.common.server.bo.AgentLifeCycleBo; import com.navercorp.pinpoint.common.hbase.HBaseTables; import com.navercorp.pinpoint.common.hbase.HbaseOperations2; import com.navercorp.pinpoint.common.hbase.ResultsExtractor; import com.navercorp.pinpoint.common.hbase.RowMapper; import com.navercorp.pinpoint.common.util.RowKeyUtils; import com.navercorp.pinpoint.common.util.TimeUtils; /** * @author HyunGil Jeong */ @Repository public class HbaseAgentLifeCycleDao implements AgentLifeCycleDao { private static final int SCANNER_CACHING = 20; @Autowired private HbaseOperations2 hbaseOperations2; @Autowired @Qualifier("agentLifeCycleMapper") private RowMapper<AgentLifeCycleBo> agentLifeCycleMapper; @Override public AgentStatus getAgentStatus(String agentId, long timestamp) { if (agentId == null) { throw new NullPointerException("agentId must not be null"); } if (timestamp < 0) { throw new IllegalArgumentException("timestamp must not be less than 0"); } long startRowTimestamp = TimeUtils.reverseTimeMillis(timestamp); long endRowTimestamp = Long.MAX_VALUE; Scan scan = createScan(agentId, startRowTimestamp, endRowTimestamp); AgentLifeCycleBo agentLifeCycleBo = this.hbaseOperations2.find(HBaseTables.AGENT_LIFECYCLE, scan, new AgentLifeCycleResultsExtractor(this.agentLifeCycleMapper, timestamp)); return createAgentStatus(agentId, agentLifeCycleBo); } @Override public void populateAgentStatus(AgentInfo agentInfo, long timestamp) { if (agentInfo == null) { return; } if (timestamp < 0) { throw new IllegalArgumentException("timestamp must not be less than 0"); } final String agentId = agentInfo.getAgentId(); final long startTimestamp = TimeUtils.reverseTimeMillis(agentInfo.getStartTimestamp()); Scan scan = createScan(agentId, startTimestamp, startTimestamp + 1); AgentLifeCycleBo agentLifeCycleBo = this.hbaseOperations2.find(HBaseTables.AGENT_LIFECYCLE, scan, new AgentLifeCycleResultsExtractor(this.agentLifeCycleMapper, timestamp)); AgentStatus agentStatus = createAgentStatus(agentId, agentLifeCycleBo); agentInfo.setStatus(agentStatus); } @Override public void populateAgentStatuses(Collection<AgentInfo> agentInfos, long timestamp) { if (CollectionUtils.isEmpty(agentInfos)) { return; } if (timestamp < 0) { throw new IllegalArgumentException("timestamp must not be less than 0"); } List<Scan> scans = new ArrayList<>(agentInfos.size()); for (AgentInfo agentInfo : agentInfos) { if (agentInfo != null) { final String agentId = agentInfo.getAgentId(); final long startTimestamp = TimeUtils.reverseTimeMillis(agentInfo.getStartTimestamp()); scans.add(createScan(agentId, startTimestamp, startTimestamp + 1)); } } List<AgentLifeCycleBo> agentLifeCycles = this.hbaseOperations2.findParallel(HBaseTables.AGENT_LIFECYCLE, scans, new AgentLifeCycleResultsExtractor(this.agentLifeCycleMapper, timestamp)); int idx = 0; for (AgentInfo agentInfo : agentInfos) { if (agentInfo != null) { AgentStatus agentStatus = createAgentStatus(agentInfo.getAgentId(), agentLifeCycles.get(idx++)); agentInfo.setStatus(agentStatus); } } } private Scan createScan(String agentId, long startTimestamp, long endTimestamp) { byte[] agentIdBytes = Bytes.toBytes(agentId); byte[] startKeyBytes = RowKeyUtils.concatFixedByteAndLong(agentIdBytes, HBaseTables.AGENT_NAME_MAX_LEN, startTimestamp); byte[] endKeyBytes = RowKeyUtils.concatFixedByteAndLong(agentIdBytes, HBaseTables.AGENT_NAME_MAX_LEN, endTimestamp); Scan scan = new Scan(startKeyBytes, endKeyBytes); scan.addColumn(HBaseTables.AGENT_LIFECYCLE_CF_STATUS, HBaseTables.AGENT_LIFECYCLE_CF_STATUS_QUALI_STATES); scan.setMaxVersions(1); scan.setCaching(SCANNER_CACHING); return scan; } private AgentStatus createAgentStatus(String agentId, AgentLifeCycleBo agentLifeCycle) { if (agentLifeCycle == null) { AgentStatus agentStatus = new AgentStatus(agentId); agentStatus.setState(AgentLifeCycleState.UNKNOWN); return agentStatus; } else { return new AgentStatus(agentLifeCycle); } } static class AgentLifeCycleResultsExtractor implements ResultsExtractor<AgentLifeCycleBo> { private final RowMapper<AgentLifeCycleBo> agentLifeCycleMapper; private final long queryTimestamp; AgentLifeCycleResultsExtractor(RowMapper<AgentLifeCycleBo> agentLifeCycleMapper, long queryTimestamp) { this.agentLifeCycleMapper = agentLifeCycleMapper; this.queryTimestamp = queryTimestamp; } @Override public AgentLifeCycleBo extractData(ResultScanner results) throws Exception { int found = 0; for (Result result : results) { AgentLifeCycleBo agentLifeCycle = this.agentLifeCycleMapper.mapRow(result, found++); if (agentLifeCycle.getEventTimestamp() < this.queryTimestamp) { return agentLifeCycle; } } return null; } } }