Refactor some small TAPI rpcs
[transportpce.git] / tapi / src / main / java / org / opendaylight / transportpce / tapi / impl / rpc / GetConnectivityServiceListImpl.java
1 /*
2  * Copyright © 2024 Orange, 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.transportpce.tapi.impl.rpc;
9
10 import com.google.common.util.concurrent.ListenableFuture;
11 import java.util.HashMap;
12 import java.util.Map;
13 import org.opendaylight.transportpce.tapi.utils.TapiContext;
14 import org.opendaylight.yang.gen.v1.urn.onf.otcc.yang.tapi.connectivity.rev221121.GetConnectivityServiceList;
15 import org.opendaylight.yang.gen.v1.urn.onf.otcc.yang.tapi.connectivity.rev221121.GetConnectivityServiceListInput;
16 import org.opendaylight.yang.gen.v1.urn.onf.otcc.yang.tapi.connectivity.rev221121.GetConnectivityServiceListOutput;
17 import org.opendaylight.yang.gen.v1.urn.onf.otcc.yang.tapi.connectivity.rev221121.GetConnectivityServiceListOutputBuilder;
18 import org.opendaylight.yang.gen.v1.urn.onf.otcc.yang.tapi.connectivity.rev221121.connectivity.context.ConnectivityService;
19 import org.opendaylight.yang.gen.v1.urn.onf.otcc.yang.tapi.connectivity.rev221121.connectivity.context.ConnectivityServiceKey;
20 import org.opendaylight.yang.gen.v1.urn.onf.otcc.yang.tapi.connectivity.rev221121.get.connectivity.service.list.output.Service;
21 import org.opendaylight.yang.gen.v1.urn.onf.otcc.yang.tapi.connectivity.rev221121.get.connectivity.service.list.output.ServiceBuilder;
22 import org.opendaylight.yang.gen.v1.urn.onf.otcc.yang.tapi.connectivity.rev221121.get.connectivity.service.list.output.ServiceKey;
23 import org.opendaylight.yangtools.yang.common.ErrorType;
24 import org.opendaylight.yangtools.yang.common.RpcResult;
25 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
26 import org.slf4j.Logger;
27 import org.slf4j.LoggerFactory;
28
29
30 public class GetConnectivityServiceListImpl implements GetConnectivityServiceList {
31     private static final Logger LOG = LoggerFactory.getLogger(GetConnectivityServiceListImpl.class);
32
33     private final TapiContext tapiContext;
34
35     public GetConnectivityServiceListImpl(TapiContext tapiContext) {
36         this.tapiContext = tapiContext;
37     }
38
39     @Override
40     public ListenableFuture<RpcResult<GetConnectivityServiceListOutput>> invoke(GetConnectivityServiceListInput input) {
41         // TODO Auto-generated method stub
42         Map<ConnectivityServiceKey, ConnectivityService> connMap = this.tapiContext.getConnectivityServices();
43         if (connMap == null) {
44             LOG.error("No services in tapi context");
45             return RpcResultBuilder.<GetConnectivityServiceListOutput>failed()
46                 .withError(ErrorType.RPC, "No services exist in datastore")
47                 .buildFuture();
48         }
49
50         Map<ServiceKey, Service> serviceMap = new HashMap<>();
51         for (ConnectivityService connectivityService: connMap.values()) {
52             Service service = new ServiceBuilder(connectivityService).build();
53             serviceMap.put(service.key(), service);
54         }
55         return RpcResultBuilder
56             .success(new GetConnectivityServiceListOutputBuilder().setService(serviceMap).build())
57             .buildFuture();
58     }
59
60 }