io.github.microcks.listener.DailyStatisticsFeeder.java Source code

Introduction

Here is the source code for io.github.microcks.listener.DailyStatisticsFeeder.java

Source

/*
 * Licensed to Laurent Broudoux (the "Author") under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. Author 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 io.github.microcks.listener;

import io.github.microcks.domain.DailyStatistic;
import io.github.microcks.event.MockInvocationEvent;
import io.github.microcks.repository.DailyStatisticRepository;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationListener;
import org.springframework.stereotype.Component;

import java.util.Calendar;
import java.util.HashMap;
import java.util.Map;

/**
 * Application event listener that updates daily statistics on incoming event.
 * @author laurent
 */
@Component
public class DailyStatisticsFeeder implements ApplicationListener<MockInvocationEvent> {

    /** A commons logger for diagnostic messages. */
    private static Log log = LogFactory.getLog(DailyStatisticsFeeder.class);

    @Autowired
    private DailyStatisticRepository statisticsRepository;

    @Override
    public void onApplicationEvent(MockInvocationEvent event) {
        log.debug(
                "Received a MockInvocationEvent on " + event.getServiceName() + " - v" + event.getServiceVersion());

        // Compute day string representation.
        Calendar calendar = Calendar.getInstance();
        calendar.setTime(event.getInvocationTimestamp());

        // Computing keys based on invocation date.
        int month = calendar.get(Calendar.MONTH) + 1;
        String monthStr = (month < 10 ? "0" : "") + String.valueOf(month);
        int dayOfMonth = calendar.get(Calendar.DAY_OF_MONTH);
        String dayOfMonthStr = (dayOfMonth < 10 ? "0" : "") + String.valueOf(dayOfMonth);

        String day = String.valueOf(calendar.get(Calendar.YEAR)) + monthStr + dayOfMonthStr;
        String hourKey = String.valueOf(calendar.get(Calendar.HOUR_OF_DAY));
        String minuteKey = String
                .valueOf((60 * calendar.get(Calendar.HOUR_OF_DAY)) + calendar.get(Calendar.MINUTE));
        if (log.isDebugEnabled()) {
            log.debug("hourKey for statistic is " + hourKey);
            log.debug("minuteKey for statistic is " + minuteKey);
        }

        // First check if there's a statistic document for invocation day.
        DailyStatistic statistic = statisticsRepository.findByDayAndServiceNameAndServiceVersion(day,
                event.getServiceName(), event.getServiceVersion());

        if (statistic == null) {
            // No statistic's yet...
            log.debug("There's no statistics for " + day + " yet. Create one....");
            // Initialize a new 0 filled structure.
            statistic = new DailyStatistic();
            statistic.setDay(day);
            statistic.setServiceName(event.getServiceName());
            statistic.setServiceVersion(event.getServiceVersion());
            statistic.setHourlyCount(initializeHourlyMap());
            statistic.setMinuteCount(initializeMinuteMap());
            // Now set first values before saving.
            statistic.setDailyCount(1);
            statistic.getHourlyCount().put(hourKey, 1);
            statistic.getMinuteCount().put(minuteKey, 1);
            statisticsRepository.save(statistic);
        } else {
            // Already a statistic document for this day, increment fields.
            log.debug("Found an existing statistic document for " + day);
            statisticsRepository.incrementDailyStatistic(day, event.getServiceName(), event.getServiceVersion(),
                    hourKey, minuteKey);
        }

        log.debug("Processing of MockInvocationEvent done !");
    }

    private Map<String, Integer> initializeHourlyMap() {
        Map<String, Integer> result = new HashMap<String, Integer>(24);
        for (int i = 0; i < 24; i++) {
            result.put(String.valueOf(i), 0);
        }
        return result;
    }

    private Map<String, Integer> initializeMinuteMap() {
        Map<String, Integer> result = new HashMap<String, Integer>(24 * 60);
        for (int i = 0; i < 24 * 60; i++) {
            result.put(String.valueOf(i), 0);
        }
        return result;
    }
}