Examples with CacheManager com.baidu.hugegraph.backend.cache.CacheManager used on opensource projects

Example 1 with CacheManager

use of com.baidu.hugegraph.backend.cache.CacheManager in project incubator-hugegraph by apache.

the class CacheManagerTest method testCacheInstance.

@Test
public void testCacheInstance() {
    // Don't mock
    teardown();
    CacheManager manager = CacheManager.instance();
    Cache<Id, Object> c1 = manager.cache("c1");
    Cache<Id, Object> c12 = manager.cache("c1");
    Cache<Id, Object> c13 = manager.cache("c1", 123);
    Assert.assertEquals(c1, c12);
    Assert.assertEquals(c1, c13);
    Assert.assertEquals(c1.capacity(), c13.capacity());
    Cache<Id, Object> c2 = manager.offheapCache(null, "c2", 1, 11);
    Cache<Id, Object> c22 = manager.offheapCache(null, "c2", 2, 22);
    Cache<Id, Object> c23 = manager.offheapCache(null, "c2", 3, 33);
    Assert.assertEquals(c2, c22);
    Assert.assertEquals(c2, c23);
    Assert.assertEquals(c2.capacity(), c23.capacity());
    Cache<Id, Object> c3 = manager.levelCache(null, "c3", 1, 1, 11);
    Cache<Id, Object> c32 = manager.levelCache(null, "c3", 2, 2, 22);
    Cache<Id, Object> c33 = manager.levelCache(null, "c3", 3, 3, 33);
    Assert.assertEquals(c3, c32);
    Assert.assertEquals(c3, c33);
    Assert.assertEquals(c3.capacity(), c33.capacity());
    Assert.assertThrows(IllegalArgumentException.class, () -> {
        manager.cache("c2");
    }, e -> {
        Assert.assertContains("Invalid cache implement:", e.getMessage());
        Assert.assertContains("OffheapCache", e.getMessage());
    });
    Assert.assertThrows(IllegalArgumentException.class, () -> {
        manager.cache("c3");
    }, e -> {
        Assert.assertContains("Invalid cache implement:", e.getMessage());
        Assert.assertContains("LevelCache", e.getMessage());
    });
    Assert.assertThrows(IllegalArgumentException.class, () -> {
        manager.offheapCache(null, "c1", 1, 11);
    }, e -> {
        Assert.assertContains("Invalid cache implement:", e.getMessage());
        Assert.assertContains("RamCache", e.getMessage());
    });
    Assert.assertThrows(IllegalArgumentException.class, () -> {
        manager.offheapCache(null, "c3", 1, 11);
    }, e -> {
        Assert.assertContains("Invalid cache implement:", e.getMessage());
        Assert.assertContains("LevelCache", e.getMessage());
    });
    Assert.assertThrows(IllegalArgumentException.class, () -> {
        manager.levelCache(null, "c1", 1, 1, 11);
    }, e -> {
        Assert.assertContains("Invalid cache implement:", e.getMessage());
        Assert.assertContains("RamCache", e.getMessage());
    });
    Assert.assertThrows(IllegalArgumentException.class, () -> {
        manager.levelCache(null, "c2", 1, 1, 11);
    }, e -> {
        Assert.assertContains("Invalid cache implement:", e.getMessage());
        Assert.assertContains("OffheapCache", e.getMessage());
    });
    this.originCaches.remove("c1");
    this.originCaches.remove("c2");
    this.originCaches.remove("c3");
}

Example 2 with CacheManager

use of com.baidu.hugegraph.backend.cache.CacheManager in project incubator-hugegraph by apache.

the class CacheManagerTest method setup.

@SuppressWarnings({ "unchecked", "rawtypes" })
@Before
public void setup() {
    CacheManager manager = CacheManager.instance();
    // Mock caches
    this.mockCaches = Mockito.mock(ConcurrentHashMap.class);
    this.originCaches = (Map) Whitebox.getInternalState(manager, "caches");
    Whitebox.setInternalState(manager, "caches", this.mockCaches);
}

Example 3 with CacheManager

use of com.baidu.hugegraph.backend.cache.CacheManager in project incubator-hugegraph by apache.

the class CacheManagerTest method testCacheEnableMetrics.

@Test
public void testCacheEnableMetrics() {
    // Don't mock
    teardown();
    CacheManager manager = CacheManager.instance();
    Cache<Id, Object> c1 = manager.cache("m1");
    Cache<Id, Object> c2 = manager.cache("m2");
    Cache<Id, Object> c3 = manager.offheapCache(null, "m3", 1, 11);
    Cache<Id, Object> c4 = manager.levelCache(null, "m4", 1, 1, 11);
    Assert.assertEquals(false, c1.enableMetrics(false));
    Assert.assertEquals(false, c2.enableMetrics(false));
    Assert.assertEquals(false, c3.enableMetrics(false));
    Assert.assertEquals(false, c4.enableMetrics(false));
    Assert.assertEquals(false, CacheManager.cacheEnableMetrics("m1", true));
    Assert.assertEquals(true, c1.enableMetrics(true));
    Assert.assertEquals(false, CacheManager.cacheEnableMetrics("m2", true));
    Assert.assertEquals(true, c2.enableMetrics(true));
    Assert.assertEquals(false, CacheManager.cacheEnableMetrics("m3", true));
    Assert.assertEquals(true, c3.enableMetrics(true));
    Assert.assertEquals(false, CacheManager.cacheEnableMetrics("m4", true));
    Assert.assertEquals(true, c4.enableMetrics(true));
    Assert.assertEquals(true, CacheManager.cacheEnableMetrics("m1", false));
    Assert.assertEquals(false, c1.enableMetrics(true));
    Assert.assertEquals(true, CacheManager.cacheEnableMetrics("m2", false));
    Assert.assertEquals(false, c2.enableMetrics(true));
    Assert.assertEquals(true, CacheManager.cacheEnableMetrics("m3", false));
    Assert.assertEquals(false, c3.enableMetrics(true));
    Assert.assertEquals(true, CacheManager.cacheEnableMetrics("m4", false));
    Assert.assertEquals(false, c4.enableMetrics(true));
}

Example 4 with CacheManager

use of com.baidu.hugegraph.backend.cache.CacheManager in project incubator-hugegraph by apache.

the class CacheManagerTest method testCachePutGetWithCapacity.

@Test
public void testCachePutGetWithCapacity() {
    final String name = "test-cache";
    final int capacity = 12345;
    CacheManager manager = CacheManager.instance();
    Mockito.when(this.mockCaches.containsKey(name)).thenReturn(false);
    @SuppressWarnings("rawtypes") final Cache[] cache = new Cache[1];
    Mockito.when(this.mockCaches.putIfAbsent(Mockito.anyString(), Mockito.any())).thenAnswer(i -> cache[0] = (Cache<?, ?>) i.getArguments()[1]);
    Mockito.when(this.mockCaches.get(name)).thenAnswer(i -> cache[0]);
    Cache<Id, Object> cache1 = manager.cache(name, capacity);
    Assert.assertNotNull(cache1);
    Assert.assertEquals(capacity, cache1.capacity());
    Mockito.verify(this.mockCaches).putIfAbsent(name, cache1);
    Mockito.when(this.mockCaches.containsKey(name)).thenReturn(true);
    Mockito.when(this.mockCaches.get(name)).thenReturn(cache1);
    Cache<Id, Object> cache2 = manager.cache(name, capacity);
    Assert.assertEquals(capacity, cache2.capacity());
    Assert.assertSame(cache1, cache2);
    Assert.assertSame(cache1, manager.cache(name));
    Assert.assertSame(cache1, manager.cache(name, 0));
    Assert.assertSame(cache1, manager.cache(name, 1));
    Assert.assertSame(cache1, manager.cache(name, capacity));
    Assert.assertSame(cache1, manager.cache(name, capacity + 10));
    Mockito.verify(this.mockCaches, Mockito.atMost(1)).putIfAbsent(Mockito.anyString(), Mockito.any());
}

Example 5 with CacheManager

use of com.baidu.hugegraph.backend.cache.CacheManager in project incubator-hugegraph by apache.

the class CacheManagerTest method testCacheList.

@Test
public void testCacheList() {
    CacheManager manager = CacheManager.instance();
    Cache<Id, Object> cache1 = Mockito.mock(RamCache.class);
    Cache<Id, Object> cache2 = Mockito.mock(OffheapCache.class);
    Mockito.when(this.mockCaches.get("cache-1")).thenReturn(cache1);
    Mockito.when(this.mockCaches.get("cache-2")).thenReturn(cache2);
    Mockito.when(this.mockCaches.size()).thenReturn(2);
    Map<String, Cache<Id, Object>> caches = manager.caches();
    Assert.assertEquals(2, caches.size());
    Assert.assertSame(cache1, caches.get("cache-1"));
    Assert.assertSame(cache2, caches.get("cache-2"));
    Assert.assertArrayEquals(this.mockCaches.values().toArray(), caches.values().toArray());
}