基于Java的Stream流实现对某些列的求和（适用于财务计算）

基于Java的Stream流实现对某些列的求和（适用于财务计算）

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import org.apache.commons.lang3.StringUtils;
import org.springframework.lang.*;

import java.math.BigDecimal;
import java.util.*;

public class PageSumUtils {

    /**
     * data：JSONArray 或 List<Map<String,Object>> 要求和的数据
     * total_column：String[] 要求和的列
     * no_total_column：不需要求和的列
     * decimal: 保留几位小数
     * total_index_column：合计、求和名称所处与那一列，值为列名
     * total_name：合计、求和的名称
     *
     * @return
     * @params
     */
    public static JSONObject getSummary(@NonNull Object data, @NonNull String[] total_column, @Nullable String[] no_total_column, String decimal, String total_index_column, String total_name) {
        JSONObject totalJson = new JSONObject();
        List<JSONObject> maps = JSONObject.parseArray(JSON.toJSONString(data), JSONObject.class);
        //四舍五入，默认保留两位小数
        if (StringUtils.isEmpty(decimal)) {
            decimal = "%.2f";
        }
        //合计名称
        if (total_index_column != null && total_name != null) {
            totalJson.put(total_index_column, total_name);
        }
        //需要求和的列
        for (String key : total_column) {
            Optional<BigDecimal> reduce = maps.stream()
                    .filter(temp -> !Objects.equals(temp.get(key), ""))
                    .map(temp -> new BigDecimal(temp.get(key).toString()))
                    .reduce(BigDecimal::add);
            if (reduce.isPresent()) {
                totalJson.put(key, String.format(decimal, reduce.get()));
            } else {
                //空列合计为 0
                totalJson.put(key, 0);
            }
        }
        if (no_total_column != null) {
            //不需要求和的列
            for (String key : no_total_column) {
                totalJson.put(key, "");
            }
        }
        return totalJson;
    }
}