IDSerialGenerator_mysql.java

/*
 * GovWay - A customizable API Gateway 
 * https://govway.org
 * 
 * Copyright (c) 2005-2024 Link.it srl (https://link.it). 
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 3, as published by
 * the Free Software Foundation.
 *
 * 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
package org.openspcoop2.utils.id.serial;

import java.lang.reflect.Method;
import java.sql.Connection;
import java.sql.Statement;

import org.openspcoop2.utils.UtilsException;
import org.slf4j.Logger;

/**
 * IDSerialGenerator_mysql
 *
 * @author Andrea Poli (apoli@link.it)
 * @author $Author$
 * @version $Rev$, $Date$
 */
public class IDSerialGenerator_mysql {

	public static String generate(Connection conDB,IDSerialGeneratorParameter param, Logger log, InfoStatistics infoStatistics) throws UtilsException{
		
		Statement stmtUpdate = null;
		String identificativoUnivoco = null;
		try{
			if(conDB==null) {
				throw new Exception("Connection is null");
			}
			
			String protocollo = param.getProtocollo();
			
			// Incremento
			stmtUpdate= conDB.createStatement();
			
			String table = param.getTableName();
			if(table==null){
				if(param.getInformazioneAssociataAlProgressivo()!=null){
					table = Constants.TABELLA_ID_RELATIVO_AS_LONG;
				}
				else{
					table = Constants.TABELLA_ID_AS_LONG;
				}
			}
			
			String columnInfoAssociata = param.getColumnRelativeInfo();
			if(columnInfoAssociata==null){
				columnInfoAssociata = Constants.TABELLA_ID_COLONNA_INFO_ASSOCIATA;
			}
			
			String columnPrg = param.getColumnPrg();
			if(columnPrg==null){
				columnPrg = Constants.TABELLA_ID_COLONNA_COUNTER;
			}
			
			String columnProtocollo = param.getColumnProtocol();
			if(columnProtocollo==null){
				columnProtocollo = Constants.TABELLA_ID_COLONNA_PROTOCOLLO;
			}

			String condition = " WHERE "+columnProtocollo+"='"+protocollo+"'";
			if(param.getInformazioneAssociataAlProgressivo()!=null){
				condition = " AND "+columnInfoAssociata+"='"+param.getInformazioneAssociataAlProgressivo()+"'";
			}
			
			String query = "UPDATE " + table  + " SET "+columnPrg+
					" = LAST_INSERT_ID("+columnPrg+"+1)"+condition;
			
			stmtUpdate.executeUpdate(query );
							
			/*			
			if (stmtUpdate instanceof com.mysql.jdbc.Statement){
			com.mysql.jdbc.Statement temp=(com.mysql.jdbc.Statement) stmtUpdate;
			//counter = ((com.mysql.jdbc.Statement) stmtUpdate).getLastInsertID();
			counter=temp.getLastInsertID();
			} else {
			org.jboss.resource.adapter.jdbc.WrappedStatement jbossStatement=(org.jboss.resource.adapter.jdbc.WrappedStatement)stmtUpdate;
			com.mysql.jdbc.Statement temp=(com.mysql.jdbc.Statement)jbossStatement.getUnderlyingStatement();
			counter=temp.getLastInsertID();
			}
			 */
		
			// TODO REFLECTION VERIFY
			Class<?> cMysqlStatement = Class.forName("com.mysql.jdbc.Statement");
			Method m = cMysqlStatement.getClass().getMethod("getLastInsertID");
			if(stmtUpdate.getClass().isAssignableFrom(cMysqlStatement)){
				Object oReturn = m.invoke(stmtUpdate);
				if(oReturn!=null){
					Long counterAsLong = (Long) oReturn;
					identificativoUnivoco = counterAsLong.longValue()+"";
				}
			}
			else{
				Method mUnder = cMysqlStatement.getClass().getMethod("getUnderlyingStatement");
				Object stmtUnder = mUnder.invoke(stmtUpdate);
				Object oReturn = m.invoke(stmtUnder);
				if(oReturn!=null){
					Long counterAsLong = (Long) oReturn;
					identificativoUnivoco = counterAsLong.longValue()+"";
				}
			}
			
			stmtUpdate.close();	
		} catch(Exception e) {
			try{
				if( stmtUpdate != null )
					stmtUpdate.close();
			} catch(Exception er) {}
			try{
				if(conDB!=null) {
					conDB.rollback();
				}
			} catch(Exception er) {
				// close
			}
		}
		
		return identificativoUnivoco;
	}
	
}