/*
* Copyright (C) 2012, 2016 higherfrequencytrading.com
* Copyright (C) 2016 Roman Leventov
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package net.openhft.chronicle.map;
import net.openhft.chronicle.hash.Data;
import net.openhft.chronicle.hash.serialization.ListMarshaller;
import net.openhft.chronicle.hash.serialization.impl.CharSequenceBytesReader;
import net.openhft.chronicle.hash.serialization.impl.CharSequenceBytesWriter;
import net.openhft.chronicle.set.*;
import org.junit.Test;
import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;
import java.util.concurrent.ThreadLocalRandom;
import static java.util.concurrent.TimeUnit.MINUTES;
import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertEquals;
public class Issue63Test {
// Right now no corresponding "knownUsers" object
private ChronicleMap<CharSequence, List<CharSequence>> knownItems;
private ChronicleMap<CharSequence, float[]> xVectors;
private ChronicleSet<CharSequence> xRecentIDs;
private ChronicleMap<CharSequence, float[]> yVectors;
private ChronicleSet<CharSequence> yRecentIDs;
public static void main(String[] args) throws Exception {
new Issue63Test().testChronicleMap();
}
private static void putIntoMapAndRecentSet(
ChronicleMap<CharSequence, float[]> map, ChronicleSet<CharSequence> recentSet,
String key, float[] value) {
try (ExternalMapQueryContext<CharSequence, float[], ?> c = map.queryContext(key);
ExternalSetQueryContext<CharSequence, ?> setC = recentSet.queryContext(key)) {
if (c.writeLock().tryLock(1, MINUTES) && setC.writeLock().tryLock(1, MINUTES)) {
putNoReturn(c, value);
addNoReturn(setC);
} else {
throw new RuntimeException("Dead lock");
}
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
}
private static <K> void addNoReturn(SetQueryContext<K, ?> c) {
SetAbsentEntry<K> setAbsentEntry = c.absentEntry();
if (setAbsentEntry != null) {
c.insert(setAbsentEntry);
}
}
private static <K, V> void putNoReturn(MapQueryContext<K, V, ?> c, V value) {
MapEntry<K, V> entry = c.entry();
Data<V> newValue = c.wrapValueAsData(value);
if (entry != null) {
c.replaceValue(entry, newValue);
} else {
MapAbsentEntry<K, V> absentEntry = c.absentEntry();
assert absentEntry != null;
c.insert(absentEntry, newValue);
}
}
@Test
public void issue63test() throws IOException {
Path path = Paths.get(System.getProperty("java.io.tmpdir") + "/test-vectors1.dat");
if (Files.exists(path)) Files.delete(path);
File mapFile = path.toFile();
mapFile.deleteOnExit();
ChronicleMap<CharSequence, float[]> xVectors = ChronicleMap
.of(CharSequence.class, float[].class)
// use the actual UUID size
.constantKeySizeBySample("2B6EC73CD63ACA5D93F4D5A710AD9CFE")
.constantValueSizeBySample(new float[100])
.putReturnsNull(true)
.entries(10)
.createPersistedTo(mapFile);
float[] exp1 = new float[]{
(float) -0.4200737, (float) -0.5428019, (float) 0.25542524, (float) -0.10631648,
(float) 0.12206168, (float) 0.0411969, (float) 0.9899967, (float) 0.15887073,
(float) -0.09775953, (float) 0.21812996, (float) -0.2724478, (float) 1.1872392,
(float) -0.57449555, (float) 0.5036392, (float) 0.1658725, (float) 0.26468855,
(float) -0.3454932, (float) 0.61344844, (float) -0.058357887, (float) 0.41589612,
(float) -0.30034602, (float) -0.065557495, (float) -0.5450994, (float) 0.24787773,
(float) -0.49933347, (float) -0.34362262, (float) -0.116148725, (float) 0.1267731,
(float) -0.021314947, (float) 0.4289211, (float) 0.018796312, (float) 1.1027592,
(float) 0.26406515, (float) -0.364442, (float) 0.032301463, (float) 0.7497238,
(float) 0.022618806, (float) 0.44369924, (float) -0.3347779, (float) -0.21492186,
(float) -0.16348012, (float) -0.07863602, (float) 0.22218524, (float) 0.13798094,
(float) 0.9739758, (float) 0.18799895, (float) 0.16804655, (float) -0.94723654,
(float) -0.09069447, (float) 1.0777866, (float) 0.45763463, (float) -0.99949086,
(float) 0.1130747, (float) 1.1800445, (float) -0.7469727, (float) -1.480476,
(float) 0.21458353, (float) 0.5420289, (float) 0.44423282, (float) -0.73524255,
(float) -0.86806494, (float) 0.77911025, (float) 0.43587336, (float) 0.45608798,
(float) -0.52584565, (float) 0.5979028, (float) 0.18747452, (float) -0.9211639,
(float) 0.2969087, (float) -0.17334144, (float) -0.30227816, (float) 0.6624411,
(float) -1.445531, (float) 0.068452656, (float) -0.54010916, (float) 0.7997881,
(float) -1.1808084, (float) 1.0036258, (float) 0.23763403, (float) -0.95869386,
(float) 0.2150584, (float) 0.16237195, (float) 0.35550624, (float) -0.59370506,
(float) 0.977463, (float) -0.14227587, (float) -1.1346477, (float) -0.29077065,
(float) -0.7924145, (float) -0.05505234, (float) -0.4519053, (float) 0.8662279,
(float) 0.056166444, (float) -0.6824282, (float) -0.28487095, (float) -0.28058794,
(float) -0.868858, (float) 0.4946002, (float) 0.61442167, (float) 0.70633507
};
float[] exp2 = new float[]{
(float) -0.0043417793, (float) -0.004025369, (float) 1.8009785E-4,
(float) 5.522854E-4, (float) -2.9725596E-4, (float) 0.0038219264,
(float) 0.0057955547, (float) -0.0036915164, (float) 1.2905941E-5,
(float) -0.0012608414, (float) 0.0075167217, (float) 1.2714228E-4,
(float) 0.004510221, (float) -0.0030373763, (float) -0.0033150043,
(float) -0.0027220408, (float) 0.0049406015, (float) 0.007475855,
(float) -0.0039889063, (float) 5.387217E-4, (float) 3.014746E-4,
(float) -0.0025138916, (float) -0.0014927724, (float) 0.0033432362,
(float) 0.0027196375, (float) -0.001453709, (float) -0.004362245,
(float) 0.0062709767, (float) 5.681349E-4, (float) 2.963205E-4, (float) 0.002127562,
(float) -0.0025758513, (float) -0.0015946038, (float) 0.0020683268,
(float) 0.004608029, (float) -0.006912731, (float) -0.003569094,
(float) 0.0029314745, (float) -0.0044829296, (float) -0.004087928,
(float) -3.7728698E-4, (float) -0.0040272907, (float) -0.006466153,
(float) 2.1587547E-4, (float) -4.334211E-5, (float) 0.0013268286,
(float) -1.1723964E-4, (float) 0.0017377065, (float) -0.009606785,
(float) -0.0059685633, (float) 0.0061167465, (float) 0.00976628,
(float) 0.0045020734, (float) 0.0072684726, (float) -0.002317661,
(float) 0.0030898168, (float) 0.0013212592, (float) 0.0017718632,
(float) 0.002785933, (float) 4.135881E-4, (float) -0.007407679,
(float) -0.008016254, (float) -0.0015525677, (float) -5.22596E-4,
(float) 0.003450544, (float) -1.4363142E-4, (float) -0.0055779675,
(float) -0.002204401, (float) 3.5834382E-4, (float) -0.0043447977,
(float) 0.0052861, (float) 0.0024472543, (float) 0.0019035664,
(float) -0.0010579216, (float) 0.008568893, (float) -0.0025444124,
(float) 0.0041700895, (float) 0.002440465, (float) -9.898118E-4,
(float) -0.004972163, (float) 0.00445475, (float) 0.0028563882,
(float) -6.568626E-4, (float) 0.0019806502, (float) 0.0021152704,
(float) -8.9459366E-4, (float) -5.853446E-4, (float) 0.006775423,
(float) -6.2033796E-5, (float) -0.0016326059, (float) 0.0028676696,
(float) -0.0020935084, (float) 0.0012473571, (float) -0.00658647,
(float) -2.9175522E-4, (float) -0.004172817, (float) -9.5688103E-4,
(float) 0.0029572574, (float) 0.0013865299, (float) -0.001356384
};
String key1 = "A2E2CD3EEFF31AE7A2EC455D2D23F8B2";
String key2 = "28C711B859926E05576CAF5084B4D66C";
xVectors.put(key1, exp1);
xVectors.put(key2, exp2);
xVectors.close();
ChronicleMap<CharSequence, float[]> xVectors2 = ChronicleMap
.of(CharSequence.class, float[].class)
// use the actual UUID size
.constantKeySizeBySample("2B6EC73CD63ACA5D93F4D5A710AD9CFE")
.constantValueSizeBySample(new float[100])
.entries(10)
.putReturnsNull(true)
.recoverPersistedTo(mapFile, true);
assertArrayEquals(exp1, xVectors2.get(key1), 0.0f);
assertArrayEquals(exp2, xVectors2.get(key2), 0.0f);
}
void testChronicleMap() throws IOException {
int num = 1_000_000;
testChronicleMap(System.getProperty("java.io.tmpdir"), num, num);
ThreadLocalRandom random = ThreadLocalRandom.current();
for (int i = 0; i < num; i++) {
String id = UUID.randomUUID().toString().substring(0, 32);
setUserVector(id, new float[random.nextInt(50, 150)]);
String id2 = UUID.randomUUID().toString().substring(0, 9);
setItemVector(id2, new float[random.nextInt(50, 150)]);
ArrayList<CharSequence> items = new ArrayList<>();
for (int j = 0; j < random.nextInt(50, 150); j++) {
items.add("average sized known item");
}
addKnownItems(knownItems, id, items);
}
}
void testChronicleMap(String persistToDir,
int numXIDs,
int numYIDs) throws IOException {
if (!Files.exists(Paths.get(persistToDir))) Files.createDirectory(Paths.get(persistToDir));
Path knownItemsPath = Paths.get(persistToDir + "/I-known-items.dat");
System.err.println("Loading knownItems");
ArrayList<CharSequence> averageKnownItems = new ArrayList<>();
for (int i = 0; i < 100; i++) {
averageKnownItems.add("average sized known item");
}
ChronicleMapBuilder<CharSequence, List<CharSequence>> knownItemsBuilder = ChronicleMap
.of(CharSequence.class, (Class<List<CharSequence>>) ((Class) List.class))
.averageKey("2B6EC73CD63ACA5D93F4D5A710AD9CFE")
.averageValue(averageKnownItems)
.valueMarshaller(ListMarshaller.of(
CharSequenceBytesReader.INSTANCE, CharSequenceBytesWriter.INSTANCE))
.entries(numXIDs)
.maxBloatFactor(5.0)
.putReturnsNull(true);
if (Files.exists(knownItemsPath)) {
knownItems = knownItemsBuilder.recoverPersistedTo(knownItemsPath.toFile(), true);
} else {
knownItems = knownItemsBuilder.createPersistedTo(knownItemsPath.toFile());
}
System.err.println("Loading xVectors");
Path xVectorsPath = Paths.get(persistToDir + "/" + "X-vectors.dat");
ChronicleMapBuilder<CharSequence, float[]> xVectorsBuilder = ChronicleMap
.of(CharSequence.class, float[].class)
.averageKey("2B6EC73CD63ACA5D93F4D5A710AD9CFE") //use the actual UUID size
.averageValue(new float[100])
.putReturnsNull(true)
.entries(numXIDs);
if (Files.exists(xVectorsPath)) {
xVectors = xVectorsBuilder.recoverPersistedTo(xVectorsPath.toFile(), true);
} else {
xVectors = xVectorsBuilder.createPersistedTo(xVectorsPath.toFile());
}
System.err.println("Loading xRecentIds");
Path xRecentIDsPath = Paths.get(persistToDir + "/" + "X-recent-ids.dat");
ChronicleSetBuilder<CharSequence> xRecentBuilder = ChronicleSet
.of(CharSequence.class)
.entries(numXIDs)
.averageKey("2B6EC73CD63ACA5D93F4D5A710AD9CFE"); //use the actual UUID size
if (Files.exists(xRecentIDsPath)) {
xRecentIDs = xRecentBuilder.recoverPersistedTo(xRecentIDsPath.toFile(), true);
} else {
xRecentIDs = xRecentBuilder.createPersistedTo(xRecentIDsPath.toFile());
}
System.err.println("Loading yVectors");
Path yVectorsPath = Paths.get(persistToDir + "/" + "Y-vectors.dat");
ChronicleMapBuilder<CharSequence, float[]> yVectorsBuilder = ChronicleMap
.of(CharSequence.class, float[].class)
.averageKey("198321433")
.averageValue(new float[100])
.putReturnsNull(true)
.entries(numYIDs);
if (Files.exists(yVectorsPath)) {
yVectors = yVectorsBuilder.recoverPersistedTo(yVectorsPath.toFile(), true);
} else {
yVectors = yVectorsBuilder.createPersistedTo(yVectorsPath.toFile());
}
System.err.println("Loading yRecentIDs");
Path yRecentIDsPath = Paths.get(persistToDir + "/" + "Y-recent-ids.dat");
ChronicleSetBuilder<CharSequence> yRecentBuilder = ChronicleSet
.of(CharSequence.class)
.averageKey("198321433")
.entries(numYIDs);
if (Files.exists(yRecentIDsPath)) {
yRecentIDs = yRecentBuilder.recoverPersistedTo(yRecentIDsPath.toFile(), true);
} else {
yRecentIDs = yRecentBuilder.createPersistedTo(yRecentIDsPath.toFile());
}
}
public void setUserVector(String id, float[] arr) {
putIntoMapAndRecentSet(xVectors, xRecentIDs, id, arr);
}
public void setItemVector(String id, float[] arr) {
putIntoMapAndRecentSet(yVectors, yRecentIDs, id, arr);
}
public void addKnownItems(ChronicleMap<CharSequence, List<CharSequence>> knownItems,
String id, List<CharSequence> items) {
try (ExternalMapQueryContext<CharSequence, List<CharSequence>, ?> c =
knownItems.queryContext(id)) {
if (c.writeLock().tryLock(1, MINUTES)) {
putNoReturn(c, items);
} else {
throw new RuntimeException("Dead lock");
}
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
}
@Test
public void testKnownItems() throws IOException {
ArrayList<CharSequence> averageKnownItems = new ArrayList<>();
for (int i = 0; i < 100; i++) {
averageKnownItems.add("average sized known item");
}
Path knownItemsPath = Paths.get(
System.getProperty("java.io.tmpdir") + "/test-vectors2.dat");
Files.deleteIfExists(knownItemsPath);
ChronicleMap<CharSequence, List<CharSequence>> knownItems;
ChronicleMapBuilder<CharSequence, List<CharSequence>> knownItemsBuilder = ChronicleMap
.of(CharSequence.class, (Class<List<CharSequence>>) ((Class) List.class))
.averageKey("2B6EC73CD63ACA5D93F4D5A710AD9CFE")
.averageValue(averageKnownItems)
.valueMarshaller(new ListMarshaller<>(
CharSequenceBytesReader.INSTANCE, CharSequenceBytesWriter.INSTANCE))
.entries(20)
.maxBloatFactor(5.0)
.putReturnsNull(true);
File mapFile = knownItemsPath.toFile();
mapFile.deleteOnExit();
knownItems = knownItemsBuilder.createPersistedTo(mapFile);
ArrayList<CharSequence> ids = new ArrayList<>();
for (int i = 0; i < 5; i++) {
String id = UUID.randomUUID().toString();
ids.add(id);
addKnownItems(knownItems, id, averageKnownItems);
}
knownItems.close();
final ChronicleMap<CharSequence, List<CharSequence>> knownItems2 =
knownItemsBuilder.recoverPersistedTo(mapFile, true);
assertEquals(5, knownItems2.size());
/* ids.forEach((id) -> {
System.out.println(knownItems2.get(id.subSequence(0, id.length())));
});*/
knownItems2.forEach((id, list) -> {
System.out.println(id + " : " + String.join(",", list));
});
}
}