Examples with ResourceQuotaCache com.yahoo.pulsar.broker.cache.ResourceQuotaCache used on opensource projects

Example 1 with ResourceQuotaCache

use of com.yahoo.pulsar.broker.cache.ResourceQuotaCache in project pulsar by yahoo.

the class ResourceQuotaCacheTest method testGetSetBundleQuota.

@Test
public void testGetSetBundleQuota() throws Exception {
    ResourceQuotaCache cache = new ResourceQuotaCache(zkCache);
    NamespaceBundle testBundle = bundleFactory.getFullBundle(new NamespaceName("pulsar/test/ns-2"));
    ResourceQuota quota1 = ResourceQuotaCache.getInitialQuotaValue();
    ResourceQuota quota2 = new ResourceQuota();
    quota2.setMsgRateIn(10);
    quota2.setMsgRateOut(20);
    quota2.setBandwidthIn(10000);
    quota2.setBandwidthOut(20000);
    quota2.setMemory(100);
    quota2.setDynamic(false);
    assertEquals(cache.getQuota(testBundle), quota1);
    cache.setQuota(testBundle, quota2);
    assertEquals(cache.getQuota(testBundle), quota2);
    cache.unsetQuota(testBundle);
    assertEquals(cache.getQuota(testBundle), quota1);
}

Example 2 with ResourceQuotaCache

use of com.yahoo.pulsar.broker.cache.ResourceQuotaCache in project pulsar by yahoo.

the class ResourceQuotaCacheTest method testGetSetDefaultQuota.

@Test
public void testGetSetDefaultQuota() throws Exception {
    ResourceQuotaCache cache = new ResourceQuotaCache(zkCache);
    ResourceQuota quota1 = ResourceQuotaCache.getInitialQuotaValue();
    ResourceQuota quota2 = new ResourceQuota();
    quota2.setMsgRateIn(10);
    quota2.setMsgRateOut(20);
    quota2.setBandwidthIn(10000);
    quota2.setBandwidthOut(20000);
    quota2.setMemory(100);
    quota2.setDynamic(false);
    assertEquals(cache.getDefaultQuota(), quota1);
    cache.setDefaultQuota(quota2);
    assertEquals(cache.getDefaultQuota(), quota2);
}