PostgreSQLThreshold.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.pdd.core.threshold;

import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Properties;

import javax.sql.DataSource;
import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;

import org.openspcoop2.core.commons.CoreException;
import org.openspcoop2.pdd.config.DBManager;
import org.openspcoop2.pdd.config.OpenSPCoop2Properties;
import org.openspcoop2.pdd.config.Resource;
import org.openspcoop2.pdd.logger.MsgDiagnostico;
import org.openspcoop2.pdd.logger.OpenSPCoop2Logger;
import org.openspcoop2.protocol.basic.BasicComponentFactory;
import org.openspcoop2.utils.jdbc.JDBCUtilities;

/**
 * Implementazione che definisce un meccanismo di Soglia sullo spazio libero rimasto
 * nelle risorse utilizzate dalla PdD, per un database MySQL. 
 *  
 * @author Andrea Manca (manca@link.it)
 * @author $Author$
 * @version $Rev$, $Date$
 */


public class PostgreSQLThreshold implements IThreshold {

    private static OpenSPCoop2Properties properties = OpenSPCoop2Properties.getInstance();
    public static final String ID_MODULO = "PostgreSQLThreshold";
    
    
    /**
     * @param parametri puo essere del tipo 
     *  "valore=<valore di soglia> [datasource=<nome datasource>]"
     *  se nome datasource non e' indicato verra' utilizzata la connessione
     *  verso il database openspcoop
     *  
     *  Il valore di soglia e' interpretato in byte a meno che non sia esplici
     *  tamente indicata l'unita di misura. 
     *  NOTA: la query per il controllo dello spazio necessita dei diritti
     *  di superutente per essere eseguita
     */
    @Override
    public boolean check(Properties parametri) throws ThresholdException {
        DBManager dbManager = DBManager.getInstance();
        Resource resource = null;
        long threshold=-1;
        long size=0L;
        long factor=1L;
        boolean result=false;
        Statement s=null;
        ResultSet rs=null;
        
        String debugS = parametri.getProperty("debug");
        boolean debug = false;
        if(debugS!=null){
            debug = Boolean.valueOf(debugS);
        }
        
        String query = parametri.getProperty("query");
        if(query==null){
            throw new ThresholdException("Parametro 'query' non presente");
        }
        
        String valoreSoglia = parametri.getProperty("valore");
        if(valoreSoglia==null){
            throw new ThresholdException("Parametro 'valore' non presente");
        }
        else{
            valoreSoglia = valoreSoglia.trim();
        }
        
        String datasource = parametri.getProperty("datasource");
        if(datasource!=null){
            datasource = datasource.trim();
        }
        
        String soglia=valoreSoglia.toLowerCase();
        StringBuilder valore=new StringBuilder();
        char ch=soglia.charAt(0);
        int i=1;
        while (Character.isDigit(ch))
        {
            valore.append(ch);
            if(i<soglia.length())
                ch=soglia.charAt(i++);
            else
                ch='f';
        }
        if ( soglia.endsWith("kb") || soglia.endsWith("k") )
            factor=1024L;
        else if ( soglia.endsWith("mb") || soglia.endsWith("m") )
            factor=1024L * 1024L;
        else if ( soglia.endsWith("gb") || soglia.endsWith("g") )
            factor=1024L * 1024L * 1024L;
        //else
            //Utilizzo del fattore di default 1
        
        MsgDiagnostico msgDiag = MsgDiagnostico.newInstance(PostgreSQLThreshold.ID_MODULO);
        Connection connection =null;
        try{
            if(datasource==null) {
                resource = getConnection(dbManager);
                connection = (Connection) resource.getResource();
            }
            else
            {
                connection = getConnection(datasource);
            }

            // controllo validita parametro
            threshold=Long.parseLong(valore.toString())*factor;
            if(threshold<0L)
                throw new CoreException("Valore di soglia negativo");
            
            //Interrogazione del database
            s=connection.createStatement();
            if(!s.execute(query))
                throw new CoreException("Impossibile verficare lo spazio occupato");
            
            rs=s.getResultSet();
            
            if(rs==null)
                throw new CoreException("Nessun risultato disponibile per la verifica di soglia");
            if(!rs.next())
                throw new CoreException("Nessun risultato disponibile per la verifica di soglia");
            
            size=rs.getLong(1);
            if(size==0)
                throw new CoreException("La quantita' di spazio occupata dai DB e' NULL");

                        
            result=(size<threshold);
            
        }catch(Exception e){
            throw new ThresholdException("PostgreSQLThreshold error: "+e.getMessage(),e);
        }finally{
            JDBCUtilities.closeResources(rs, s);
            rs=null;s=null;
            try {
                if(datasource==null)
                    dbManager.releaseResource(PostgreSQLThreshold.properties.getIdentitaPortaDefaultWithoutProtocol(), PostgreSQLThreshold.ID_MODULO, resource);
                else {
                    if ((connection != null) && (!connection.isClosed()))
                        JDBCUtilities.closeConnection(BasicComponentFactory.getCheckLogger(), connection, BasicComponentFactory.isCheckAutocommit(), BasicComponentFactory.isCheckIsClosed());
                }
            }catch(SQLException ex){
                // ignore
            }       
        }
        
        String prefix = "";
        if(datasource!=null) {
            prefix = "["+datasource+"] ";
        }
        String msg = prefix+"Spazio occupato: "+size+"; soglia: "+result+"; risultato:"+(result?"ok":"ko");
        if(debug) {
            OpenSPCoop2Logger.getLoggerOpenSPCoopResources().info(msg);
        }
        msgDiag.highDebug(msg);
        
        return result;
        
    }
    
    private Resource getConnection(DBManager dbManager) throws CoreException {
        Resource resource = null;
        try{
            resource = dbManager.getResource(PostgreSQLThreshold.properties.getIdentitaPortaDefaultWithoutProtocol(), PostgreSQLThreshold.ID_MODULO,null);
        }catch(Exception e){
            throw new CoreException("Impossibile ottenere una Risorsa dal DBManager",e);
        }
        if(resource==null)
            throw new CoreException("Risorsa is null");
        if(resource.getResource() == null)
            throw new CoreException("Connessione is null");
        return resource;
    }
    private Connection getConnection(String datasource) throws NamingException, SQLException {
        Context c = new InitialContext();
        DataSource ds= (DataSource)c.lookup(datasource);
        c.close();  
        return ds.getConnection();
    }

}