/* * 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)); }); } }