From b59c307db321200806caa9b9b01c4c728fbcb45e Mon Sep 17 00:00:00 2001 From: Jacky Hu Date: Tue, 22 Jul 2025 15:10:39 -0700 Subject: [PATCH 1/2] fix(csharp/src/Drivers/Hive2): Fix unnecessary CloseOperation for Metadata Query --- csharp/src/Drivers/Apache/Hive2/HiveServer2Statement.cs | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/csharp/src/Drivers/Apache/Hive2/HiveServer2Statement.cs b/csharp/src/Drivers/Apache/Hive2/HiveServer2Statement.cs index 31bb234b8a..e91c740b7c 100644 --- a/csharp/src/Drivers/Apache/Hive2/HiveServer2Statement.cs +++ b/csharp/src/Drivers/Apache/Hive2/HiveServer2Statement.cs @@ -457,6 +457,7 @@ protected virtual async Task GetCrossReferenceAsync(CancellationTok ForeignTableName, cancellationToken); OperationHandle = resp.OperationHandle; + _directResults = resp.DirectResults; return await GetQueryResult(resp.DirectResults, cancellationToken); } @@ -474,6 +475,7 @@ protected virtual async Task GetPrimaryKeysAsync(CancellationToken TableName, cancellationToken); OperationHandle = resp.OperationHandle; + _directResults = resp.DirectResults; return await GetQueryResult(resp.DirectResults, cancellationToken); } @@ -482,6 +484,7 @@ protected virtual async Task GetCatalogsAsync(CancellationToken can { TGetCatalogsResp resp = await Connection.GetCatalogsAsync(cancellationToken); OperationHandle = resp.OperationHandle; + _directResults = resp.DirectResults; return await GetQueryResult(resp.DirectResults, cancellationToken); } @@ -493,6 +496,7 @@ protected virtual async Task GetSchemasAsync(CancellationToken canc EscapePatternWildcardsInName(SchemaName), cancellationToken); OperationHandle = resp.OperationHandle; + _directResults = resp.DirectResults; return await GetQueryResult(resp.DirectResults, cancellationToken); } @@ -507,6 +511,7 @@ protected virtual async Task GetTablesAsync(CancellationToken cance tableTypesList, cancellationToken); OperationHandle = resp.OperationHandle; + _directResults = resp.DirectResults; return await GetQueryResult(resp.DirectResults, cancellationToken); } From e5b155a1758a47ca1177f8e2d47c291a41850431 Mon Sep 17 00:00:00 2001 From: Jacky Hu Date: Tue, 22 Jul 2025 17:13:01 -0700 Subject: [PATCH 2/2] Set directResult in GetCrossReferenceAsForeignTableAsync --- csharp/src/Drivers/Apache/Hive2/HiveServer2Statement.cs | 1 + 1 file changed, 1 insertion(+) diff --git a/csharp/src/Drivers/Apache/Hive2/HiveServer2Statement.cs b/csharp/src/Drivers/Apache/Hive2/HiveServer2Statement.cs index e91c740b7c..23d0e53e57 100644 --- a/csharp/src/Drivers/Apache/Hive2/HiveServer2Statement.cs +++ b/csharp/src/Drivers/Apache/Hive2/HiveServer2Statement.cs @@ -437,6 +437,7 @@ protected virtual async Task GetCrossReferenceAsForeignTableAsync(C TableName, cancellationToken); OperationHandle = resp.OperationHandle; + _directResults = resp.DirectResults; return await GetQueryResult(resp.DirectResults, cancellationToken); }