Add wrapper to builder

ls-urs-keller · Jan 10, 2024 · a7adfd6 · a7adfd6
1 parent 1a5fc11
commit a7adfd6
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/java-client/src/main/java/org/opensearch/client/opensearch/_types/query_dsl/Query.java b/java-client/src/main/java/org/opensearch/client/opensearch/_types/query_dsl/Query.java
@@ -1153,6 +1153,23 @@ public WildcardQuery wildcard() {
         return TaggedUnionUtils.get(this, Kind.Wildcard);
     }
 
+    /**
+     * Is this variant instance of kind {@code wrapper}?
+     */
+    public boolean isWrapper() {
+        return this._kind == Query.Kind.Wrapper;
+    }
+
+    /**
+     * Get the {@code wrapper} variant value.
+     *
+     * @throws IllegalStateException
+     *             if the current variant is not of the {@code wrapper} kind.
+     */
+    public WrapperQuery wrapper() {
+        return (WrapperQuery) TaggedUnionUtils.get(this, Query.Kind.Wrapper);
+    }
+
     /**
      * Is this variant instance of kind {@code type}?
      */
@@ -1743,6 +1760,16 @@ public ObjectBuilder<Query> wildcard(Function<WildcardQuery.Builder, ObjectBuild
             return this.wildcard(fn.apply(new WildcardQuery.Builder()).build());
         }
 
+        public ObjectBuilder<Query> wrapper(WrapperQuery v) {
+            this._kind = Query.Kind.Wrapper;
+            this._value = v;
+            return this;
+        }
+
+        public ObjectBuilder<Query> wrapper(Function<WrapperQuery.Builder, ObjectBuilder<WrapperQuery>> fn) {
+            return this.wrapper(fn.apply(new WrapperQuery.Builder()).build());
+        }
+
         public ObjectBuilder<Query> type(TypeQuery v) {
             this._kind = Kind.Type;
             this._value = v;