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

Added extra table

Browse source
This commit is contained in:
Leon Mika 2022-07-28 21:32:32 +10:00
parent 64c43d6b73
commit 2a35667cdb
2 changed files with 38 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
internal/dynamo-browse/ui/model.go
View file

@ -36,7 +36,7 @@ func NewModel(rc *controllers.TableReadController, wc *controllers.TableWriteCon
dtv := dynamotableview.New(uiStyles)
div := dynamoitemview.New(uiStyles)
mainView := layout.NewVBox(layout.LastChildFixedAt(17), dtv, div)
mainView := layout.NewVBox(layout.LastChildFixedAt(13), dtv, div)
itemEdit := dynamoitemedit.NewModel(mainView)
statusAndPrompt := statusandprompt.New(itemEdit, "", uiStyles.StatusAndPrompt)

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

@ -13,6 +13,7 @@ import (
"github.com/lmika/awstools/internal/dynamo-browse/providers/dynamo"
"github.com/lmika/awstools/internal/dynamo-browse/services/tables"
"github.com/lmika/gopkgs/cli"
"github.com/pkg/errors"
"log"
)
@ -29,28 +30,17 @@ func main() {
dynamoClient := dynamodb.NewFromConfig(cfg,
dynamodb.WithEndpointResolver(dynamodb.EndpointResolverFromURL("http://localhost:4566")))
if _, err = dynamoClient.DeleteTable(ctx, &dynamodb.DeleteTableInput{
TableName: aws.String(tableName),
}); err != nil {
log.Printf("warn: cannot delete table: %v: %v", tableName, err)
// Other tables
if err := createTable(ctx, dynamoClient, "user-accounts"); err != nil {
log.Fatal(err)
}
if _, err = dynamoClient.CreateTable(ctx, &dynamodb.CreateTableInput{
TableName: aws.String(tableName),
KeySchema: []types.KeySchemaElement{
{AttributeName: aws.String("pk"), KeyType: types.KeyTypeHash},
{AttributeName: aws.String("sk"), KeyType: types.KeyTypeRange},
},
AttributeDefinitions: []types.AttributeDefinition{
{AttributeName: aws.String("pk"), AttributeType: types.ScalarAttributeTypeS},
{AttributeName: aws.String("sk"), AttributeType: types.ScalarAttributeTypeS},
},
ProvisionedThroughput: &types.ProvisionedThroughput{
ReadCapacityUnits: aws.Int64(100),
WriteCapacityUnits: aws.Int64(100),
},
}); err != nil {
log.Fatalf("warn: cannot create table: %v", tableName)
if err := createTable(ctx, dynamoClient, "inventory"); err != nil {
log.Fatal(err)
}
if err := createTable(ctx, dynamoClient, tableName); err != nil {
log.Fatal(err)
}
tableInfo := &models.TableInfo{
@ -86,3 +76,30 @@ func main() {
log.Printf("table '%v' created with %v items", tableName, totalItems)
}
func createTable(ctx context.Context, dynamoClient *dynamodb.Client, tableName string) error {
if _, err := dynamoClient.DeleteTable(ctx, &dynamodb.DeleteTableInput{
TableName: aws.String(tableName),
}); err != nil {
log.Printf("warn: cannot delete table: %v: %v", tableName, err)
}
if _, err := dynamoClient.CreateTable(ctx, &dynamodb.CreateTableInput{
TableName: aws.String(tableName),
KeySchema: []types.KeySchemaElement{
{AttributeName: aws.String("pk"), KeyType: types.KeyTypeHash},
{AttributeName: aws.String("sk"), KeyType: types.KeyTypeRange},
},
AttributeDefinitions: []types.AttributeDefinition{
{AttributeName: aws.String("pk"), AttributeType: types.ScalarAttributeTypeS},
{AttributeName: aws.String("sk"), AttributeType: types.ScalarAttributeTypeS},
},
ProvisionedThroughput: &types.ProvisionedThroughput{
ReadCapacityUnits: aws.Int64(100),
WriteCapacityUnits: aws.Int64(100),
},
}); err != nil {
return errors.Wrapf(err, "cannot create table: %v", tableName)
}
return nil
}