From 7586f77e98e25187f0a29116c45d2d913cb5f0da Mon Sep 17 00:00:00 2001
From: Houston Putman <houston@apache.org>
Date: Wed, 8 Jan 2025 12:22:27 -0600
Subject: [PATCH] Remove Boolean classes. use client.unwrap()

---
 .../framework/imps/GetDataBuilderImpl.java    |  6 ++--
 .../api/AsyncTransactionCreateBuilder.java    |  4 +--
 .../async/details/AsyncTransactionOpImpl.java | 10 ++-----
 .../modeled/details/ModeledFrameworkImpl.java | 29 +++++++------------
 4 files changed, 18 insertions(+), 31 deletions(-)

diff --git a/curator-framework/src/main/java/org/apache/curator/framework/imps/GetDataBuilderImpl.java b/curator-framework/src/main/java/org/apache/curator/framework/imps/GetDataBuilderImpl.java
index 3468ce6be..df6d15cbd 100644
--- a/curator-framework/src/main/java/org/apache/curator/framework/imps/GetDataBuilderImpl.java
+++ b/curator-framework/src/main/java/org/apache/curator/framework/imps/GetDataBuilderImpl.java
@@ -39,7 +39,7 @@ public class GetDataBuilderImpl implements GetDataBuilder, BackgroundOperation<S
     private Stat responseStat;
     private Watching watching;
     private Backgrounding backgrounding;
-    private Boolean decompress;
+    private boolean decompress;
 
     GetDataBuilderImpl(CuratorFrameworkImpl client) {
         this.client = client;
@@ -54,12 +54,12 @@ public GetDataBuilderImpl(
             Stat responseStat,
             Watcher watcher,
             Backgrounding backgrounding,
-            Boolean decompress) {
+            boolean decompress) {
         this.client = client;
         this.responseStat = responseStat;
         this.watching = new Watching(client, watcher);
         this.backgrounding = backgrounding;
-        this.decompress = decompress != null ? decompress : client.compressionEnabled();
+        this.decompress = decompress;
     }
 
     @Override
diff --git a/curator-x-async/src/main/java/org/apache/curator/x/async/api/AsyncTransactionCreateBuilder.java b/curator-x-async/src/main/java/org/apache/curator/x/async/api/AsyncTransactionCreateBuilder.java
index 674da31cb..fc9bd77d8 100644
--- a/curator-x-async/src/main/java/org/apache/curator/x/async/api/AsyncTransactionCreateBuilder.java
+++ b/curator-x-async/src/main/java/org/apache/curator/x/async/api/AsyncTransactionCreateBuilder.java
@@ -82,7 +82,7 @@ public interface AsyncTransactionCreateBuilder extends AsyncPathAndBytesable<Cur
      * @see #compressed()
      * @return this
      */
-    AsyncPathAndBytesable<CuratorOp> withOptions(CreateMode createMode, List<ACL> aclList, Boolean compressed);
+    AsyncPathAndBytesable<CuratorOp> withOptions(CreateMode createMode, List<ACL> aclList, boolean compressed);
 
     /**
      * Specify mode, acl list, compression and ttl
@@ -97,5 +97,5 @@ public interface AsyncTransactionCreateBuilder extends AsyncPathAndBytesable<Cur
      * @return this
      */
     AsyncPathAndBytesable<CuratorOp> withOptions(
-            CreateMode createMode, List<ACL> aclList, Boolean compressed, long ttl);
+            CreateMode createMode, List<ACL> aclList, boolean compressed, long ttl);
 }
diff --git a/curator-x-async/src/main/java/org/apache/curator/x/async/details/AsyncTransactionOpImpl.java b/curator-x-async/src/main/java/org/apache/curator/x/async/details/AsyncTransactionOpImpl.java
index 7de1ff4e6..6d1d44f43 100644
--- a/curator-x-async/src/main/java/org/apache/curator/x/async/details/AsyncTransactionOpImpl.java
+++ b/curator-x-async/src/main/java/org/apache/curator/x/async/details/AsyncTransactionOpImpl.java
@@ -85,20 +85,16 @@ public AsyncPathAndBytesable<CuratorOp> withTtl(long ttl) {
 
             @Override
             public AsyncPathAndBytesable<CuratorOp> withOptions(
-                    CreateMode createMode, List<ACL> aclList, Boolean compressed) {
+                    CreateMode createMode, List<ACL> aclList, boolean compressed) {
                 return withOptions(createMode, aclList, compressed, ttl);
             }
 
             @Override
             public AsyncPathAndBytesable<CuratorOp> withOptions(
-                    CreateMode createMode, List<ACL> aclList, Boolean compressed, long ttl) {
+                    CreateMode createMode, List<ACL> aclList, boolean compressed, long ttl) {
                 this.createMode = Objects.requireNonNull(createMode, "createMode cannot be null");
                 this.aclList = aclList;
-                if (compressed == Boolean.TRUE) {
-                    this.compressed = true;
-                } else if (compressed == Boolean.FALSE) {
-                    this.compressed = false;
-                }
+                this.compressed = compressed;
                 this.ttl = ttl;
                 return this;
             }
diff --git a/curator-x-async/src/main/java/org/apache/curator/x/async/modeled/details/ModeledFrameworkImpl.java b/curator-x-async/src/main/java/org/apache/curator/x/async/modeled/details/ModeledFrameworkImpl.java
index 02a2bb39f..c02c93127 100644
--- a/curator-x-async/src/main/java/org/apache/curator/x/async/modeled/details/ModeledFrameworkImpl.java
+++ b/curator-x-async/src/main/java/org/apache/curator/x/async/modeled/details/ModeledFrameworkImpl.java
@@ -214,13 +214,10 @@ public AsyncStage<Stat> update(T item, int version) {
             byte[] bytes = modelSpec.serializer().serialize(item);
             AsyncSetDataBuilder dataBuilder = dslClient.setData();
             AsyncPathAndBytesable<AsyncStage<Stat>> next;
-            Boolean isCompressed = isCompressed();
-            if (isCompressed == Boolean.TRUE) {
+            if (isCompressed()) {
                 next = dataBuilder.compressedWithVersion(version);
-            } else if (isCompressed == Boolean.FALSE) {
-                next = dataBuilder.uncompressedWithVersion(version);
             } else {
-                next = dataBuilder.withVersion(version);
+                next = dataBuilder.uncompressedWithVersion(version);
             }
             return next.forPath(resolveForSet(item), bytes);
         } catch (Exception e) {
@@ -371,14 +368,11 @@ public CuratorOp updateOp(T model) {
     @Override
     public CuratorOp updateOp(T model, int version) {
         AsyncTransactionSetDataBuilder builder = client.transactionOp().setData();
-        Boolean isCompressed = isCompressed();
         AsyncPathAndBytesable<CuratorOp> builder2;
-        if (isCompressed == Boolean.TRUE) {
+        if (isCompressed()) {
             builder2 = builder.withVersionCompressed(version);
-        } else if (isCompressed == Boolean.FALSE) {
-            builder2 = builder.withVersionUncompressed(version);
         } else {
-            builder2 = builder.withVersion(version);
+            builder2 = builder.withVersionUncompressed(version);
         }
         return builder2.forPath(resolveForSet(model), modelSpec.serializer().serialize(model));
     }
@@ -414,26 +408,23 @@ public AsyncStage<List<CuratorTransactionResult>> inTransaction(List<CuratorOp>
         return client.transaction().forOperations(operations);
     }
 
-    private Boolean isCompressed() {
+    private boolean isCompressed() {
         if (modelSpec.createOptions().contains(CreateOption.compress)) {
-            return Boolean.TRUE;
+            return true;
         } else if (modelSpec.createOptions().contains(CreateOption.uncompress)) {
-            return Boolean.FALSE;
+            return false;
         } else {
-            return null;
+            return client.unwrap().compressionEnabled();
         }
     }
 
     private <U> ModelStage<U> internalRead(Function<ZNode<T>, U> resolver, Stat storingStatIn) {
         Stat stat = (storingStatIn != null) ? storingStatIn : new Stat();
         AsyncPathable<AsyncStage<byte[]>> next;
-        Boolean isCompressed = isCompressed();
-        if (isCompressed == Boolean.TRUE) {
+        if (isCompressed()) {
             next = watchableClient.getData().decompressedStoringStatIn(stat);
-        } else if (isCompressed == Boolean.FALSE) {
-            next = watchableClient.getData().undecompressedStoringStatIn(stat);
         } else {
-            next = watchableClient.getData().storingStatIn(stat);
+            next = watchableClient.getData().undecompressedStoringStatIn(stat);
         }
         AsyncStage<byte[]> asyncStage = next.forPath(modelSpec.path().fullPath());
         ModelStage<U> modelStage = ModelStage.make(asyncStage.event());