package com.oms.util;

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.oms.dto.PageResponse;

import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

public class PageUtil {

    public static <T, R> PageResponse<R> convert(Page<T> page, Function<T, R> converter) {
        List<R> items = page.getRecords().stream().map(converter).collect(Collectors.toList());
        return PageResponse.of(items, page.getTotal(), (int) page.getCurrent(), (int) page.getSize());
    }

    public static <T> PageResponse<T> of(Page<T> page) {
        return PageResponse.of(page.getRecords(), page.getTotal(), (int) page.getCurrent(), (int) page.getSize());
    }
}