com.netflix.hollow.tools.history.DiffEqualityMappingOrdinalRemapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hollow Show documentation
hollow
There is a newer version: 7.13.0
/*
 *  Copyright 2016-2019 Netflix, Inc.
 *
 *     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 com.netflix.hollow.tools.history;

import com.netflix.hollow.core.util.IntMap;
import com.netflix.hollow.tools.combine.OrdinalRemapper;
import com.netflix.hollow.tools.diff.exact.DiffEqualityMapping;
import java.util.HashMap;

public class DiffEqualityMappingOrdinalRemapper implements OrdinalRemapper {

    private final HashMap unmatchedOrdinalRemapping;
    private final DiffEqualityMapping equalityMapping;

    DiffEqualityMappingOrdinalRemapper(DiffEqualityMapping mapping) {
        this.equalityMapping = mapping;
        this.unmatchedOrdinalRemapping = new HashMap();
    }

    @Override
    public int getMappedOrdinal(String type, int originalOrdinal) {
        IntMap remapping = unmatchedOrdinalRemapping.get(type);

        if(remapping != null) {
            int remappedOrdinal = remapping.get(originalOrdinal);
            if(remappedOrdinal != -1)
                return remappedOrdinal;
        }

        int matchedOrdinal = equalityMapping.getEqualOrdinalMap(type).getIdentityFromOrdinal(originalOrdinal);
        return matchedOrdinal == -1 ? originalOrdinal : matchedOrdinal;
    }

    public void hintUnmatchedOrdinalCount(String type, int numOrdinals) {
        unmatchedOrdinalRemapping.put(type, new IntMap(numOrdinals));
    }

    @Override
    public void remapOrdinal(String type, int originalOrdinal, int mappedOrdinal) {
        IntMap remap = unmatchedOrdinalRemapping.get(type);
        if(remap == null)
            throw new IllegalStateException("Must call hintUnmatchedOrdinalCount for type " + type + " before attempting to remap unmatched ordinals");
        remap.put(originalOrdinal, mappedOrdinal);
    }

    @Override
    public boolean ordinalIsMapped(String type, int originalOrdinal) {
        throw new UnsupportedOperationException();
    }

    public DiffEqualityMapping getDiffEqualityMapping() {
        return equalityMapping;
    }

    public IntMap getUnmatchedOrdinalMapping(String type) {
        return unmatchedOrdinalRemapping.get(type);
    }

}