io.github.microcks.repository.DailyStatisticRepositoryImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of microcks-app Show documentation
The newest version!
/*
 * Copyright The Microcks Authors.
 *
 * 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 io.github.microcks.repository;

import io.github.microcks.domain.DailyStatistic;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.domain.Sort.Direction;
import org.springframework.data.domain.Sort.Order;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.aggregation.Aggregation;
import org.springframework.data.mongodb.core.aggregation.AggregationResults;
import org.springframework.data.mongodb.core.mapreduce.MapReduceResults;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;

import java.util.List;

import static org.springframework.data.domain.Sort.Direction.ASC;
import static org.springframework.data.mongodb.core.aggregation.Aggregation.*;

/**
 * Implementation of CustomDailyStatisticRepository.
 * @author laurent
 */
public class DailyStatisticRepositoryImpl implements CustomDailyStatisticRepository {

   /** A simple logger for diagnostic messages. */
   private static Logger log = LoggerFactory.getLogger(DailyStatisticRepositoryImpl.class);

   @Autowired
   private MongoTemplate template;

   @Override
   public void incrementDailyStatistic(String day, String serviceName, String serviceVersion, String hourKey,
         String minuteKey) {

      // Build a query to select specific object within collection.
      Query query = new Query(
            Criteria.where("day").is(day).and("serviceName").is(serviceName).and("serviceVersion").is(serviceVersion));

      // Other way to build a query using statically imported methods.
      //Query queryShort = query(where("day").is(day).and("serviceName").is(serviceName).and("serviceVersion").is(serviceVersion));

      // Build update to increment the 3 fields.
      Update update = new Update().inc("dailyCount", 1).inc("hourlyCount." + hourKey, 1).inc("minuteCount." + minuteKey,
            1);

      // Do an upsert with find and modify.
      template.findAndModify(query, update, DailyStatistic.class);
   }

   @Override
   public DailyStatistic aggregateDailyStatistics(String day) {

      // Build a query to pre-select the statistics that will be aggregated.
      Query query = new Query(Criteria.where("day").is(day));

      // Execute a MapReduce command.
      MapReduceResults results = template.mapReduce(query, "dailyStatistic",
            "classpath:mapDailyStatisticForADay.js", "classpath:reduceDailyStatisticForADay.js",
            WrappedDailyStatistic.class);

      // Output some debug messages.
      if (log.isDebugEnabled()) {
         log.debug("aggregateDailyStatistics mapReduce for day " + day);
         log.debug("aggregateDailyStatistics mapReduce result counts: " + results.getCounts());
         for (WrappedDailyStatistic wdt : results) {
            log.debug("aggregateDailyStatistics mapReduce result value: " + wdt.getValue());
         }
      }

      // We've got a result if we've got an output.
      if (results.getCounts().getOutputCount() > 0) {
         return results.iterator().next().getValue();
      }

      // Build and return an empty object otherwise?
      DailyStatistic statistic = new DailyStatistic();
      statistic.setDay(day);
      statistic.setDailyCount(0);
      return statistic;
   }

   @Override
   public List aggregateDailyStatistics(String afterday, String beforeday) {
      // Build a query to pre-select the statistics that will be aggregated.
      Aggregation aggregation = newAggregation(match(Criteria.where("day").gte(afterday).lte(beforeday)),
            group("day").sum("dailyCount").as("number"), project("number").and("day").previousOperation(),
            sort(ASC, "day"));
      AggregationResults results = template.aggregate(aggregation, DailyStatistic.class,
            InvocationCount.class);
      return results.getMappedResults();
   }

   @Override
   public List findTopStatistics(String day, int limit) {
      // Build a query selecting and sorting / limiting.
      Query query = new Query(Criteria.where("day").is(day)).with(Sort.by(new Order(Direction.DESC, "dailyCount")))
            .limit(limit);

      return template.find(query, DailyStatistic.class);
   }


   /** Utility class used for wrapping a DailyStatistic object within MapReduce command results. */
   public class WrappedDailyStatistic {
      private String id;
      private DailyStatistic value;

      public String getId() {
         return id;
      }

      public void setId(String id) {
         this.id = id;
      }

      public DailyStatistic getValue() {
         return value;
      }

      public void setValue(DailyStatistic value) {
         this.value = value;
      }
   }
}