Code cleaning

This commit is contained in:
Riccardo
2022-07-25 19:05:37 +02:00
parent b7eb930bd8
commit 9f5d892e5b
4 changed files with 36 additions and 19 deletions

View File

@@ -70,8 +70,10 @@ func CreateTable(basics Table) (*types.TableDescription, error) {
if err != nil {
log.Printf("Wait for table exists failed. Here's why: %v\n", err)
}
tableDesc = table.TableDescription
}
return tableDesc, err
}
@@ -87,19 +89,24 @@ func AddNewsBatch(basics Table, news []News, batchSize int) (int, error) {
written := 0
start := 0
end := start + batchSize
for start < len(news) {
var writeReqs []types.WriteRequest
if end > len(news) {
end = len(news)
}
for _, entry := range news[start:end] {
item, err = attributevalue.MarshalMap(entry)
if err != nil {
log.Printf("Couldn't marshal news %v for batch writing: %v\n", entry.Title, err)
} else {
writeReqs = append(writeReqs, types.WriteRequest{PutRequest: &types.PutRequest{Item: item}})
}
}
_, err = basics.DynamoDbClient.BatchWriteItem(context.TODO(), &dynamodb.BatchWriteItemInput{
RequestItems: map[string][]types.WriteRequest{basics.TableName: writeReqs}})
if err != nil {
@@ -107,6 +114,7 @@ func AddNewsBatch(basics Table, news []News, batchSize int) (int, error) {
} else {
written += len(writeReqs)
}
start = end
end += batchSize
}
@@ -120,7 +128,7 @@ func ReadTodayNews(basics Table, timeStart int, timeEnd int) ([]News, error) {
var news []News
params, err := attributevalue.MarshalList([]interface{}{timeStart, timeEnd})
if err != nil {
panic(err)
log.Fatalln(err)
}
response, err := basics.DynamoDbClient.ExecuteStatement(context.TODO(), &dynamodb.ExecuteStatementInput{