fix - aws rds download logs

debeando · Oct 28, 2023 · 1bb05cd · 1bb05cd
1 parent 04bffd1
commit 1bb05cd
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/aws/rds/main.go b/aws/rds/main.go
@@ -76,18 +76,22 @@ func (r *RDS) PollLogs(identifier, filename string) (string, error) {
 	params := &rds.DownloadDBLogFilePortionInput{
 		DBInstanceIdentifier: aws.String(identifier),
 		LogFileName:          aws.String(filename),
+		Marker:               aws.String("0"),
 	}
 
-	result, err := r.Client.DownloadDBLogFilePortion(params)
+	var body string
+
+	err := r.Client.DownloadDBLogFilePortionPages(
+		params,
+		func(page *rds.DownloadDBLogFilePortionOutput, lastPage bool) bool {
+			body = body + aws.StringValue(page.LogFileData)
+			return !lastPage
+		})
 	if err != nil {
 		return "", err
 	}
 
-	if result.LogFileData != nil {
-		return aws.StringValue(result.LogFileData), nil
-	}
-
-	return "", nil
+	return body, nil
 }
 
 func (r *RDS) ParametersGroup() (ParametersGroup, error) {