FunnelGraph.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2022 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.web.funnel.graph;
 
import java.util.Set;
import java.util.TreeSet;
import java.util.function.Predicate;
import java.util.stream.Collectors;
import org.jgrapht.Graphs;
import org.jgrapht.graph.SimpleDirectedGraph;
import org.turro.elephant.db.ElephantPU;
import org.turro.elephant.entities.web.WebGoal;
import org.turro.elephant.entities.web.WebItem;
import org.turro.elephant.entities.web.WebItemType;
import org.turro.jpa.Dao;
import org.turro.sql.SqlClause;
import org.turro.util.Cached;
 
public class FunnelGraph extends SimpleDirectedGraph<FunnelVertex, FunnelEdge> {
 
  public WebItemType getBehaviour(WebItem item) {
    FunnelVertex vertex = getVertex(item);
    return vertex == null ? null : vertex.getActingType();
  }
 
  public Set<FunnelVertex> getTargets() {
    return getItems(vertex -> vertex.isTarget());
  }
  
  public Set<FunnelVertex> getSolutions() {
    return getItems(vertex -> vertex.isSolution());
  }
  
  public Set<FunnelVertex> getServices() {
    return getItems(vertex -> vertex.isService());
  }
  
  public Set<FunnelVertex> getChains() {
    return getItems(vertex -> vertex.isChain());
  }
  
  public Set<FunnelVertex> getItems(Predicate<FunnelVertex> filter) {
    return vertexSet().stream()
            .filter(filter)
            .collect(Collectors.toCollection(TreeSet::new));
  }
  
  public FunnelVertex getVertex(String webTag) {
    return vertexSet().stream()
            .filter(vertex -> vertex.getWebTag().equals(webTag))
            .findAny().orElse(null);
  }
  
  public FunnelVertex getVertex(WebItem item) {
    return vertexSet().stream()
            .filter(vertex -> vertex.getItem().equals(item))
            .findAny().orElse(null);
  }
  
  public WebGoal getGoal(String goalAction) {
    return vertexSet().stream()
            .flatMap(vertex -> vertex.getItem().getGoals().stream())
            .filter(goal -> goal.getGoalAction().equals(goalAction))
            .findAny().orElse(null);
  }
  
  /* Initialization */
  
  private void addVertices() {
    SqlClause.select("wi").from("WebItem wi")
            .orderBy("ordering")
            .dao(dao.get())
            .resultList(WebItem.class)
            .forEach(item -> {
              addVertex(new FunnelVertex(this, item));
            });
  }
  
  private void addEdges() {
    vertexSet().stream().sorted().forEach(vertex -> {
      vertex.getItem().getNexts().forEach(item -> {
        FunnelVertex vitem = new FunnelVertex(this, item);
        if(containsVertex(vitem)) {
          Graphs.addEdgeWithVertices(this, vertex, vitem);
        }
      });
    });
  }
  
  /* Utils */
  
  public static WebItemType behaveAs(WebItem item) {
    return load().getBehaviour(item);
  }
  
  /* Factory */
 
  public static FunnelGraph load() {
    return new FunnelGraph();
  }
  
  private FunnelGraph() {
    super(FunnelEdge.class);
    addVertices();
    addEdges();
  }
 
  /* Dao */
  
  protected final Cached<Dao> dao = Cached.instance(() -> new ElephantPU());
 
}