public static final Logger s_logger = Logger.getLogger(StatsCollector.class.getName());

private static StatsCollector s_instance = null;

private ScheduledExecutorService _executor = null;

@Inject private AgentManager _agentMgr;

@Inject private UserVmManager _userVmMgr;

@Inject private HostDao _hostDao;

@Inject private UserVmDao _userVmDao;

@Inject private VolumeDao _volsDao;

@Inject private PrimaryDataStoreDao _storagePoolDao;

@Inject private ImageStoreDao _imageStoreDao;

@Inject private StorageManager _storageManager;

@Inject private StoragePoolHostDao _storagePoolHostDao;

@Inject private DataStoreManager _dataStoreMgr;

@Inject private ResourceManager _resourceMgr;

@Inject private ConfigurationDao _configDao;

@Inject private EndPointSelector _epSelector;

@Inject private VmDiskStatisticsDao _vmDiskStatsDao;

@Inject private ManagementServerHostDao _msHostDao;

private ConcurrentHashMap<Long, HostStats> _hostStats = new ConcurrentHashMap<Long, HostStats>();

private final ConcurrentHashMap<Long, VmStats> _VmStats = new ConcurrentHashMap<Long, VmStats>();

private ConcurrentHashMap<Long, VolumeStats> _volumeStats = new ConcurrentHashMap<Long, VolumeStats>();

private ConcurrentHashMap<Long, StorageStats> _storageStats = new ConcurrentHashMap<Long, StorageStats>();

private ConcurrentHashMap<Long, StorageStats> _storagePoolStats = new ConcurrentHashMap<Long, StorageStats>();

long hostStatsInterval = -1L;

long hostAndVmStatsInterval = -1L;

long storageStatsInterval = -1L;

long volumeStatsInterval = -1L;

int vmDiskStatsInterval = 0;

private ScheduledExecutorService _diskStatsUpdateExecutor;

private int _usageAggregationRange = 1440;

private String _usageTimeZone = "GMT";

private final long mgmtSrvrId = MacAddress.getMacAddress().toLong();

private static final int ACQUIRE_GLOBAL_LOCK_TIMEOUT_FOR_COOPERATION = 5; // 5 seconds

private static final int USAGE_AGGREGATION_RANGE_MIN = 10; // 10 minutes, same to com.cloud.usage.UsageManagerImpl.USAGE_AGGREGATION_RANGE_MIN

private boolean _dailyOrHourly = false;

//private final GlobalLock m_capacityCheckLock = GlobalLock.getInternLock("capacity.check");

public static StatsCollector getInstance() {

return s_instance;

}

public static StatsCollector getInstance(Map<String, String> configs) {

s_instance.init(configs);

return s_instance;

}

public StatsCollector() {

s_instance = this;

}

@Override

public boolean start() {

init(_configDao.getConfiguration());

return true;

}

private void init(Map<String, String> configs) {

_executor = Executors.newScheduledThreadPool(3, new NamedThreadFactory("StatsCollector"));

hostStatsInterval = NumbersUtil.parseLong(configs.get("host.stats.interval"), 60000L);

hostAndVmStatsInterval = NumbersUtil.parseLong(configs.get("vm.stats.interval"), 60000L);

storageStatsInterval = NumbersUtil.parseLong(configs.get("storage.stats.interval"), 60000L);

volumeStatsInterval = NumbersUtil.parseLong(configs.get("volume.stats.interval"), -1L);

vmDiskStatsInterval = NumbersUtil.parseInt(configs.get("vm.disk.stats.interval"), 0);

if (hostStatsInterval > 0) {

_executor.scheduleWithFixedDelay(new HostCollector(), 15000L, hostStatsInterval, TimeUnit.MILLISECONDS);

}

if (hostAndVmStatsInterval > 0) {

_executor.scheduleWithFixedDelay(new VmStatsCollector(), 15000L, hostAndVmStatsInterval, TimeUnit.MILLISECONDS);

}

if (storageStatsInterval > 0) {

_executor.scheduleWithFixedDelay(new StorageCollector(), 15000L, storageStatsInterval, TimeUnit.MILLISECONDS);

}

if (vmDiskStatsInterval > 0) {

if (vmDiskStatsInterval < 300)

vmDiskStatsInterval = 300;

_executor.scheduleAtFixedRate(new VmDiskStatsTask(), vmDiskStatsInterval, vmDiskStatsInterval, TimeUnit.SECONDS);

}

//Schedule disk stats update task

_diskStatsUpdateExecutor = Executors.newScheduledThreadPool(1, new NamedThreadFactory("DiskStatsUpdater"));

String aggregationRange = configs.get("usage.stats.job.aggregation.range");

_usageAggregationRange = NumbersUtil.parseInt(aggregationRange, 1440);

_usageTimeZone = configs.get("usage.aggregation.timezone");

if(_usageTimeZone == null){

_usageTimeZone = "GMT";

}

TimeZone usageTimezone = TimeZone.getTimeZone(_usageTimeZone);

Calendar cal = Calendar.getInstance(usageTimezone);

cal.setTime(new Date());

long endDate = 0;

int HOURLY_TIME = 60;

final int DAILY_TIME = 60 * 24;

if (_usageAggregationRange == DAILY_TIME) {

cal.set(Calendar.HOUR_OF_DAY, 0);

cal.set(Calendar.MINUTE, 0);

cal.set(Calendar.SECOND, 0);

cal.set(Calendar.MILLISECOND, 0);

cal.roll(Calendar.DAY_OF_YEAR, true);

cal.add(Calendar.MILLISECOND, -1);

endDate = cal.getTime().getTime();

_dailyOrHourly = true;

} else if (_usageAggregationRange == HOURLY_TIME) {

cal.set(Calendar.MINUTE, 0);

cal.set(Calendar.SECOND, 0);

cal.set(Calendar.MILLISECOND, 0);

cal.roll(Calendar.HOUR_OF_DAY, true);

cal.add(Calendar.MILLISECOND, -1);

endDate = cal.getTime().getTime();

_dailyOrHourly = true;

} else {

endDate = cal.getTime().getTime();

_dailyOrHourly = false;

}

if (_usageAggregationRange < USAGE_AGGREGATION_RANGE_MIN) {

s_logger.warn("Usage stats job aggregation range is to small, using the minimum value of " + USAGE_AGGREGATION_RANGE_MIN);

_usageAggregationRange = USAGE_AGGREGATION_RANGE_MIN;

}

_diskStatsUpdateExecutor.scheduleAtFixedRate(new VmDiskStatsUpdaterTask(), (endDate - System.currentTimeMillis()),

(_usageAggregationRange * 60 * 1000), TimeUnit.MILLISECONDS);

}

class HostCollector implements Runnable {

@Override

public void run() {

try {

s_logger.debug("HostStatsCollector is running...");

SearchCriteria<HostVO> sc = _hostDao.createSearchCriteria();

sc.addAnd("status", SearchCriteria.Op.EQ, Status.Up.toString());

sc.addAnd("resourceState", SearchCriteria.Op.NIN, ResourceState.Maintenance, ResourceState.PrepareForMaintenance, ResourceState.ErrorInMaintenance);

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.Storage.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.ConsoleProxy.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.SecondaryStorage.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.LocalSecondaryStorage.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.TrafficMonitor.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.SecondaryStorageVM.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.ExternalFirewall.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.ExternalLoadBalancer.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.L2Networking.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.BaremetalDhcp.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.BaremetalPxe.toString());

ConcurrentHashMap<Long, HostStats> hostStats = new ConcurrentHashMap<Long, HostStats>();

List<HostVO> hosts = _hostDao.search(sc, null);

for (HostVO host : hosts)

{

HostStatsEntry stats = (HostStatsEntry) _resourceMgr.getHostStatistics(host.getId());

if (stats != null)

{

hostStats.put(host.getId(), stats);

}

else

{

s_logger.warn("Received invalid host stats for host: " + host.getId());

}

}

_hostStats = hostStats;

}

catch (Throwable t)

{

s_logger.error("Error trying to retrieve host stats", t);

}

}

}

class VmStatsCollector implements Runnable {

@Override

public void run() {

try {

s_logger.debug("VmStatsCollector is running...");

SearchCriteria<HostVO> sc = _hostDao.createSearchCriteria();

sc.addAnd("status", SearchCriteria.Op.EQ, Status.Up.toString());

sc.addAnd("resourceState", SearchCriteria.Op.NIN, ResourceState.Maintenance, ResourceState.PrepareForMaintenance, ResourceState.ErrorInMaintenance);

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.Storage.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.ConsoleProxy.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.SecondaryStorage.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.LocalSecondaryStorage.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.TrafficMonitor.toString());

sc.addAnd("type", SearchCriteria.Op.NEQ, Host.Type.SecondaryStorageVM.toString());

List<HostVO> hosts = _hostDao.search(sc, null);

for (HostVO host : hosts) {

List<UserVmVO> vms = _userVmDao.listRunningByHostId(host.getId());

List<Long> vmIds = new ArrayList<Long>();

for (UserVmVO vm : vms) {

vmIds.add(vm.getId());

}

try

{

HashMap<Long, VmStatsEntry> vmStatsById = _userVmMgr.getVirtualMachineStatistics(host.getId(), host.getName(), vmIds);

if(vmStatsById != null)

{

VmStatsEntry statsInMemory = null;

Set<Long> vmIdSet = vmStatsById.keySet();

for(Long vmId : vmIdSet)

{

VmStatsEntry statsForCurrentIteration = vmStatsById.get(vmId);

statsInMemory = (VmStatsEntry) _VmStats.get(vmId);

if(statsInMemory == null)

{

//no stats exist for this vm, directly persist

_VmStats.put(vmId, statsForCurrentIteration);

}

else

{

//update each field

statsInMemory.setCPUUtilization(statsForCurrentIteration.getCPUUtilization());

statsInMemory.setNumCPUs(statsForCurrentIteration.getNumCPUs());

statsInMemory.setNetworkReadKBs(statsInMemory.getNetworkReadKBs() + statsForCurrentIteration.getNetworkReadKBs());

statsInMemory.setNetworkWriteKBs(statsInMemory.getNetworkWriteKBs() + statsForCurrentIteration.getNetworkWriteKBs());

statsInMemory.setDiskWriteKBs(statsInMemory.getDiskWriteKBs() + statsForCurrentIteration.getDiskWriteKBs());

statsInMemory.setDiskReadIOs(statsInMemory.getDiskReadIOs() + statsForCurrentIteration.getDiskReadIOs());

statsInMemory.setDiskWriteIOs(statsInMemory.getDiskWriteIOs() + statsForCurrentIteration.getDiskWriteIOs());

statsInMemory.setDiskReadKBs(statsInMemory.getDiskReadKBs() + statsForCurrentIteration.getDiskReadKBs());

_VmStats.put(vmId, statsInMemory);

}

}

}

} catch (Exception e) {

s_logger.debug("Failed to get VM stats for host with ID: " + host.getId());

continue;

}

}

} catch (Throwable t) {

s_logger.error("Error trying to retrieve VM stats", t);

}

}

}

public VmStats getVmStats(long id) {

return _VmStats.get(id);

}

class VmDiskStatsUpdaterTask implements Runnable {

@Override

public void run() {

GlobalLock scanLock = GlobalLock.getInternLock("vm.disk.stats");

try {

if(scanLock.lock(ACQUIRE_GLOBAL_LOCK_TIMEOUT_FOR_COOPERATION)) {

//Check for ownership

//msHost in UP state with min id should run the job

ManagementServerHostVO msHost = _msHostDao.findOneInUpState(new Filter(ManagementServerHostVO.class, "id", true, 0L, 1L));

if(msHost == null || (msHost.getMsid() != mgmtSrvrId)){

s_logger.debug("Skipping aggregate disk stats update");

scanLock.unlock();

return;

}

Transaction txn = Transaction.open(Transaction.CLOUD_DB);

try {

txn.start();

//get all stats with delta > 0

List<VmDiskStatisticsVO> updatedVmNetStats = _vmDiskStatsDao.listUpdatedStats();

for(VmDiskStatisticsVO stat : updatedVmNetStats){

if (_dailyOrHourly) {

//update agg bytes

stat.setAggBytesRead(stat.getCurrentBytesRead() + stat.getNetBytesRead());

stat.setAggBytesWrite(stat.getCurrentBytesWrite() + stat.getNetBytesWrite());

stat.setAggIORead(stat.getCurrentIORead() + stat.getNetIORead());

stat.setAggIOWrite(stat.getCurrentIOWrite() + stat.getNetIOWrite());

_vmDiskStatsDao.update(stat.getId(), stat);

}

}

s_logger.debug("Successfully updated aggregate vm disk stats");

txn.commit();

} catch (Exception e){

txn.rollback();

s_logger.debug("Failed to update aggregate disk stats", e);

} finally {

scanLock.unlock();

txn.close();

}

}

} catch (Exception e){

s_logger.debug("Exception while trying to acquire disk stats lock", e);

} finally {

scanLock.releaseRef();

}

}

}

class VmDiskStatsTask implements Runnable {

@Override

public void run() {

// collect the vm disk statistics(total) from hypervisor. added by weizhou, 2013.03.

Transaction txn = Transaction.open(Transaction.CLOUD_DB);

try {

txn.start();

SearchCriteria<HostVO> sc = _hostDao.createSearchCriteria();

sc.addAnd("status", SearchCriteria.Op.EQ, Status.Up.toString());

sc.addAnd("resourceState", SearchCriteria.Op.NIN, ResourceState.Maintenance, ResourceState.PrepareForMaintenance, ResourceState.ErrorInMaintenance);

sc.addAnd("type", SearchCriteria.Op.EQ, Host.Type.Routing.toString());

sc.addAnd("hypervisorType", SearchCriteria.Op.EQ, HypervisorType.KVM); // support KVM only util 2013.06.25

List<HostVO> hosts = _hostDao.search(sc, null);

for (HostVO host : hosts) {

List<UserVmVO> vms = _userVmDao.listRunningByHostId(host.getId());

List<Long> vmIds = new ArrayList<Long>();

for (UserVmVO vm : vms) {

if (vm.getType() == VirtualMachine.Type.User) // user vm

vmIds.add(vm.getId());

}

HashMap<Long, List<VmDiskStatsEntry>> vmDiskStatsById = _userVmMgr.getVmDiskStatistics(host.getId(), host.getName(), vmIds);

if (vmDiskStatsById == null)

continue;

Set<Long> vmIdSet = vmDiskStatsById.keySet();

for(Long vmId : vmIdSet)

{

List<VmDiskStatsEntry> vmDiskStats = vmDiskStatsById.get(vmId);

if (vmDiskStats == null)

continue;

UserVmVO userVm = _userVmDao.findById(vmId);

for (VmDiskStatsEntry vmDiskStat:vmDiskStats) {

SearchCriteria<VolumeVO> sc_volume = _volsDao.createSearchCriteria();

sc_volume.addAnd("path", SearchCriteria.Op.EQ, vmDiskStat.getPath());

VolumeVO volume = _volsDao.search(sc_volume, null).get(0);

VmDiskStatisticsVO previousVmDiskStats = _vmDiskStatsDao.findBy(userVm.getAccountId(), userVm.getDataCenterId(), vmId, volume.getId());

VmDiskStatisticsVO vmDiskStat_lock = _vmDiskStatsDao.lock(userVm.getAccountId(), userVm.getDataCenterId(), vmId, volume.getId());

if ((vmDiskStat.getBytesRead() == 0) && (vmDiskStat.getBytesWrite() == 0)

&& (vmDiskStat.getIORead() == 0) && (vmDiskStat.getIOWrite() == 0)) {

s_logger.debug("IO/bytes read and write are all 0. Not updating vm_disk_statistics");

continue;

}

if (vmDiskStat_lock == null) {

s_logger.warn("unable to find vm disk stats from host for account: " + userVm.getAccountId() + " with vmId: " + userVm.getId()+ " and volumeId:" + volume.getId());

continue;

}

if (previousVmDiskStats != null

&& ((previousVmDiskStats.getCurrentBytesRead() != vmDiskStat_lock.getCurrentBytesRead())

|| (previousVmDiskStats.getCurrentBytesWrite() != vmDiskStat_lock.getCurrentBytesWrite())

|| (previousVmDiskStats.getCurrentIORead() != vmDiskStat_lock.getCurrentIORead())

|| (previousVmDiskStats.getCurrentIOWrite() != vmDiskStat_lock.getCurrentIOWrite()))) {

s_logger.debug("vm disk stats changed from the time GetVmDiskStatsCommand was sent. " +

"Ignoring current answer. Host: " + host.getName() + " . VM: " + vmDiskStat.getVmName() +

" Read(Bytes): " + vmDiskStat.getBytesRead() + " write(Bytes): " + vmDiskStat.getBytesWrite() +

" Read(IO): " + vmDiskStat.getIORead() + " write(IO): " + vmDiskStat.getIOWrite());

continue;

}

if (vmDiskStat_lock.getCurrentBytesRead() > vmDiskStat.getBytesRead()) {

if (s_logger.isDebugEnabled()) {

s_logger.debug("Read # of bytes that's less than the last one. " +

"Assuming something went wrong and persisting it. Host: " + host.getName() + " . VM: " + vmDiskStat.getVmName() +

" Reported: " + vmDiskStat.getBytesRead() + " Stored: " + vmDiskStat_lock.getCurrentBytesRead());

}

vmDiskStat_lock.setNetBytesRead(vmDiskStat_lock.getNetBytesRead() + vmDiskStat_lock.getCurrentBytesRead());

}

vmDiskStat_lock.setCurrentBytesRead(vmDiskStat.getBytesRead());

if (vmDiskStat_lock.getCurrentBytesWrite() > vmDiskStat.getBytesWrite()) {

if (s_logger.isDebugEnabled()) {

s_logger.debug("Write # of bytes that's less than the last one. " +

"Assuming something went wrong and persisting it. Host: " + host.getName() + " . VM: " + vmDiskStat.getVmName() +

" Reported: " + vmDiskStat.getBytesWrite() + " Stored: " + vmDiskStat_lock.getCurrentBytesWrite());

}

vmDiskStat_lock.setNetBytesWrite(vmDiskStat_lock.getNetBytesWrite() + vmDiskStat_lock.getCurrentBytesWrite());

}

vmDiskStat_lock.setCurrentBytesWrite(vmDiskStat.getBytesWrite());

if (vmDiskStat_lock.getCurrentIORead() > vmDiskStat.getIORead()) {

if (s_logger.isDebugEnabled()) {

s_logger.debug("Read # of IO that's less than the last one. " +

"Assuming something went wrong and persisting it. Host: " + host.getName() + " . VM: " + vmDiskStat.getVmName() +

" Reported: " + vmDiskStat.getIORead() + " Stored: " + vmDiskStat_lock.getCurrentIORead());

}

vmDiskStat_lock.setNetIORead(vmDiskStat_lock.getNetIORead() + vmDiskStat_lock.getCurrentIORead());

}

vmDiskStat_lock.setCurrentIORead(vmDiskStat.getIORead());

if (vmDiskStat_lock.getCurrentIOWrite() > vmDiskStat.getIOWrite()) {

if (s_logger.isDebugEnabled()) {

s_logger.debug("Write # of IO that's less than the last one. " +

"Assuming something went wrong and persisting it. Host: " + host.getName() + " . VM: " + vmDiskStat.getVmName() +

" Reported: " + vmDiskStat.getIOWrite() + " Stored: " + vmDiskStat_lock.getCurrentIOWrite());

}

vmDiskStat_lock.setNetIOWrite(vmDiskStat_lock.getNetIOWrite() + vmDiskStat_lock.getCurrentIOWrite());

}

vmDiskStat_lock.setCurrentIOWrite(vmDiskStat.getIOWrite());

if (! _dailyOrHourly) {

//update agg bytes

vmDiskStat_lock.setAggBytesWrite(vmDiskStat_lock.getNetBytesWrite() + vmDiskStat_lock.getCurrentBytesWrite());

vmDiskStat_lock.setAggBytesRead(vmDiskStat_lock.getNetBytesRead() + vmDiskStat_lock.getCurrentBytesRead());

vmDiskStat_lock.setAggIOWrite(vmDiskStat_lock.getNetIOWrite() + vmDiskStat_lock.getCurrentIOWrite());

vmDiskStat_lock.setAggIORead(vmDiskStat_lock.getNetIORead() + vmDiskStat_lock.getCurrentIORead());

}

_vmDiskStatsDao.update(vmDiskStat_lock.getId(), vmDiskStat_lock);

}

}

}

txn.commit();

} catch (Exception e) {

s_logger.warn("Error while collecting vm disk stats from hosts", e);

} finally {

txn.close();

}

}

}

class StorageCollector implements Runnable {

@Override

public void run() {

try {

if (s_logger.isDebugEnabled()) {

s_logger.debug("StorageCollector is running...");

}

List<DataStore> stores = _dataStoreMgr.listImageStores();

ConcurrentHashMap<Long, StorageStats> storageStats = new ConcurrentHashMap<Long, StorageStats>();

for (DataStore store : stores) {

if ( store.getUri() == null ) {

continue;

}

GetStorageStatsCommand command = new GetStorageStatsCommand(store.getUri());

EndPoint ssAhost = _epSelector.select(store);

if (ssAhost == null) {

s_logger.debug("There is no secondary storage VM for secondary storage host " + store.getName());

continue;

}

long storeId = store.getId();

Answer answer = ssAhost.sendMessage(command);

if (answer != null && answer.getResult()) {

storageStats.put(storeId, (StorageStats)answer);

s_logger.trace("HostId: "+storeId+ " Used: " + ((StorageStats)answer).getByteUsed() + " Total Available: " + ((StorageStats)answer).getCapacityBytes());

}

}

_storageStats = storageStats;

ConcurrentHashMap<Long, StorageStats> storagePoolStats = new ConcurrentHashMap<Long, StorageStats>();

List<StoragePoolVO> storagePools = _storagePoolDao.listAll();

for (StoragePoolVO pool: storagePools) {

// check if the pool has enabled hosts

List<Long> hostIds = _storageManager.getUpHostsInPool(pool.getId());

if (hostIds == null || hostIds.isEmpty()) continue;

GetStorageStatsCommand command = new GetStorageStatsCommand(pool.getUuid(), pool.getPoolType(), pool.getPath());

long poolId = pool.getId();

try {

Answer answer = _storageManager.sendToPool(pool, command);

if (answer != null && answer.getResult()) {

storagePoolStats.put(pool.getId(), (StorageStats)answer);

// Seems like we have dynamically updated the pool size since the prev. size and the current do not match

if (_storagePoolStats.get(poolId)!= null &&

_storagePoolStats.get(poolId).getCapacityBytes() != ((StorageStats)answer).getCapacityBytes()){

pool.setCapacityBytes(((StorageStats)answer).getCapacityBytes());

_storagePoolDao.update(pool.getId(), pool);

}

}

} catch (StorageUnavailableException e) {

s_logger.info("Unable to reach " + pool, e);

} catch (Exception e) {

s_logger.warn("Unable to get stats for " + pool, e);

}

}

_storagePoolStats = storagePoolStats;

} catch (Throwable t) {

s_logger.error("Error trying to retrieve storage stats", t);

}

}

}

public StorageStats getStorageStats(long id) {

return _storageStats.get(id);

}

public HostStats getHostStats(long hostId){

return _hostStats.get(hostId);

}

public StorageStats getStoragePoolStats(long id) {

return _storagePoolStats.get(id);

}