Examples with DefaultServiceFilter com.alipay.sofa.ark.container.registry.DefaultServiceFilter used on opensource projects

Example 1 with DefaultServiceFilter

use of com.alipay.sofa.ark.container.registry.DefaultServiceFilter in project sofa-ark by alipay.

the class ServiceRegistrationTest method testFilter.

@Test
@SuppressWarnings("unchecked")
public void testFilter() {
    final PluginModel pluginA = new PluginModel();
    pluginA.setPluginName("plugin A").setPriority("10");
    PluginModel pluginB = new PluginModel();
    pluginB.setPluginName("plugin B").setPriority("1");
    pluginManagerService.registerPlugin(pluginA);
    pluginManagerService.registerPlugin(pluginB);
    registryService.publishService(ITest.class, new TestObjectA(), new PluginServiceProvider(pluginA));
    registryService.publishService(ITest.class, new TestObjectB(), new PluginServiceProvider(pluginB));
    registryService.publishService(ITest.class, new TestObjectC(), new ContainerServiceProvider());
    List<ServiceReference> references = registryService.referenceServices(new DefaultServiceFilter().setServiceInterface(ITest.class).setProviderType(ServiceProviderType.ARK_PLUGIN));
    Assert.assertTrue(2 == references.size());
    PluginServiceProvider provider = (PluginServiceProvider) references.get(0).getServiceMetadata().getServiceProvider();
    Assert.assertEquals(pluginB.getPluginName(), provider.getPluginName());
    references = registryService.referenceServices(new ServiceFilter() {

        @Override
        public boolean match(ServiceReference serviceReference) {
            ServiceProvider serviceProvider = serviceReference.getServiceMetadata().getServiceProvider();
            if (serviceProvider instanceof PluginServiceProvider) {
                if (((PluginServiceProvider) serviceProvider).getPluginName().equals(pluginA.getPluginName())) {
                    return true;
                }
            }
            return false;
        }
    });
    Assert.assertTrue(1 == references.size());
    provider = (PluginServiceProvider) references.get(0).getServiceMetadata().getServiceProvider();
    Assert.assertEquals(pluginA.getPluginName(), provider.getPluginName());
    references = registryService.referenceServices(new DefaultServiceFilter().setServiceInterface(ITest.class));
    Assert.assertTrue(3 == references.size());
    references = registryService.referenceServices(new DefaultServiceFilter().setProviderType(ServiceProviderType.ARK_CONTAINER).setServiceInterface(ITest.class));
    Assert.assertTrue(1 == references.size());
    Assert.assertEquals("TestObject C", ((TestObjectC) references.get(0).getService()).test());
}

Example 2 with DefaultServiceFilter

use of com.alipay.sofa.ark.container.registry.DefaultServiceFilter in project sofa-ark by alipay.

the class ServiceRegistrationTest method testPublishService.

@Test
@SuppressWarnings("unchecked")
public void testPublishService() {
    ServiceReference<ITest> iTestServiceReference = registryService.publishService(ITest.class, new TestObjectA(), new ContainerServiceProvider());
    Assert.assertNotNull(iTestServiceReference);
    Assert.assertEquals(TestObjectA.OUTPUT, iTestServiceReference.getService().test());
    int c = registryService.unPublishServices(new DefaultServiceFilter().setServiceInterface(ITest.class).setProviderType(ServiceProviderType.ARK_CONTAINER));
    Assert.assertTrue(c == 1);
    iTestServiceReference = registryService.referenceService(ITest.class);
    Assert.assertNull(iTestServiceReference);
}

Example 3 with DefaultServiceFilter

use of com.alipay.sofa.ark.container.registry.DefaultServiceFilter in project sofa-ark by alipay.

the class ServiceRegistrationTest method testMultipleService.

@Test
@SuppressWarnings("unchecked")
public void testMultipleService() throws Exception {
    // 非插件发布的服务，优先级别最低
    registryService.publishService(ITest.class, new TestObjectA(), new ContainerServiceProvider());
    PluginModel pluginA = new PluginModel();
    pluginA.setPluginName("plugin A").setPriority("10").setClassPath(new URL[] { classPathURL }).setImportClasses(StringUtils.EMPTY_STRING).setImportPackages(StringUtils.EMPTY_STRING).setExportPackages(ClassUtils.getPackageName(INTERFACE_CLASS)).setExportClasses("").setImportResources(StringUtils.EMPTY_STRING).setExportResources(StringUtils.EMPTY_STRING).setPluginActivator(PluginActivatorA.class.getName()).setPluginClassLoader(new PluginClassLoader(pluginA.getPluginName(), pluginA.getClassPath())).setPluginContext(new PluginContextImpl(pluginA));
    PluginModel pluginB = new PluginModel();
    pluginB.setPluginName("plugin B").setPriority("1").setClassPath(new URL[] { classPathURL }).setImportClasses(INTERFACE_CLASS).setImportPackages(StringUtils.EMPTY_STRING).setExportPackages("").setExportClasses("").setImportResources(StringUtils.EMPTY_STRING).setExportResources(StringUtils.EMPTY_STRING).setPluginActivator(PluginActivatorB.class.getName()).setPluginClassLoader(new PluginClassLoader(pluginB.getPluginName(), pluginB.getClassPath())).setPluginContext(new PluginContextImpl(pluginB));
    PluginModel pluginC = new PluginModel();
    pluginC.setPluginName("plugin C").setPriority("100").setClassPath(new URL[] { classPathURL }).setImportClasses(INTERFACE_CLASS).setImportPackages(StringUtils.EMPTY_STRING).setExportPackages("").setExportClasses("").setImportResources(StringUtils.EMPTY_STRING).setExportResources(StringUtils.EMPTY_STRING).setPluginActivator(PluginActivatorC.class.getName()).setPluginClassLoader(new PluginClassLoader(pluginC.getPluginName(), pluginC.getClassPath())).setPluginContext(new PluginContextImpl(pluginC));
    pluginManagerService.registerPlugin(pluginA);
    pluginManagerService.registerPlugin(pluginB);
    pluginManagerService.registerPlugin(pluginC);
    classloaderService.prepareExportClassAndResourceCache();
    pluginDeployService.deploy();
    Class iTest = pluginA.getPluginClassLoader().loadClass(ITest.class.getCanonicalName());
    Assert.assertEquals(3, pluginA.getPluginContext().referenceServices(new DefaultServiceFilter().setServiceInterface(iTest)).size());
    // 应该获取到优先级别比较高的服务
    ServiceReference reference = pluginC.getPluginContext().referenceService(iTest);
    PluginServiceProvider provider = (PluginServiceProvider) reference.getServiceMetadata().getServiceProvider();
    Assert.assertEquals(pluginB.getPluginName(), provider.getPluginName());
    List<ServiceReference> references = pluginC.getPluginContext().referenceServices(new DefaultServiceFilter().setServiceInterface(iTest));
    provider = (PluginServiceProvider) references.get(0).getServiceMetadata().getServiceProvider();
    Assert.assertEquals(pluginB.getPluginName(), provider.getPluginName());
}

Example 4 with DefaultServiceFilter

use of com.alipay.sofa.ark.container.registry.DefaultServiceFilter in project sofa-ark by alipay.

the class ServiceRegistrationTest method testPublishDuplicateService.

@Test
@SuppressWarnings("unchecked")
public void testPublishDuplicateService() {
    registryService.publishService(ITest.class, new TestObjectA(), new ContainerServiceProvider());
    registryService.publishService(ITest.class, new TestObjectB(), new ContainerServiceProvider());
    // 只有第一个服务发布成功
    Assert.assertEquals(1, registryService.referenceServices(ITest.class).size());
    Assert.assertEquals(TestObjectA.OUTPUT, registryService.referenceService(ITest.class).getService().test());
    registryService.unPublishServices(new DefaultServiceFilter().setServiceInterface(ITest.class));
    Assert.assertEquals(0, registryService.referenceServices(ITest.class).size());
    registryService.publishService(ITest.class, new TestObjectA(), "testA", new ContainerServiceProvider());
    registryService.publishService(ITest.class, new TestObjectB(), "testB", new ContainerServiceProvider());
    Assert.assertEquals(2, registryService.referenceServices(new DefaultServiceFilter().setServiceInterface(ITest.class)).size());
    Assert.assertEquals(TestObjectA.OUTPUT, registryService.referenceService(ITest.class, "testA").getService().test());
    Assert.assertEquals(TestObjectB.OUTPUT, registryService.referenceService(ITest.class, "testB").getService().test());
    int c = registryService.unPublishServices(new DefaultServiceFilter().setUniqueId("testA"));
    Assert.assertTrue(c == 1);
    c = registryService.unPublishServices(new DefaultServiceFilter().setProviderType(ServiceProviderType.ARK_CONTAINER).setServiceInterface(ITest.class));
    Assert.assertTrue(c == 1);
    Assert.assertEquals(0, registryService.referenceServices(ITest.class).size());
}

Example 5 with DefaultServiceFilter

use of com.alipay.sofa.ark.container.registry.DefaultServiceFilter in project sofa-ark by alipay.

the class ServiceRegistrationTest method testPublishDuplicateServiceInPlugin.

@Test
public void testPublishDuplicateServiceInPlugin() throws Exception {
    PluginModel pluginA = new PluginModel();
    pluginA.setPluginName("plugin A").setPriority("10").setClassPath(new URL[] { classPathURL }).setImportClasses(StringUtils.EMPTY_STRING).setImportPackages(StringUtils.EMPTY_STRING).setExportPackages(ClassUtils.getPackageName(INTERFACE_CLASS)).setExportClasses(StringUtils.EMPTY_STRING).setImportResources(StringUtils.EMPTY_STRING).setExportResources(StringUtils.EMPTY_STRING).setPluginActivator(PluginActivatorADup.class.getName()).setPluginClassLoader(new PluginClassLoader(pluginA.getPluginName(), pluginA.getClassPath())).setPluginContext(new PluginContextImpl(pluginA));
    pluginManagerService.registerPlugin(pluginA);
    classloaderService.prepareExportClassAndResourceCache();
    pluginDeployService.deploy();
    Assert.assertEquals(1, registryService.referenceServices(pluginA.getPluginClassLoader().loadClass(ITest.class.getCanonicalName())).size());
    int c = registryService.unPublishServices(new DefaultServiceFilter().setProviderType(ServiceProviderType.ARK_PLUGIN));
    Assert.assertTrue(c == 1);
}