fix: the counts displayed in the console

shuntaka9576 · Jun 12, 2022 · 8e6d57c · 8e6d57c
1 parent 541931b
commit 8e6d57c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/client.go b/client.go
@@ -116,10 +116,10 @@ func (d *DDBClient) BatchWrite(ctx context.Context, req BatchRequest) (output Ba
 				}
 
 				output.UnprocessedRecord = append(output.UnprocessedRecord, string(jsonByte))
-				output.SuccessCount = req.Number() - unprocessedNum
-
-				return output, err
 			}
+			output.SuccessCount = req.Number() - len(output.UnprocessedRecord)
+
+			return output, err
 		} else {
 			return output, errors.New("undefined retry value")
 		}