Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright 2010-2012 Ning, Inc.
*
* Ning licenses this file to you 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.ning.billing.meter.api.user;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Collection;
import java.util.Map;
import javax.annotation.Nullable;
import javax.inject.Inject;
import org.joda.time.DateTime;
import com.ning.billing.meter.api.DecimationMode;
import com.ning.billing.meter.api.TimeAggregationMode;
import com.ning.billing.meter.timeline.TimelineEventHandler;
import com.ning.billing.meter.timeline.persistent.TimelineDao;
import com.ning.billing.util.callcontext.CallContext;
import com.ning.billing.util.callcontext.TenantContext;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableMap.Builder;
public class DefaultMeterUserApi implements MeterUserApi {
private static final String AGGREGATE_METRIC_NAME = "__AGGREGATE__";
private final TimelineEventHandler timelineEventHandler;
private final TimelineDao timelineDao;
@Inject
public DefaultMeterUserApi(final TimelineEventHandler timelineEventHandler,
final TimelineDao timelineDao) {
this.timelineEventHandler = timelineEventHandler;
this.timelineDao = timelineDao;
}
@Override
public void getUsage(final OutputStream outputStream, final TimeAggregationMode timeAggregationMode,
final String source, final Collection categories,
final DateTime fromTimestamp, final DateTime toTimestamp, final TenantContext context) throws IOException {
final ImmutableMap.Builder> metricsPerCategory = new Builder>();
for (final String category : categories) {
metricsPerCategory.put(category, ImmutableList.of(AGGREGATE_METRIC_NAME));
}
getUsage(outputStream, timeAggregationMode, source, metricsPerCategory.build(), fromTimestamp, toTimestamp, context);
}
@Override
public void getUsage(final OutputStream outputStream, final TimeAggregationMode timeAggregationMode,
final String source, final Map> metricsPerCategory,
final DateTime fromTimestamp, final DateTime toTimestamp, final TenantContext context) throws IOException {
final JsonSamplesOutputer outputerJson = new AccumulatingJsonSamplesOutputer(timeAggregationMode, timelineEventHandler, timelineDao, context);
outputerJson.output(outputStream, ImmutableList.of(source), metricsPerCategory, fromTimestamp, toTimestamp);
}
@Override
public void getUsage(final OutputStream outputStream, final DecimationMode decimationMode, @Nullable final Integer outputCount,
final String source, final Map> metricsPerCategory,
final DateTime fromTimestamp, final DateTime toTimestamp, final TenantContext context) throws IOException {
final JsonSamplesOutputer outputerJson = new DecimatingJsonSamplesOutputer(decimationMode, outputCount, timelineEventHandler, timelineDao, context);
outputerJson.output(outputStream, ImmutableList.of(source), metricsPerCategory, fromTimestamp, toTimestamp);
}
@Override
public void getUsage(final OutputStream outputStream, final String source, final Collection categories,
final DateTime fromTimestamp, final DateTime toTimestamp, final TenantContext context) throws IOException {
final ImmutableMap.Builder> metricsPerCategory = new Builder>();
for (final String category : categories) {
metricsPerCategory.put(category, ImmutableList.of(AGGREGATE_METRIC_NAME));
}
getUsage(outputStream, source, metricsPerCategory.build(), fromTimestamp, toTimestamp, context);
}
@Override
public void getUsage(final OutputStream outputStream, final String source, final Map> metricsPerCategory,
final DateTime fromTimestamp, final DateTime toTimestamp, final TenantContext context) throws IOException {
final JsonSamplesOutputer outputerJson = new DefaultJsonSamplesOutputer(timelineEventHandler, timelineDao, context);
outputerJson.output(outputStream, ImmutableList.of(source), metricsPerCategory, fromTimestamp, toTimestamp);
}
@Override
public void incrementUsage(final String source, final String categoryName, final String metricName,
final DateTime timestamp, final CallContext context) {
recordUsage(source,
ImmutableMap.>of(categoryName, ImmutableMap.of(metricName, (short) 1)),
timestamp,
context);
}
@Override
public void incrementUsageAndAggregate(final String source, final String categoryName, final String metricName,
final DateTime timestamp, final CallContext context) {
recordUsage(source,
ImmutableMap.>of(categoryName, ImmutableMap.of(metricName, (short) 1, AGGREGATE_METRIC_NAME, (short) 1)),
timestamp,
context);
}
@Override
public void recordUsage(final String source, final Map> samplesForCategoriesAndMetrics,
final DateTime timestamp, final CallContext context) {
for (final String category : samplesForCategoriesAndMetrics.keySet()) {
timelineEventHandler.record(source, category, timestamp, samplesForCategoriesAndMetrics.get(category), context);
}
}
}
