cmd/dynamo-browse
1
0
Fork 0
Code Issues Pull requests Projects Releases 2 Packages Wiki Activity Actions

Added paging and the ability to specify the table in query script method (#44)

Browse source 
* paging: added controller for paging through results
* paging: added command and key binding for going to the next page
* paging: added the ability to specify the table in the query script method
* paging: have got exclusive start key written to backstack
This commit is contained in:
Leon Mika 2023-01-23 21:23:46 +11:00 • committed by GitHub
parent 9e658b8619
commit ad1a77a257
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
20 changed files with 444 additions and 95 deletions
Download patch file Download diff file Expand all files Collapse all files

30
test/cmd/load-test-table/main.go
View file

@ -35,15 +35,15 @@ func main() {
dynamodb.WithEndpointResolver(dynamodb.EndpointResolverFromURL("http://localhost:4566")))
// Other tables
if err := createTable(ctx, dynamoClient, "user-accounts"); err != nil {
if err := createTable(ctx, dynamoClient, "user-accounts", false); err != nil {
log.Fatal(err)
}
if err := createTable(ctx, dynamoClient, "inventory"); err != nil {
if err := createTable(ctx, dynamoClient, "inventory", true); err != nil {
log.Fatal(err)
}
if err := createTable(ctx, dynamoClient, tableName); err != nil {
if err := createTable(ctx, dynamoClient, tableName, false); err != nil {
log.Fatal(err)
}
@ -51,6 +51,10 @@ func main() {
Name: tableName,
Keys: models.KeyAttribute{PartitionKey: "pk", SortKey: "sk"},
}
inventoryTableInfo := &models.TableInfo{
Name: "inventory",
Keys: models.KeyAttribute{PartitionKey: "pk", SortKey: "sk"},
}
dynamoProvider := dynamo.NewProvider(dynamoClient)
tableService := tables.NewService(dynamoProvider, notROService{})
@ -87,16 +91,32 @@ func main() {
}
}
key := gofakeit.UUID()
for i := 0; i < totalItems; i++ {
if err := tableService.Put(ctx, inventoryTableInfo, models.Item{
"pk": &types.AttributeValueMemberS{Value: key},
"sk": &types.AttributeValueMemberN{Value: fmt.Sprint(i)},
"uuid": &types.AttributeValueMemberS{Value: gofakeit.UUID()},
}); err != nil {
log.Fatalln(err)
}
}
log.Printf("table '%v' created with %v items", tableName, totalItems)
}
func createTable(ctx context.Context, dynamoClient *dynamodb.Client, tableName string) error {
func createTable(ctx context.Context, dynamoClient *dynamodb.Client, tableName string, skNumber bool) error {
if _, err := dynamoClient.DeleteTable(ctx, &dynamodb.DeleteTableInput{
TableName: aws.String(tableName),
}); err != nil {
log.Printf("warn: cannot delete table: %v: %v", tableName, err)
}
var skType = types.ScalarAttributeTypeS
if skNumber {
skType = types.ScalarAttributeTypeN
}
if _, err := dynamoClient.CreateTable(ctx, &dynamodb.CreateTableInput{
TableName: aws.String(tableName),
KeySchema: []types.KeySchemaElement{
@ -105,7 +125,7 @@ func createTable(ctx context.Context, dynamoClient *dynamodb.Client, tableName s
},
AttributeDefinitions: []types.AttributeDefinition{
{AttributeName: aws.String("pk"), AttributeType: types.ScalarAttributeTypeS},
{AttributeName: aws.String("sk"), AttributeType: types.ScalarAttributeTypeS},
{AttributeName: aws.String("sk"), AttributeType: skType},
},
ProvisionedThroughput: &types.ProvisionedThroughput{
ReadCapacityUnits: aws.Int64(100),