Skip to content

Commit

Permalink
Revert "Block cross-db 'SELECT ... INTO table FROM ...' statement (ba…
Browse files Browse the repository at this point in the history 
…belfish-for-postgresql#2744) (babelfish-for-postgresql#2780)"

This reverts commit 0b401ee.
  • Loading branch information
ANJU BHARTI committed Nov 13, 2024
1 parent 55aec82 commit b9d7a05
Show file tree
Hide file tree
Showing 4 changed files with 0 additions and 581 deletions.
16 changes: 0 additions & 16 deletions contrib/babelfishpg_tsql/src/tsqlIface.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1790,22 +1790,6 @@ class tsqlBuilder : public tsqlCommonMutator
if (is_schema_specified)
stmt->is_schema_specified = true;

if (is_cross_db)
{
if (ctx->select_statement_standalone() &&
ctx->select_statement_standalone()->select_statement() &&
ctx->select_statement_standalone()->select_statement()->query_expression() &&
ctx->select_statement_standalone()->select_statement()->query_expression()->query_specification() &&
ctx->select_statement_standalone()->select_statement()->query_expression()->query_specification()->INTO() &&
ctx->select_statement_standalone()->select_statement()->query_expression()->query_specification()->table_name())
{
throw PGErrorWrapperException(ERROR,
ERRCODE_FEATURE_NOT_SUPPORTED,
"cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.",
getLineAndPos(ctx->select_statement_standalone()));
}
}

if (is_compiling_create_function())
{
/* select without destination should be blocked. We can use already information about desitnation, which is already processed. */
Expand Down
203 changes: 0 additions & 203 deletions test/JDBC/expected/BABEL-CROSS-DB.out
View file
Original file line number Diff line number Diff line change
Expand Up @@ -523,206 +523,3 @@ GO

DROP DATABASE db2;
GO

-- BABEL-4934 Test blocking cross-db SELECT-INTO statement
CREATE DATABASE db_4934_1;
GO

CREATE DATABASE db_4934_2;
GO

USE master;
GO

CREATE TABLE t1(a int);
GO

CREATE TABLE t2(b int);
GO

USE db_4934_1;
GO

CREATE TABLE t1(a int);
GO

CREATE TABLE t2(b int);
GO

USE db_4934_2;
GO

CREATE TABLE t1(a int);
GO

CREATE TABLE t2(b int);
GO

USE db_4934_1;
GO

-- #4934.1 It should be blocked
SELECT * INTO t222 FROM master.dbo.t1;
GO
~~ERROR (Code: 33557097)~~

~~ERROR (Message: cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.)~~


SELECT * INTO t222 FROM master..t1;
GO
~~ERROR (Code: 33557097)~~

~~ERROR (Message: cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.)~~


SELECT * INTO t222 FROM db_4934_2..t1;
GO
~~ERROR (Code: 33557097)~~

~~ERROR (Message: cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.)~~


SELECT * INTO t222 FROM db_4934_2.dbo.t1;
GO
~~ERROR (Code: 33557097)~~

~~ERROR (Message: cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.)~~


SELECT * INTO t222 FROM db_4934_2..t1, t1;
GO
~~ERROR (Code: 33557097)~~

~~ERROR (Message: cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.)~~


SELECT * INTO t222 FROM t1, db_4934_1..t2;
GO

SELECT * INTO t222 FROM master..t1, db_4934_2..t1;
GO
~~ERROR (Code: 33557097)~~

~~ERROR (Message: cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.)~~


SELECT * INTO t222 FROM (SELECT * FROM master..t1) AS Subquery;
GO
~~ERROR (Code: 33557097)~~

~~ERROR (Message: cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.)~~


SELECT * INTO t222 FROM (SELECT * FROM t1, master..t1) AS Subquery;
GO
~~ERROR (Code: 33557097)~~

~~ERROR (Message: cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.)~~


SELECT * INTO t222 FROM (SELECT *, (SELECT * FROM master..t1) FROM t1) AS Subquery;
GO
~~ERROR (Code: 33557097)~~

~~ERROR (Message: cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.)~~


SELECT * INTO t222 FROM (SELECT *, (SELECT * FROM t1) FROM master..t3) AS Subquery;
GO
~~ERROR (Code: 33557097)~~

~~ERROR (Message: cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.)~~


SELECT * INTO master..t222 FROM db_4934_2..t1;
GO
~~ERROR (Code: 33557097)~~

~~ERROR (Message: cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.)~~


SELECT * INTO db_4934_1..t222 FROM db_4934_2..t1;
GO
~~ERROR (Code: 33557097)~~

~~ERROR (Message: cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.)~~


SELECT * INTO master..t222 FROM master..t1;
GO
~~ERROR (Code: 33557097)~~

~~ERROR (Message: cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.)~~


-- #4934.2 Following statements will succeed across same database
SELECT * INTO t3 FROM db_4934_1.dbo.t1;
GO

SELECT * INTO t4 FROM dbo.t1;
GO

SELECT * INTO t5 FROM db_4934_1..t1;
GO

SELECT * INTO t6 FROM db_4934_1..t1, db_4934_1..t2;
GO

SELECT * INTO t7 FROM (SELECT * FROM db_4934_1..t1) AS Subquery;
GO

SELECT * INTO t8 FROM (SELECT *, (SELECT * FROM db_4934_1..t2) FROM db_4934_1..t1) AS Subquery;
GO

-- validate the access
SELECT * FROM t3, t4, t5, t6, t7, t8;
GO
~~START~~
int#!#int#!#int#!#int#!#int#!#int#!#int#!#int
~~END~~


DROP TABLE t3, t4, t5, t6, t7, t8;
GO

-- #4934.3 Temporary table should not be blocked
SELECT * INTO #t1 FROM db_4934_1.dbo.t1;
GO

SELECT * INTO #t2 FROM (SELECT * FROM db_4934_1.dbo.t1) AS Subquery;
GO

-- validate the access
SELECT * FROM #t1, #t2;
GO
~~START~~
int#!#int
~~END~~


DROP TABLE #t1, #t2;
GO

-- Even though this is same as above statement, this will still fail since
-- internally it considers as cross-db statement (This behaviour is general to
-- all of the applicable DMLs)
SELECT * INTO tempdb..#t3 FROM db_4934_1.dbo.t1;
GO
~~ERROR (Code: 33557097)~~

~~ERROR (Message: cross-db 'SELECT INTO' statement is not supported in Babelfish. As a workaround, consider running CREATE TABLE and INSERT-SELECT statements.)~~



USE master;
GO

DROP TABLE t1, t2;
GO

DROP DATABASE db_4934_1;
GO

DROP DATABASE db_4934_2;
GO
Loading

0 comments on commit b9d7a05

Please sign in to comment.