org.picketlink.idm.query.RelationshipCriteria Maven / Gradle / Ivy

package org.picketlink.idm.query;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;

import org.picketlink.idm.model.IdentityType;
import org.picketlink.idm.model.Relationship;

/**
 *
 * @author Shane Bryzak
 */
public class RelationshipCriteria {
    private Map criteria = new HashMap();

    private Class relationshipClass;

    public RelationshipCriteria(Class relationshipClass) {
        this.relationshipClass = relationshipClass;
    }

    public void addCriteria(String propertyName, IdentityType identity) {
        criteria.put(propertyName, identity);
    }

    public Class getRelationshipClass() {
        return relationshipClass;
    }

    public Set getPropertyNames() {
        return criteria.keySet();
    }

    public IdentityType getCriteria(String propertyName) {
        return criteria.get(propertyName);
    }
}