From 7d4b5319cd3d1ff2b8a75f13c875bc218a7cbe40 Mon Sep 17 00:00:00 2001
From: liaozan <378024053@qq.com>
Date: Tue, 15 Aug 2023 16:29:22 +0800
Subject: [PATCH] Polish

---
 .../java/com/schbrain/common/util/TreeUtils.java    |  2 +-
 .../java/com/schbrain/common/enums/BooleanEnum.java | 13 +++++++++++++
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/commons/common-util/src/main/java/com/schbrain/common/util/TreeUtils.java b/commons/common-util/src/main/java/com/schbrain/common/util/TreeUtils.java
index c18b6bd..358b626 100644
--- a/commons/common-util/src/main/java/com/schbrain/common/util/TreeUtils.java
+++ b/commons/common-util/src/main/java/com/schbrain/common/util/TreeUtils.java
@@ -41,7 +41,7 @@ public class TreeUtils {
                                               Function<T, K> parentKeyExtractor,
                                               Function<T, E> childMapper,
                                               BiConsumer<E, List<E>> childrenSetter,
-                                              Comparator<E> childrenComparator,
+                                              @Nullable Comparator<E> childrenComparator,
                                               @Nullable K parentId) {
         if (CollectionUtils.isEmpty(nodes)) {
             return new ArrayList<>();
diff --git a/commons/common/src/main/java/com/schbrain/common/enums/BooleanEnum.java b/commons/common/src/main/java/com/schbrain/common/enums/BooleanEnum.java
index 9b2acd0..99d9830 100644
--- a/commons/common/src/main/java/com/schbrain/common/enums/BooleanEnum.java
+++ b/commons/common/src/main/java/com/schbrain/common/enums/BooleanEnum.java
@@ -3,7 +3,10 @@ package com.schbrain.common.enums;
 import lombok.AllArgsConstructor;
 import lombok.Getter;
 
+import java.util.Arrays;
 import java.util.Objects;
+import java.util.Set;
+import java.util.stream.Collectors;
 
 /**
  * @author huangxi
@@ -22,10 +25,20 @@ public enum BooleanEnum {
      */
     FALSE(0);
 
+    private static final Set<Integer> VALUES = Arrays.stream(values()).map(BooleanEnum::getValue).collect(Collectors.toSet());
+
     private final Integer value;
 
     public static boolean validate(Integer value) {
+        return VALUES.contains(value);
+    }
+
+    public static boolean isTrue(Integer value) {
         return Objects.equals(TRUE.value, value);
     }
 
+    public static boolean isFalse(Integer value) {
+        return Objects.equals(FALSE.value, value);
+    }
+
 }
-- 
GitLab