Java tutorial
/************************************************************************* * Copyright 2009-2014 Eucalyptus Systems, Inc. * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; version 3 of the License. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see http://www.gnu.org/licenses/. * * Please contact Eucalyptus Systems, Inc., 6755 Hollister Ave., Goleta * CA 93117, USA or visit http://www.eucalyptus.com/licenses/ if you need * additional information or have any questions. * * This file may incorporate work covered under the following copyright * and permission notice: * * Copyright 2010-2014 Amazon.com, Inc. or its affiliates. All Rights * Reserved. * * Licensed under the Apache License, Version 2.0 (the "License"). * You may not use this file except in compliance with the License. * A copy of the License is located at * * http://aws.amazon.com/apache2.0 * * or in the "license" file accompanying this file. This file 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.eucalyptus.cloudwatch.workflow.alarms; import java.util.Iterator; import java.util.List; import java.util.Set; import java.util.concurrent.CountDownLatch; import java.util.concurrent.ExecutionException; import java.util.concurrent.ExecutorCompletionService; import java.util.concurrent.ExecutorService; import java.util.concurrent.Future; import com.eucalyptus.cloudwatch.common.config.CloudWatchConfigProperties; import com.eucalyptus.cloudwatch.common.internal.domain.alarms.AlarmEntity; import com.eucalyptus.cloudwatch.common.internal.domain.metricdata.MetricEntityFactory; import com.eucalyptus.cloudwatch.common.internal.domain.metricdata.MetricManager; import com.eucalyptus.entities.TransactionResource; import com.google.common.collect.Iterables; import com.google.common.collect.LinkedListMultimap; import com.google.common.collect.Lists; import com.google.common.collect.Multimap; import com.google.common.collect.Sets; import org.apache.log4j.Logger; import org.hibernate.Criteria; import com.eucalyptus.bootstrap.Bootstrap; import com.eucalyptus.cloudwatch.common.CloudWatchBackend; import com.eucalyptus.component.Topology; import com.eucalyptus.entities.Entities; public class AlarmStateEvaluationDispatcher implements Runnable { private static final Logger LOG = Logger.getLogger(AlarmStateEvaluationDispatcher.class); ExecutorService executorService; public AlarmStateEvaluationDispatcher(ExecutorService executorService) { super(); this.executorService = executorService; } @Override public void run() { if (!CloudWatchConfigProperties.isDisabledCloudWatchService() && Bootstrap.isOperational() && Topology.isEnabledLocally(CloudWatchBackend.class)) { LOG.debug("Kicking off AlarmStateEvaluationDispatcher"); int size = 0; long before = 0; List<List<AlarmEntity>> resultsList = null; try (final TransactionResource db = Entities.transactionFor(AlarmEntity.class)) { Criteria criteria = Entities.createCriteria(AlarmEntity.class); List<AlarmEntity> results = (List<AlarmEntity>) criteria.list(); resultsList = makeResultsList(results); before = System.currentTimeMillis(); size = results.size(); } try { if (resultsList != null) { ExecutorCompletionService executorCompletionService = new ExecutorCompletionService( executorService); Set<Future> futures = Sets.newHashSet(); for (List<AlarmEntity> alarmEntityList : resultsList) { futures.add(executorCompletionService .submit(new AlarmStateEvaluationWorker(alarmEntityList), new Object())); } Future completedFuture; while (futures.size() > 0) { completedFuture = executorCompletionService.take(); futures.remove(completedFuture); try { completedFuture.get(); } catch (ExecutionException e) { Throwable cause = e.getCause(); LOG.error(cause); } } long after = System.currentTimeMillis(); LOG.debug("Done evaluating " + size + " alarms, time = " + (after - before) + " ms"); } } catch (InterruptedException e) { LOG.debug(e); } } } private List<List<AlarmEntity>> makeResultsList(List<AlarmEntity> results) { Multimap<Class, AlarmEntity> classMultiMap = LinkedListMultimap.create(); for (AlarmEntity alarmEntity : results) { classMultiMap.put(MetricEntityFactory.getClassForEntitiesGet(alarmEntity.getMetricType(), MetricManager.hash(alarmEntity.getDimensionMap())), alarmEntity); } List<Iterator<List<AlarmEntity>>> iterators = Lists.newArrayList(); for (Class clazz : classMultiMap.keySet()) { iterators.add(Iterables.partition(classMultiMap.get(clazz), 100).iterator()); } List<List<AlarmEntity>> retVal = Lists.newArrayList(); boolean atLeastOneMightHaveMore = true; while (atLeastOneMightHaveMore) { atLeastOneMightHaveMore = false; for (Iterator<List<AlarmEntity>> iterator : iterators) { if (iterator.hasNext()) { atLeastOneMightHaveMore = true; retVal.add(iterator.next()); } } } return retVal; } }