xref: /trunk/main/connectivity/java/sdbc_postgresql/src/com/sun/star/sdbcx/comp/postgresql/PostgresqlTables.java (revision 60ba3965)

/**************************************************************
 *
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 *
 *************************************************************/

package com.sun.star.sdbcx.comp.postgresql;

import java.util.List;

import org.apache.openoffice.comp.sdbc.dbtools.comphelper.CompHelper;
import org.apache.openoffice.comp.sdbc.dbtools.sdbcx.OContainer;
import org.apache.openoffice.comp.sdbc.dbtools.sdbcx.descriptors.SdbcxTableDescriptor;
import org.apache.openoffice.comp.sdbc.dbtools.util.ComposeRule;
import org.apache.openoffice.comp.sdbc.dbtools.util.DbTools;
import org.apache.openoffice.comp.sdbc.dbtools.util.DbTools.NameComponents;
import org.apache.openoffice.comp.sdbc.dbtools.util.PropertyIds;
import org.apache.openoffice.comp.sdbc.dbtools.util.StandardSQLState;

import com.sun.star.beans.UnknownPropertyException;
import com.sun.star.beans.XPropertySet;
import com.sun.star.container.ElementExistException;
import com.sun.star.container.NoSuchElementException;
import com.sun.star.lang.IllegalArgumentException;
import com.sun.star.lang.WrappedTargetException;
import com.sun.star.sdbc.SQLException;
import com.sun.star.sdbc.XDatabaseMetaData;
import com.sun.star.sdbc.XResultSet;
import com.sun.star.sdbc.XRow;
import com.sun.star.sdbc.XStatement;
import com.sun.star.sdbcx.XDrop;
import com.sun.star.uno.Any;
import com.sun.star.uno.AnyConverter;
import com.sun.star.uno.UnoRuntime;

public class PostgresqlTables extends OContainer {
    private XDatabaseMetaData metadata;
    private PostgresqlCatalog catalog;

    public PostgresqlTables(Object lock, XDatabaseMetaData metadata, PostgresqlCatalog catalog, List<String> names) throws ElementExistException {
        super(lock, true, names);
        this.metadata = metadata;
        this.catalog = catalog;
    }

    @Override
    public XPropertySet createObject(String name) throws SQLException {
        NameComponents nameComponents = DbTools.qualifiedNameComponents(metadata, name, ComposeRule.InDataManipulation);
        Object queryCatalog = nameComponents.getCatalog().isEmpty() ? Any.VOID : nameComponents.getCatalog();
        XPropertySet ret = null;
        XResultSet results = null;
        try {
            results = metadata.getTables(
                    queryCatalog, nameComponents.getSchema(), nameComponents.getTable(), new String[] { "VIEW", "TABLE", "%" });
            if (results != null) {
                XRow row = UnoRuntime.queryInterface(XRow.class, results);
                if (results.next()) {
                    String type = row.getString(4);
                    String remarks = row.getString(5);
                    ret = new PostgresqlTable(metadata.getConnection(), this, nameComponents.getTable(),
                            nameComponents.getCatalog(), nameComponents.getSchema(), remarks, type);
                }
            }
        } finally {
            CompHelper.disposeComponent(results);
        }
        return ret;

    }

    @Override
    public void dropObject(int index, String name) throws SQLException {
        try {
            Object object = getObject(index);

            NameComponents nameComponents = DbTools.qualifiedNameComponents(metadata, name, ComposeRule.InDataManipulation);

            boolean isView = false;
            XPropertySet propertySet = UnoRuntime.queryInterface(XPropertySet.class, object);
            if (propertySet != null) {
                isView = AnyConverter.toString(propertySet.getPropertyValue(PropertyIds.TYPE.name)).equals("VIEW");
            }

            String composedName = DbTools.composeTableName(metadata, nameComponents.getCatalog(), nameComponents.getSchema(), nameComponents.getTable(),
                    true, ComposeRule.InDataManipulation);
            if (isView) {
                XDrop dropView = UnoRuntime.queryInterface(XDrop.class, catalog.getViews());
                String unquotedName = DbTools.composeTableName(metadata, nameComponents.getCatalog(), nameComponents.getSchema(), nameComponents.getTable(),
                        false, ComposeRule.InDataManipulation);
                dropView.dropByName(unquotedName);
                return;
            }

            String sql = "DROP TABLE " + composedName;

            XStatement statement = null;
            try {
                statement = metadata.getConnection().createStatement();
                statement.execute(sql);
            } finally {
                CompHelper.disposeComponent(statement);
            }
        } catch (IllegalArgumentException | UnknownPropertyException | WrappedTargetException | NoSuchElementException wrappedTargetException) {
            throw new SQLException("Error", this, StandardSQLState.SQL_GENERAL_ERROR.text(), 0, wrappedTargetException);
        }
    }

    @Override
    public void impl_refresh() {
        catalog.refreshObjects();
    }

    @Override
    public XPropertySet createDescriptor() {
        return new SdbcxTableDescriptor(true);
    }

    @Override
    public XPropertySet appendObject(String name, XPropertySet descriptor) throws SQLException {
        createTable(descriptor);
        return createObject(name);
    }

    void createTable(XPropertySet descriptor) throws SQLException {
        XStatement statement = null;
        try {
            String sql = DbTools.createSqlCreateTableStatement(descriptor, metadata.getConnection(), null, "(M,D)");
            statement = metadata.getConnection().createStatement();
            statement.execute(sql);
        } finally {
            CompHelper.disposeComponent(statement);
        }
    }
}