Add getter for record field of RecordSet and RecordSequenceRecordSet

aerospike · Mar 28, 2024 · 343aeb3 · 343aeb3
1 parent 0c77b29
commit 343aeb3
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/client/src/com/aerospike/client/query/RecordSet.java b/client/src/com/aerospike/client/query/RecordSet.java
@@ -131,6 +131,13 @@ public Record getRecord() {
 		return record.record;
 	}
 
+	/**
+	 * Get KeyRecord.
+	 */
+	public KeyRecord getKeyRecord() {
+		return record;
+	}
+
 	//-------------------------------------------------------
 	// Methods for internal use only.
 	//-------------------------------------------------------
@@ -206,7 +213,7 @@ public boolean hasNext() {
 
 		@Override
 		public KeyRecord next() {
-			KeyRecord kr = recordSet.record;
+			KeyRecord kr = recordSet.getKeyRecord();
 			more = recordSet.next();
 			return kr;
 		}

diff --git a/proxy/src/com/aerospike/client/proxy/RecordSequenceRecordSet.java b/proxy/src/com/aerospike/client/proxy/RecordSequenceRecordSet.java
@@ -34,7 +34,7 @@ public class RecordSequenceRecordSet extends RecordSet implements RecordSequence
 	private final long taskId;
 	private volatile boolean valid = true;
 	private final BlockingQueue<KeyRecord> queue;
-	private volatile KeyRecord record;
+	protected volatile KeyRecord record;
 	private volatile AerospikeException exception;
 
 	public RecordSequenceRecordSet(long taskId, int capacity) {
@@ -109,6 +109,11 @@ public Record getRecord() {
 		return record.record;
 	}
 
+	@Override
+	public KeyRecord getKeyRecord() {
+		return record;
+	}
+
 	@Override
 	public Key getKey() {
 		return record.key;