jprdonnelly/airbyte
1
0
Fork 0
mirror of synced 2025-12-25 02:09:19 -05:00
Code Issues Packages Projects Releases Wiki Activity

[source-dynamo] filter sort key from primary key detection (#42433)

Browse Source
This commit is contained in:
Dan Aronson
2024-07-23 22:30:41 +03:00
committed by GitHub
parent 8b0da8bfd7
commit 49e0cf3443
3 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
airbyte-integrations/connectors/source-dynamodb/metadata.yaml
View File

@@ -5,7 +5,7 @@ data:
connectorSubtype: api
connectorType: source
definitionId: 50401137-8871-4c5a-abb7-1f5fda35545a
dockerImageTag: 0.3.3
dockerImageTag: 0.3.5
dockerRepository: airbyte/source-dynamodb
documentationUrl: https://docs.airbyte.com/integrations/sources/dynamodb
githubIssueLabel: source-dynamodb

2
airbyte-integrations/connectors/source-dynamodb/src/main/java/io/airbyte/integrations/source/dynamodb/DynamodbOperations.java
View File

@@ -22,6 +22,7 @@ import software.amazon.awssdk.services.dynamodb.DynamoDbClient;
import software.amazon.awssdk.services.dynamodb.model.AttributeValue;
import software.amazon.awssdk.services.dynamodb.model.DescribeTableRequest;
import software.amazon.awssdk.services.dynamodb.model.KeySchemaElement;
import software.amazon.awssdk.services.dynamodb.model.KeyType;
import software.amazon.awssdk.services.dynamodb.model.ListTablesRequest;
import software.amazon.awssdk.services.dynamodb.model.ListTablesResponse;
import software.amazon.awssdk.services.dynamodb.model.ScanRequest;
@@ -81,6 +82,7 @@ public class DynamodbOperations extends AbstractDatabase implements Closeable {
public List<String> primaryKey(String tableName) {
DescribeTableRequest describeTableRequest = DescribeTableRequest.builder().tableName(tableName).build();
return dynamoDbClient.describeTable(describeTableRequest).table().keySchema().stream()
.filter(element -> element.keyType().equals(KeyType.HASH))
.map(KeySchemaElement::attributeName)
.toList();
}