DocumentModel.java

package be.redtree.model;

import java.io.Serializable;
import java.util.Iterator;
import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

public class DocumentModel extends ListDataModel<Document> implements SelectableDataModel<Document> ,Serializable{

	public DocumentModel() {

	}

	public DocumentModel(List<Document> documents) {
		super(documents);
	}

	public Document getRowData(String value) {

		Iterator<Document> itr = super.iterator();
		while (itr.hasNext()) {

			Document document = itr.next();

			if (value.equals(document.getId())) {
				return document;
			}
		}
		return null;
	}

	public Object getRowKey(Document document) {
		return document.getId();
	}

}