/* * Copyright 2005 Red Hat, Inc. and/or its affiliates. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.drools.core.common; import java.io.IOException; import java.io.ObjectInput; import java.io.ObjectOutput; import org.drools.core.util.AbstractHashTable; import org.drools.core.util.AbstractHashTable.AbstractObjectComparator; import org.kie.api.runtime.rule.FactHandle; public class EqualityAssertMapComparator extends AbstractObjectComparator { private static final long serialVersionUID = 510l; public EqualityAssertMapComparator() { } @Override public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException { } @Override public void writeExternal(ObjectOutput out) throws IOException { } @Override public int hashCodeOf(final Object obj) { if ( obj instanceof FactHandle) { return AbstractHashTable.rehash( ((InternalFactHandle) obj).getObjectHashCode() ); } return AbstractHashTable.rehash( obj.hashCode() ); } /** * Special comparator that allows FactHandles to be keys, but always checks * equals with the identity of the objects involved */ @Override public boolean areEqual(final Object o1, Object o2) { if ( o1 == o2 ) { return true; } // o1 is a FactHandle, so just check their id's are the same if ( o1 instanceof FactHandle ) { return ((InternalFactHandle)o1).getId() == ((InternalFactHandle)o2).getId() ; } // o1 is an object, so unwrap o2 for comparison final InternalFactHandle handle = ((InternalFactHandle) o2); o2 = handle.getObject(); return o1 == o2 || o2.equals( o1 ); } public int compare(final Object o1, final Object o2) { return ((Comparable) o1).compareTo( o2 ); } @Override public String toString() { return "equality"; } }