Application: gvSIG desktop » gvSIG data provider for MS SQLServer

package org.gvsig.mssqlserver.dal.operations;

import org.gvsig.fmap.dal.store.jdbc2.spi.operations.*;

import java.sql.Connection;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Statement;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.store.jdbc.exception.JDBCSQLException;

import org.gvsig.fmap.dal.store.jdbc2.JDBCHelper;

import org.gvsig.fmap.dal.store.jdbc2.JDBCUtils;

import org.gvsig.fmap.dal.store.jdbc2.OperationsFactory.TableReference;

public class MSSQLServerCanModifyTableOperation extends CanModifyTableOperation {

    public MSSQLServerCanModifyTableOperation(

            JDBCHelper helper

        ) {

        this(helper, null);

    }

    public MSSQLServerCanModifyTableOperation(

            JDBCHelper helper,

            TableReference table

        ) {

        super(helper,table);

    }

    private String getFullTableName( 

            TableReference table

    ) {

        StringBuilder builder = new StringBuilder();

        if( table.hasDatabase() ) {

            builder.append(table.getDatabase());

        }

        if( table.hasSchema() ) {

            if( builder.length()>0 ) {

                builder.append(".");

            }

            builder.append(table.getSchema());

        }

        if( table.hasTable() ) {

            if( builder.length()>0 ) {

                builder.append(".");

            }

            builder.append(table.getTable());

        }

        return builder.toString();

    }

    @Override

    public boolean canModifyTable(Connection conn,

            TableReference table

        ) throws DataException {

        String sql = "SELECT count(*) " + 

                     "FROM fn_my_permissions('"+getFullTableName(table)+"', 'OBJECT') " +

                     "WHERE permission_name in ('UPDATE', 'INSERT', 'DELETE') AND subentity_name = ''";

        Statement st = null;

        ResultSet rs = null;

        try {

            st = conn.createStatement();

            rs = JDBCUtils.executeQuery(st, sql);

            if (!rs.next()) {

                return false;

            }

            return rs.getInt(1)==3;

        } catch (SQLException ex) {

            throw new JDBCSQLException(ex);

        } finally {

            JDBCUtils.closeQuietly(st);

            JDBCUtils.closeQuietly(rs);

        }

    }

}