ElementDB.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2011 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.elephant.db;
 
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Collection;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.turro.elephant.context.ElephantContext;
import org.turro.elephant.context.IElement;
import org.turro.elephant.layout.IManageable;
 
public class ElementDB {
  protected IElement element;
  protected Connection conn = null;
  protected int lastFirst, lastCurrent, lastCount, totalCount;
  
  public ElementDB(IElement element) {    
    this.element = element;
  }
  
  public Connection getConnection() throws SQLException {
    if(conn == null || conn.isClosed()) {
      String driverClass = ((IManageable)element).getAttributes().getAttributeValue("attrib:jdbc-driver", null),
             url = ((IManageable)element).getAttributes().getAttributeValue("attrib:jdbc-url", null),
             user = ((IManageable)element).getAttributes().getAttributeValue("attrib:jdbc-user", null),
             password = ((IManageable)element).getAttributes().getAttributeValue("attrib:jdbc-password", null);
      if(driverClass != null && url != null) {
        // TODO: optimize
        url = url.replaceAll("\\#idel", element.getId());
        url = url.replaceAll("\\#ctx", 
                ElephantContext.getRealPath(
                  element.getContext().getPath()
                ));
        url = url.replaceAll("\\#root", ElephantContext.getRealPath("/"));
        url = url.replaceAll("\\\\","/");
        try {
          Class.forName(driverClass);
          if(user != null && user.length() > 0 && password != null) {
            conn = DriverManager.getConnection(url, user, password);
          }
          else {
            conn = DriverManager.getConnection(url);
          }
          return conn;
        } catch (ClassNotFoundException | SQLException ex) {
          Logger.getLogger(ElementDB.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
          return null;
        }
      }
      return null;
    }
    return conn;
  }
 
  public boolean positionFirst(ResultSet rs, int first) throws SQLException {
    lastFirst = first;
    boolean hasNext = true;
    int current = 0;
    while((current < first) && hasNext) {
      current++;
      hasNext = rs.next();
    }
    return hasNext;
  }
 
  public boolean iterate(ResultSet rs, int count, int current) throws SQLException {
    lastCount = count;
    lastCurrent = current;
    if(current - lastFirst < count) {
      return rs.next();
    }
    return false;
  }
  
  public int findLast(ResultSet rs) throws SQLException {
    totalCount = lastCurrent;
    while(rs.next()) {
      totalCount++;
    }
    return totalCount;
  }
  
  public ResultSet query(String sql) throws SQLException {
    Connection con = getConnection();
    return con.createStatement().executeQuery(sql);
  }
 
  public ResultSet query(String sql, Collection params) throws SQLException {
    return query(sql, params.toArray());
  }
 
  public ResultSet query(String sql, Object[] params) throws SQLException {
    if(params == null) {
      return query(sql);
    }
    Connection con = getConnection();
    PreparedStatement ps = con.prepareStatement(sql);
    for(int i = 0; i < params.length; i++) {
      if(params[i] != null) {
        ps.setObject(i + 1, params[i]);
      }
      else {
        ps.setNull(i + 1, java.sql.Types.VARCHAR);
      }
    }      
    return ps.executeQuery();
  }
 
  public int execute(String sql) throws SQLException {
    Connection con = getConnection();
    return con.createStatement().executeUpdate(sql);
  }
 
  public int execute(String sql, Collection params) throws SQLException {
    return execute(sql, params.toArray());
  }
 
  public int execute(String sql, Object[] params) throws SQLException {
    if(params == null) {
      return execute(sql);
    }
    Connection con = getConnection();
    PreparedStatement ps = con.prepareStatement(sql);
    for(int i = 0; i < params.length; i++) {
      if(params[i] != null) {
        ps.setObject(i + 1, params[i]);
      }
      else {
        ps.setNull(i + 1, java.sql.Types.VARCHAR);
      }
    }      
    return ps.executeUpdate();
  }
 
  public void execute(String commands[]) throws SQLException {
    for (String command : commands) {
      execute(command);
    }
  }
  
  public void close(ResultSet rs) throws SQLException {
    rs.close();
  }
  
  public void close() throws SQLException {
    if(conn != null && !conn.isClosed())
      conn.close();
  }
 
  public void renderNavigator(PrintWriter out, String param) {
    if(totalCount <= lastCount) return;
    out.print("<div class='dbnav'>");
    out.print("<ul>");
    if(lastFirst > 0) {
      out.print("<li>");
      out.print("<a href='?" + param + "=" + (lastFirst - lastCount) + "'>");
      out.print("&lt;&lt;");
      out.print("</a>");
      out.print("</li>");
    }
    int iter = (int)Math.ceil(((double)totalCount) / ((double)lastCount));
    for(int i = 0; i < iter; i++) {
      if(lastFirst >= i * lastCount && lastFirst < (i+1) * lastCount) {
        out.print("<li class='active'>");
      }
      else {
        out.print("<li>");
      }
      out.print("<a href='?" + param + "=" + (i * lastCount) + "'>");
      out.print(i + 1);
      out.print("</a>");
      out.print("</li>");
    }
    if(lastFirst + lastCount < totalCount) {
      out.print("<li>");
      out.print("<a href='?" + param + "=" + (lastFirst + lastCount) + "'>");
      out.print("&gt;&gt;");
      out.print("</a>");
      out.print("</li>");
    }
    out.print("</ul>");
    out.print("</div>");
  }
 
  public java.util.Date toDate(java.sql.Timestamp date) {
    if(date == null) return null;
    return new java.util.Date(date.getTime());
  }
 
  public java.sql.Timestamp toTimestamp(java.util.Date date) {
    if(date == null) return null;
    return new java.sql.Timestamp(date.getTime());
  }
  
  public java.util.Date toDate(java.sql.Date date) {
    if(date == null) return null;
    return new java.util.Date(date.getTime());
  }
 
  public java.sql.Date toSqlDate(java.util.Date date) {
    if(date == null) return null;
    return new java.sql.Date(date.getTime());
  }
 
}