DocumentComparator.java

package com.tradecloud.domain.document;

import java.util.Comparator;

/**
 * Compares two documents. Order by date of creation
 */
public class DocumentComparator implements Comparator<Document> {

    @Override
    public int compare(Document document1, Document document2) {
        if (document1.getDocumentType().equals(document2.getDocumentType())) {
            return document1.getDateCreated().compareTo(document2.getDateCreated());
        }
        return document1.getDocumentType().compareTo(document2.getDocumentType());
    }
}