From 7a9e8457300781a62a4122c303fca05f398966c5 Mon Sep 17 00:00:00 2001
From: yixiaco <627428179@qq.com>
Date: Fri, 15 Mar 2024 17:11:03 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=96=B0=E5=A2=9E=E9=9B=86=E5=90=88?=
 =?UTF-8?q?=E5=B7=AE=E5=BC=82=E6=AF=94=E8=BE=83=E5=B7=A5=E5=85=B7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../common/core/utils/CollDiffOps.java        | 165 ++++++++++++++++++
 1 file changed, 165 insertions(+)
 create mode 100644 ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/utils/CollDiffOps.java

diff --git a/ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/utils/CollDiffOps.java b/ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/utils/CollDiffOps.java
new file mode 100644
index 00000000..a6475afb
--- /dev/null
+++ b/ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/utils/CollDiffOps.java
@@ -0,0 +1,165 @@
+package org.dromara.common.core.utils;
+
+import java.util.Collection;
+import java.util.List;
+import java.util.Objects;
+import java.util.Set;
+import java.util.function.Function;
+import java.util.stream.Collectors;
+
+/**
+ * 集合差异比较工具
+ *
+ * @author hexm
+ * @date 2024/03/15 15:12
+ */
+public class CollDiffOps<T, K, R> {
+
+    /** Left数据 */
+    private final Collection<T> left;
+    /** Left数据转为比较对象 */
+    private final Function<T, R> leftConvert;
+    /** Right数据 */
+    private final Collection<K> right;
+    /** Right数据转为比较对象 */
+    private final Function<K, R> rightConvert;
+
+    public CollDiffOps(Collection<T> left, Function<T, R> leftConvert, Collection<K> right, Function<K, R> rightConvert) {
+        this.left = left;
+        this.leftConvert = leftConvert;
+        this.right = right;
+        this.rightConvert = rightConvert;
+    }
+
+    public static <T> CollDiffOps<T, T, T> getOps(Collection<T> left, Collection<T> right) {
+        return new CollDiffOps<>(left, Function.identity(), right, Function.identity());
+    }
+
+    public static <T, R> CollDiffOps<T, R, R> getOps(Collection<T> left, Function<T, R> leftConvert, Collection<R> right) {
+        return new CollDiffOps<>(left, leftConvert, right, Function.identity());
+    }
+
+    public static <K, R> CollDiffOps<R, K, R> getOps(Collection<R> left, Collection<K> right, Function<K, R> rightConvert) {
+        return new CollDiffOps<>(left, Function.identity(), right, rightConvert);
+    }
+
+    public static <T, K, R> CollDiffOps<T, K, R> getOps(Collection<T> left, Function<T, R> leftConvert, Collection<K> right, Function<K, R> rightConvert) {
+        return new CollDiffOps<>(left, leftConvert, right, rightConvert);
+    }
+
+    public static <T, R> CollDiffOps<T, T, R> getUniOps(Collection<T> left, Collection<T> right, Function<T, R> convert) {
+        return new CollDiffOps<>(left, convert, right, convert);
+    }
+
+    /**
+     * 获取差异的Left对象
+     *
+     * @return
+     */
+    public List<T> getDiffLeft() {
+        return getDiff(left, leftConvert, right, rightConvert, Function.identity());
+    }
+
+    /**
+     * 获取差异的Left转换后对象
+     *
+     * @return
+     */
+    public List<R> getDiffConvertLeft() {
+        return getDiff(left, leftConvert, right, rightConvert, leftConvert);
+    }
+
+    /**
+     * 获取差异的Right对象
+     *
+     * @return
+     */
+    public List<K> getDiffRight() {
+        return getDiff(right, rightConvert, left, leftConvert, Function.identity());
+    }
+
+    /**
+     * 获取差异的Right转换后对象
+     *
+     * @return
+     */
+    public List<R> getDiffConvertRight() {
+        return getDiff(right, rightConvert, left, leftConvert, rightConvert);
+    }
+
+    /**
+     * 获取相同的Left对象
+     *
+     * @return
+     */
+    public List<T> getSameLeft() {
+        return getSame(left, leftConvert, right, rightConvert, Function.identity());
+    }
+
+    /**
+     * 获取相同的Left转换后对象
+     *
+     * @return
+     */
+    public List<R> getSameConvertLeft() {
+        return getSame(left, leftConvert, right, rightConvert, leftConvert);
+    }
+
+    /**
+     * 获取相同的Right对象
+     *
+     * @return
+     */
+    public List<K> getSameRight() {
+        return getSame(right, rightConvert, left, leftConvert, Function.identity());
+    }
+
+    /**
+     * 获取相同的Right转换后对象
+     *
+     * @return
+     */
+    public List<R> getSameConvertRight() {
+        return getSame(right, rightConvert, left, leftConvert, rightConvert);
+    }
+
+    public <T0> CollDiffOps<T0, K, R> newLeft(Collection<T0> left, Function<T0, R> leftConvert) {
+        return new CollDiffOps<>(left, leftConvert, right, rightConvert);
+    }
+
+    public <K0> CollDiffOps<T, K0, R> newRight(Collection<K0> right, Function<K0, R> rightConvert) {
+        return new CollDiffOps<>(left, leftConvert, right, rightConvert);
+    }
+
+    /**
+     * 得到差异的left数据
+     */
+    private static <T, K, I, R> List<R> getDiff(Collection<T> left, Function<T, I> leftConvert,
+                                                Collection<K> right, Function<K, I> rightConvert,
+                                                Function<T, R> resultConvert) {
+        Set<I> rightSet = right.stream()
+            .map(rightConvert)
+            .filter(Objects::nonNull)
+            .collect(Collectors.toSet());
+        return left.stream()
+            .filter(t -> !rightSet.contains(leftConvert.apply(t)))
+            .map(resultConvert)
+            .collect(Collectors.toList());
+    }
+
+    /**
+     * 得到相同的left数据
+     */
+    private static <T, K, I, R> List<R> getSame(Collection<T> left, Function<T, I> leftConvert,
+                                                Collection<K> right, Function<K, I> rightConvert,
+                                                Function<T, R> resultConvert) {
+        Set<I> rightSet = right.stream()
+            .map(rightConvert)
+            .filter(Objects::nonNull)
+            .collect(Collectors.toSet());
+        return left.stream()
+            .filter(t -> rightSet.contains(leftConvert.apply(t)))
+            .map(resultConvert)
+            .collect(Collectors.toList());
+    }
+}