| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more contributor license |
| * agreements. See the NOTICE file distributed with this work for additional information regarding |
| * copyright ownership. The ASF licenses this file to You 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.apache.geode.internal.cache.persistence; |
| |
| import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap; |
| import it.unimi.dsi.fastutil.objects.Object2IntOpenHashMap; |
| import org.apache.logging.log4j.Logger; |
| |
| import org.apache.geode.logging.internal.log4j.api.LogService; |
| |
| /** |
| * This class manages in memory copy of the canonical ids held in the disk init file. It's used by |
| * the init file to assign numbers to new ids and retrieve existing ids. |
| * |
| * This class is not thread safe, so it should be synchronized externally. |
| * |
| * |
| */ |
| public class CanonicalIdHolder { |
| private static final Logger logger = LogService.getLogger(); |
| |
| /** |
| * Map of integer representation to canonicalized member ids. |
| */ |
| private Int2ObjectOpenHashMap idToObject = new Int2ObjectOpenHashMap(); |
| |
| /** |
| * Map of canonicalized member ids to integer representation. |
| */ |
| private Object2IntOpenHashMap objectToID = new Object2IntOpenHashMap(); |
| |
| private int highestID = 0; |
| |
| /** |
| * Add a mapping that we have recovered from disk |
| */ |
| public void addMapping(int id, Object object) { |
| // Store the mapping |
| idToObject.put(id, object); |
| objectToID.put(object, id); |
| |
| // increase the next canonical id the recovered id is higher than it. |
| highestID = highestID < id ? id : highestID; |
| } |
| |
| /** |
| * Get the id for a given object |
| */ |
| public int getId(Object object) { |
| return objectToID.getInt(object); |
| } |
| |
| /** |
| * Get the object for a given id. |
| */ |
| public Object getObject(int id) { |
| return idToObject.get(id); |
| } |
| |
| /** |
| * Create an id of the given object. |
| * |
| * @return the id generated for this object. |
| */ |
| public int createId(Object object) { |
| assert !objectToID.containsKey(object); |
| int id = ++highestID; |
| objectToID.put(object, id); |
| idToObject.put(id, object); |
| return id; |
| } |
| |
| /** |
| * Get all of the objects that are mapped. |
| * |
| * @return a map of id to object for all objects held by this canonical id holder. |
| */ |
| public Int2ObjectOpenHashMap getAllMappings() { |
| return idToObject; |
| } |
| |
| |
| |
| } |