数据库代码自动生成

zgqc-common/src/main/java/com/panda/zgqc/common/support/ArrayCollector.java

+/*
+ *
+ * Copyright (c) 2020-2022, Java知识图谱 (http://www.altitude.xin).
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+package com.panda.zgqc.common.support;
+
+import java.lang.reflect.Array;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.Set;
+import java.util.function.BiConsumer;
+import java.util.function.BinaryOperator;
+import java.util.function.Function;
+import java.util.function.Supplier;
+import java.util.stream.Collector;
+
+/**
+ * 数组收集器
+ *
+ * @author <a href="http://www.altitude.xin" target="_blank">Java知识图谱</a>
+ * @author <a href="https://gitee.com/decsa/ucode-cms-vue" target="_blank">UCode CMS</a>
+ * @author <a href="https://space.bilibili.com/1936685014" target="_blank">B站视频</a>
+ **/
+public class ArrayCollector<T> implements Collector<T, List<T>, T[]> {
+    private static final Set<Characteristics> CHARACTERISTICS = Collections.emptySet();
+    private final Class<T> elementType;
+
+
+    public ArrayCollector(final Class<T> clazz) {
+        this.elementType = clazz;
+    }
+
+    @Override
+    public BiConsumer<List<T>, T> accumulator() {
+        return List::add;
+    }
+
+    @Override
+    public Set<Characteristics> characteristics() {
+        return CHARACTERISTICS;
+    }
+
+    @Override
+    public BinaryOperator<List<T>> combiner() {
+        return (left, right) -> {
+            left.addAll(right);
+            return left;
+        };
+    }
+
+    @Override
+    @SuppressWarnings("unchecked")
+    public Function<List<T>, T[]> finisher() {
+        return e -> e.toArray((T[]) Array.newInstance(elementType, e.size()));
+    }
+
+    @Override
+    public Supplier<List<T>> supplier() {
+        return ArrayList::new;
+    }
+}