fr.ms.log4jdbc.sql.impl.QuerySQL Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of log4jdbc-file Show documentation
There is a newer version: 0.0.5
/*
 * This file is part of Log4Jdbc.
 *
 * Log4Jdbc is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Log4Jdbc 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Log4Jdbc.  If not, see .
 *
 */
package fr.ms.log4jdbc.sql.impl;

import java.util.HashMap;
import java.util.Map;

import fr.ms.log4jdbc.rdbms.DataRdbms;
import fr.ms.log4jdbc.rdbms.RdbmsSpecifics;
import fr.ms.log4jdbc.sql.FormatQuery;

/**
 * 
 * @see Marco4J
 * 
 * 
 * @author Marco Semiao
 * 
 */
public class QuerySQL {

  private final RdbmsSpecifics rdbms;

  private final String jdbcQuery;

  private final Map params = new HashMap();

  private boolean sqlUpdate;

  private String sql;

  private String typeQuery;

  QuerySQL(final RdbmsSpecifics rdbms, final String jdbcQuery) {
    this.rdbms = rdbms;
    this.jdbcQuery = jdbcQuery;
  }

  public Object putParams(final Object key, final Object value) {
    sqlUpdate = false;
    return params.put(key, value);
  }

  public String getJDBCQuery() {
    return jdbcQuery;
  }

  public String getJDBCQuery(final FormatQuery formatQuery) {
    if (formatQuery == null) {
      return getJDBCQuery();
    }
    return formatQuery.format(getJDBCQuery(), rdbms);
  }

  public Map getJDBCParameters() {
    return params;
  }

  public String getTypeQuery() {
    if (typeQuery == null) {
      typeQuery = rdbms.getTypeQuery(getSQLQuery());
    }
    return typeQuery;
  }

  public String getSQLQuery() {
    if (!sqlUpdate) {
      sql = addQueryParameters(jdbcQuery);
      sqlUpdate = true;
    }

    return sql;
  }

  public String getSQLQuery(final FormatQuery formatQuery) {
    if (formatQuery == null) {
      return getSQLQuery();
    }
    return formatQuery.format(getSQLQuery(), rdbms);
  }

  protected String addQueryParameters(String sql) {
    sql = sql.trim();

    if (params == null || params.isEmpty()) {
      return sql;
    }
    final StringBuffer query = new StringBuffer();
    int index = 1;
    int lastPos = 0;
    int position = sql.indexOf('?', lastPos);

    while (position != -1) {
      query.append(sql.substring(lastPos, position));
      final Integer indexCast = new Integer(index);
      final Object param = params.get(indexCast);
      final DataRdbms data = rdbms.getData(param);
      final String paramFormat = data.getParameter();
      query.append(paramFormat);

      lastPos = position + 1;
      position = sql.indexOf('?', lastPos);
      index++;
    }

    if (lastPos < sql.length()) {
      query.append(sql.substring(lastPos, sql.length()));
    }

    return query.toString();
  }

  public String toString() {
    final String result = "QuerySQL [sql=" + getSQLQuery() + "]";

    return result;
  }
}