IdUtils.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.util.logging.Level;
import java.util.logging.Logger;
import org.turro.string.Strings;
import org.turro.elephant.context.ElephantContext;
import org.turro.elephant.impl.util.StringParser;
import org.turro.jpa.Dao;
 
public class IdUtils {
 
  public static long getNewLongIdFromLong(Dao dao, String table, String field) {
    return getNewLongIdFromLong(dao, table, field, new String[] {});
  }
 
  public static long getNewLongIdFromLong(Dao dao, String table, String field, String fieldConstraints[]) {
    Long result = getMaxLongIdFromLong(dao, table, field, fieldConstraints, true);
    if(result == 0) {
      WhereClause wc = new WhereClause();
      wc.addClause("select max(a." + field + ") + 1 from " + table + " a");
      wc.addClause("where a." + field + " = (");
      wc.addClause("  select count(*) from " + table + " b");
      wc.addClause("  where " + putPrefix(field, "b.") + " <= " + putPrefix(field, "a."));
      wc.addClause("  and " + putPrefix(field, "b.") + " > 0");
      for(String fc : fieldConstraints) {
        if(!Strings.isBlank(fc)) wc.addClause("  and " + putPrefix(fc, "b."));
      }
      wc.addClause(")");
      for(String fc : fieldConstraints) {
        if(!Strings.isBlank(fc)) wc.addClause("  and " + putPrefix(fc, "a."));
      }
      result = (Long) dao.getSingleResult(wc.getClause());
      if(result == null || result == 0) {
        result = getMaxLongIdFromLong(dao, table, field, fieldConstraints);
      }
    }
    return result;
  }
 
  public static long getNewLongIdFromString(Dao dao, String table, String field) {
    return getNewLongIdFromString(dao, table, field, new String[] {});
  }
 
  public static long getNewLongIdFromString(Dao dao, String table, String field, String fieldConstraints[]) {
    try {
      WhereClause wc = new WhereClause();
      wc.addClause("select max(cast(a." + field + " as signed)) + 1 from " + table + " a");
      wc.addClause("where cast(" + putPrefix(field, "a.") + " as signed) = (");
      wc.addClause("  select count(*) from " + table + " b");
      wc.addClause("  where cast(" + putPrefix(field, "b.") + " as signed) <= cast(" + putPrefix(field, "a.") + " as signed)");
      wc.addClause("  and cast(" + putPrefix(field, "b.") + " as signed) > 0");
      for(String fc : fieldConstraints) {
        if(!Strings.isBlank(fc)) wc.addClause("  and " + putPrefix(fc, "b."));
      }
      wc.addClause(")");
      for(String fc : fieldConstraints) {
        if(!Strings.isBlank(fc)) wc.addClause("  and " + putPrefix(fc, "a."));
      }
      Long result = (Long) StringParser.convertToClass(Long.class, dao.getSingleNativeResult(wc.getClause()));
      if(result == null || result == 0) {
        result = getMaxLongIdFromString(dao, table, field, fieldConstraints);
      }
      return result;
    } catch(Exception ex) {
      Logger.getLogger(IdUtils.class.getName()).log(Level.INFO, ElephantContext.logMsg(null), ex);
      return 1;
    }
  }
 
  public static long getMaxLongIdFromLong(Dao dao, String table, String field) {
    return getMaxLongIdFromLong(dao, table, field, new String[] {});
  }
 
  public static long getMaxLongIdFromLong(Dao dao, String table, String field, String fieldConstraints[]) {
    return getMaxLongIdFromLong(dao, table, field, fieldConstraints, false);
  }
 
  public static long getMaxLongIdFromLong(Dao dao, String table, String field, String fieldConstraints[], boolean count) {
    WhereClause wc = new WhereClause();
    wc.addClause("select count(a." + field + "), max(a." + field + ") from " + table + " a");
    wc.addClause("where 1=1");
    for(String fc : fieldConstraints) {
      if(!Strings.isBlank(fc)) wc.addClause("  and " + putPrefix(fc, "a."));
    }
    Object result[] = (Object[]) dao.getSingleResult(wc.getClause());
    Long regs = (Long) result[0], id = (Long) result[1];
    if(id == null || id == 0) {
      regs = 0L;
      id = 0L;
    }
    return count ? (regs.equals(id) ? id + 1L : 0L) : id + 1L;
  }
 
  public static long getMaxLongIdFromString(Dao dao, String table, String field) {
    return getMaxLongIdFromString(dao, table, field, new String[] {});
  }
 
  public static long getMaxLongIdFromString(Dao dao, String table, String field, String fieldConstraints[]) {
    try {
      WhereClause wc = new WhereClause();
      wc.addClause("select max(cast(a." + field + " as signed)) from " + table + " a");
      wc.addClause("where 1=1");
      for(String fc : fieldConstraints) {
        if(!Strings.isBlank(fc)) wc.addClause("  and " + putPrefix(fc, "a."));
      }
      Long result = (Long) StringParser.convertToClass(Long.class, dao.getSingleNativeResult(wc.getClause()));
      if(result == null || result == 0) {
        result = Long.valueOf(0);
      }
      return result + 1;
    } catch(Exception ex) {
      Logger.getLogger(IdUtils.class.getName()).log(Level.INFO, ElephantContext.logMsg(null), ex);
      return 1;
    }
  }
 
  public static String putPrefix(String field, String prefix) {
    int p = Math.max(field.lastIndexOf("("), 0);
    if(p > 0) {
      field = field.substring(0, p + 1) + prefix + field.substring(p + 1);
    } else {
      field = prefix + field;
    }
    return field;
  }
 
  private IdUtils() {
  }
 
}