You are viewing a plain text version of this content. The canonical link for it is here.
Posted to github@arrow.apache.org by "WillAyd (via GitHub)" <gi...@apache.org> on 2023/06/09 21:00:57 UTC
[GitHub] [arrow-adbc] WillAyd opened a new pull request, #757: feat(c/driver/postgresql): Implement Foreign Key information for GetObjects
WillAyd opened a new pull request, #757:
URL: https://github.com/apache/arrow-adbc/pull/757
(no comment)
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [arrow-adbc] lidavidm commented on pull request #757: feat(c/driver/postgresql): Implement Foreign Key information for GetObjects
Posted by "lidavidm (via GitHub)" <gi...@apache.org>.
lidavidm commented on PR #757:
URL: https://github.com/apache/arrow-adbc/pull/757#issuecomment-1587696885
(...though it would be very unfortunate to have duplicate APIs for this.)
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [arrow-adbc] lidavidm commented on pull request #757: feat(c/driver/postgresql): Implement Foreign Key information for GetObjects
Posted by "lidavidm (via GitHub)" <gi...@apache.org>.
lidavidm commented on PR #757:
URL: https://github.com/apache/arrow-adbc/pull/757#issuecomment-1587292947
I kicked the job but if it doesn't pass I don't think we have to consider it a blocker (Dewey has been working on the R stuff anyways)
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [arrow-adbc] lidavidm commented on pull request #757: feat(c/driver/postgresql): Implement Foreign Key information for GetObjects
Posted by "lidavidm (via GitHub)" <gi...@apache.org>.
lidavidm commented on PR #757:
URL: https://github.com/apache/arrow-adbc/pull/757#issuecomment-1587696623
Frankly given https://github.com/apache/arrow-adbc/issues/320 and https://github.com/apache/arrow-adbc/issues/621 I'm wondering if we shouldn't retire GetObjects in favor of non-nested alternatives. The original design perhaps favored DuckDB and its ability to query nested data too much.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [arrow-adbc] lidavidm commented on a diff in pull request #757: feat(c/driver/postgresql): Implement Foreign Key information for GetObjects
Posted by "lidavidm (via GitHub)" <gi...@apache.org>.
lidavidm commented on code in PR #757:
URL: https://github.com/apache/arrow-adbc/pull/757#discussion_r1226627001
##########
c/driver/postgresql/postgresql_test.cc:
##########
@@ -402,6 +402,193 @@ TEST_F(PostgresConnectionTest, GetObjectsGetAllFindsPrimaryKey) {
ASSERT_TRUE(seen_primary_key) << "could not find primary key for adbc_pkey_test";
}
+TEST_F(PostgresConnectionTest, GetObjectsGetAllFindsForeignKey) {
Review Comment:
Could possibly factor out a helper that just drops the given table + runs the given DDL so you can `ASSERT_NO_FATAL_FAILURE(CreateTable("adbc_fkey_test", "CREATE TABLE ..."));`
##########
c/driver/postgresql/postgresql_test.cc:
##########
@@ -402,6 +402,193 @@ TEST_F(PostgresConnectionTest, GetObjectsGetAllFindsPrimaryKey) {
ASSERT_TRUE(seen_primary_key) << "could not find primary key for adbc_pkey_test";
}
+TEST_F(PostgresConnectionTest, GetObjectsGetAllFindsForeignKey) {
Review Comment:
The nested loop is...messy, I have had some ideas about how to simplify this sort of thing (basically, I think Arrow could use an IDL like Protobuf, where you can take a predefined schema and generate higher-level accessors/iterators/validators)
##########
c/driver/postgresql/postgresql_test.cc:
##########
@@ -402,6 +402,193 @@ TEST_F(PostgresConnectionTest, GetObjectsGetAllFindsPrimaryKey) {
ASSERT_TRUE(seen_primary_key) << "could not find primary key for adbc_pkey_test";
}
+TEST_F(PostgresConnectionTest, GetObjectsGetAllFindsForeignKey) {
Review Comment:
I've also thought about whether some complicated tests could belong in Python instead (with the caveats around that development loop)
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [arrow-adbc] WillAyd commented on pull request #757: feat(c/driver/postgresql): Implement Foreign Key information for GetObjects
Posted by "WillAyd (via GitHub)" <gi...@apache.org>.
WillAyd commented on PR #757:
URL: https://github.com/apache/arrow-adbc/pull/757#issuecomment-1587638539
I think defer any changes to subsequent PRs; not pushing those off forever just think it would be too much to solve alongside implementing this
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [arrow-adbc] lidavidm merged pull request #757: feat(c/driver/postgresql): Implement Foreign Key information for GetObjects
Posted by "lidavidm (via GitHub)" <gi...@apache.org>.
lidavidm merged PR #757:
URL: https://github.com/apache/arrow-adbc/pull/757
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [arrow-adbc] WillAyd commented on a diff in pull request #757: feat(c/driver/postgresql): Implement Foreign Key information for GetObjects
Posted by "WillAyd (via GitHub)" <gi...@apache.org>.
WillAyd commented on code in PR #757:
URL: https://github.com/apache/arrow-adbc/pull/757#discussion_r1226774089
##########
c/driver/postgresql/postgresql_test.cc:
##########
@@ -402,6 +402,193 @@ TEST_F(PostgresConnectionTest, GetObjectsGetAllFindsPrimaryKey) {
ASSERT_TRUE(seen_primary_key) << "could not find primary key for adbc_pkey_test";
}
+TEST_F(PostgresConnectionTest, GetObjectsGetAllFindsForeignKey) {
Review Comment:
Yea I don't like the loops. The IDL would be nice. Also been toying with some util helpers to abstract it in another branch, just hasn't fully materialized yet
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [arrow-adbc] lidavidm commented on a diff in pull request #757: feat(c/driver/postgresql): Implement Foreign Key information for GetObjects
Posted by "lidavidm (via GitHub)" <gi...@apache.org>.
lidavidm commented on code in PR #757:
URL: https://github.com/apache/arrow-adbc/pull/757#discussion_r1226572547
##########
c/driver/postgresql/postgresql_test.cc:
##########
@@ -402,6 +402,193 @@ TEST_F(PostgresConnectionTest, GetObjectsGetAllFindsPrimaryKey) {
ASSERT_TRUE(seen_primary_key) << "could not find primary key for adbc_pkey_test";
}
+TEST_F(PostgresConnectionTest, GetObjectsGetAllFindsForeignKey) {
+ ASSERT_THAT(AdbcConnectionNew(&connection, &error), IsOkStatus(&error));
+ ASSERT_THAT(AdbcConnectionInit(&connection, &database, &error), IsOkStatus(&error));
+
+ if (!quirks()->supports_get_objects()) {
+ GTEST_SKIP();
+ }
+
+ ASSERT_THAT(quirks()->DropTable(&connection, "adbc_fkey_test", &error),
+ IsOkStatus(&error));
+ ASSERT_THAT(quirks()->DropTable(&connection, "adbc_fkey_test_base", &error),
+ IsOkStatus(&error));
+
+ struct AdbcStatement statement;
+ ASSERT_THAT(AdbcStatementNew(&connection, &statement, &error), IsOkStatus(&error));
+ {
+ ASSERT_THAT(
+ AdbcStatementSetSqlQuery(&statement,
+ "CREATE TABLE adbc_fkey_test_base (id1 INT, id2 INT, "
+ "PRIMARY KEY (id1, id2))",
+ &error),
+ IsOkStatus(&error));
+ adbc_validation::StreamReader reader;
+ ASSERT_THAT(AdbcStatementExecuteQuery(&statement, &reader.stream.value,
+ &reader.rows_affected, &error),
+ IsOkStatus(&error));
+ ASSERT_EQ(reader.rows_affected, 0);
+ ASSERT_NO_FATAL_FAILURE(reader.GetSchema());
+ ASSERT_NO_FATAL_FAILURE(reader.Next());
+ ASSERT_EQ(reader.array->release, nullptr);
+ }
+
+ ASSERT_THAT(AdbcStatementNew(&connection, &statement, &error), IsOkStatus(&error));
+ {
+ ASSERT_THAT(AdbcStatementSetSqlQuery(
+ &statement,
+ "CREATE TABLE adbc_fkey_test (fid1 INT, fid2 INT, "
+ "FOREIGN KEY (fid1, fid2) REFERENCES adbc_fkey_test_base(id1, id2))",
+ &error),
+ IsOkStatus(&error));
+ adbc_validation::StreamReader reader;
+ ASSERT_THAT(AdbcStatementExecuteQuery(&statement, &reader.stream.value,
+ &reader.rows_affected, &error),
+ IsOkStatus(&error));
+ ASSERT_EQ(reader.rows_affected, 0);
+ ASSERT_NO_FATAL_FAILURE(reader.GetSchema());
+ ASSERT_NO_FATAL_FAILURE(reader.Next());
+ ASSERT_EQ(reader.array->release, nullptr);
+ }
+
+ adbc_validation::StreamReader reader;
+ ASSERT_THAT(
+ AdbcConnectionGetObjects(&connection, ADBC_OBJECT_DEPTH_ALL, nullptr, nullptr,
+ nullptr, nullptr, nullptr, &reader.stream.value, &error),
+ IsOkStatus(&error));
+ ASSERT_NO_FATAL_FAILURE(reader.GetSchema());
+ ASSERT_NO_FATAL_FAILURE(reader.Next());
+ ASSERT_NE(nullptr, reader.array->release);
+ ASSERT_GT(reader.array->length, 0);
+
+ bool seen_fid1 = false;
+ bool seen_fid2 = false;
+
+ struct ArrowArrayView* catalog_db_schemas_list = reader.array_view->children[1];
+ struct ArrowArrayView* catalog_db_schemas_items = catalog_db_schemas_list->children[0];
+ struct ArrowArrayView* db_schema_name_col = catalog_db_schemas_items->children[0];
+ struct ArrowArrayView* db_schema_tables_col = catalog_db_schemas_items->children[1];
+
+ struct ArrowArrayView* schema_table_items = db_schema_tables_col->children[0];
+ struct ArrowArrayView* table_name_col = schema_table_items->children[0];
+ // struct ArrowArrayView* table_columns_col = schema_table_items->children[2];
+ struct ArrowArrayView* table_constraints_col = schema_table_items->children[3];
+
+ // struct ArrowArrayView* table_columns_items = table_columns_col->children[0];
+ // struct ArrowArrayView* column_name_col = table_columns_items->children[0];
+
+ struct ArrowArrayView* table_constraints_items = table_constraints_col->children[0];
+ struct ArrowArrayView* constraint_name_col = table_constraints_items->children[0];
+ struct ArrowArrayView* constraint_type_col = table_constraints_items->children[1];
+ // struct ArrowArrayView* constraint_column_names_col =
+ // table_constraints_items->children[2];
+
+ struct ArrowArrayView* constraint_column_usages_col =
+ table_constraints_items->children[3];
+ struct ArrowArrayView* constraint_column_usage_items =
+ constraint_column_usages_col->children[0];
+ struct ArrowArrayView* fk_catalog_col = constraint_column_usage_items->children[0];
+ struct ArrowArrayView* fk_db_schema_col = constraint_column_usage_items->children[1];
+ struct ArrowArrayView* fk_table_col = constraint_column_usage_items->children[2];
+ struct ArrowArrayView* fk_column_name_col = constraint_column_usage_items->children[3];
+
+ do {
+ for (int64_t catalog_idx = 0; catalog_idx < reader.array->length; catalog_idx++) {
+ ArrowStringView db_name =
+ ArrowArrayViewGetStringUnsafe(reader.array_view->children[0], catalog_idx);
+ auto db_str = std::string(db_name.data, db_name.size_bytes);
+
+ auto schema_list_start =
+ ArrowArrayViewListChildOffset(catalog_db_schemas_list, catalog_idx);
+ auto schema_list_end =
+ ArrowArrayViewListChildOffset(catalog_db_schemas_list, catalog_idx + 1);
+
+ if (db_str == "postgres") {
+ for (auto db_schemas_index = schema_list_start;
+ db_schemas_index < schema_list_end; db_schemas_index++) {
+ ArrowStringView schema_name =
+ ArrowArrayViewGetStringUnsafe(db_schema_name_col, db_schemas_index);
+ auto schema_str = std::string(schema_name.data, schema_name.size_bytes);
+ if (schema_str == "public") {
+ for (auto tables_index = ArrowArrayViewListChildOffset(db_schema_tables_col,
+ db_schemas_index);
+ tables_index < ArrowArrayViewListChildOffset(db_schema_tables_col,
+ db_schemas_index + 1);
+ tables_index++) {
+ ArrowStringView table_name =
+ ArrowArrayViewGetStringUnsafe(table_name_col, tables_index);
+ auto table_str = std::string(table_name.data, table_name.size_bytes);
+ if (table_str == "adbc_fkey_test") {
+ for (auto constraints_index = ArrowArrayViewListChildOffset(
+ table_constraints_col, tables_index);
+ constraints_index < ArrowArrayViewListChildOffset(
+ table_constraints_col, tables_index + 1);
+ constraints_index++) {
+ ArrowStringView constraint_name = ArrowArrayViewGetStringUnsafe(
+ constraint_name_col, constraints_index);
+ auto constraint_name_str =
+ std::string(constraint_name.data, constraint_name.size_bytes);
+ ArrowStringView constraint_type = ArrowArrayViewGetStringUnsafe(
+ constraint_type_col, constraints_index);
+ auto constraint_type_str =
+ std::string(constraint_type.data, constraint_type.size_bytes);
+
+ if (constraint_type_str == "FOREIGN KEY") {
+ for (auto usage_index = ArrowArrayViewListChildOffset(
+ constraint_column_usages_col, constraints_index);
+ usage_index <
+ ArrowArrayViewListChildOffset(constraint_column_usages_col,
+ constraints_index + 1);
+ usage_index++) {
+ ArrowStringView fk_catalog_name =
+ ArrowArrayViewGetStringUnsafe(fk_catalog_col, usage_index);
+ auto fk_catalog_name_str =
+ std::string(fk_catalog_name.data, fk_catalog_name.size_bytes);
+ ArrowStringView fk_schema_name =
+ ArrowArrayViewGetStringUnsafe(fk_db_schema_col, usage_index);
+ auto fk_schema_name_str =
+ std::string(fk_schema_name.data, fk_schema_name.size_bytes);
+ ArrowStringView fk_table_name =
+ ArrowArrayViewGetStringUnsafe(fk_table_col, usage_index);
+ auto fk_table_name_str =
+ std::string(fk_table_name.data, fk_table_name.size_bytes);
+ ArrowStringView fk_column_name =
+ ArrowArrayViewGetStringUnsafe(fk_column_name_col, usage_index);
+ auto fk_column_name_str =
+ std::string(fk_column_name.data, fk_column_name.size_bytes);
+ if ((fk_catalog_name_str == "postgres") &&
+ (fk_schema_name_str == "public") &&
+ (fk_table_name_str == "adbc_fkey_test_base")) {
+ // TODO: the current implementation makes it so the length of
Review Comment:
Yes, if it's a foreign key it should be 1:1 so we know what column is being referenced by each column in the foreign key constraint.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [arrow-adbc] WillAyd commented on a diff in pull request #757: feat(c/driver/postgresql): Implement Foreign Key information for GetObjects
Posted by "WillAyd (via GitHub)" <gi...@apache.org>.
WillAyd commented on code in PR #757:
URL: https://github.com/apache/arrow-adbc/pull/757#discussion_r1224773064
##########
c/driver/postgresql/postgresql_test.cc:
##########
@@ -402,6 +402,193 @@ TEST_F(PostgresConnectionTest, GetObjectsGetAllFindsPrimaryKey) {
ASSERT_TRUE(seen_primary_key) << "could not find primary key for adbc_pkey_test";
}
+TEST_F(PostgresConnectionTest, GetObjectsGetAllFindsForeignKey) {
Review Comment:
I initially tried to combine this with the primary key test, but eventually moved away doing that as this was easier to express with its own set of tables. Still a good deal of shared elements that could be shared between the tests though
##########
c/driver/postgresql/postgresql_test.cc:
##########
@@ -402,6 +402,193 @@ TEST_F(PostgresConnectionTest, GetObjectsGetAllFindsPrimaryKey) {
ASSERT_TRUE(seen_primary_key) << "could not find primary key for adbc_pkey_test";
}
+TEST_F(PostgresConnectionTest, GetObjectsGetAllFindsForeignKey) {
+ ASSERT_THAT(AdbcConnectionNew(&connection, &error), IsOkStatus(&error));
+ ASSERT_THAT(AdbcConnectionInit(&connection, &database, &error), IsOkStatus(&error));
+
+ if (!quirks()->supports_get_objects()) {
+ GTEST_SKIP();
+ }
+
+ ASSERT_THAT(quirks()->DropTable(&connection, "adbc_fkey_test", &error),
+ IsOkStatus(&error));
+ ASSERT_THAT(quirks()->DropTable(&connection, "adbc_fkey_test_base", &error),
+ IsOkStatus(&error));
+
+ struct AdbcStatement statement;
+ ASSERT_THAT(AdbcStatementNew(&connection, &statement, &error), IsOkStatus(&error));
+ {
+ ASSERT_THAT(
+ AdbcStatementSetSqlQuery(&statement,
+ "CREATE TABLE adbc_fkey_test_base (id1 INT, id2 INT, "
+ "PRIMARY KEY (id1, id2))",
+ &error),
+ IsOkStatus(&error));
+ adbc_validation::StreamReader reader;
+ ASSERT_THAT(AdbcStatementExecuteQuery(&statement, &reader.stream.value,
+ &reader.rows_affected, &error),
+ IsOkStatus(&error));
+ ASSERT_EQ(reader.rows_affected, 0);
+ ASSERT_NO_FATAL_FAILURE(reader.GetSchema());
+ ASSERT_NO_FATAL_FAILURE(reader.Next());
+ ASSERT_EQ(reader.array->release, nullptr);
+ }
+
+ ASSERT_THAT(AdbcStatementNew(&connection, &statement, &error), IsOkStatus(&error));
+ {
+ ASSERT_THAT(AdbcStatementSetSqlQuery(
+ &statement,
+ "CREATE TABLE adbc_fkey_test (fid1 INT, fid2 INT, "
+ "FOREIGN KEY (fid1, fid2) REFERENCES adbc_fkey_test_base(id1, id2))",
+ &error),
+ IsOkStatus(&error));
+ adbc_validation::StreamReader reader;
+ ASSERT_THAT(AdbcStatementExecuteQuery(&statement, &reader.stream.value,
+ &reader.rows_affected, &error),
+ IsOkStatus(&error));
+ ASSERT_EQ(reader.rows_affected, 0);
+ ASSERT_NO_FATAL_FAILURE(reader.GetSchema());
+ ASSERT_NO_FATAL_FAILURE(reader.Next());
+ ASSERT_EQ(reader.array->release, nullptr);
+ }
+
+ adbc_validation::StreamReader reader;
+ ASSERT_THAT(
+ AdbcConnectionGetObjects(&connection, ADBC_OBJECT_DEPTH_ALL, nullptr, nullptr,
+ nullptr, nullptr, nullptr, &reader.stream.value, &error),
+ IsOkStatus(&error));
+ ASSERT_NO_FATAL_FAILURE(reader.GetSchema());
+ ASSERT_NO_FATAL_FAILURE(reader.Next());
+ ASSERT_NE(nullptr, reader.array->release);
+ ASSERT_GT(reader.array->length, 0);
+
+ bool seen_fid1 = false;
+ bool seen_fid2 = false;
+
+ struct ArrowArrayView* catalog_db_schemas_list = reader.array_view->children[1];
+ struct ArrowArrayView* catalog_db_schemas_items = catalog_db_schemas_list->children[0];
+ struct ArrowArrayView* db_schema_name_col = catalog_db_schemas_items->children[0];
+ struct ArrowArrayView* db_schema_tables_col = catalog_db_schemas_items->children[1];
+
+ struct ArrowArrayView* schema_table_items = db_schema_tables_col->children[0];
+ struct ArrowArrayView* table_name_col = schema_table_items->children[0];
+ // struct ArrowArrayView* table_columns_col = schema_table_items->children[2];
+ struct ArrowArrayView* table_constraints_col = schema_table_items->children[3];
+
+ // struct ArrowArrayView* table_columns_items = table_columns_col->children[0];
+ // struct ArrowArrayView* column_name_col = table_columns_items->children[0];
+
+ struct ArrowArrayView* table_constraints_items = table_constraints_col->children[0];
+ struct ArrowArrayView* constraint_name_col = table_constraints_items->children[0];
+ struct ArrowArrayView* constraint_type_col = table_constraints_items->children[1];
+ // struct ArrowArrayView* constraint_column_names_col =
+ // table_constraints_items->children[2];
+
+ struct ArrowArrayView* constraint_column_usages_col =
+ table_constraints_items->children[3];
+ struct ArrowArrayView* constraint_column_usage_items =
+ constraint_column_usages_col->children[0];
+ struct ArrowArrayView* fk_catalog_col = constraint_column_usage_items->children[0];
+ struct ArrowArrayView* fk_db_schema_col = constraint_column_usage_items->children[1];
+ struct ArrowArrayView* fk_table_col = constraint_column_usage_items->children[2];
+ struct ArrowArrayView* fk_column_name_col = constraint_column_usage_items->children[3];
+
+ do {
+ for (int64_t catalog_idx = 0; catalog_idx < reader.array->length; catalog_idx++) {
+ ArrowStringView db_name =
+ ArrowArrayViewGetStringUnsafe(reader.array_view->children[0], catalog_idx);
+ auto db_str = std::string(db_name.data, db_name.size_bytes);
+
+ auto schema_list_start =
+ ArrowArrayViewListChildOffset(catalog_db_schemas_list, catalog_idx);
+ auto schema_list_end =
+ ArrowArrayViewListChildOffset(catalog_db_schemas_list, catalog_idx + 1);
+
+ if (db_str == "postgres") {
+ for (auto db_schemas_index = schema_list_start;
+ db_schemas_index < schema_list_end; db_schemas_index++) {
+ ArrowStringView schema_name =
+ ArrowArrayViewGetStringUnsafe(db_schema_name_col, db_schemas_index);
+ auto schema_str = std::string(schema_name.data, schema_name.size_bytes);
+ if (schema_str == "public") {
+ for (auto tables_index = ArrowArrayViewListChildOffset(db_schema_tables_col,
+ db_schemas_index);
+ tables_index < ArrowArrayViewListChildOffset(db_schema_tables_col,
+ db_schemas_index + 1);
+ tables_index++) {
+ ArrowStringView table_name =
+ ArrowArrayViewGetStringUnsafe(table_name_col, tables_index);
+ auto table_str = std::string(table_name.data, table_name.size_bytes);
+ if (table_str == "adbc_fkey_test") {
+ for (auto constraints_index = ArrowArrayViewListChildOffset(
+ table_constraints_col, tables_index);
+ constraints_index < ArrowArrayViewListChildOffset(
+ table_constraints_col, tables_index + 1);
+ constraints_index++) {
+ ArrowStringView constraint_name = ArrowArrayViewGetStringUnsafe(
+ constraint_name_col, constraints_index);
+ auto constraint_name_str =
+ std::string(constraint_name.data, constraint_name.size_bytes);
+ ArrowStringView constraint_type = ArrowArrayViewGetStringUnsafe(
+ constraint_type_col, constraints_index);
+ auto constraint_type_str =
+ std::string(constraint_type.data, constraint_type.size_bytes);
+
+ if (constraint_type_str == "FOREIGN KEY") {
+ for (auto usage_index = ArrowArrayViewListChildOffset(
+ constraint_column_usages_col, constraints_index);
+ usage_index <
+ ArrowArrayViewListChildOffset(constraint_column_usages_col,
+ constraints_index + 1);
+ usage_index++) {
+ ArrowStringView fk_catalog_name =
+ ArrowArrayViewGetStringUnsafe(fk_catalog_col, usage_index);
+ auto fk_catalog_name_str =
+ std::string(fk_catalog_name.data, fk_catalog_name.size_bytes);
+ ArrowStringView fk_schema_name =
+ ArrowArrayViewGetStringUnsafe(fk_db_schema_col, usage_index);
+ auto fk_schema_name_str =
+ std::string(fk_schema_name.data, fk_schema_name.size_bytes);
+ ArrowStringView fk_table_name =
+ ArrowArrayViewGetStringUnsafe(fk_table_col, usage_index);
+ auto fk_table_name_str =
+ std::string(fk_table_name.data, fk_table_name.size_bytes);
+ ArrowStringView fk_column_name =
+ ArrowArrayViewGetStringUnsafe(fk_column_name_col, usage_index);
+ auto fk_column_name_str =
+ std::string(fk_column_name.data, fk_column_name.size_bytes);
+ if ((fk_catalog_name_str == "postgres") &&
+ (fk_schema_name_str == "public") &&
+ (fk_table_name_str == "adbc_fkey_test_base")) {
+ // TODO: the current implementation makes it so the length of
Review Comment:
I might be overthinking this but we have for instance 5 constraints that added right now in the test suite, of which only 2 are foreign keys. The fact that the adbc specification treats the column names of the source table and the usages as separate lists I think means we would have to ensure they are always the same length to know which source column maps to which foreign key table column. Is that the intent?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [arrow-adbc] WillAyd commented on pull request #757: feat(c/driver/postgresql): Implement Foreign Key information for GetObjects
Posted by "WillAyd (via GitHub)" <gi...@apache.org>.
WillAyd commented on PR #757:
URL: https://github.com/apache/arrow-adbc/pull/757#issuecomment-1585144143
I don't think the R failure is related to this PR?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [arrow-adbc] lidavidm commented on pull request #757: feat(c/driver/postgresql): Implement Foreign Key information for GetObjects
Posted by "lidavidm (via GitHub)" <gi...@apache.org>.
lidavidm commented on PR #757:
URL: https://github.com/apache/arrow-adbc/pull/757#issuecomment-1587376084
Alright, it passes. Do you want to add any changes here, or defer those for later PRs?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org
[GitHub] [arrow-adbc] lidavidm commented on pull request #757: feat(c/driver/postgresql): Implement Foreign Key information for GetObjects
Posted by "lidavidm (via GitHub)" <gi...@apache.org>.
lidavidm commented on PR #757:
URL: https://github.com/apache/arrow-adbc/pull/757#issuecomment-1587292154
I think the R failure is because the upstream nanoarrow package must've gotten out of sync
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: github-unsubscribe@arrow.apache.org
For queries about this service, please contact Infrastructure at:
users@infra.apache.org