Allow getting schema field name (#3576)

redis · Oct 11, 2023 · cef8e40 · cef8e40
1 parent 2bb3989
commit cef8e40
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 0 deletions.
diff --git a/src/main/java/redis/clients/jedis/search/FieldName.java b/src/main/java/redis/clients/jedis/search/FieldName.java
@@ -30,6 +30,14 @@ public FieldName as(String attribute) {
     return this;
   }
 
+  public final String getName() {
+    return name;
+  }
+
+  public final String getAttribute() {
+    return attribute;
+  }
+
   public int addCommandArguments(List<Object> args) {
     args.add(name);
     if (attribute == null) {

diff --git a/src/main/java/redis/clients/jedis/search/schemafields/SchemaField.java b/src/main/java/redis/clients/jedis/search/schemafields/SchemaField.java
@@ -19,4 +19,12 @@ public SchemaField as(String attribute) {
     fieldName.as(attribute);
     return this;
   }
+
+  public final FieldName getFieldName() {
+    return fieldName;
+  }
+
+  public final String getName() {
+    return fieldName.getName();
+  }
 }
diff --git a/src/test/java/redis/clients/jedis/modules/search/UtilTest.java b/src/test/java/redis/clients/jedis/modules/search/UtilTest.java
@@ -1,9 +1,15 @@
 package redis.clients.jedis.modules.search;
 
+import static org.junit.Assert.assertEquals;
+
 import org.junit.Assert;
 import org.junit.Test;
+
 import redis.clients.jedis.search.RediSearchUtil;
 
+import redis.clients.jedis.search.schemafields.NumericField;
+import redis.clients.jedis.search.schemafields.SchemaField;
+
 public class UtilTest {
 
   @Test
@@ -13,4 +19,14 @@ public void floatArrayToByteArray() {
     byte[] expected = new byte[]{-51, -52, 76, 62};
     Assert.assertArrayEquals(expected, bytes);
   }
+
+  @Test
+  public void getSchemaFieldName() {
+    SchemaField field = NumericField.of("$.num").as("num");
+
+    assertEquals("$.num", field.getFieldName().getName());
+    assertEquals("num", field.getFieldName().getAttribute());
+
+    assertEquals("$.num", field.getName());
+  }
 }