ServiceIndicator.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2020 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
package org.turro.indicator;
 
import java.time.Duration;
import java.time.Instant;
import java.util.stream.Stream;
import org.turro.string.Strings;
import org.turro.annotation.ElephantIndicator;
import org.turro.contacts.ContactService;
import org.turro.contacts.db.ContactsPU;
import org.turro.elephant.db.WhereClause;
import org.turro.jpa.Dao;
import org.turro.matching.GenericMatching;
import org.turro.matching.IMatching;
 
@ElephantIndicator
public class ServiceIndicator extends AbstractIndicator {
 
  public ServiceIndicator() {
    root = "service";
  }
 
  @Override
  protected void initVariables() {
    addVariable("participation", true, VariableType.MATCHING_VARIABLE);
    addVariable("daysCreation", VariableType.RANKING_VARIABLE);
    addVariable("daysStart", VariableType.RANKING_VARIABLE);
    addVariable("daysEnd", VariableType.RANKING_VARIABLE);
    addVariable("words", VariableType.RANKING_VARIABLE);
  }
 
  @Override
  protected void initExternalVariables() {
    addExternalVariable("generic", "stars");
    addExternalVariable("generic", "comments");
    addExternalVariable("forum", "topics");
    addExternalVariable("forum", "posts");
    addExternalVariable("commons", "followed");
    addExternalVariable("commons", "seen");
    addExternalVariable("commons", "like");
  }
 
  @Override
  protected Dao createDao() {
    return new ContactsPU();
  }
 
  @Override
  public double getValue(IndicatorVariable variable, Object entity) {
    if((entity instanceof ContactService) && itsMine(variable)) {
      ContactService service = (ContactService) entity;
      switch(variable.getName()) {
        case "daysCreation":
          return service.getCreation() != null ? Duration.between(service.getCreation().toInstant(), Instant.now()).toDays() : 9999.9;
        case "daysStart":
          return service.getStartDate()!= null ? Duration.between(service.getStartDate().toInstant(), Instant.now()).toDays() : 9999.9;
        case "daysEnd":
          return service.getEndDate()!= null ? Duration.between(service.getEndDate().toInstant(), Instant.now()).toDays() : 9999.9;
        case "words":
          return countWords(service);
      }
    }
    return 0.0;
  }
 
  @Override
  protected void processRankingVariable(Dao dao, IPreprocessor preprocessor, IndicatorVariable variable, String entityRoot) {
  }
 
  @Override
  protected void processMatchingVariable(Dao dao, IPreprocessor preprocessor, IndicatorVariable variable, String entityRoot, String relatedRoot) {
    WhereClause wc = null;
    if("contact".equals(relatedRoot)) {
      switch(variable.getName()) {
        case "participation":
          wc = PreprocessClause.load("ContactService")
                  .setRankingRoot(entityRoot)
                  .setMatchingRoot(relatedRoot)
                  .setRankingField("concat('/service/',id)")
                  .setMatchingField("responsible.id")
                  .setVariable(variable)
                  .setAggregate("count(id)")
                  .getClause();
          break;
      }
    }
    if(wc != null) {
      try(Stream<GenericMatching> stream = dao.stream(GenericMatching.class, wc, 1000)) {
        stream.forEach(r -> {
          preprocessor.poolInstance(r.copyTo((IMatching) preprocessor.createInstance()));
        });
        preprocessor.finishPreprocessor();
      }
    }
  }
 
  private double countWords(ContactService service) {
    double count = 0;
    if(service.getPlainText() != null) {
      count += Strings.countWords(service.getPlainText());
    }
    return count;
  }
 
}