Adjust for Binding RPC codegen changes
[controller.git] / opendaylight / md-sal / samples / toaster-consumer / src / main / java / org / opendaylight / controller / sample / kitchen / impl / KitchenServiceImpl.java
1 /*
2  * Copyright (c) 2014, 2015 Brocade Communications Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.controller.sample.kitchen.impl;
9
10 import com.google.common.collect.ImmutableList;
11 import com.google.common.collect.ImmutableList.Builder;
12 import com.google.common.util.concurrent.Futures;
13 import com.google.common.util.concurrent.ListenableFuture;
14 import com.google.common.util.concurrent.ListeningExecutorService;
15 import com.google.common.util.concurrent.MoreExecutors;
16 import java.util.List;
17 import java.util.concurrent.ExecutionException;
18 import java.util.concurrent.Executors;
19 import java.util.concurrent.Future;
20 import org.opendaylight.controller.md.sal.common.util.jmx.AbstractMXBean;
21 import org.opendaylight.controller.sample.kitchen.api.EggsType;
22 import org.opendaylight.controller.sample.kitchen.api.KitchenService;
23 import org.opendaylight.controller.sample.kitchen.api.KitchenServiceRuntimeMXBean;
24 import org.opendaylight.yang.gen.v1.http.netconfcentral.org.ns.toaster.rev091120.MakeToastInput;
25 import org.opendaylight.yang.gen.v1.http.netconfcentral.org.ns.toaster.rev091120.MakeToastInputBuilder;
26 import org.opendaylight.yang.gen.v1.http.netconfcentral.org.ns.toaster.rev091120.MakeToastOutput;
27 import org.opendaylight.yang.gen.v1.http.netconfcentral.org.ns.toaster.rev091120.MakeToastOutputBuilder;
28 import org.opendaylight.yang.gen.v1.http.netconfcentral.org.ns.toaster.rev091120.ToastType;
29 import org.opendaylight.yang.gen.v1.http.netconfcentral.org.ns.toaster.rev091120.ToasterListener;
30 import org.opendaylight.yang.gen.v1.http.netconfcentral.org.ns.toaster.rev091120.ToasterOutOfBread;
31 import org.opendaylight.yang.gen.v1.http.netconfcentral.org.ns.toaster.rev091120.ToasterRestocked;
32 import org.opendaylight.yang.gen.v1.http.netconfcentral.org.ns.toaster.rev091120.ToasterService;
33 import org.opendaylight.yang.gen.v1.http.netconfcentral.org.ns.toaster.rev091120.WheatBread;
34 import org.opendaylight.yangtools.yang.common.RpcError;
35 import org.opendaylight.yangtools.yang.common.RpcError.ErrorType;
36 import org.opendaylight.yangtools.yang.common.RpcResult;
37 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
38 import org.slf4j.Logger;
39 import org.slf4j.LoggerFactory;
40
41 public class KitchenServiceImpl extends AbstractMXBean
42         implements KitchenService, KitchenServiceRuntimeMXBean, ToasterListener {
43
44     private static final Logger LOG = LoggerFactory.getLogger(KitchenServiceImpl.class);
45     private static final MakeToastOutput EMPTY_MAKE_OUTPUT = new MakeToastOutputBuilder().build();
46
47     private final ToasterService toaster;
48
49     private final ListeningExecutorService executor = MoreExecutors.listeningDecorator(Executors.newCachedThreadPool());
50
51     private volatile boolean toasterOutOfBread;
52
53     public KitchenServiceImpl(final ToasterService toaster) {
54         super("KitchenService", "toaster-consumer", null);
55         this.toaster = toaster;
56     }
57
58     @Override
59     public Future<RpcResult<Void>> makeBreakfast(final EggsType eggsType, final Class<? extends ToastType> toastType,
60             final int toastDoneness) {
61         // Call makeToast, The OpendaylightToaster impl already returns a ListenableFuture so the conversion is
62         // actually a no-op.
63
64         ListenableFuture<RpcResult<MakeToastOutput>> makeToastFuture = makeToast(toastType, toastDoneness);
65
66         ListenableFuture<RpcResult<Void>> makeEggsFuture = makeEggs(eggsType);
67
68         // Combine the 2 ListenableFutures into 1 containing a list RpcResults.
69
70         ListenableFuture<List<RpcResult<? extends Object>>> combinedFutures = Futures
71                 .allAsList(ImmutableList.of(makeToastFuture, makeEggsFuture));
72
73         // Then transform the RpcResults into 1.
74
75         return Futures.transformAsync(combinedFutures, results -> {
76             boolean atLeastOneSucceeded = false;
77             Builder<RpcError> errorList = ImmutableList.builder();
78             for (RpcResult<? extends Object> result : results) {
79                 if (result.isSuccessful()) {
80                     atLeastOneSucceeded = true;
81                 }
82
83                 if (result.getErrors() != null) {
84                     errorList.addAll(result.getErrors());
85                 }
86             }
87
88             return Futures.immediateFuture(RpcResultBuilder.<Void>status(atLeastOneSucceeded)
89                     .withRpcErrors(errorList.build()).build());
90         }, MoreExecutors.directExecutor());
91     }
92
93     private ListenableFuture<RpcResult<Void>> makeEggs(final EggsType eggsType) {
94         return executor.submit(() -> RpcResultBuilder.<Void>success().build());
95     }
96
97     private ListenableFuture<RpcResult<MakeToastOutput>> makeToast(final Class<? extends ToastType> toastType,
98             final int toastDoneness) {
99
100         if (toasterOutOfBread) {
101             LOG.info("We're out of toast but we can make eggs");
102             return Futures.immediateFuture(RpcResultBuilder.success(EMPTY_MAKE_OUTPUT)
103                 .withWarning(ErrorType.APPLICATION, "partial-operation",
104                     "Toaster is out of bread but we can make you eggs").build());
105         }
106
107         // Access the ToasterService to make the toast.
108
109         MakeToastInput toastInput = new MakeToastInputBuilder().setToasterDoneness((long) toastDoneness)
110                 .setToasterToastType(toastType).build();
111
112         return toaster.makeToast(toastInput);
113     }
114
115     @Override
116     public Boolean makeScrambledWithWheat() {
117         try {
118             // This call has to block since we must return a result to the JMX client.
119             RpcResult<Void> result = makeBreakfast(EggsType.SCRAMBLED, WheatBread.class, 2).get();
120             if (result.isSuccessful()) {
121                 LOG.info("makeBreakfast succeeded");
122             } else {
123                 LOG.warn("makeBreakfast failed: " + result.getErrors());
124             }
125
126             return result.isSuccessful();
127
128         } catch (InterruptedException | ExecutionException e) {
129             LOG.warn("An error occurred while maing breakfast: " + e);
130         }
131
132         return Boolean.FALSE;
133     }
134
135     /**
136      * Implemented from the ToasterListener interface.
137      */
138     @Override
139     public void onToasterOutOfBread(final ToasterOutOfBread notification) {
140         LOG.info("ToasterOutOfBread notification");
141         toasterOutOfBread = true;
142     }
143
144     /**
145      * Implemented from the ToasterListener interface.
146      */
147     @Override
148     public void onToasterRestocked(final ToasterRestocked notification) {
149         LOG.info("ToasterRestocked notification - amountOfBread: " + notification.getAmountOfBread());
150         toasterOutOfBread = false;
151     }
152 }

©2013 OpenDaylight, A Linux Foundation Collaborative Project. All Rights Reserved.
OpenDaylight is a registered trademark of The OpenDaylight Project, Inc.
Linux Foundation and OpenDaylight are registered trademarks of the Linux Foundation.
Linux is a registered trademark of Linus Torvalds.