chore: change hint prexfix to x-greptime-hint- (#44)

GreptimeTeam · Jul 31, 2024 · 1da56f5 · 1da56f5
1 parent 83096a0
commit 1da56f5
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/ingester-rpc/src/main/java/io/greptime/rpc/Context.java b/ingester-rpc/src/main/java/io/greptime/rpc/Context.java
@@ -28,7 +28,7 @@
 @SuppressWarnings({"unchecked", "unused"})
 public class Context implements Copiable<Context> {
 
-    private static final String HINT_PREFIX = "x-greptime-hint:";
+    private static final String HINT_PREFIX = "x-greptime-hint-";
 
     private final Map<String, Object> ctx = new HashMap<>();
 
@@ -99,6 +99,17 @@ public <T> T get(String key) {
         }
     }
 
+    /**
+     * Gets the value of the specified hint key.
+     *
+     * @param key the hint key
+     * @return the value
+     * @param <T> the type of the value
+     */
+    public <T> T getHint(String key) {
+        return get(HINT_PREFIX + key);
+    }
+
     /**
      * Removes the specified key and its value form this {@link Context}.
      *

diff --git a/ingester-rpc/src/test/java/io/greptime/rpc/ContextTest.java b/ingester-rpc/src/test/java/io/greptime/rpc/ContextTest.java
@@ -38,7 +38,8 @@ public void ofShouldReturnContextWithOneEntryTest() {
     @Test
     public void hintShouldReturnContextWithHintEntryTest() {
         Context context = Context.hint("key", "value");
-        Assert.assertEquals("value", context.get("x-greptime-hint:key"));
+        Assert.assertEquals("value", context.getHint("key"));
+        Assert.assertEquals("value", context.get("x-greptime-hint-key"));
     }
 
     @Test
@@ -52,7 +53,8 @@ public void withShouldAddEntryToContextTest() {
     public void withHintShouldAddHintEntryToContextTest() {
         Context context = Context.newDefault();
         context.withHint("key", "value").with("key2", "value");
-        Assert.assertEquals("value", context.get("x-greptime-hint:key"));
+        Assert.assertEquals("value", context.get("x-greptime-hint-key"));
+        Assert.assertEquals("value", context.getHint("key"));
         Assert.assertEquals("value", context.get("key2"));
     }