add static records_count field

apache · Jul 17, 2022 · b84ab8c · b84ab8c
1 parent e952d9a
commit b84ab8c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/...receiver/src/test/java/org/apache/beam/sdk/io/sparkreceiver/CustomReceiverWithOffset.java b/...receiver/src/test/java/org/apache/beam/sdk/io/sparkreceiver/CustomReceiverWithOffset.java
@@ -35,8 +35,8 @@ public class CustomReceiverWithOffset extends Receiver<String> implements HasOff
 
  private static final Logger LOG = LoggerFactory.getLogger(CustomReceiverWithOffset.class);
  private static final int TIMEOUT_MS = 500;
-
  private static final List<String> STORED_RECORDS = new ArrayList<>();
+ private static final int RECORDS_COUNT = 20;
  private Long startOffset;
 
  CustomReceiverWithOffset() {
@@ -67,7 +67,7 @@ public Long getEndOffset() {
  private void receive() {
  Long currentOffset = startOffset;
  while (!isStopped()) {
- if (currentOffset <= 20) {
+ if (currentOffset <= RECORDS_COUNT) {
  STORED_RECORDS.add(currentOffset.toString());
  store((currentOffset++).toString());
  }

diff --git a/...eiver/src/test/java/org/apache/beam/sdk/io/sparkreceiver/CustomReceiverWithoutOffset.java b/...eiver/src/test/java/org/apache/beam/sdk/io/sparkreceiver/CustomReceiverWithoutOffset.java
@@ -36,6 +36,7 @@ public class CustomReceiverWithoutOffset extends Receiver<String> {
  private static final Logger LOG = LoggerFactory.getLogger(CustomReceiverWithoutOffset.class);
  private static final int TIMEOUT_MS = 500;
  private static final List<String> STORED_RECORDS = new ArrayList<>();
+ private static final int RECORDS_COUNT = 20;
 
  CustomReceiverWithoutOffset() {
  super(StorageLevel.MEMORY_AND_DISK_2());
@@ -53,7 +54,7 @@ public void onStop() {}
  private void receive() {
  Long currentOffset = 0L;
  while (!isStopped()) {
- if (currentOffset <= 20) {
+ if (currentOffset <= RECORDS_COUNT) {
  STORED_RECORDS.add(currentOffset.toString());
  store((currentOffset++).toString());
  }