PostgreSQLKeyGenerator.java

/*
 * GovWay - A customizable API Gateway 
 * https://govway.org
 * 
 * Copyright (c) 2005-2025 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.jdbc;

import java.sql.Connection;

import org.openspcoop2.utils.TipiDatabase;

/**
 * Contiene la definizione di un KeyGenerator per Postgresql
 *
 * @author Poli Andrea (apoli@link.it)
 * @author $Author$
 * @version $Rev$, $Date$
 */

public class PostgreSQLKeyGenerator extends AbstractNoReturnGeneratedKeyGenerator {

    public PostgreSQLKeyGenerator(Connection connection,IKeyGeneratorObject type) throws KeyGeneratorException {
        super(connection,type);
    }
    

    @Override
    public String getSequenceQuery() throws KeyGeneratorException {
        String sequence = null;
        switch (this.type.getType()) {
        case DEFAULT:
            sequence = ((DefaultKeyGeneratorObject)this.type).getPrefix()+"_id_seq";
            break;
        case CUSTOM:
            CustomKeyGeneratorObject c = (CustomKeyGeneratorObject)this.type; 
            if(c.existsCustomTableObjectsForDatabaseType(TipiDatabase.POSTGRESQL)){
                sequence = c.getCustomTableObjectsForDatabaseType(TipiDatabase.POSTGRESQL).getSequenceName();
            }else{
                sequence = c.getDefaultTableObjects().getSequenceName();
            }
            break;
        default:
            throw new KeyGeneratorException("Tipo di KeyGeneratorObjects non gestito: "+this.type);
        }
        return "select nextval('"+sequence+"'::text) as nextval";
    }
    
}