GetImportedForeignKeysRequest.java

/*
 * JBoss, Home of Professional Open Source.
 *
 * See the LEGAL.txt file distributed with this work for information regarding copyright ownership and licensing.
 *
 * See the AUTHORS.txt file distributed with this work for a full listing of individual contributors.
 */
package org.teiid.designer.jdbc.metadata.impl;

import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.List;
import org.eclipse.core.runtime.IStatus;
import org.teiid.designer.jdbc.JdbcPlugin;
import org.teiid.designer.jdbc.JdbcUtil;
import org.teiid.designer.jdbc.data.MetadataRequest;
import org.teiid.designer.jdbc.data.Response;
import org.teiid.designer.jdbc.data.TupleValidator;

/**
 * GetImportedForeignKeysRequest
 *
 * @since 8.0
 */
public class GetImportedForeignKeysRequest extends MetadataRequest {
    
    public static final String NAME = JdbcPlugin.Util.getString("GetImportedForeignKeysRequestName"); //$NON-NLS-1$
    private static final String METHOD_NAME = "getImportedKeys"; //$NON-NLS-1$

    /**
     * Construct an instance of GetImportedForeignKeysRequest.
     * @param name
     * @param target
     * @param methodName
     * @param params
     */
    public GetImportedForeignKeysRequest( final DatabaseMetaData metadata,
                              final String catalogNamePattern, 
                              final String schemaNamePattern,
                              final String tableNamePattern ) {
        super(NAME, metadata, METHOD_NAME, 
              new Object[]{catalogNamePattern,schemaNamePattern,tableNamePattern});
    }
    
    /** 
     * This method is overridden to optimize performance.
     * @see org.teiid.designer.jdbc.data.MethodRequest#performInvocation(org.teiid.designer.jdbc.data.Response)
     * @since 4.2
     */
    @Override
    protected IStatus performInvocation(final Response results) {
        // Override to optimize ...
        final DatabaseMetaData dbmd = this.getDatabaseMetaData();
        ResultSet resultSet = null;
        IStatus status = null;
        try {
            final String catalogPattern = (String)getParameters()[0];
            final String schemaPattern  = (String)getParameters()[1];
            final String tablePattern   = (String)getParameters()[2];
            resultSet = dbmd.getImportedKeys(catalogPattern,schemaPattern,tablePattern);
            Response.addResults(results,resultSet,this.isMetadataRequested(),
                                new TupleValidator() {
                                    @Override
									public boolean isTupleValid(List tuple) {
                                        if(schemaPattern == null || schemaPattern.length() == 0) {
                                            return true;
                                        }                               
                                        String schemaValue = (String)tuple.get(1);
                                        if(schemaValue != null
                                            && schemaPattern.equalsIgnoreCase(schemaValue)){
                                            
                                            return true;
                                        }
                                        return false;
                                    }
                                });
        } catch ( SQLException e ) {
            status = JdbcUtil.createIStatus(e,e.getLocalizedMessage());
        } finally {
            if ( resultSet != null ) {
                try {
                    resultSet.close();
                } catch (SQLException e1) {
                }
            }
        }
        return status;
    }
    
}