openflowplugin-impl/src/main/java/org/opendaylight/openflowplugin/impl/services/SalTableServiceImpl.java

   1 /**
   2  * Copyright (c) 2015 Cisco 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.openflowplugin.impl.services;
   9
  10 import com.google.common.base.Function;
  11 import com.google.common.util.concurrent.FutureCallback;
  12 import com.google.common.util.concurrent.Futures;
  13 import com.google.common.util.concurrent.ListenableFuture;
  14 import com.google.common.util.concurrent.SettableFuture;
  15 import java.math.BigInteger;
  16 import java.util.ArrayList;
  17 import java.util.List;
  18 import java.util.concurrent.Future;
  19 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  20 import org.opendaylight.openflowjava.protocol.api.connection.OutboundQueue;
  21 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
  22 import org.opendaylight.openflowplugin.api.openflow.device.RequestContext;
  23 import org.opendaylight.openflowplugin.api.openflow.device.RequestContextStack;
  24 import org.opendaylight.openflowplugin.api.openflow.device.Xid;
  25 import org.opendaylight.openflowplugin.api.openflow.device.handlers.MultiMsgCollector;
  26 import org.opendaylight.openflowplugin.api.openflow.statistics.ofpspecific.MessageSpy;
  27 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.TableFeaturesConvertor;
  28 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.TableFeaturesReplyConvertor;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.Table;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.TableKey;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.TransactionId;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartRequestFlags;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReply;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartRequestInput;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartRequestInputBuilder;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.OfHeader;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.MultipartReplyBody;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyTableFeaturesCase;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.table.features._case.MultipartReplyTableFeatures;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.MultipartRequestTableFeaturesCaseBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.multipart.request.table.features._case.MultipartRequestTableFeaturesBuilder;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.request.multipart.request.body.multipart.request.table.features._case.multipart.request.table.features.TableFeatures;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.table.service.rev131026.SalTableService;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.table.service.rev131026.UpdateTableInput;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.table.service.rev131026.UpdateTableOutput;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.table.service.rev131026.UpdateTableOutputBuilder;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.table.types.rev131026.table.features.TableFeaturesKey;
  54 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  55 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
  56 import org.opendaylight.yangtools.yang.common.RpcError;
  57 import org.opendaylight.yangtools.yang.common.RpcError.ErrorType;
  58 import org.opendaylight.yangtools.yang.common.RpcResult;
  59 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
  60 import org.slf4j.Logger;
  61
  62 public class SalTableServiceImpl extends CommonService implements SalTableService {
  63
  64     private static final Logger LOG = org.slf4j.LoggerFactory.getLogger(SalTableServiceImpl.class);
  65
  66     public SalTableServiceImpl(final RequestContextStack requestContextStack, final DeviceContext deviceContext) {
  67         super(requestContextStack, deviceContext);
  68     }
  69
  70     @Override
  71     public Future<RpcResult<UpdateTableOutput>> updateTable(final UpdateTableInput input) {
  72         class FunctionImpl implements
  73                 Function<RequestContext<List<MultipartReply>>, ListenableFuture<RpcResult<List<MultipartReply>>>> {
  74
  75             @Override
  76             public ListenableFuture<RpcResult<List<MultipartReply>>> apply(final RequestContext<List<MultipartReply>> requestContext) {
  77                 getMessageSpy().spyMessage(input.getImplementedInterface(),
  78                         MessageSpy.STATISTIC_GROUP.TO_SWITCH_SUBMIT_SUCCESS);
  79
  80                 final SettableFuture<RpcResult<List<MultipartReply>>> result = SettableFuture.create();
  81
  82                 final MultipartRequestTableFeaturesCaseBuilder caseBuilder = new MultipartRequestTableFeaturesCaseBuilder();
  83                 final MultipartRequestTableFeaturesBuilder requestBuilder = new MultipartRequestTableFeaturesBuilder();
  84                 final List<TableFeatures> ofTableFeatureList = TableFeaturesConvertor.toTableFeaturesRequest(input
  85                         .getUpdatedTable());
  86                 requestBuilder.setTableFeatures(ofTableFeatureList);
  87                 caseBuilder.setMultipartRequestTableFeatures(requestBuilder.build());
  88
  89                 // Set request body to main multipart request
  90                 final Xid xid = requestContext.getXid();
  91                 final MultipartRequestInputBuilder mprInput = createMultipartHeader(MultipartType.OFPMPTABLEFEATURES,
  92                         xid.getValue());
  93                 mprInput.setMultipartRequestBody(caseBuilder.build());
  94                 final OutboundQueue outboundQueue = getDeviceContext().getPrimaryConnectionContext().getOutboundQueueProvider();
  95
  96                 final MultiMsgCollector multiMsgCollector = getDeviceContext().getMultiMsgCollector();
  97                 multiMsgCollector.registerMultipartRequestContext(requestContext);
  98
  99                 final MultipartRequestInput multipartRequestInput = mprInput.build();
 100                 outboundQueue.commitEntry(xid.getValue(), multipartRequestInput, new FutureCallback<OfHeader>() {
 101                     @Override
 102                     public void onSuccess(final OfHeader ofHeader) {
 103                         if (ofHeader instanceof MultipartReply) {
 104                             MultipartReply multipartReply = (MultipartReply) ofHeader;
 105                             multiMsgCollector.addMultipartMsg(multipartReply);
 106                         } else {
 107                             if (null != ofHeader) {
 108                                 LOG.info("Unexpected response type received {}.", ofHeader.getClass());
 109                             } else {
 110                                 LOG.info("Response received is null.");
 111                             }
 112                         }
 113                         getMessageSpy().spyMessage(multipartRequestInput.getImplementedInterface(), MessageSpy.STATISTIC_GROUP.TO_SWITCH_SUBMIT_SUCCESS);
 114                     }
 115
 116                     @Override
 117                     public void onFailure(final Throwable throwable) {
 118                         RpcResultBuilder<List<MultipartReply>> rpcResultBuilder = RpcResultBuilder.<List<MultipartReply>>failed().withError(RpcError.ErrorType.APPLICATION, throwable.getMessage(), throwable);
 119                         RequestContextUtil.closeRequstContext(requestContext);
 120                         multiMsgCollector.registerMultipartRequestContext(requestContext);
 121                         getMessageSpy().spyMessage(multipartRequestInput.getImplementedInterface(), MessageSpy.STATISTIC_GROUP.TO_SWITCH_SUBMIT_FAILURE);
 122                     }
 123                 });
 124                 return requestContext.getFuture();
 125             }
 126         }
 127
 128         final ListenableFuture<RpcResult<List<MultipartReply>>> multipartFuture = handleServiceCall(new FunctionImpl());
 129         final SettableFuture<RpcResult<UpdateTableOutput>> finalFuture = SettableFuture.create();
 130
 131         class CallBackImpl implements FutureCallback<RpcResult<List<MultipartReply>>> {
 132             private final Logger LOGGER = org.slf4j.LoggerFactory.getLogger(CallBackImpl.class);
 133
 134             @Override
 135             public void onSuccess(final RpcResult<List<MultipartReply>> result) {
 136
 137                 if (result.isSuccessful()) {
 138                     final List<MultipartReply> multipartReplies = result.getResult();
 139                     if (multipartReplies.isEmpty()) {
 140                         LOGGER.debug("Multipart reply to table features request shouldn't be empty list.");
 141                         finalFuture.set(RpcResultBuilder.<UpdateTableOutput>failed()
 142                                 .withError(ErrorType.RPC, "Multipart reply list is empty.").build());
 143                     } else {
 144                         final Long xid = multipartReplies.get(0).getXid();
 145                         LOGGER.debug(
 146                                 "OnSuccess, rpc result successful, multipart response for rpc update-table with xid {} obtained.",
 147                                 xid);
 148                         final UpdateTableOutputBuilder updateTableOutputBuilder = new UpdateTableOutputBuilder();
 149                         updateTableOutputBuilder.setTransactionId(new TransactionId(BigInteger.valueOf(xid)));
 150                         finalFuture.set(RpcResultBuilder.success(updateTableOutputBuilder.build()).build());
 151                         writeResponseToOperationalDatastore(multipartReplies);
 152                     }
 153                 } else {
 154                     LOGGER.debug("OnSuccess, rpc result unsuccessful, multipart response for rpc update-table was unsuccessful.");
 155                     finalFuture.set(RpcResultBuilder.<UpdateTableOutput>failed().withRpcErrors(result.getErrors())
 156                             .build());
 157                 }
 158             }
 159
 160             @Override
 161             public void onFailure(final Throwable t) {
 162                 LOGGER.debug("Failure multipart response for table features request. Exception: {}", t);
 163                 finalFuture.set(RpcResultBuilder.<UpdateTableOutput>failed()
 164                         .withError(ErrorType.RPC, "Future error", t).build());
 165             }
 166
 167             /**
 168              * @param multipartReplies
 169              */
 170             private void writeResponseToOperationalDatastore(final List<MultipartReply> multipartReplies) {
 171
 172                 final List<org.opendaylight.yang.gen.v1.urn.opendaylight.table.types.rev131026.table.features.TableFeatures> salTableFeatures = convertToSalTableFeatures(multipartReplies);
 173
 174                 final DeviceContext deviceContext = getDeviceContext();
 175                 final NodeId nodeId = deviceContext.getPrimaryConnectionContext().getNodeId();
 176                 final InstanceIdentifier<FlowCapableNode> flowCapableNodeII = InstanceIdentifier.create(Nodes.class)
 177                         .child(Node.class, new NodeKey(nodeId)).augmentation(FlowCapableNode.class);
 178                 for (org.opendaylight.yang.gen.v1.urn.opendaylight.table.types.rev131026.table.features.TableFeatures tableFeatureData : salTableFeatures) {
 179                     final Short tableId = tableFeatureData.getTableId();
 180                     KeyedInstanceIdentifier<org.opendaylight.yang.gen.v1.urn.opendaylight.table.types.rev131026.table.features.TableFeatures, TableFeaturesKey> tableFeaturesII = flowCapableNodeII
 181                             .child(Table.class, new TableKey(tableId))
 182                             .child(org.opendaylight.yang.gen.v1.urn.opendaylight.table.types.rev131026.table.features.TableFeatures.class,
 183                                     new TableFeaturesKey(tableId));
 184                     deviceContext.writeToTransaction(LogicalDatastoreType.OPERATIONAL, tableFeaturesII,
 185                             tableFeatureData);
 186                 }
 187
 188             }
 189
 190             private List<org.opendaylight.yang.gen.v1.urn.opendaylight.table.types.rev131026.table.features.TableFeatures> convertToSalTableFeatures(
 191                     final List<MultipartReply> multipartReplies) {
 192                 final List<org.opendaylight.yang.gen.v1.urn.opendaylight.table.types.rev131026.table.features.TableFeatures> salTableFeaturesAll = new ArrayList<>();
 193                 for (MultipartReply multipartReply : multipartReplies) {
 194                     if (multipartReply.getType().equals(MultipartType.OFPMPTABLEFEATURES)) {
 195                         MultipartReplyBody multipartReplyBody = multipartReply.getMultipartReplyBody();
 196                         if (multipartReplyBody instanceof MultipartReplyTableFeaturesCase) {
 197                             MultipartReplyTableFeaturesCase tableFeaturesCase = ((MultipartReplyTableFeaturesCase) multipartReplyBody);
 198                             MultipartReplyTableFeatures salTableFeatures = tableFeaturesCase
 199                                     .getMultipartReplyTableFeatures();
 200                             List<org.opendaylight.yang.gen.v1.urn.opendaylight.table.types.rev131026.table.features.TableFeatures> salTableFeaturesPartial = TableFeaturesReplyConvertor
 201                                     .toTableFeaturesReply(salTableFeatures);
 202                             salTableFeaturesAll.addAll(salTableFeaturesPartial);
 203                             LOGGER.debug("TableFeature {} for xid {}.", salTableFeatures, multipartReply.getXid());
 204                         }
 205                     }
 206                 }
 207                 return salTableFeaturesAll;
 208             }
 209
 210         }
 211
 212         Futures.addCallback(multipartFuture, new CallBackImpl());
 213
 214         return finalFuture;
 215     }
 216
 217     private MultipartRequestInputBuilder createMultipartHeader(final MultipartType multipart, final Long xid) {
 218         final MultipartRequestInputBuilder mprInput = new MultipartRequestInputBuilder();
 219         mprInput.setType(multipart);
 220         mprInput.setVersion(getVersion());
 221         mprInput.setXid(xid);
 222         mprInput.setFlags(new MultipartRequestFlags(false));
 223         return mprInput;
 224     }
 225
 226 }