PreprocessClause.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.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.turro.string.Strings;
import org.turro.elephant.db.WhereClause;
 
public class PreprocessClause {
  
  private String table, entityRoot, relatedRoot, pathField, 
          relatedField, aggregate, filterEntity;
  private boolean matchingPath;
  private IndicatorVariable variable;
  private final List<String> joins = new ArrayList<>();
  private final List<String> conditions = new ArrayList<>();
  private final Map<String, Object> namedValues = new HashMap<>();
  
  public static PreprocessClause load(String table) {
    return new PreprocessClause(table);
  }
 
  public PreprocessClause setRankingRoot(String entityRoot) {
    this.entityRoot = entityRoot;
    return this;
  }
  
  public PreprocessClause setMatchingRoot(String entityRoot) {
    this.relatedRoot = entityRoot;
    return this;
  }
  
  public PreprocessClause setRankingField(String pathField) {
    this.pathField = pathField;
    return this;
  }
  
  public PreprocessClause setMatchingField(String pathField) {
    this.relatedField = pathField;
    return this;
  }
  
  public PreprocessClause setMatchingIsPath() {
    this.matchingPath = true;
    return this;
  }
  
  public PreprocessClause setAggregate(String aggregate) {
    this.aggregate = aggregate;
    return this;
  }
  
  public PreprocessClause setFilter(String filterEntity) {
    this.filterEntity = filterEntity;
    return this;
  }
  
  public PreprocessClause setVariable(IndicatorVariable variable) {
    this.variable = variable;
    return this;
  }
  
  public PreprocessClause addJoin(String join) {
    joins.add(join);
    return this;
  }
  
  public PreprocessClause addCondition(String condition) {
    conditions.add(condition);
    return this;
  }
  
  public PreprocessClause addNamedValue(String name, Object value) {
    namedValues.put(name, value);
    return this;
  }
  
  public WhereClause getClause() {
    if(Strings.isBlank(relatedRoot)) {
      return getRankingClause();
    } else {
      return getMatchingClause();
    }
  }
  
  public WhereClause getRankingClause() {
    WhereClause wc = new WhereClause();
    wc.addClause("select new org.turro.ranking.GenericRanking(");
    wc.addClause(pathField + ", '" + variable.getName() + "', " + aggregate + ")");
    wc.addClause("from " + table);
    for(String join : joins) {
      wc.addClause(join);
    }
    if(!Strings.isBlank(filterEntity)) {
      wc.addClause("where " + pathField + " like concat(:entity, '%')");
      wc.addNamedValue("entity", filterEntity);
    } else {
      wc.addClause("where " + pathField + " like concat('/', :root, '/%')");
      wc.addNamedValue("root", entityRoot);
    }
    for(String condition : conditions) {
      wc.addClause(condition);
    }
    namedValues.keySet().forEach(k -> wc.addNamedValue(k, namedValues.get(k)));
    wc.addClause("group by " + pathField);
    wc.addClause("having " + aggregate + " <> 0");
    return wc;
  }
  
  public WhereClause getMatchingClause() {
    WhereClause wc = new WhereClause();
    wc.addClause("select new org.turro.matching.GenericMatching(");
    String relatedConcat = matchingPath ? relatedField :
            "concat('/" + relatedRoot + "/'," + relatedField + ")";
    wc.addClause(pathField + ", " + relatedConcat + ", '" + variable.getName() + "', " + aggregate + ")");
    wc.addClause("from " + table);
    for(String join : joins) {
      wc.addClause(join);
    }
    if(!Strings.isBlank(filterEntity)) {
      wc.addClause("where " + pathField + " like concat(:entity, '%')");
      wc.addNamedValue("entity", filterEntity);
    } else {
      wc.addClause("where " + pathField + " like concat('/', :root, '/%')");
      wc.addNamedValue("root", entityRoot);
    }
    wc.addClause("and " + relatedConcat + " like concat('/', :rroot, '/%')");
    wc.addNamedValue("rroot", relatedRoot);
    for(String condition : conditions) {
      wc.addClause(condition);
    }
    namedValues.keySet().forEach(k -> wc.addNamedValue(k, namedValues.get(k)));
    wc.addClause("group by " + pathField + "," + relatedConcat);
    wc.addClause("having " + aggregate + " <> 0");
    return wc;
  }
  
  private PreprocessClause(String table) {
    this.table = table;
  }
 
}