工具類——自定義Collections集合方法

執筆記憶的空白發表於2017-05-18

專案中有需要多次統計 某些集合中 的某個屬性值,所以考慮封裝一個方法,讓其其定義實現計算方式。 話不多說,看程式碼:


1、封裝的自定義集合工具類:CollectionsCustom

package com.test.util;

import java.util.Collection;

import org.apache.commons.collections.CollectionUtils;

/**
 * 自定義集合處理類
 * @author : shijing
 * 2017年5月18日下午2:00:10
 */
public class CollectionsCustom {

	
	/**
     * 將傳入的collection內物件進行計算後得出結果
      * @author : shijing  
      * 2017年5月18日下午3:09:53
     * @param original       計算前collection
     * @param reduceFunction 計算方式
     * @param initValue      計算結果初始值
     * @param <Input>        collection物件型別
     * @param <Output>       結果型別
     * @return
     */
    public static <Input, Output> Output reduce(Collection<Input> original, Output initValue, ReduceFunction<Input, Output> reduceFunction) {
        Output result = initValue;
        if (CollectionUtils.isEmpty(original)) {
            return result;
        }
        if (reduceFunction == null) {
            return result;
        }
        for (Input input : original) {
            result = reduceFunction.apply(input, result);
        }
        return result;
    }
    
    /**
     * 自定義計算介面
     * @author : shijing
     * 2017年5月18日下午3:09:53
     * @param <Input>
     * @param <Result>
     */
    public interface ReduceFunction<Input, Result> {

        Result apply(Input input, Result lastResult);

    }
    
}


2、測試類TestCollections

package com.test;

import java.math.BigDecimal;
import java.util.Arrays;
import java.util.List;

import com.test.util.CollectionsCustom;

public class TestCollection {

	private static  List<User> list = Arrays.asList(
			new User("張三", BigDecimal.valueOf(35.6), 18),
			new User("李四", BigDecimal.valueOf(85), 30),
			new User("趙六", BigDecimal.valueOf(66.55), 25));
	
	public static void main(String[] args) {
		//統計集合內分數之和
		testTotalScore();
		//統計集合內年齡之和
		testTotalAge();
		
	}
	
	private static void testTotalScore(){
		//統計集合內分數之和
		BigDecimal totalScore = CollectionsCustom.reduce(list, BigDecimal.ZERO, new CollectionsCustom.ReduceFunction<User, BigDecimal>() {
			@Override
			public BigDecimal apply(User input, BigDecimal lastResult) {
				// TODO Auto-generated method stub
				return lastResult.add(input.getScore());
			}
		});
		System.out.println("總共分數:" + totalScore);
	}
	
	private static void testTotalAge(){
		//統計集合內年齡之和
		Integer totalAge = CollectionsCustom.reduce(list, 0, new CollectionsCustom.ReduceFunction<User, Integer>() {
			@Override
			public Integer apply(User input, Integer lastResult) {
				// TODO Auto-generated method stub
				return lastResult += input.getAge();
			}
		});
		System.out.println("總共年齡:" + totalAge);
	}
	
	static class User{
		private String userName; //姓名
		private BigDecimal score;//分數
		private Integer age;
		public String getUserName() {
			return userName;
		}
		public void setUserName(String userName) {
			this.userName = userName;
		}
		public BigDecimal getScore() {
			return score;
		}
		public void setScore(BigDecimal score) {
			this.score = score;
		}
		public Integer getAge() {
			return age;
		}
		public void setAge(Integer age) {
			this.age = age;
		}
		public User(String userName, BigDecimal score, Integer age) {
			super();
			this.userName = userName;
			this.score = score;
			this.age = age;
		}
		public User() {
			// TODO Auto-generated constructor stub
		}
	}
	
}


3、測試輸出結果:

總共分數:187.15
總共年齡:73

這裡如果傳入的是封裝型別Integer等,最好自己做下非空處理。相信高質量的封裝程式碼能為你自己加分的!






相關文章