| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| package org.apache.felix.dm.lambda.itest; |
| |
| import static org.apache.felix.dm.lambda.DependencyManagerActivator.aspect; |
| import static org.apache.felix.dm.lambda.DependencyManagerActivator.component; |
| |
| import org.apache.felix.dm.Component; |
| import org.apache.felix.dm.DependencyManager; |
| |
| |
| /** |
| * @author <a href="mailto:dev@felix.apache.org">Felix Project Team</a> |
| */ |
| public class AspectWithCallbacksServiceDependencyTest extends TestBase { |
| public void testServiceRegistrationAndConsumption() { |
| DependencyManager m = getDM(); |
| // helper class that ensures certain steps get executed in sequence |
| Ensure e = new Ensure(); |
| // create a service provider and consumer |
| Component sp = component(m).impl(new ServiceProvider(e)).provides(ServiceInterface.class.getName()).build(); |
| Component sc = component(m).impl(new ServiceConsumer(e)).withSrv(ServiceInterface.class, s->s.cb("add", "remove")).build(); |
| Component asp = aspect(m, ServiceInterface.class).rank(100).cb("add", null, "remove", "swap").impl(ServiceProviderAspect.class).build(); |
| |
| m.add(sp); |
| m.add(sc); |
| m.add(asp); |
| m.remove(asp); |
| m.remove(sc); |
| m.remove(sp); |
| |
| // ensure we executed all steps inside the component instance |
| e.step(8); |
| } |
| |
| public void testServiceRegistrationAndConsumptionRef() { |
| DependencyManager m = getDM(); |
| // helper class that ensures certain steps get executed in sequence |
| Ensure e = new Ensure(); |
| // create a service provider and consumer |
| Component sp = component(m).impl(new ServiceProvider(e)).provides(ServiceInterface.class.getName()).build(); |
| Component sc = component(m).impl(new ServiceConsumer(e)).withSrv(ServiceInterface.class, s->s.cb(ServiceConsumer::add, ServiceConsumer::remove)).build(); |
| Component asp = aspect(m, ServiceInterface.class) |
| .impl(ServiceProviderAspect.class).rank(100) |
| .cb(ServiceProviderAspect::add, ServiceProviderAspect::remove).sw(ServiceProviderAspect::swap) |
| .build(); |
| |
| m.add(sp); |
| m.add(sc); |
| m.add(asp); |
| m.remove(asp); |
| m.remove(sc); |
| m.remove(sp); |
| |
| // ensure we executed all steps inside the component instance |
| e.step(8); |
| } |
| |
| public void testServiceRegistrationAndConsumptionWithAspectCallbackInstance() { |
| DependencyManager m = getDM(); |
| // helper class that ensures certain steps get executed in sequence |
| Ensure e = new Ensure(); |
| // create a service provider and consumer |
| Component sp = component(m).impl(new ServiceProvider(e)).provides(ServiceInterface.class).build(); |
| Component sc = component(m).impl(new ServiceConsumer(e)).withSrv(ServiceInterface.class, s->s.cb("add", "remove")).build(); |
| |
| ServiceProviderAspect providerAspect = new ServiceProviderAspect(); |
| ServiceProviderAspectCallbackInstance aspectCb = new ServiceProviderAspectCallbackInstance(providerAspect); |
| Component asp = aspect(m, ServiceInterface.class).rank(100).impl(providerAspect).cbi(aspectCb::add, aspectCb::remove).swi(aspectCb::swap).build(); |
| |
| m.add(sp); |
| m.add(sc); |
| m.add(asp); |
| m.remove(asp); |
| m.remove(sc); |
| m.remove(sp); |
| |
| // ensure we executed all steps inside the component instance |
| e.step(8); |
| } |
| |
| static interface ServiceInterface { |
| public void invoke(String caller); |
| } |
| |
| static class ServiceProvider implements ServiceInterface { |
| private final Ensure m_ensure; |
| public ServiceProvider(Ensure e) { |
| m_ensure = e; |
| } |
| public void invoke(String caller) { |
| if (caller.equals("consumer.init")) { |
| m_ensure.step(3); |
| } else if (caller.equals("aspect.consumer.add")) { |
| m_ensure.step(5); |
| } |
| } |
| } |
| |
| public static class ServiceProviderAspectCallbackInstance { |
| private final ServiceProviderAspect m_aspect; |
| |
| ServiceProviderAspectCallbackInstance(ServiceProviderAspect aspect) { |
| m_aspect = aspect; |
| } |
| |
| public void add(ServiceInterface service) { |
| m_aspect.add(service); |
| } |
| |
| public void remove(ServiceInterface service) { |
| m_aspect.remove(service); |
| } |
| |
| public void swap(ServiceInterface previous, ServiceInterface current) { |
| m_aspect.swap(previous, current); |
| } |
| } |
| |
| static class ServiceProviderAspect implements ServiceInterface { |
| private volatile ServiceInterface m_service; |
| |
| public ServiceProviderAspect() { |
| } |
| |
| @Override |
| public void invoke(String caller) { |
| m_service.invoke("aspect." + caller); |
| } |
| |
| public void add(ServiceInterface service) { |
| m_service = service; |
| } |
| |
| public void remove(ServiceInterface service) { |
| m_service = null; |
| } |
| |
| public void swap(ServiceInterface previous, ServiceInterface current) { |
| m_service = current; |
| } |
| } |
| |
| static class ServiceConsumer { |
| private volatile ServiceInterface m_service; |
| private final Ensure m_ensure; |
| private int addCount = 0; |
| |
| public ServiceConsumer(Ensure e) { |
| m_ensure = e; |
| } |
| |
| public void init() { |
| m_ensure.step(2); |
| m_service.invoke("consumer.init"); |
| } |
| |
| public void destroy() { |
| m_ensure.step(7); |
| } |
| |
| public void add(ServiceInterface service) { |
| m_service = service; |
| switch (addCount) { |
| case 0: m_ensure.step(1); |
| break; |
| case 1: m_ensure.step(4); |
| // aspect had been added |
| m_service.invoke("consumer.add"); |
| break; |
| case 2: m_ensure.step(6); |
| break; |
| default: |
| } |
| addCount ++; |
| } |
| public void remove(ServiceInterface service) { |
| m_service = null; |
| } |
| } |
| |
| } |