com.fatwire.dta.sscrawler.reporting.reporters.PageletTimingsStatisticsReporter.java Source code

Introduction

Here is the source code for com.fatwire.dta.sscrawler.reporting.reporters.PageletTimingsStatisticsReporter.java

Source

/*
 * Copyright 2008 FatWire Corporation. 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.
 * 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.fatwire.dta.sscrawler.reporting.reporters;

import java.text.DecimalFormat;
import java.util.Arrays;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentSkipListMap;
import java.util.concurrent.atomic.AtomicInteger;

import org.apache.commons.math.stat.descriptive.SynchronizedSummaryStatistics;

import com.fatwire.dta.sscrawler.ResultPage;
import com.fatwire.dta.sscrawler.reporting.Report;
import com.fatwire.dta.sscrawler.util.CacheHelper;

public class PageletTimingsStatisticsReporter extends ReportDelegatingReporter {

    private final Map<String, SynchronizedSummaryStatistics> stats = new ConcurrentSkipListMap<String, SynchronizedSummaryStatistics>();

    private final SynchronizedSummaryStatistics total = new SynchronizedSummaryStatistics();

    private final AtomicInteger pagesDone = new AtomicInteger();

    private final Map<String, Boolean> cached = new ConcurrentHashMap<String, Boolean>();

    /**
     * @param report
     */
    public PageletTimingsStatisticsReporter(final Report report) {
        super(report);

    }

    public void addToReport(final ResultPage page) {
        pagesDone.incrementAndGet();
        total.addValue(page.getReadTime());
        final String pagename = page.getPageName();
        if (pagename != null) {
            SynchronizedSummaryStatistics ss = stats.get(pagename);
            if (ss == null) {
                ss = new SynchronizedSummaryStatistics();
                stats.put(pagename, ss);
                cached.put(pagename, CacheHelper.shouldCache(page.getResponseHeaders()));
            }
            ss.addValue(page.getReadTime());
        }

    }

    @Override
    public synchronized void endCollecting() {
        if (pagesDone.get() == 0)
            return;
        report.startReport();
        final DecimalFormat df = new DecimalFormat("###0.00");
        final DecimalFormat lf = new DecimalFormat("##0");
        int l = 0;
        for (final String s : stats.keySet()) {
            l = Math.max(s.length(), l);
        }
        l += 3;
        final char[] blank = new char[l];
        Arrays.fill(blank, ' ');
        report.addHeader("pagename" + new String(blank, 0, l - 8), "invocations", "average", "min", "max",
                "standard-deviation");

        for (final Map.Entry<String, SynchronizedSummaryStatistics> e : stats.entrySet()) {
            final SynchronizedSummaryStatistics s = e.getValue();

            final String n = e.getKey() + new String(blank, 0, l - e.getKey().length())
                    + (Boolean.FALSE.equals(cached.get(e.getKey())) ? " * " : "   ");

            report.addRow(n, Long.toString(s.getN()), df.format(s.getMean()), lf.format(s.getMin()),
                    lf.format(s.getMax()), df.format(s.getStandardDeviation()));

        }
        final String n = "total" + new String(blank, 0, l - "total".length());
        report.addRow(n, Long.toString(total.getN()), df.format(total.getMean()), lf.format(total.getMin()),
                lf.format(total.getMax()), df.format(total.getStandardDeviation()));

        report.finishReport();

    }

    @Override
    protected String[] getHeader() {
        return new String[0];
    }

    @Override
    public synchronized void startCollecting() {
        stats.clear();
        pagesDone.set(0);
        total.clear();

    }

    public Verdict getVerdict() {
        return total.getMean() > 200 ? Verdict.RED : total.getMean() > 100 ? Verdict.AMBER : Verdict.GREEN;
    }

}