package com.biubiu.myboot.util;

import org.springframework.beans.BeanUtils;
import org.springframework.util.CollectionUtils;

import java.util.ArrayList;
import java.util.List;
import java.util.function.Supplier;
import java.util.stream.Collectors;

/**
 * <p>
 * DtoCopyUtil Dto拷贝工具类
 * </p>
 *
 * @author biubiu
 * @date 2022/3/29 0:11
 */
public class DtoCopyUtil {

    private DtoCopyUtil() {
    }

    /**
     * 单个对象的拷贝
     *
     * @param src      源对象
     * @param supplier 泛型对象
     * @param <T>      泛型 T
     * @param <R>      泛型 R
     * @return 目标对象
     */
    public static <T, R> R copyFrom(T src, Supplier<R> supplier) {
        if (src == null) {
            return null;
        }
        R dst = supplier.get();

        BeanUtils.copyProperties(src, dst);
        return dst;
    }

    /**
     * list中对象的拷贝
     *
     * @param srcList  源list
     * @param supplier 泛型得实例
     * @param <T>      泛型 T
     * @param <R>      泛型 R
     * @return List<R>
     */
    public static <T, R> List<R> copyFrom(List<T> srcList, Supplier<R> supplier) {
        if (CollectionUtils.isEmpty(srcList)) {
            return new ArrayList<>();
        }

        List<R> destList = srcList.stream().map(src -> copyFrom(src, supplier)).collect(Collectors.toList());
        return destList;
    }

    /**
     * 单个对象的拷贝
     *
     * @param src      源对象
     * @param supplier 泛型对象
     * @param <T>      泛型 T
     * @param <R>      泛型 R
     * @return 目标对象
     */
    public static <T, R> R copyFrom(T src, R supplier) {
        if (src == null || supplier == null) {
            return null;
        }
        R dst = supplier;
        BeanUtils.copyProperties(src, dst);
        return dst;
    }

}