com.sysunite.coinsweb.graphset.QueryFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of validator-cli Show documentation
The newest version!
package com.sysunite.coinsweb.graphset;

import com.sysunite.coinsweb.parser.config.pojo.GraphVarImpl;
import com.sysunite.coinsweb.parser.profile.factory.ProfileFactory;
import com.sysunite.coinsweb.parser.profile.pojo.Bundle;
import com.sysunite.coinsweb.parser.profile.pojo.ProfileFile;
import com.sysunite.coinsweb.parser.profile.pojo.Query;
import freemarker.cache.StringTemplateLoader;
import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.io.StringWriter;
import java.io.Writer;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

/**
 * @author bastbijl, Sysunite 2017
 */
public class QueryFactory {

  private static final Logger log = LoggerFactory.getLogger(QueryFactory.class);

  public static final String VALIDATOR_HOST = "http://www.coinsweb.nl/";
  public static final String VALIDATOR_NS = "http://www.coinsweb.nl/2017/07/validator#";


  public static String buildQuery(Query query, Map data) {
    return buildQuery(query, data, null, 0);
  }
  public static String buildQuery(Query query, Map data, String prefixes) {
    return buildQuery(query, data, prefixes, 0);
  }
  public static String buildQuery(Query query, Map data, String prefixes, int maxResults) {
    String cleanQuery = query.getQuery();
    if(prefixes != null && !prefixes.isEmpty()) {
      cleanQuery = prefixes + '\n' + cleanQuery;
    }
    String finalQuery = parseFreemarker(cleanQuery, data);
    if(maxResults > 0) {
      finalQuery += " LIMIT " + maxResults;
    }
    log.trace(finalQuery.replace("\n", " "));
    return finalQuery;
  }

  public static Set usedVars(ProfileFile profileFile) {
    HashSet result = new HashSet();
    for(String graphVar : ProfileFactory.usedVars(profileFile)) {
      result.add(new GraphVarImpl(graphVar));
    }
    return result;
  }

  public static Set usedVars(Bundle bundle) {
    HashSet result = new HashSet();
    for(String graphVar : ProfileFactory.usedVars(bundle)) {
      result.add(new GraphVarImpl(graphVar));
    }
    return result;
  }





  public static String toSelectQuery(String insertQuery) {

    if(!insertQuery.toLowerCase().contains("insert")) {
      throw new RuntimeException("Please only call this for queries with an insert");
    }

    String mappedQuery = "";

    boolean atHeader = true;
    boolean atInsert = false;
    boolean atWhere = false;

    String[] lines = insertQuery.split("\n");
    for(String line : lines) {
      if(line == null || line.isEmpty()) {
        continue;
      }
      String cleanLine = line.trim().toLowerCase();

      // State switches
      if(atHeader && cleanLine.startsWith("insert")) {
        atHeader = false;
        atInsert = true;

        mappedQuery += "SELECT * \n";
      }
      if(atInsert && cleanLine.startsWith("where")) {
        atInsert = false;
        atWhere = true;
      }

      if(atHeader || atWhere) {
        mappedQuery += line+"\n";
      }
    }
    return mappedQuery;
  }



  private static String parseFreemarker(String queryTemplate, Map data) {
    Configuration cfg = new Configuration();
    StringTemplateLoader templateLoader = new StringTemplateLoader();
    cfg.setTemplateLoader(templateLoader);
    templateLoader.putTemplate("queryTemplate", queryTemplate);
    try {
      Template template = cfg.getTemplate("queryTemplate");
      Writer writer = new StringWriter();
      template.process(data, writer);
      return writer.toString();

    } catch (IOException e) {
      log.error(e.getMessage(), e);
    } catch (TemplateException e) {
      log.error(e.getMessage(), e);
    }

    throw new RuntimeException("Something went wrong building query.");
  }


}