forked from RuedigerMoeller/fast-serialization
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Making Int2ObjectMap configurable through a factory.
This change allows to register a new Int2ObjectMap factory to the configuration while keeping the older implementation as the default. To set up a new factory clients of the library will have to create a factory and set it through the configuration. In addition clearing of the default configuration is now responsible for deciding how to clean itself while maintining implementation details encapsulated.
- Loading branch information
Showing
7 changed files
with
222 additions
and
179 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
184 changes: 184 additions & 0 deletions
184
src/main/java/org/nustaq/serialization/util/DefaultFSTInt2ObjectMap.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,184 @@ | ||
| /* | ||
| * Copyright 2014 Ruediger Moeller. | ||
| * | ||
| * 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.nustaq.serialization.util; | ||
|
|
||
| public class DefaultFSTInt2ObjectMap<V> implements FSTInt2ObjectMap <V>{ | ||
|
|
||
| private int mKeys[]; | ||
| private Object mValues[]; | ||
| private int mNumberOfElements; | ||
| private DefaultFSTInt2ObjectMap<V> next; | ||
| private static final int GROWFAC = 2; | ||
|
|
||
| public DefaultFSTInt2ObjectMap(int initialSize) { | ||
| if (initialSize < 2) { | ||
| initialSize = 2; | ||
| } | ||
|
|
||
| initialSize = FSTObject2IntMap.adjustSize(initialSize * 2); | ||
|
|
||
| mKeys = new int[initialSize]; | ||
| mValues = new Object[initialSize]; | ||
| mNumberOfElements = 0; | ||
| } | ||
|
|
||
| public int size() { | ||
| return mNumberOfElements + (next != null ? next.size() : 0); | ||
| } | ||
|
|
||
| final public void put(int key, V value) { | ||
| int hash = key & 0x7FFFFFFF; | ||
| if (key == 0 && value == null) { | ||
| throw new RuntimeException("key value pair not supported " + key + " " + value); | ||
| } | ||
| putHash(key, value, hash, this); | ||
| } | ||
|
|
||
| final private static <V> void putHash(int key, V value, int hash, DefaultFSTInt2ObjectMap<V> current, DefaultFSTInt2ObjectMap<V> parent) { | ||
| int count = 0; | ||
| while(true){ | ||
| if (current.mNumberOfElements * GROWFAC > current.mKeys.length) { | ||
| if (parent != null) { | ||
| if ((parent.mNumberOfElements + current.mNumberOfElements) * GROWFAC > parent.mKeys.length) { | ||
| parent.resize(parent.mKeys.length * GROWFAC); | ||
| parent.put(key, value); | ||
| return; | ||
| } else { | ||
| current.resize(current.mKeys.length * GROWFAC); | ||
| } | ||
| } else { | ||
| current.resize(current.mKeys.length * GROWFAC); | ||
| } | ||
| } | ||
|
|
||
| int idx = hash % current.mKeys.length; | ||
|
|
||
| if (current.mKeys[idx] == 0 && current.mValues[idx] == null) // new | ||
| { | ||
| current.mNumberOfElements++; | ||
| current.mValues[idx] = value; | ||
| current.mKeys[idx] = key; | ||
| return; | ||
| } else if (current.mKeys[idx] == key) // overwrite | ||
| { | ||
| current.mValues[idx] = value; | ||
| return; | ||
| } else { | ||
| if (current.next == null) { | ||
| // try break edge cases leading to long chains of maps | ||
| if ( count > 4 && current.mNumberOfElements < 5 ) { | ||
| int newSiz = current.mNumberOfElements*2+1; | ||
| current.next = new DefaultFSTInt2ObjectMap<V>(newSiz); | ||
| count = 0; | ||
| } else { | ||
| int newSiz = current.mNumberOfElements / 3; | ||
| current.next = new DefaultFSTInt2ObjectMap<V>(newSiz); | ||
| } | ||
| } | ||
| parent = current; | ||
| current = current.next; | ||
| count ++; | ||
| } | ||
| } | ||
| } | ||
|
|
||
| final void putHash(int key, V value, int hash, DefaultFSTInt2ObjectMap<V> parent) { | ||
| putHash(key, value, hash,this, parent); | ||
| } | ||
|
|
||
| final public V get(int key) { | ||
| int hash = key & 0x7FFFFFFF; | ||
| return getHash(key, hash); | ||
| } | ||
|
|
||
| final V getHash(int key, int hash) { | ||
| final int idx = hash % mKeys.length; | ||
|
|
||
| final int mKey = mKeys[idx]; | ||
| final Object mValue = mValues[idx]; | ||
| if (mKey == 0 && mValue == null) // not found | ||
| { | ||
| // hit++; | ||
| return null; | ||
| } else if (mKey == key) // found | ||
| { | ||
| // hit++; | ||
| return (V) mValue; | ||
| } else { | ||
| if (next == null) { | ||
| return null; | ||
| } | ||
| // miss++; | ||
| return next.getHash(key, hash); | ||
| } | ||
| } | ||
|
|
||
| final void resize(int newSize) { | ||
| newSize = FSTObject2IntMap.adjustSize(newSize); | ||
| int[] oldTabKey = mKeys; | ||
| Object[] oldTabVal = mValues; | ||
|
|
||
| mKeys = new int[newSize]; | ||
| mValues = new Object[newSize]; | ||
| mNumberOfElements = 0; | ||
|
|
||
| for (int n = 0; n < oldTabKey.length; n++) { | ||
| if (oldTabKey[n] != 0 || oldTabVal[n] != null) { | ||
| put(oldTabKey[n], (V) oldTabVal[n]); | ||
| } | ||
| } | ||
| if (next != null) { | ||
| DefaultFSTInt2ObjectMap oldNext = next; | ||
| next = null; | ||
| oldNext.rePut(this); | ||
| } | ||
| } | ||
|
|
||
| private void rePut(DefaultFSTInt2ObjectMap<V> kfstObject2IntMap) { | ||
| for (int i = 0; i < mKeys.length; i++) { | ||
| int mKey = mKeys[i]; | ||
| if (mKey != 0 || mValues[i] != null) { | ||
| kfstObject2IntMap.put(mKey, (V) mValues[i]); | ||
| } | ||
| } | ||
| if (next != null) { | ||
| next.rePut(kfstObject2IntMap); | ||
| } | ||
| } | ||
|
|
||
| public void clear() { | ||
| int size = size(); | ||
| if (size == 0) | ||
| return; | ||
| if (mKeys.length > 6 * size() && size > 0) { | ||
| // avoid cleaning huge mem areas after having written a large object | ||
| if (size < 2) { | ||
| size = 2; | ||
| } | ||
| size = FSTObject2IntMap.adjustSize(size * 2); | ||
| mKeys = new int[size]; | ||
| mValues = new Object[size]; | ||
| mNumberOfElements = 0; | ||
| } else { | ||
| FSTUtil.clear(mKeys); | ||
| FSTUtil.clear(mValues); | ||
| mNumberOfElements = 0; | ||
| if (next != null) { | ||
| next.clear(); | ||
| } | ||
| } | ||
| } | ||
| } |
8 changes: 8 additions & 0 deletions
8
src/main/java/org/nustaq/serialization/util/DefaultFSTInt2ObjectMapFactory.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,8 @@ | ||
| package org.nustaq.serialization.util; | ||
|
|
||
| public class DefaultFSTInt2ObjectMapFactory implements FSTInt2ObjectMapFactory{ | ||
| @Override | ||
| public <V> FSTInt2ObjectMap<V> createMap(int size) { | ||
| return new DefaultFSTInt2ObjectMap<>(size); | ||
| } | ||
| } |
Oops, something went wrong.