-
Notifications
You must be signed in to change notification settings - Fork 20
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #466 from niveathika/2201.2.x
[2201.2.x Backport] Add support for XA transaction
- Loading branch information
Showing
12 changed files
with
229 additions
and
21 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,166 @@ | ||
| // Copyright (c) 2022 WSO2 LLC. (http://www.wso2.org) All Rights Reserved. | ||
| // | ||
| // WSO2 LLC. 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. | ||
|
|
||
| import ballerina/test; | ||
| import ballerina/sql; | ||
|
|
||
| int DBCLIENT2_PORT = 1522; | ||
|
|
||
| type XAResultCount record { | ||
| int COUNTVAL; | ||
| }; | ||
|
|
||
| // Following test cases are disabled, since they need 2 db instances to run. | ||
| // Due to a constraint 2 Oracle Docker containers cannot be run on the same machine. | ||
| // Currently verified it manually with a local instance running on port 1522 and docker image running on port 1521 | ||
|
|
||
| @test:Config { | ||
| enable: false, | ||
| groups: ["transaction", "xa-transaction"] | ||
| } | ||
| function testXATransactionSuccess() returns error? { | ||
| Client dbClient1 = check new (HOST, USER, PASSWORD, DATABASE, PORT, connectionPool = {maxOpenConnections: 1}, | ||
| options = { | ||
| useXADatasource: true | ||
| } | ||
| ); | ||
| Client dbClient2 = check new (HOST, USER, PASSWORD, DATABASE, DBCLIENT2_PORT, connectionPool = {maxOpenConnections: 1}, | ||
| options = { | ||
| useXADatasource: true | ||
| } | ||
| ); | ||
|
|
||
| transaction { | ||
| _ = check dbClient1->execute(`insert into Customers (customerId, name, creditLimit, country) | ||
| values (1, 'Anne', 1000, 'UK')`); | ||
| _ = check dbClient2->execute(`insert into Salary (id, value) values (1, 1000)`); | ||
| check commit; | ||
| } on fail { | ||
| test:assertFail(msg = "Transaction failed"); | ||
| } | ||
|
|
||
| int count1 = check getCustomerCount(dbClient1, 1); | ||
| int count2 = check getSalaryCount(dbClient2, 1); | ||
| test:assertEquals(count1, 1, "First transaction failed"); | ||
| test:assertEquals(count2, 1, "Second transaction failed"); | ||
|
|
||
| check dbClient1.close(); | ||
| check dbClient2.close(); | ||
| } | ||
|
|
||
| @test:Config { | ||
| enable: false, | ||
| groups: ["transaction", "xa-transaction"] | ||
| } | ||
| function testXATransactionFailureWithDataSource() returns error? { | ||
| Client dbClient1 = check new (HOST, USER, PASSWORD, DATABASE, PORT, connectionPool = {maxOpenConnections: 1}, | ||
| options = { | ||
| useXADatasource: true | ||
| } | ||
| ); | ||
| Client dbClient2 = check new (HOST, USER, PASSWORD, DATABASE, DBCLIENT2_PORT, connectionPool = {maxOpenConnections: 1}, | ||
| options = { | ||
| useXADatasource: true | ||
| } | ||
| ); | ||
|
|
||
| transaction { | ||
| // Intentionally fail first statement | ||
| _ = check dbClient1->execute(`insert into CustomersTrx (customerId, name, creditLimit, country) | ||
| values (30, 'Anne', 1000, 'UK')`); | ||
| _ = check dbClient2->execute(`insert into Salary (id, value) values (10, 1000)`); | ||
| check commit; | ||
| } on fail error e { | ||
| test:assertTrue(e.message().includes("Duplicate"), msg = "Transaction failed as expected"); | ||
| } | ||
|
|
||
| int count1 = check getCustomerTrxCount(dbClient1, 30); | ||
| int count2 = check getSalaryCount(dbClient2, 20); | ||
| test:assertEquals(count1, 1, "First transaction should have failed"); | ||
| test:assertEquals(count2, 0, "Second transaction should not have been executed"); | ||
|
|
||
| check dbClient1.close(); | ||
| check dbClient2.close(); | ||
| } | ||
|
|
||
| @test:Config { | ||
| enable: false, | ||
| groups: ["transaction", "xa-transaction"] | ||
| } | ||
| function testXATransactionPartialSuccessWithDataSource() returns error? { | ||
| Client dbClient1 = check new (HOST, USER, PASSWORD, DATABASE, PORT, connectionPool = {maxOpenConnections: 1}, | ||
| options = { | ||
| useXADatasource: true | ||
| } | ||
| ); | ||
| Client dbClient2 = check new (HOST, USER, PASSWORD, DATABASE, DBCLIENT2_PORT, connectionPool = {maxOpenConnections: 1}, | ||
| options = { | ||
| useXADatasource: true | ||
| } | ||
| ); | ||
|
|
||
| transaction { | ||
| _ = check dbClient1->execute(`insert into Customers (customerId, name, creditLimit, country) | ||
| values (30, 'Anne', 1000, 'UK')`); | ||
| // Intentionally fail second statement | ||
| _ = check dbClient2->execute(`insert into SalaryTrx (id, value) values (20, 1000)`); | ||
| check commit; | ||
| } on fail error e { | ||
| test:assertTrue(e.message().includes("Duplicate"), msg = "Transaction failed as expected"); | ||
| } | ||
|
|
||
| int count1 = check getCustomerCount(dbClient1, 30); | ||
| int count2 = check getSalaryTrxCount(dbClient2, 20); | ||
| test:assertEquals(count1, 0, "First transaction is not rolledback"); | ||
| test:assertEquals(count2, 1, "Second transaction has succeeded"); | ||
|
|
||
| check dbClient1.close(); | ||
| check dbClient2.close(); | ||
| } | ||
|
|
||
| isolated function getCustomerCount(Client dbClient, int id) returns int|error { | ||
| stream<XAResultCount, sql:Error?> streamData = dbClient->query(`Select COUNT(*) as | ||
| countVal from Customers where customerId = ${id}`); | ||
| return getResult(streamData); | ||
| } | ||
|
|
||
| isolated function getCustomerTrxCount(Client dbClient, int id) returns int|error { | ||
| stream<XAResultCount, sql:Error?> streamData = dbClient->query(`Select COUNT(*) as | ||
| countVal from CustomersTrx where customerId = ${id}`); | ||
| return getResult(streamData); | ||
| } | ||
|
|
||
| isolated function getSalaryCount(Client dbClient, int id) returns int|error { | ||
| stream<XAResultCount, sql:Error?> streamData = dbClient->query(`Select COUNT(*) as countval | ||
| from Salary where id = ${id}`); | ||
| return getResult(streamData); | ||
| } | ||
|
|
||
| isolated function getSalaryTrxCount(Client dbClient, int id) returns int|error { | ||
| stream<XAResultCount, sql:Error?> streamData = dbClient->query(`Select COUNT(*) as countval | ||
| from SalaryTrx where id = ${id}`); | ||
| return getResult(streamData); | ||
| } | ||
|
|
||
| isolated function getResult(stream<XAResultCount, sql:Error?> streamData) returns int|error { | ||
| record {|XAResultCount value;|}? data = check streamData.next(); | ||
| check streamData.close(); | ||
| XAResultCount? value = data?.value; | ||
| if value is XAResultCount { | ||
| return value.COUNTVAL; | ||
| } | ||
| return 0; | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
16 changes: 16 additions & 0 deletions
16
ballerina/tests/resources/sql-scripts/transaction/xa-transaction-test-data-1.sql
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,16 @@ | ||
| CREATE TABLE Customers( | ||
| customerId NUMBER, | ||
| name VARCHAR(300), | ||
| creditLimit DOUBLE PRECISION, | ||
| country VARCHAR(300) | ||
| ); | ||
|
|
||
| CREATE TABLE CustomersTrx( | ||
| customerId INTEGER, | ||
| name VARCHAR(300), | ||
| creditLimit DOUBLE PRECISION, | ||
| country VARCHAR(300), | ||
| PRIMARY KEY (customerId) | ||
| ); | ||
|
|
||
| INSERT INTO CustomersTrx VALUES (30, 'Oliver', 200000, 'UK'); |
13 changes: 13 additions & 0 deletions
13
ballerina/tests/resources/sql-scripts/transaction/xa-transaction-test-data-2.sql
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,13 @@ | ||
| CREATE TABLE Salary ( | ||
| ID INTEGER, | ||
| VALUE DOUBLE PRECISION | ||
| ); | ||
|
|
||
| CREATE TABLE SalaryTrx ( | ||
| ID INTEGER, | ||
| VALUE DOUBLE PRECISION, | ||
| PRIMARY KEY (ID) | ||
| ); | ||
|
|
||
| INSERT INTO SalaryTrx VALUES (20, 30000); | ||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters